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"
|
|
|
|
)
|
|
|
|
|
2013-08-06 17:54:14 -04:00
|
|
|
type stepCreateAMI struct{}
|
2013-05-21 03:55:32 -04:00
|
|
|
|
2013-06-04 13:00:06 -04:00
|
|
|
func (s *stepCreateAMI) Run(state map[string]interface{}) multistep.StepAction {
|
2013-05-21 03:55:32 -04:00
|
|
|
config := state["config"].(config)
|
|
|
|
ec2conn := state["ec2"].(*ec2.EC2)
|
|
|
|
instance := state["instance"].(*ec2.Instance)
|
|
|
|
ui := state["ui"].(packer.Ui)
|
|
|
|
|
|
|
|
// 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{
|
|
|
|
InstanceId: instance.InstanceId,
|
2013-08-08 17:57:38 -04:00
|
|
|
Name: config.AMIName,
|
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)
|
|
|
|
state["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
|
2013-05-27 18:15:42 -04:00
|
|
|
ui.Say(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-05-22 01:28:41 -04:00
|
|
|
state["amis"] = amis
|
2013-05-21 03:55:32 -04:00
|
|
|
|
|
|
|
// Wait for the image to become ready
|
|
|
|
ui.Say("Waiting for AMI to become ready...")
|
2013-07-25 01:56:37 -04:00
|
|
|
if err := awscommon.WaitForAMI(ec2conn, createResp.ImageId); err != nil {
|
|
|
|
err := fmt.Errorf("Error waiting for AMI: %s", err)
|
|
|
|
state["error"] = err
|
|
|
|
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
|
|
|
}
|
|
|
|
|
|
|
|
func (s *stepCreateAMI) Cleanup(map[string]interface{}) {
|
|
|
|
// No cleanup...
|
|
|
|
}
|