ekristen/aws-nuke

package resources

import (
    "context"

    "github.com/aws/aws-sdk-go/aws"
    "github.com/aws/aws-sdk-go/service/autoscaling"

    "github.com/ekristen/libnuke/pkg/registry"
    "github.com/ekristen/libnuke/pkg/resource"
    "github.com/ekristen/libnuke/pkg/types"

    "github.com/ekristen/aws-nuke/v3/pkg/nuke"
)

const AutoScalingGroupResource = "AutoScalingGroup"

func init() {
    registry.Register(&registry.Registration{
        Name:     AutoScalingGroupResource,
        Scope:    nuke.Account,
        Resource: &AutoScalingGroup{},
        Lister:   &AutoScalingGroupLister{},
    })
}

type AutoScalingGroupLister struct{}

func (l *AutoScalingGroupLister) List(_ context.Context, o interface{}) ([]resource.Resource, error) {
    opts := o.(*nuke.ListerOpts)

    svc := autoscaling.New(opts.Session)
    resources := make([]resource.Resource, 0)

    params := &autoscaling.DescribeAutoScalingGroupsInput{}
    err := svc.DescribeAutoScalingGroupsPages(params,
        func(page *autoscaling.DescribeAutoScalingGroupsOutput, lastPage bool) bool {
            for _, asg := range page.AutoScalingGroups {
                resources = append(resources, &AutoScalingGroup{
                    group: asg,
                    svc:   svc,
                    tags:  asg.Tags,
                })
            }
            return !lastPage
        })

    if err != nil {
        return nil, err
    }

    return resources, nil
}

type AutoScalingGroup struct {
    svc   *autoscaling.AutoScaling
    group *autoscaling.Group
    tags  []*autoscaling.TagDescription
}

func (asg *AutoScalingGroup) Remove(_ context.Context) error {
    params := &autoscaling.DeleteAutoScalingGroupInput{
        AutoScalingGroupName: asg.group.AutoScalingGroupName,
        ForceDelete:          aws.Bool(true),
    }

    _, err := asg.svc.DeleteAutoScalingGroup(params)
    if err != nil {
        return err
    }

    return nil
}

func (asg *AutoScalingGroup) String() string {
    return *asg.group.AutoScalingGroupName
}

func (asg *AutoScalingGroup) Properties() types.Properties {
    properties := types.NewProperties()
    for _, tag := range asg.tags {
        properties.SetTag(tag.Key, tag.Value)
    }

    properties.Set("CreatedTime", asg.group.CreatedTime)
    properties.Set("Name", asg.group.AutoScalingGroupName)

    return properties
}