awslimitchecker.go

package awslimitchecker

import (
	"fmt"

	"github.com/sebasrp/awslimitchecker/internal/services"
)

var SupportedAwsServices = map[string]func() services.Svcquota{
	"acm":            services.NewAcmChecker,
	"autoscaling":    services.NewAutoscalingChecker,
	"cloudformation": services.NewCloudformationChecker,
	"dynamodb":       services.NewDynamoDbChecker,
	"ebs":            services.NewEbsChecker,
	"eks":            services.NewEksChecker,
	"elasticache":    services.NewElastiCacheChecker,
	"elb":            services.NewElbChecker,
	"iam":            services.NewIamChecker,
	"kinesis":        services.NewKinesisChecker,
	"rds":            services.NewRdsChecker,
	"s3":             services.NewS3Checker,
	"sns":            services.NewSnsChecker,
}

func GetUsage(awsService string, awsprofile string, region string, overrides []services.AWSQuotaOverride) (ret []services.AWSQuotaInfo) {
	_, err := services.InitializeConfig(awsprofile, region)
	if err != nil {
		fmt.Printf("Unable to create AWS session, %v", err)
		return
	}

	if awsService == "all" {
		for _, checker := range SupportedAwsServices {
			service := checker()
			service.SetQuotasOverride(overrides)
			ret = append(ret, service.GetUsage()...)
		}
	} else if val, ok := SupportedAwsServices[awsService]; ok {
		service := val()
		service.SetQuotasOverride(overrides)
		ret = service.GetUsage()
	}
	return
}

func GetIamPolicies() (ret []string) {
	for _, checker := range SupportedAwsServices {
		service := checker()
		ret = append(ret, service.GetRequiredPermissions()...)
	}
	return
}

func IsValidAwsService(service string) bool {
	if _, ok := SupportedAwsServices[service]; ok || service == "all" {
		return true
	} else {
		return false
	}
}