91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
package ebs
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/mitchellh/goamz/ec2"
|
|
"github.com/mitchellh/multistep"
|
|
awscommon "github.com/mitchellh/packer/builder/amazon/common"
|
|
"github.com/mitchellh/packer/packer"
|
|
)
|
|
|
|
type stepCreateAMI struct {
|
|
image *ec2.Image
|
|
}
|
|
|
|
func (s *stepCreateAMI) Run(state multistep.StateBag) multistep.StepAction {
|
|
config := state.Get("config").(config)
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
instance := state.Get("instance").(*ec2.Instance)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
// Create the image
|
|
ui.Say(fmt.Sprintf("Creating the AMI: %s", config.AMIName))
|
|
createOpts := &ec2.CreateImage{
|
|
InstanceId: instance.InstanceId,
|
|
Name: config.AMIName,
|
|
BlockDevices: config.BlockDevices.BuildAMIDevices(),
|
|
}
|
|
|
|
createResp, err := ec2conn.CreateImage(createOpts)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error creating AMI: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
// Set the AMI ID in the state
|
|
ui.Message(fmt.Sprintf("AMI: %s", createResp.ImageId))
|
|
amis := make(map[string]string)
|
|
amis[ec2conn.Region.Name] = createResp.ImageId
|
|
state.Put("amis", amis)
|
|
|
|
// Wait for the image to become ready
|
|
stateChange := awscommon.StateChangeConf{
|
|
Pending: []string{"pending"},
|
|
Target: "available",
|
|
Refresh: awscommon.AMIStateRefreshFunc(ec2conn, createResp.ImageId),
|
|
StepState: state,
|
|
}
|
|
|
|
ui.Say("Waiting for AMI to become ready...")
|
|
if _, err := awscommon.WaitForState(&stateChange); err != nil {
|
|
err := fmt.Errorf("Error waiting for AMI: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
imagesResp, err := ec2conn.Images([]string{createResp.ImageId}, nil)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error searching for AMI: %s", err)
|
|
state.Put("error", err)
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
s.image = &imagesResp.Images[0]
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *stepCreateAMI) Cleanup(state multistep.StateBag) {
|
|
_, cancelled := state.GetOk(multistep.StateCancelled)
|
|
_, halted := state.GetOk(multistep.StateHalted)
|
|
|
|
if !cancelled && !halted {
|
|
return
|
|
}
|
|
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
ui.Say("Deregistering the AMI because cancelation or error...")
|
|
if resp, err := ec2conn.DeregisterImage(s.image.Id); err != nil {
|
|
ui.Error(fmt.Sprintf("Error deregistering AMI, may still be around: %s", err))
|
|
return
|
|
} else if resp.Return == false {
|
|
ui.Error(fmt.Sprintf("Error deregistering AMI, may still be around: %s", resp.Return))
|
|
return
|
|
}
|
|
}
|