budgets-budget.go

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,
		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
}