1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
|
package resources
import (
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/imagebuilder"
"github.com/rebuy-de/aws-nuke/pkg/types"
)
type ImageBuilderComponent struct {
svc *imagebuilder.Imagebuilder
arn string
}
func init() {
register("ImageBuilderComponent", ListImageBuilderComponents)
}
func ListImageBuilderComponents(sess *session.Session) ([]Resource, error) {
svc := imagebuilder.New(sess)
params := &imagebuilder.ListComponentsInput{}
resources := make([]Resource, 0)
for {
resp, err := svc.ListComponents(params)
if err != nil {
return nil, err
}
for _, out := range resp.ComponentVersionList {
resources, err = ListImageBuilderComponentVersions(svc, out.Arn, resources)
if err != nil {
return nil, err
}
}
if resp.NextToken == nil {
break
}
params = &imagebuilder.ListComponentsInput{
NextToken: resp.NextToken,
}
}
return resources, nil
}
func ListImageBuilderComponentVersions(svc *imagebuilder.Imagebuilder, componentVersionArn *string, resources []Resource) ([]Resource, error) {
params := &imagebuilder.ListComponentBuildVersionsInput{
ComponentVersionArn: componentVersionArn,
}
for {
resp, err := svc.ListComponentBuildVersions(params)
if err != nil {
return nil, err
}
for _, out := range resp.ComponentSummaryList {
resources = append(resources, &ImageBuilderComponent{
svc: svc,
arn: *out.Arn,
})
}
if resp.NextToken == nil {
break
}
params = &imagebuilder.ListComponentBuildVersionsInput{
ComponentVersionArn: componentVersionArn,
NextToken: resp.NextToken,
}
}
return resources, nil
}
func (e *ImageBuilderComponent) Remove() error {
_, err := e.svc.DeleteComponent(&imagebuilder.DeleteComponentInput{
ComponentBuildVersionArn: &e.arn,
})
return err
}
func (e *ImageBuilderComponent) Properties() types.Properties {
properties := types.NewProperties()
properties.Set("arn", e.arn)
return properties
}
func (e *ImageBuilderComponent) String() string {
return e.arn
}
|