Add selftest tui commands for manual prompt verification

cmd/selftest/tui/ask.go

@@ -0,0 +1,45 @@
+package tui
+
+import (
+	"github.com/databricks/cli/libs/cmdio"
+	"github.com/spf13/cobra"
+)
+
+func newAskCmd() *cobra.Command {
+	var defaultVal string
+	cmd := &cobra.Command{
+		Use:   "ask",
+		Short: "cmdio.Ask (single-line text input with optional default)",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			ans, err := cmdio.Ask(ctx, "Enter a value", defaultVal)
+			if err != nil {
+				return err
+			}
+			cmdio.LogString(ctx, "Entered: "+ans)
+			return nil
+		},
+	}
+	cmd.Flags().StringVar(&defaultVal, "default", "", "default returned if the user just presses Enter")
+	return cmd
+}
+
+func newAskYesOrNoCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "ask-yes-no",
+		Short: "cmdio.AskYesOrNo (yes/no question)",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			ans, err := cmdio.AskYesOrNo(ctx, "Continue")
+			if err != nil {
+				return err
+			}
+			if ans {
+				cmdio.LogString(ctx, "Answer: yes")
+			} else {
+				cmdio.LogString(ctx, "Answer: no")
+			}
+			return nil
+		},
+	}
+}

cmd/selftest/tui/colors.go

@@ -0,0 +1,25 @@
+package tui
+
+import (
+	"github.com/databricks/cli/libs/cmdio"
+	"github.com/spf13/cobra"
+)
+
+func newColorsCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "colors",
+		Short: "Print colored text to verify color support",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			swatch := "the quick brown fox jumps over the lazy dog"
+			cmdio.LogString(ctx, "red:     "+cmdio.Red(ctx, swatch))
+			cmdio.LogString(ctx, "green:   "+cmdio.Green(ctx, swatch))
+			cmdio.LogString(ctx, "yellow:  "+cmdio.Yellow(ctx, swatch))
+			cmdio.LogString(ctx, "blue:    "+cmdio.Blue(ctx, swatch))
+			cmdio.LogString(ctx, "cyan:    "+cmdio.Cyan(ctx, swatch))
+			cmdio.LogString(ctx, "hiblack: "+cmdio.HiBlack(ctx, swatch))
+			cmdio.LogString(ctx, "hiblue:  "+cmdio.HiBlue(ctx, swatch))
+			return nil
+		},
+	}
+}

cmd/selftest/tui/fixtures.go

