exec fixed

Author: teaguru

cmd/exec.go

@@ -18,13 +18,11 @@ var execCmd = &cobra.Command{
     Args: cobra.MinimumNArgs(1),
     Run: func(cmd *cobra.Command, args []string) {
         viper.SetDefault("run.launch_type", "FARGATE")
-        var containerName string
+        //var containerName string
         var commandArgs []string
         if name := viper.GetString("container_name"); name == "" {
-            containerName = args[0]
             commandArgs = args[1:]
         } else {
-            containerName = name
             commandArgs = args
         }
 
@@ -34,8 +32,6 @@ var execCmd = &cobra.Command{
         err := lib.ExecFargate(
             viper.GetString("profile"),
             viper.GetString("cluster"),
-            viper.GetString("run.service"),
-            containerName,
             commandString, // Pass the combined command string
         )
         if err != nil {

lib/exec.go

@@ -8,6 +8,7 @@ import (
     "github.com/aws/aws-sdk-go-v2/config"
     "github.com/aws/aws-sdk-go-v2/service/ecs"
     "github.com/fujiwara/ecsta"
+    "os"
 )
 
 var sessionInstance *ecs.Client
@@ -22,14 +23,16 @@ func InitAWS(profile string) error {
         if err != nil {
             return fmt.Errorf("failed to load configuration: %w", err)
         }
+        os.Setenv("AWS_PROFILE", profile) //required for aws sdk
         sessionInstance = ecs.NewFromConfig(cfg)
         sessionConfig = cfg // Save session configuration
     }
     return nil
 }
 
 // ExecFargate executes a command in a specified container on an ECS Fargate service
-func ExecFargate(profile, cluster, service, containerName, command string) error {
+func ExecFargate(profile, cluster, command string) error {
+    
     if err := InitAWS(profile); err != nil {
         return fmt.Errorf("failed to initialize AWS session: %w", err)
     }
@@ -39,15 +42,13 @@ func ExecFargate(profile, cluster, service, containerName, command string) error
     if err != nil {
         return fmt.Errorf("failed to create ecsta application: %w", err)
     }
-    service = "app"
 
     entrypoint := "/usr/bin/ssm-parent"
     configPath := "/app/.ssm-parent.yaml"
     fullCommand := fmt.Sprintf("%s -c %s run -- %s", entrypoint, configPath, command)
     execOpt := ecsta.ExecOption{
         Command:   fullCommand,
     }
-    fmt.Println(execOpt)
     if err := ecstaApp.RunExec(context.Background(), &execOpt); err != nil {
         return fmt.Errorf("failed to execute command: %w", err)
     }

lib/runFargate.go

@@ -63,7 +63,7 @@ securityGroups, err := fetchSecurityGroupsByName(svcEC2, securityGroupFilter)
     var foundContainerName bool
     if err := modifyContainerDefinitionImages(imageTag, imageTags, workDir, taskDefinition.ContainerDefinitions, ctx); err != nil {
         return 1, err
-    }
+    } 
     for n, containerDefinition := range taskDefinition.ContainerDefinitions {
     if aws.StringValue(containerDefinition.Name) == containerName {
         foundContainerName = true