List postgres databases during Lakebase plugin selection

Author: pkosiec

libs/apps/prompt/listers.go

@@ -397,6 +397,28 @@ func ListPostgresBranches(ctx context.Context, projectName string) ([]ListItem,
 	return out, nil
 }
 
+// ListPostgresDatabases returns databases within a Lakebase Autoscaling branch as selectable items.
+func ListPostgresDatabases(ctx context.Context, branchName string) ([]ListItem, error) {
+	w, err := workspaceClient(ctx)
+	if err != nil {
+		return nil, err
+	}
+	iter := w.Postgres.ListDatabases(ctx, postgres.ListDatabasesRequest{Parent: branchName})
+	databases, err := listing.ToSlice(ctx, iter)
+	if err != nil {
+		return nil, err
+	}
+	out := make([]ListItem, 0, len(databases))
+	for _, db := range databases {
+		label := extractIDFromName(db.Name, "databases")
+		if db.Status != nil && db.Status.PostgresDatabase != "" {
+			label = db.Status.PostgresDatabase
+		}
+		out = append(out, ListItem{ID: db.Name, Label: label})
+	}
+	return out, nil
+}
+
 // ListGenieSpaces returns Genie spaces as selectable items.
 func ListGenieSpaces(ctx context.Context) ([]ListItem, error) {
 	w, err := workspaceClient(ctx)

libs/apps/prompt/prompt.go

@@ -562,22 +562,23 @@ func PromptForPostgres(ctx context.Context, r manifest.Resource, required bool)
 		return nil, nil
 	}
 
-	// Step 3: enter a database name (pre-filled with default)
-	dbName := "databricks_postgres"
-	theme := AppkitTheme()
-	err = huh.NewInput().
-		Title("Database name").
-		Description("Enter the database name to connect to").
-		Value(&dbName).
-		WithTheme(theme).
-		Run()
+	// Step 3: pick a database within the branch
+	var databases []ListItem
+	err = RunWithSpinnerCtx(ctx, "Fetching databases...", func() error {
+		var fetchErr error
+		databases, fetchErr = ListPostgresDatabases(ctx, branchName)
+		return fetchErr
+	})
+	if err != nil {
+		return nil, err
+	}
+	dbName, err := PromptFromList(ctx, "Select Database", "no databases found in branch "+branchName, databases, required)
 	if err != nil {
 		return nil, err
 	}
 	if dbName == "" {
 		return nil, nil
 	}
-	printAnswered(ctx, "Database", dbName)
 
 	return map[string]string{
 		r.Key() + ".branch":   branchName,