rgt.go

package cloud

import (
	"context"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/resourcegroupstaggingapi"
	"github.com/aws/aws-sdk-go/service/resourcegroupstaggingapi/resourcegroupstaggingapiiface"
)

const (
	ResourceTypeELBLoadBalancer  = "elasticloadbalancing:loadbalancer"
	ResourceTypeELBTargetGroup   = "elasticloadbalancing:targetgroup"
	ResourceTypeEC2SecurityGroup = "ec2:security-group"
	ResourceTypeEC2Subnet        = "ec2:subnet"
)

type RGT interface {
	resourcegroupstaggingapiiface.ResourceGroupsTaggingAPIAPI

	GetResourcesAsList(ctx context.Context, input *resourcegroupstaggingapi.GetResourcesInput) ([]*resourcegroupstaggingapi.ResourceTagMapping, error)
}

func NewRGT(session *session.Session) RGT {
	return &defaultRGT{
		resourcegroupstaggingapi.New(session),
	}
}

var _ RGT = (*defaultRGT)(nil)

type defaultRGT struct {
	resourcegroupstaggingapiiface.ResourceGroupsTaggingAPIAPI
}

func (c *defaultRGT) GetResourcesAsList(ctx context.Context, input *resourcegroupstaggingapi.GetResourcesInput) ([]*resourcegroupstaggingapi.ResourceTagMapping, error) {
	var result []*resourcegroupstaggingapi.ResourceTagMapping
	if err := c.GetResourcesPagesWithContext(ctx, input, func(output *resourcegroupstaggingapi.GetResourcesOutput, _ bool) bool {
		for _, i := range output.ResourceTagMappingList {
			result = append(result, i)
		}
		return true
	}); err != nil {
		return nil, err
	}
	return result, nil
}

func NewRGTTagFilters(tags map[string]string) []*resourcegroupstaggingapi.TagFilter {
	tagFilters := make([]*resourcegroupstaggingapi.TagFilter, 0, len(tags))
	for k, v := range tags {
		tagFilters = append(tagFilters, &resourcegroupstaggingapi.TagFilter{
			Key:    aws.String(k),
			Values: aws.StringSlice([]string{v}),
		})
	}
	return tagFilters
}

func NewRGTTagFiltersV2(tags map[string][]string) []*resourcegroupstaggingapi.TagFilter {
	tagFilters := make([]*resourcegroupstaggingapi.TagFilter, 0, len(tags))
	for k, v := range tags {
		tagFilters = append(tagFilters, &resourcegroupstaggingapi.TagFilter{
			Key:    aws.String(k),
			Values: aws.StringSlice(v),
		})
	}
	return tagFilters
}

func ParseRGTTags(tags []*resourcegroupstaggingapi.Tag) map[string]string {
	result := make(map[string]string, len(tags))
	for _, tag := range tags {
		result[aws.StringValue(tag.Key)] = aws.StringValue(tag.Value)
	}
	return result
}