Tagging utils

PettitWesley · PettitWesley · commit 021e10457e66 · 2019-02-05T14:06:00.000-08:00
diff --git a/ecs-cli/modules/commands/flags/flags.go b/ecs-cli/modules/commands/flags/flags.go
@@ -140,6 +140,8 @@ const (
 	OutputDirFlag             = "output-dir"
 
 	DesiredTaskStatus = "desired-status"
+
+	ResourceTagsFlag = "tags"
 )
 
 // OptionalRegionAndProfileFlags provides these flags:
@@ -251,7 +253,7 @@ func DebugFlag() []cli.Flag {
 func FipsEndpointFlag() []cli.Flag {
 	return []cli.Flag{
 		cli.BoolFlag{
-			Name: UseFIPSFlag + ",fips",
+			Name:  UseFIPSFlag + ",fips",
 			Usage: "[Optional] Routes calls to AWS services through FIPS endpoints.",
 		},
 	}
diff --git a/ecs-cli/modules/utils/utils.go b/ecs-cli/modules/utils/utils.go
@@ -17,8 +17,11 @@ package utils
 import (
 	"fmt"
 	"os"
+	"strings"
 
+	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/awserr"
+	"github.com/aws/aws-sdk-go/service/ecs"
 )
 
 const (
@@ -58,3 +61,20 @@ func EntityAlreadyExists(err error) bool {
 	}
 	return false
 }
+
+// GetTags parses AWS Resource tags from the flag value
+// users specify tags in this format: key1=value1,key2=value2,key3=value3
+func GetTags(flagValue string, tags []*ecs.Tag) ([]*ecs.Tag, error) {
+	keyValPairs := strings.Split(flagValue, ",")
+	for _, pair := range keyValPairs {
+		split := strings.Split(pair, "=")
+		if len(split) != 2 {
+			return nil, fmt.Errorf("Tag input not formatted correctly: %s", pair)
+		}
+		tags = append(tags, &ecs.Tag{
+			Key:   aws.String(split[0]),
+			Value: aws.String(split[1]),
+		})
+	}
+	return tags, nil
+}
