Improve input prompt for AWS profile

gtsiolis · gtsiolis · commit c0e7cc236310 · 2026-03-11T14:59:41.000+02:00
diff --git a/internal/awsconfig/awsconfig.go b/internal/awsconfig/awsconfig.go
@@ -8,7 +8,6 @@ import (
 	"net/url"
 	"os"
 	"path/filepath"
-	"strings"
 
 	"gopkg.in/ini.v1"
 
@@ -220,10 +219,9 @@ func Setup(ctx context.Context, sink output.Sink, interactive bool, resolvedHost
 		return nil
 	}
 
-	files := strings.Join(status.filesToModify(), " and ")
 	responseCh := make(chan output.InputResponse, 1)
 	output.EmitUserInputRequest(sink, output.UserInputRequestEvent{
-		Prompt:     fmt.Sprintf("Set up LocalStack AWS profile in %s?", files),
+		Prompt:     "Configure AWS profile in ~/.aws/?",
 		Options:    []output.InputOption{{Key: "y", Label: "Y"}, {Key: "n", Label: "n"}},
 		ResponseCh: responseCh,
 	})
@@ -245,7 +243,7 @@ func Setup(ctx context.Context, sink output.Sink, interactive bool, resolvedHost
 				return nil
 			}
 		}
-		output.EmitSuccess(sink, fmt.Sprintf("LocalStack AWS profile written to %s", files))
+		output.EmitSuccess(sink, "AWS profile successfully configured")
 		output.EmitNote(sink, fmt.Sprintf("Try: aws s3 mb s3://test --profile %s", profileName))
 	case <-ctx.Done():
 		return ctx.Err()
diff --git a/internal/ui/components/input_prompt.go b/internal/ui/components/input_prompt.go
@@ -1,6 +1,9 @@
 package components
 
 import (
+	"fmt"
+	"strings"
+
 	"github.com/localstack/lstk/internal/output"
 	"github.com/localstack/lstk/internal/ui/styles"
 )
@@ -36,5 +39,38 @@ func (p InputPrompt) View() string {
 		return ""
 	}
 
-	return styles.SecondaryMessage.Render(output.FormatPrompt(p.prompt, p.options))
+	lines := strings.Split(p.prompt, "\n")
+	firstLine := lines[0]
+
+	var sb strings.Builder
+
+	// "?" prefix in secondary color
+	sb.WriteString(styles.Secondary.Render("? "))
+
+	// Style trailing "?" with secondary color
+	if before, found := strings.CutSuffix(firstLine, "?"); found {
+		sb.WriteString(styles.Message.Render(before + "?"))
+	} else {
+		sb.WriteString(styles.Message.Render(firstLine))
+	}
+
+	// Style option labels with secondary color
+	labels := make([]string, 0, len(p.options))
+	for _, opt := range p.options {
+		if opt.Label != "" {
+			labels = append(labels, opt.Label)
+		}
+	}
+	if len(labels) == 1 {
+		sb.WriteString(styles.Secondary.Render(fmt.Sprintf(" (%s)", labels[0])))
+	} else if len(labels) > 1 {
+		sb.WriteString(styles.Secondary.Render(fmt.Sprintf(" [%s]", strings.Join(labels, "/"))))
+	}
+
+	if len(lines) > 1 {
+		sb.WriteString("\n")
+		sb.WriteString(styles.SecondaryMessage.Render(strings.Join(lines[1:], "\n")))
+	}
+
+	return sb.String()
 }
diff --git a/test/integration/awsconfig_test.go b/test/integration/awsconfig_test.go
@@ -57,7 +57,7 @@ func TestStartPromptsToCreateAWSProfileWhenMissing(t *testing.T) {
 
 	// Wait for the AWS profile prompt.
 	require.Eventually(t, func() bool {
-		return bytes.Contains(out.Bytes(), []byte("Set up LocalStack AWS profile"))
+		return bytes.Contains(out.Bytes(), []byte("Configure AWS profile in"))
 	}, 2*time.Minute, 200*time.Millisecond, "AWS profile prompt should appear")
 
 	// Press Y to confirm.
@@ -66,7 +66,7 @@ func TestStartPromptsToCreateAWSProfileWhenMissing(t *testing.T) {
 
 	// Wait for the success message.
 	require.Eventually(t, func() bool {
-		return bytes.Contains(out.Bytes(), []byte("LocalStack AWS profile written"))
+		return bytes.Contains(out.Bytes(), []byte("AWS profile successfully configured"))
 	}, 10*time.Second, 200*time.Millisecond, "success message should appear")
 
 	// Verify files were written to the isolated home dir, not the real one.
@@ -133,7 +133,7 @@ func TestStartSkipsAWSProfilePromptWhenAlreadyConfigured(t *testing.T) {
 	_ = cmd.Wait()
 	<-outputCh
 
-	assert.NotContains(t, out.String(), "Set up LocalStack AWS profile",
+	assert.NotContains(t, out.String(), "Configure AWS profile in",
 		"profile prompt should not appear when profile is already correctly configured")
 }
 
