Fallback to user-provided cluster from metadata (#2)

Fixes: #19, docker-archive/compose-cli#1084, docker-archive/compose-cli#2086
austindrenski · Nov 8, 2023 · 8b2912f · 8b2912f
1 parent 5e94b1a
commit 8b2912f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/ecs/aws.go b/ecs/aws.go
@@ -51,6 +51,7 @@ type API interface {
 	GetStackID(ctx context.Context, name string) (string, error)
 	ListStacks(ctx context.Context) ([]api.Stack, error)
 	GetStackClusterID(ctx context.Context, stack string) (string, error)
+	GetStackMetadataClusterID(ctx context.Context, stack string) (string, error)
 	GetServiceTaskDefinition(ctx context.Context, cluster string, serviceArns []string) (map[string]string, error)
 	ListStackServices(ctx context.Context, stack string) ([]string, error)
 	GetServiceTasks(ctx context.Context, cluster string, service string, stopped bool) ([]*ecs.Task, error)

diff --git a/ecs/list.go b/ecs/list.go
@@ -71,6 +71,12 @@ func (b *ComposeECS) checkStackState(ctx context.Context, name string) error {
 			svcNames[r.ARN] = r.LogicalID
 		}
 	}
+	if len(cluster) == 0 {
+		cluster, err = b.aws.GetStackMetadataClusterID(ctx, name)
+		if err != nil {
+			return err
+		}
+	}
 	if len(svcArns) == 0 {
 		return nil
 	}

diff --git a/ecs/sdk.go b/ecs/sdk.go
@@ -505,6 +505,10 @@ func (s sdk) GetStackClusterID(ctx context.Context, stack string) (string, error
 		token = response.NextToken
 	}
 	// stack is using user-provided cluster
+	return s.GetStackMetadataClusterID(ctx, stack)
+}
+
+func (s sdk) GetStackMetadataClusterID(ctx context.Context, stack string) (string, error) {
 	res, err := s.CF.GetTemplateSummaryWithContext(ctx, &cloudformation.GetTemplateSummaryInput{
 		StackName: aws.String(stack),
 	})