52 lines
1.4 KiB
Go
52 lines
1.4 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 stepStopInstance struct{}
|
|
|
|
func (s *stepStopInstance) Run(state map[string]interface{}) multistep.StepAction {
|
|
ec2conn := state["ec2"].(*ec2.EC2)
|
|
instance := state["instance"].(*ec2.Instance)
|
|
ui := state["ui"].(packer.Ui)
|
|
|
|
// Stop the instance so we can create an AMI from it
|
|
ui.Say("Stopping the source instance...")
|
|
_, err := ec2conn.StopInstances(instance.InstanceId)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error stopping instance: %s", err)
|
|
state["error"] = err
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
// Wait for the instance to actual stop
|
|
ui.Say("Waiting for the instance to stop...")
|
|
stateChange := awscommon.StateChangeConf{
|
|
Conn: ec2conn,
|
|
Pending: []string{"running", "stopping"},
|
|
Target: "stopped",
|
|
Refresh: awscommon.InstanceStateRefreshFunc(ec2conn, instance),
|
|
StepState: state,
|
|
}
|
|
instanceRaw, err := awscommon.WaitForState(&stateChange)
|
|
instance = instanceRaw.(*ec2.Instance)
|
|
if err != nil {
|
|
err := fmt.Errorf("Error waiting for instance to stop: %s", err)
|
|
state["error"] = err
|
|
ui.Error(err.Error())
|
|
return multistep.ActionHalt
|
|
}
|
|
|
|
return multistep.ActionContinue
|
|
}
|
|
|
|
func (s *stepStopInstance) Cleanup(map[string]interface{}) {
|
|
// No cleanup...
|
|
}
|