ekristen/aws-nuke

package resources

import (
    "context"
    "fmt"

    "github.com/gotidy/ptr"
    "github.com/sirupsen/logrus"

    "github.com/aws/aws-sdk-go/service/budgets"
    "github.com/aws/aws-sdk-go/service/budgets/budgetsiface"
    "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 BudgetsBudgetResource = "BudgetsBudget"

func init() {
    registry.Register(&registry.Registration{
        Name:     BudgetsBudgetResource,
        Scope:    nuke.Account,
        Resource: &BudgetsBudget{},
        Lister:   &BudgetsBudgetLister{},
        DeprecatedAliases: []string{
            "Budget",
        },
    })
}

type BudgetsBudgetLister struct {
    mockSvc budgetsiface.BudgetsAPI
}

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

    var svc budgetsiface.BudgetsAPI

    if l.mockSvc != nil {
        svc = l.mockSvc
    } else {
        svc = budgets.New(opts.Session)
    }

    params := &budgets.DescribeBudgetsInput{
        AccountId:  opts.AccountID,
        MaxResults: ptr.Int64(100),
    }

    buds := make([]*budgets.Budget, 0)
    err := svc.DescribeBudgetsPages(params, func(page *budgets.DescribeBudgetsOutput, lastPage bool) bool {
        buds = append(buds, page.Budgets...)
        return true
    })
    if err != nil {
        return nil, err
    }

    for _, bud := range buds {
        var resourceTags []*budgets.ResourceTag
        tags, tagsErr := svc.ListTagsForResource(&budgets.ListTagsForResourceInput{
            ResourceARN: ptr.String(fmt.Sprintf("arn:aws:budgets::%s:budget/%s", *opts.AccountID, *bud.BudgetName)),
        })
        if tagsErr != nil {
            logrus.WithError(tagsErr).Error("unable to get tags for budget")
        } else {
            resourceTags = tags.ResourceTags
        }

        resources = append(resources, &BudgetsBudget{
            svc:        svc,
            Name:       bud.BudgetName,
            BudgetType: bud.BudgetType,
            AccountID:  opts.AccountID,
            Tags:       resourceTags,
        })
    }

    return resources, nil
}

type BudgetsBudget struct {
    svc        budgetsiface.BudgetsAPI
    Name       *string
    BudgetType *string
    AccountID  *string
    Tags       []*budgets.ResourceTag
}

func (r *BudgetsBudget) Remove(_ context.Context) error {
    _, err := r.svc.DeleteBudget(&budgets.DeleteBudgetInput{
        AccountId:  r.AccountID,
        BudgetName: r.Name,
    })

    return err
}

func (r *BudgetsBudget) Properties() types.Properties {
    return types.NewPropertiesFromStruct(r)
}

func (r *BudgetsBudget) String() string {
    return *r.Name
}