0c7e73b1cf
While implementing my acceptance test, I stumbled upon a comment stating that snapshot deletion should also be implemented, so I snuck that in. I can't help but wonder if there is some generic logic that is implemented a few times throughout the packer code base that could maybe better serve us if it were abstracted to the common package.
125 lines
2.7 KiB
Go
125 lines
2.7 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/mitchellh/packer/packer"
|
|
)
|
|
|
|
// Artifact is an artifact implementation that contains built AMIs.
|
|
type Artifact struct {
|
|
// A map of regions to AMI IDs.
|
|
Amis map[string]string
|
|
|
|
// BuilderId is the unique ID for the builder that created this AMI
|
|
BuilderIdValue string
|
|
|
|
// EC2 connection for performing API stuff.
|
|
Conn *ec2.EC2
|
|
}
|
|
|
|
func (a *Artifact) BuilderId() string {
|
|
return a.BuilderIdValue
|
|
}
|
|
|
|
func (*Artifact) Files() []string {
|
|
// We have no files
|
|
return nil
|
|
}
|
|
|
|
func (a *Artifact) Id() string {
|
|
parts := make([]string, 0, len(a.Amis))
|
|
for region, amiId := range a.Amis {
|
|
parts = append(parts, fmt.Sprintf("%s:%s", region, amiId))
|
|
}
|
|
|
|
sort.Strings(parts)
|
|
return strings.Join(parts, ",")
|
|
}
|
|
|
|
func (a *Artifact) String() string {
|
|
amiStrings := make([]string, 0, len(a.Amis))
|
|
for region, id := range a.Amis {
|
|
single := fmt.Sprintf("%s: %s", region, id)
|
|
amiStrings = append(amiStrings, single)
|
|
}
|
|
|
|
sort.Strings(amiStrings)
|
|
return fmt.Sprintf("AMIs were created:\n\n%s", strings.Join(amiStrings, "\n"))
|
|
}
|
|
|
|
func (a *Artifact) State(name string) interface{} {
|
|
switch name {
|
|
case "atlas.artifact.metadata":
|
|
return a.stateAtlasMetadata()
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (a *Artifact) Destroy() error {
|
|
errors := make([]error, 0)
|
|
|
|
for region, imageId := range a.Amis {
|
|
log.Printf("Deregistering image ID (%s) from region (%s)", imageId, region)
|
|
|
|
regionConfig := &aws.Config{
|
|
Credentials: a.Conn.Config.Credentials,
|
|
Region: aws.String(region),
|
|
}
|
|
session, err := session.NewSession(regionConfig)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
regionConn := ec2.New(session)
|
|
|
|
// Get image metadata
|
|
imageResp, err := regionConn.DescribeImages(&ec2.DescribeImagesInput{
|
|
ImageIds: []*string{&imageId},
|
|
})
|
|
if err != nil {
|
|
errors = append(errors, err)
|
|
}
|
|
if len(imageResp.Images) == 0 {
|
|
err := fmt.Errorf("Error retrieving details for AMI (%s), no images found", imageId)
|
|
errors = append(errors, err)
|
|
}
|
|
|
|
// Deregister ami
|
|
input := &ec2.DeregisterImageInput{
|
|
ImageId: &imageId,
|
|
}
|
|
if _, err := regionConn.DeregisterImage(input); err != nil {
|
|
errors = append(errors, err)
|
|
}
|
|
|
|
// TODO(mitchellh): Delete the snapshots associated with an AMI too
|
|
}
|
|
|
|
if len(errors) > 0 {
|
|
if len(errors) == 1 {
|
|
return errors[0]
|
|
} else {
|
|
return &packer.MultiError{Errors: errors}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *Artifact) stateAtlasMetadata() interface{} {
|
|
metadata := make(map[string]string)
|
|
for region, imageId := range a.Amis {
|
|
k := fmt.Sprintf("region.%s", region)
|
|
metadata[k] = imageId
|
|
}
|
|
|
|
return metadata
|
|
}
|