2013-07-15 02:02:18 -04:00
|
|
|
package ebs
|
2013-05-21 03:55:32 -04:00
|
|
|
|
|
|
|
import (
|
2013-05-27 18:15:42 -04:00
|
|
|
"fmt"
|
2013-05-21 03:55:32 -04:00
|
|
|
"github.com/mitchellh/goamz/ec2"
|
2013-06-04 13:00:06 -04:00
|
|
|
"github.com/mitchellh/multistep"
|
2013-07-25 01:56:37 -04:00
|
|
|
awscommon "github.com/mitchellh/packer/builder/amazon/common"
|
2013-05-21 03:55:32 -04:00
|
|
|
"github.com/mitchellh/packer/packer"
|
|
|
|
)
|
|
|
|
|
2014-05-19 12:40:39 -04:00
|
|
|
type stepCreateAMI struct {
|
|
|
|
image *ec2.Image
|
|
|
|
}
|
2013-05-21 03:55:32 -04:00
|
|
|
|
2013-08-31 16:00:43 -04:00
|
|
|
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)
|
2013-05-21 03:55:32 -04:00
|
|
|
|
|
|
|
// Create the image
|
2013-08-08 17:57:38 -04:00
|
|
|
ui.Say(fmt.Sprintf("Creating the AMI: %s", config.AMIName))
|
2013-05-21 03:55:32 -04:00
|
|
|
createOpts := &ec2.CreateImage{
|
2013-08-15 17:05:08 -04:00
|
|
|
InstanceId: instance.InstanceId,
|
|
|
|
Name: config.AMIName,
|
|
|
|
BlockDevices: config.BlockDevices.BuildAMIDevices(),
|
2013-05-21 03:55:32 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
createResp, err := ec2conn.CreateImage(createOpts)
|
|
|
|
if err != nil {
|
2013-06-19 23:54:02 -04:00
|
|
|
err := fmt.Errorf("Error creating AMI: %s", err)
|
2013-08-31 16:00:43 -04:00
|
|
|
state.Put("error", err)
|
2013-05-21 03:55:32 -04:00
|
|
|
ui.Error(err.Error())
|
2013-06-04 13:00:06 -04:00
|
|
|
return multistep.ActionHalt
|
2013-05-21 03:55:32 -04:00
|
|
|
}
|
|
|
|
|
2013-05-22 01:28:41 -04:00
|
|
|
// Set the AMI ID in the state
|
2014-05-19 12:40:39 -04:00
|
|
|
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]
|
2013-12-12 15:24:51 -05:00
|
|
|
ui.Message(fmt.Sprintf("AMI: %s", createResp.ImageId))
|
2013-05-22 01:28:41 -04:00
|
|
|
amis := make(map[string]string)
|
2013-07-29 19:42:35 -04:00
|
|
|
amis[ec2conn.Region.Name] = createResp.ImageId
|
2013-08-31 16:00:43 -04:00
|
|
|
state.Put("amis", amis)
|
2013-05-21 03:55:32 -04:00
|
|
|
|
|
|
|
// Wait for the image to become ready
|
2013-09-12 23:33:32 -04:00
|
|
|
stateChange := awscommon.StateChangeConf{
|
|
|
|
Pending: []string{"pending"},
|
|
|
|
Target: "available",
|
|
|
|
Refresh: awscommon.AMIStateRefreshFunc(ec2conn, createResp.ImageId),
|
|
|
|
StepState: state,
|
|
|
|
}
|
|
|
|
|
2013-05-21 03:55:32 -04:00
|
|
|
ui.Say("Waiting for AMI to become ready...")
|
2013-09-12 23:33:32 -04:00
|
|
|
if _, err := awscommon.WaitForState(&stateChange); err != nil {
|
2013-07-25 01:56:37 -04:00
|
|
|
err := fmt.Errorf("Error waiting for AMI: %s", err)
|
2013-08-31 16:00:43 -04:00
|
|
|
state.Put("error", err)
|
2013-07-25 01:56:37 -04:00
|
|
|
ui.Error(err.Error())
|
|
|
|
return multistep.ActionHalt
|
2013-05-21 03:55:32 -04:00
|
|
|
}
|
|
|
|
|
2013-06-04 13:00:06 -04:00
|
|
|
return multistep.ActionContinue
|
2013-05-21 03:55:32 -04:00
|
|
|
}
|
|
|
|
|
2014-05-19 12:40:39 -04:00
|
|
|
func (s *stepCreateAMI) Cleanup(state multistep.StateBag) {
|
|
|
|
if s.image == nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
ec2conn := state.Get("ec2").(*ec2.EC2)
|
|
|
|
ui := state.Get("ui").(packer.Ui)
|
|
|
|
|
|
|
|
ui.Say("Deregistering the AMI ...")
|
|
|
|
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
|
|
|
|
}
|
2013-05-21 03:55:32 -04:00
|
|
|
}
|