From e396c88e69d487f2f201734ac4435cbe48d8bd9f Mon Sep 17 00:00:00 2001
From: Rafael Gumieri <rafael.gumieri@geekie.com.br>
Date: Fri, 7 Jun 2024 11:59:19 -0300
Subject: [PATCH] feat: get the IPv4 Private IP on tasks run

---
 cmd/task_definitions_run.go | 54 ++++++++++++++++++++++++++++++++++---
 1 file changed, 51 insertions(+), 3 deletions(-)

diff --git a/cmd/task_definitions_run.go b/cmd/task_definitions_run.go
index 51c2234..6a2632e 100644
--- a/cmd/task_definitions_run.go
+++ b/cmd/task_definitions_run.go
@@ -12,6 +12,7 @@ import (
 	"github.com/TylerBrock/colorjson"
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/service/cloudwatchlogs"
+	"github.com/aws/aws-sdk-go/service/ec2"
 	"github.com/aws/aws-sdk-go/service/ecs"
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
@@ -180,6 +181,53 @@ func taskDefinitionsRunRun(cmd *cobra.Command, args []string) {
 		t.Exitln("task failed to run")
 	}
 
+	tSplited := strings.Split(aws.StringValue(taskResult.Tasks[0].TaskArn), "/")
+	taskID := tSplited[2]
+
+	if viper.GetBool("output-ip") {
+		var tasksStatus *ecs.DescribeTasksOutput
+		var err error
+
+		for {
+			tasksStatus, err = ecsI.DescribeTasks(&ecs.DescribeTasksInput{
+				Cluster: aws.String(cluster),
+				Tasks:   []*string{aws.String(taskID)},
+			})
+			t.Must(err)
+
+			status := aws.StringValue(tasksStatus.Tasks[0].LastStatus)
+			if status != "PENDING" {
+				break
+			}
+
+			time.Sleep(1 * time.Second)
+		}
+
+		ec2ECSInstance, err := ecsI.DescribeContainerInstances(&ecs.DescribeContainerInstancesInput{
+			Cluster:            aws.String(cluster),
+			ContainerInstances: []*string{tasksStatus.Tasks[0].ContainerInstanceArn},
+		})
+		t.Must(err)
+
+		if len(ec2ECSInstance.ContainerInstances) == 0 {
+			t.Exitln("failed to find the EC2 Instance running this task")
+		}
+
+		ec2Info, err := ec2I.DescribeInstances(&ec2.DescribeInstancesInput{
+			InstanceIds: []*string{ec2ECSInstance.ContainerInstances[0].Ec2InstanceId},
+		})
+
+		if len(ec2Info.Reservations) == 0 || len(ec2Info.Reservations[0].Instances) == 0 {
+			t.Exitln("failed to describe the EC2 Instance running this task")
+		}
+
+		hostIp := ec2Info.Reservations[0].Instances[0].PrivateIpAddress
+
+		hostPort := tasksStatus.Tasks[0].Containers[0].NetworkBindings[0].HostPort
+
+		t.Outf("%s:%d\n", aws.StringValue(hostIp), aws.Int64Value(hostPort))
+	}
+
 	if !follow {
 		t.Exit(nil)
 	}
@@ -200,9 +248,6 @@ func taskDefinitionsRunRun(cmd *cobra.Command, args []string) {
 		}()
 	}
 
-	tSplited := strings.Split(aws.StringValue(taskResult.Tasks[0].TaskArn), "/")
-	taskID := tSplited[2]
-
 	logDriver := td.ContainerDefinitions[0].LogConfiguration.LogDriver
 	if aws.StringValue(logDriver) != "awslogs" {
 		t.Exit(nil)
@@ -315,6 +360,9 @@ func init() {
 
 	flags.StringVar(&groupTasks, "group", "", groupTasksSpec)
 
+	flags.Bool("output-ip", false, "Return the Private IP and Port of the Container on running Task for EC2 Instances")
+	viper.BindPFlag("output-ip", taskDefinitionsRunCmd.Flags().Lookup("output-ip"))
+
 	flags.StringP("cluster", "c", "default", clusterSpec)
 	viper.BindPFlag("cluster", taskDefinitionsRunCmd.Flags().Lookup("cluster"))
 }