64 lines
2.0 KiB
Go
64 lines
2.0 KiB
Go
package common
|
|
|
|
import (
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/hashicorp/packer/builder"
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
)
|
|
|
|
type BuildInfoTemplate struct {
|
|
BuildRegion string
|
|
SourceAMI string
|
|
SourceAMICreationDate string
|
|
SourceAMIName string
|
|
SourceAMIOwner string
|
|
SourceAMIOwnerName string
|
|
SourceAMITags map[string]string
|
|
}
|
|
|
|
func extractBuildInfo(region string, state multistep.StateBag, generatedData *builder.GeneratedData) *BuildInfoTemplate {
|
|
rawSourceAMI, hasSourceAMI := state.GetOk("source_image")
|
|
if !hasSourceAMI {
|
|
return &BuildInfoTemplate{
|
|
BuildRegion: region,
|
|
}
|
|
}
|
|
|
|
sourceAMI := rawSourceAMI.(*ec2.Image)
|
|
sourceAMITags := make(map[string]string, len(sourceAMI.Tags))
|
|
for _, tag := range sourceAMI.Tags {
|
|
sourceAMITags[aws.StringValue(tag.Key)] = aws.StringValue(tag.Value)
|
|
}
|
|
|
|
buildInfoTemplate := &BuildInfoTemplate{
|
|
BuildRegion: region,
|
|
SourceAMI: aws.StringValue(sourceAMI.ImageId),
|
|
SourceAMICreationDate: aws.StringValue(sourceAMI.CreationDate),
|
|
SourceAMIName: aws.StringValue(sourceAMI.Name),
|
|
SourceAMIOwner: aws.StringValue(sourceAMI.OwnerId),
|
|
SourceAMIOwnerName: aws.StringValue(sourceAMI.ImageOwnerAlias),
|
|
SourceAMITags: sourceAMITags,
|
|
}
|
|
|
|
generatedData.Put("BuildRegion", buildInfoTemplate.BuildRegion)
|
|
generatedData.Put("SourceAMI", buildInfoTemplate.SourceAMI)
|
|
generatedData.Put("SourceAMICreationDate", buildInfoTemplate.SourceAMICreationDate)
|
|
generatedData.Put("SourceAMIName", buildInfoTemplate.SourceAMIName)
|
|
generatedData.Put("SourceAMIOwner", buildInfoTemplate.SourceAMIOwner)
|
|
generatedData.Put("SourceAMIOwnerName", buildInfoTemplate.SourceAMIOwnerName)
|
|
|
|
return buildInfoTemplate
|
|
}
|
|
|
|
func GetGeneratedDataList() []string {
|
|
return []string{
|
|
"SourceAMIName",
|
|
"BuildRegion",
|
|
"SourceAMI",
|
|
"SourceAMICreationDate",
|
|
"SourceAMIOwner",
|
|
"SourceAMIOwnerName",
|
|
}
|
|
}
|