@@ -0,0 +1,175 @@
+package tui
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/databricks/cli/libs/cmdio"
+)
+
+type spinnerMessage struct {
+	text     string
+	duration time.Duration
+}
+
+var spinnerMessages = []spinnerMessage{
+	{"Initializing...", time.Second},
+	{"Loading configuration", time.Second},
+	{"Connecting to workspace", time.Second},
+	{"Processing files", time.Second},
+	{"Finalizing", time.Second},
+}
+
+// databricksFeatures is a stable list of Databricks product / feature names
+// used as fixture data for the prompt scenarios. Drawn from the public docs
+// (https://docs.databricks.com) so the demo data looks like something a user
+// would actually encounter.
+var databricksFeatures = []string{
+	"unity-catalog",
+	"delta-lake",
+	"delta-sharing",
+	"photon",
+	"mlflow",
+	"mosaic-ai",
+	"genie",
+	"lakeflow-connect",
+	"lakeflow-jobs",
+	"vector-search",
+	"model-serving",
+	"feature-store",
+	"databricks-sql",
+	"ai-playground",
+	"foundation-models",
+	"lakehouse-monitoring",
+	"liquid-clustering",
+	"predictive-optimization",
+	"governed-tags",
+	"lakeflow-designer",
+}
+
+// buildItems uses zero-padded ids so the alphabetical Select scenario has
+// a stable sort order.
+func buildItems(n int) []cmdio.Tuple {
+	n = min(n, len(databricksFeatures))
+	items := make([]cmdio.Tuple, 0, n)
+	for i := range n {
+		items = append(items, cmdio.Tuple{
+			Name: databricksFeatures[i],
+			Id:   fmt.Sprintf("id-%02d", i+1),
+		})
+	}
+	return items
+}
+
+// buildFilterItems returns 15 items where 5 share the substring "lake", so
+// progressive typing narrows the list, and a non-matching substring ("xyz")
+// hits the "No results" path.
+func buildFilterItems() []cmdio.Tuple {
+	names := []string{
+		"lakehouse-monitoring",
+		"lakeflow-connect",
+		"lakeflow-jobs",
+		"delta-lake",
+		"lakebase-postgres",
+		"unity-catalog",
+		"mosaic-ai",
+		"vector-search",
+		"model-serving",
+		"feature-store",
+		"ai-playground",
+		"genie-spaces",
+		"mlflow-tracking",
+		"liquid-clustering",
+		"predictive-optimization",
+	}
+	items := make([]cmdio.Tuple, 0, len(names))
+	for i, name := range names {
+		items = append(items, cmdio.Tuple{
+			Name: name,
+			Id:   fmt.Sprintf("id-%02d", i+1),
+		})
+	}
+	return items
+}
+
+// buildLongItems uses fully-qualified workspace URLs as ids so that the
+// rendered field overflows a typical terminal width.
+func buildLongItems() []cmdio.Tuple {
+	hosts := []string{
+		"https://adb-1234567890123456.78.azuredatabricks.net/?o=1234567890123456",
+		"https://adb-2345678901234567.89.azuredatabricks.net/?o=2345678901234567",
+		"https://acme-prod.cloud.databricks.com/?o=3456789012345678",
+		"https://acme-staging.cloud.databricks.com/?o=4567890123456789",
+		"https://acme-dev.cloud.databricks.com/?o=5678901234567890",
+		"https://1234567890123456.7.gcp.databricks.com/?o=6789012345678901",
+		"https://2345678901234567.8.gcp.databricks.com/?o=7890123456789012",
+		"https://field-eng-east.cloud.databricks.com/?o=8901234567890123",
+	}
+	items := make([]cmdio.Tuple, 0, len(hosts))
+	for i, host := range hosts {
+		items = append(items, cmdio.Tuple{
+			Name: fmt.Sprintf("workspace-%02d", i+1),
+			Id:   host,
+		})
+	}
+	return items
+}
+
+// clusterItem mirrors libs/databrickscfg/cfgpickers/clusters.go's
+// compatibleCluster: State, Access, and Runtime are exposed as methods so
+// the Active/Inactive templates exercise text/template's method-resolution
+// path, and State returns a pre-rendered colored string (matching the
+// renderedState cache in production) so the demo also exercises ANSI codes
+// emitted from inside a template.
+type clusterItem struct {
+	Name string
+	Id   string
+
+	access        string
+	runtimeName   string
+	renderedState string
+}
+
+func (c clusterItem) Access() string  { return c.access }
+func (c clusterItem) Runtime() string { return c.runtimeName }
+func (c clusterItem) State() string   { return c.renderedState }
+
+func buildClusterItems(ctx context.Context) []clusterItem {
+	green := func(s string) string { return cmdio.Green(ctx, s) }
+	red := func(s string) string { return cmdio.Red(ctx, s) }
+	blue := func(s string) string { return cmdio.Blue(ctx, s) }
+	return []clusterItem{
+		{Name: "shared-autoscaling-prod", Id: "0123-456789-abcdef01", access: "Shared", runtimeName: "DBR 14.3 LTS", renderedState: green("RUNNING")},
+		{Name: "ml-gpu-experiments", Id: "0123-456789-abcdef02", access: "Assigned", runtimeName: "DBR 15.0 ML", renderedState: red("TERMINATED")},
+		{Name: "job-compute-bronze-etl", Id: "0123-456789-abcdef03", access: "Shared", runtimeName: "DBR 13.3 LTS", renderedState: green("RUNNING")},
+		{Name: "interactive-analytics", Id: "0123-456789-abcdef04", access: "Assigned", runtimeName: "DBR 14.3", renderedState: blue("PENDING")},
+		{Name: "photon-streaming-realtime", Id: "0123-456789-abcdef05", access: "Shared", runtimeName: "DBR 14.3 Photon", renderedState: green("RUNNING")},
+		{Name: "single-node-dev", Id: "0123-456789-abcdef06", access: "Assigned", runtimeName: "DBR 14.3 LTS", renderedState: red("TERMINATED")},
+		{Name: "all-purpose-shared", Id: "0123-456789-abcdef07", access: "Shared", runtimeName: "DBR 15.0", renderedState: green("RUNNING")},
+		{Name: "legacy-data-eng", Id: "0123-456789-abcdef08", access: "Assigned", runtimeName: "DBR 12.2 LTS", renderedState: red("TERMINATED")},
+	}
+}
+
+// profileItem mirrors the profile picker in cmd/auth/token.go: regular items
+// have a Host, the trailing meta items do not (so the {{if .Host}} branch fires).
+type profileItem struct {
+	Name string
+	Host string
+}
+
+// buildProfileItems returns 6 profile-shaped items across AWS / Azure / GCP
+// hosts plus the two trailing meta-rows ("Create a new profile", "Enter a
+// host URL manually") used by the real profile picker.
+func buildProfileItems() []profileItem {
+	return []profileItem{
+		{Name: "DEFAULT", Host: "https://acme.cloud.databricks.com"},
+		{Name: "production", Host: "https://acme-prod.cloud.databricks.com"},
+		{Name: "staging", Host: "https://acme-stg.cloud.databricks.com"},
+		{Name: "field-eng", Host: "https://field-eng.cloud.databricks.com"},
+		{Name: "azure-personal", Host: "https://adb-1234567890123456.78.azuredatabricks.net"},
+		{Name: "gcp-sandbox", Host: "https://1234567890123456.7.gcp.databricks.com"},
+		{Name: "Create a new profile"},
+		{Name: "Enter a host URL manually"},
+	}
+}

