From a5a925173c9c12364d8bf23fd716500dd930db21 Mon Sep 17 00:00:00 2001
From: Nicolas De Loof <nicolas.deloof@gmail.com>
Date: Mon, 18 May 2020 16:01:00 +0200
Subject: [PATCH] SDK methods to query service tasks and retrieve public IP

Signed-off-by: Nicolas De Loof <nicolas.deloof@gmail.com>
---
 ecs/pkg/amazon/sdk.go | 52 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/ecs/pkg/amazon/sdk.go b/ecs/pkg/amazon/sdk.go
index b29751309..2df26efd6 100644
--- a/ecs/pkg/amazon/sdk.go
+++ b/ecs/pkg/amazon/sdk.go
@@ -338,3 +338,55 @@ func (s sdk) GetLogs(ctx context.Context, name string) error {
 		time.Sleep(500 * time.Millisecond)
 	}
 }
+
+func (s sdk) GetTasks(ctx context.Context, cluster string, name string) ([]string, error) {
+	tasks, err := s.ECS.ListTasksWithContext(ctx, &ecs.ListTasksInput{
+		Cluster:     aws.String(cluster),
+		ServiceName: aws.String(name),
+	})
+	if err != nil {
+		return nil, err
+	}
+	arns := []string{}
+	for _, arn := range tasks.TaskArns {
+		arns = append(arns, *arn)
+	}
+	return arns, nil
+}
+
+func (s sdk) GetNetworkInterfaces(ctx context.Context, cluster string, arns ...string) ([]string, error) {
+	tasks, err := s.ECS.DescribeTasksWithContext(ctx, &ecs.DescribeTasksInput{
+		Cluster: aws.String(cluster),
+		Tasks:   aws.StringSlice(arns),
+	})
+	if err != nil {
+		return nil, err
+	}
+	interfaces := []string{}
+	for _, task := range tasks.Tasks {
+		for _, attachement := range task.Attachments {
+			if *attachement.Type == "ElasticNetworkInterface" {
+				for _, pair := range attachement.Details {
+					if *pair.Name == "networkInterfaceId" {
+						interfaces = append(interfaces, *pair.Value)
+					}
+				}
+			}
+		}
+	}
+	return interfaces, nil
+}
+
+func (s sdk) GetPublicIPs(ctx context.Context, interfaces ...string) ([]string, error) {
+	desc, err := s.EC2.DescribeNetworkInterfaces(&ec2.DescribeNetworkInterfacesInput{
+		NetworkInterfaceIds: aws.StringSlice(interfaces),
+	})
+	if err != nil {
+		return nil, err
+	}
+	publicIPs := []string{}
+	for _, interf := range desc.NetworkInterfaces {
+		publicIPs = append(publicIPs, *interf.Association.PublicIp)
+	}
+	return publicIPs, nil
+}