mirror of https://github.com/hashicorp/packer
parent
420d0ebe3e
commit
9f310045fc
@ -0,0 +1,83 @@
|
||||
package classic
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/hashicorp/packer/helper/multistep"
|
||||
"github.com/hashicorp/packer/packer"
|
||||
"github.com/hashicorp/packer/template/interpolate"
|
||||
)
|
||||
|
||||
type stepUploadImage struct {
|
||||
uploadImageCommand string
|
||||
}
|
||||
|
||||
type uploadCmdData struct {
|
||||
Username string
|
||||
Password string
|
||||
AccountID string
|
||||
ImageFile string
|
||||
SegmentPath string
|
||||
}
|
||||
|
||||
func (s *stepUploadImage) Run(_ context.Context, state multistep.StateBag) multistep.StepAction {
|
||||
ui := state.Get("ui").(packer.Ui)
|
||||
comm := state.Get("communicator").(packer.Communicator)
|
||||
config := state.Get("config").(*Config)
|
||||
runID := state.Get("run_id").(string)
|
||||
|
||||
imageFile := fmt.Sprintf("%s.tar.gz", config.ImageName)
|
||||
state.Put("image_file", imageFile)
|
||||
|
||||
config.ctx.Data = uploadCmdData{
|
||||
Username: config.Username,
|
||||
Password: config.Password,
|
||||
AccountID: config.IdentityDomain,
|
||||
ImageFile: imageFile,
|
||||
SegmentPath: fmt.Sprintf("compute_images_segments/%s/_segment_/%s", imageFile, runID),
|
||||
}
|
||||
uploadImageCmd, err := interpolate.Render(s.uploadImageCommand, &config.ctx)
|
||||
if err != nil {
|
||||
err := fmt.Errorf("Error processing image upload command: %s", err)
|
||||
state.Put("error", err)
|
||||
ui.Error(err.Error())
|
||||
return multistep.ActionHalt
|
||||
}
|
||||
|
||||
command := fmt.Sprintf(`#!/bin/sh
|
||||
set -e
|
||||
set -x
|
||||
mkdir /builder
|
||||
mkfs -t ext3 /dev/xvdb
|
||||
mount /dev/xvdb /builder
|
||||
chown opc:opc /builder
|
||||
cd /builder
|
||||
dd if=/dev/xvdc bs=8M status=progress | cp --sparse=always /dev/stdin diskimage.raw
|
||||
tar czSf ./diskimage.tar.gz ./diskimage.raw
|
||||
rm diskimage.raw
|
||||
%s`, uploadImageCmd)
|
||||
|
||||
dest := "/tmp/create-packer-diskimage.sh"
|
||||
comm.Upload(dest, strings.NewReader(command), nil)
|
||||
cmd := &packer.RemoteCmd{
|
||||
Command: fmt.Sprintf("sudo /bin/sh %s", dest),
|
||||
}
|
||||
if err := cmd.StartWithUi(comm, ui); err != nil {
|
||||
err = fmt.Errorf("Problem creating image`: %s", err)
|
||||
ui.Error(err.Error())
|
||||
state.Put("error", err)
|
||||
return multistep.ActionHalt
|
||||
}
|
||||
|
||||
if cmd.ExitStatus != 0 {
|
||||
err = fmt.Errorf("Create Disk Image command failed with exit code %d", cmd.ExitStatus)
|
||||
ui.Error(err.Error())
|
||||
state.Put("error", err)
|
||||
return multistep.ActionHalt
|
||||
}
|
||||
return multistep.ActionContinue
|
||||
}
|
||||
|
||||
func (s *stepUploadImage) Cleanup(state multistep.StateBag) {}
|
||||
Loading…
Reference in new issue