cmd/selftest/tui/prompt.go

@@ -0,0 +1,70 @@
+package tui
+
+import (
+	"errors"
+	"fmt"
+	"strings"
+
+	"github.com/databricks/cli/libs/cmdio"
+	"github.com/spf13/cobra"
+)
+
+func newPromptCmd() *cobra.Command {
+	var (
+		defaultVal string
+		mask       bool
+		validate   bool
+	)
+	cmd := &cobra.Command{
+		Use:   "prompt",
+		Short: "cmdio.RunPrompt (single-line text input)",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			opts := cmdio.PromptOptions{
+				Label:   "Enter a value",
+				Default: defaultVal,
+			}
+			if mask {
+				opts.Mask = '*'
+			}
+			if validate {
+				opts.Validate = func(input string) error {
+					if !strings.Contains(input, "://") {
+						return errors.New("value must contain '://'")
+					}
+					return nil
+				}
+			}
+			value, err := cmdio.RunPrompt(ctx, opts)
+			if err != nil {
+				return err
+			}
+			if mask {
+				cmdio.LogString(ctx, fmt.Sprintf("Entered %d characters", len(value)))
+				return nil
+			}
+			cmdio.LogString(ctx, "Entered: "+value)
+			return nil
+		},
+	}
+	cmd.Flags().StringVar(&defaultVal, "default", "", "pre-fill the input with this value")
+	cmd.Flags().BoolVar(&mask, "mask", false, "echo input as '*'")
+	cmd.Flags().BoolVar(&validate, "validate", false, "require '://' in input")
+	return cmd
+}
+
+func newSecretCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "secret",
+		Short: "cmdio.Secret (masked password input)",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			value, err := cmdio.Secret(ctx, "Personal access token")
+			if err != nil {
+				return err
+			}
+			cmdio.LogString(ctx, fmt.Sprintf("Entered %d characters", len(value)))
+			return nil
+		},
+	}
+}