f555e7a9f2
* I had to contextualise Communicator.Start and RemoteCmd.StartWithUi NOTE: Communicator.Start starts a RemoteCmd but RemoteCmd.StartWithUi will run the cmd and wait for a return, so I renamed StartWithUi to RunWithUi so that the intent is clearer. Ideally in the future RunWithUi will be named back to StartWithUi and the exit status or wait funcs of the command will allow to wait for a return. If you do so please read carrefully https://golang.org/pkg/os/exec/#Cmd.Stdout to avoid a deadlock * cmd.ExitStatus to cmd.ExitStatus() is now blocking to avoid race conditions * also had to simplify StartWithUi
99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
package instance
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
"github.com/hashicorp/packer/packer"
|
|
"github.com/hashicorp/packer/template/interpolate"
|
|
)
|
|
|
|
type uploadCmdData struct {
|
|
AccessKey string
|
|
BucketName string
|
|
BundleDirectory string
|
|
ManifestPath string
|
|
Region string
|
|
SecretKey string
|
|
Token string
|
|
}
|
|
|
|
type StepUploadBundle struct {
|
|
Debug bool
|
|
}
|
|
|
|
func (s *StepUploadBundle) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
|
|
comm := state.Get("communicator").(packer.Communicator)
|
|
config := state.Get("config").(*Config)
|
|
manifestName := state.Get("manifest_name").(string)
|
|
manifestPath := state.Get("manifest_path").(string)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
accessKey := config.AccessKey
|
|
secretKey := config.SecretKey
|
|
session, err := config.AccessConfig.Session()
|
|
region := *session.Config.Region
|
|
accessConfig := session.Config
|
|
var token string
|
|
if err == nil && accessKey == "" && secretKey == "" {
|
|
credentials, err := accessConfig.Credentials.Get()
|
|
if err == nil {
|
|
accessKey = credentials.AccessKeyID
|
|
secretKey = credentials.SecretAccessKey
|
|
token = credentials.SessionToken
|
|
}
|
|
}
|
|
|
|
config.ctx.Data = uploadCmdData{
|
|
AccessKey: accessKey,
|
|
BucketName: config.S3Bucket,
|
|
BundleDirectory: config.BundleDestination,
|
|
ManifestPath: manifestPath,
|
|
Region: region,
|
|
SecretKey: secretKey,
|
|
Token: token,
|
|
}
|
|
config.BundleUploadCommand, err = interpolate.Render(config.BundleUploadCommand, &config.ctx)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error processing bundle upload command: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
ui.Say("Uploading the bundle...")
|
|
cmd := &packer.RemoteCmd{Command: config.BundleUploadCommand}
|
|
|
|
if s.Debug {
|
|
ui.Say(fmt.Sprintf("Running: %s", config.BundleUploadCommand))
|
|
}
|
|
|
|
if err := cmd.RunWithUi(ctx, comm, ui); err != nil {
|
|
state.Put("error", fmt.Errorf("Error uploading volume: %s", err))
|
|
ui.Error(state.Get("error").(error).Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
if cmd.ExitStatus() != 0 {
|
|
if cmd.ExitStatus() == 3 {
|
|
ui.Error(fmt.Sprintf("Please check that the bucket `%s` "+
|
|
"does not exist, or exists and is writable. This error "+
|
|
"indicates that the bucket may be owned by somebody else.",
|
|
config.S3Bucket))
|
|
}
|
|
state.Put("error", fmt.Errorf(
|
|
"Bundle upload failed. Please see the output above for more\n"+
|
|
"details on what went wrong."))
|
|
ui.Error(state.Get("error").(error).Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
state.Put("remote_manifest_path", fmt.Sprintf(
|
|
"%s/%s", config.S3Bucket, manifestName))
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *StepUploadBundle) Cleanup(state multistep.StateBag) {}
|