github.phpd.cn/hashicorp/packer@v1.3.2/builder/amazon/ebs/step_create_ami.go (about)

     1  package ebs
     2  
     3  import (
     4  	"context"
     5  	"fmt"
     6  	"log"
     7  
     8  	"github.com/aws/aws-sdk-go/service/ec2"
     9  	awscommon "github.com/hashicorp/packer/builder/amazon/common"
    10  	"github.com/hashicorp/packer/common/random"
    11  	"github.com/hashicorp/packer/helper/multistep"
    12  	"github.com/hashicorp/packer/packer"
    13  )
    14  
    15  type stepCreateAMI struct {
    16  	image *ec2.Image
    17  }
    18  
    19  func (s *stepCreateAMI) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction {
    20  	config := state.Get("config").(*Config)
    21  	ec2conn := state.Get("ec2").(*ec2.EC2)
    22  	instance := state.Get("instance").(*ec2.Instance)
    23  	ui := state.Get("ui").(packer.Ui)
    24  
    25  	// Create the image
    26  	amiName := config.AMIName
    27  	if config.AMIEncryptBootVolume {
    28  		// to avoid having a temporary unencrypted
    29  		// image named config.AMIName
    30  		amiName = random.AlphaNum(7)
    31  	}
    32  
    33  	ui.Say(fmt.Sprintf("Creating unencrypted AMI %s from instance %s", amiName, *instance.InstanceId))
    34  	createOpts := &ec2.CreateImageInput{
    35  		InstanceId:          instance.InstanceId,
    36  		Name:                &amiName,
    37  		BlockDeviceMappings: config.BlockDevices.BuildAMIDevices(),
    38  	}
    39  
    40  	createResp, err := ec2conn.CreateImage(createOpts)
    41  	if err != nil {
    42  		err := fmt.Errorf("Error creating AMI: %s", err)
    43  		state.Put("error", err)
    44  		ui.Error(err.Error())
    45  		return multistep.ActionHalt
    46  	}
    47  
    48  	// Set the AMI ID in the state
    49  	ui.Message(fmt.Sprintf("AMI: %s", *createResp.ImageId))
    50  	amis := make(map[string]string)
    51  	amis[*ec2conn.Config.Region] = *createResp.ImageId
    52  	state.Put("amis", amis)
    53  
    54  	// Wait for the image to become ready
    55  	ui.Say("Waiting for AMI to become ready...")
    56  	if err := awscommon.WaitUntilAMIAvailable(ctx, ec2conn, *createResp.ImageId); err != nil {
    57  		log.Printf("Error waiting for AMI: %s", err)
    58  		imagesResp, err := ec2conn.DescribeImages(&ec2.DescribeImagesInput{ImageIds: []*string{createResp.ImageId}})
    59  		if err != nil {
    60  			log.Printf("Unable to determine reason waiting for AMI failed: %s", err)
    61  			err = fmt.Errorf("Unknown error waiting for AMI.")
    62  		} else {
    63  			stateReason := imagesResp.Images[0].StateReason
    64  			err = fmt.Errorf("Error waiting for AMI. Reason: %s", stateReason)
    65  		}
    66  
    67  		state.Put("error", err)
    68  		ui.Error(err.Error())
    69  		return multistep.ActionHalt
    70  	}
    71  
    72  	imagesResp, err := ec2conn.DescribeImages(&ec2.DescribeImagesInput{ImageIds: []*string{createResp.ImageId}})
    73  	if err != nil {
    74  		err := fmt.Errorf("Error searching for AMI: %s", err)
    75  		state.Put("error", err)
    76  		ui.Error(err.Error())
    77  		return multistep.ActionHalt
    78  	}
    79  	s.image = imagesResp.Images[0]
    80  
    81  	snapshots := make(map[string][]string)
    82  	for _, blockDeviceMapping := range imagesResp.Images[0].BlockDeviceMappings {
    83  		if blockDeviceMapping.Ebs != nil && blockDeviceMapping.Ebs.SnapshotId != nil {
    84  
    85  			snapshots[*ec2conn.Config.Region] = append(snapshots[*ec2conn.Config.Region], *blockDeviceMapping.Ebs.SnapshotId)
    86  		}
    87  	}
    88  	state.Put("snapshots", snapshots)
    89  
    90  	return multistep.ActionContinue
    91  }
    92  
    93  func (s *stepCreateAMI) Cleanup(state multistep.StateBag) {
    94  	if s.image == nil {
    95  		return
    96  	}
    97  
    98  	_, cancelled := state.GetOk(multistep.StateCancelled)
    99  	_, halted := state.GetOk(multistep.StateHalted)
   100  	if !cancelled && !halted {
   101  		return
   102  	}
   103  
   104  	ec2conn := state.Get("ec2").(*ec2.EC2)
   105  	ui := state.Get("ui").(packer.Ui)
   106  
   107  	ui.Say("Deregistering the AMI because cancellation or error...")
   108  	deregisterOpts := &ec2.DeregisterImageInput{ImageId: s.image.ImageId}
   109  	if _, err := ec2conn.DeregisterImage(deregisterOpts); err != nil {
   110  		ui.Error(fmt.Sprintf("Error deregistering AMI, may still be around: %s", err))
   111  		return
   112  	}
   113  }