feat: add workspace command

Author: yjp20

README.md

@@ -30,3 +30,23 @@ stainless-api-cli builds create \
 ```
 
 For details about specific commands, use the `--help` flag.
+
+## Workspace Configuration
+
+The CLI supports workspace configuration to avoid repeatedly specifying the project name. When you run a command, the CLI will:
+
+1. Check if a project name is provided via command-line flag
+2. If not, look for a `stainless-workspace.json` file in the current directory or any parent directory
+3. Use the project name from the workspace configuration if found
+
+### Initializing a Workspace
+
+You can initialize a workspace configuration with:
+
+```sh
+stainless-api-cli init --project-name your-project-name
+```
+
+If you don't provide the `--project-name` flag, you'll be prompted to enter a project name.
+
+Additionally, when you run a command with a project name flag in an interactive terminal, the CLI will offer to initialize a workspace configuration for you.

auth.go

@@ -58,7 +58,7 @@ func handleAuthLogin(ctx context.Context, cmd *cli.Command) error {
 	if err := SaveAuthConfig(config); err != nil {
 		return fmt.Errorf("%s", au.Red(fmt.Sprintf("Failed to save authentication: %v", err)))
 	}
-	fmt.Printf("%s %s\n", au.BrightGreen("✓"), "Authentication successful! Your credentials have been saved.")
+	fmt.Printf("%s %s\n", au.BrightGreen("✱"), "Authentication successful! Your credentials have been saved.")
 	return nil
 }
 
@@ -142,14 +142,14 @@ func handleAuthLogout(ctx context.Context, cmd *cli.Command) error {
 		return fmt.Errorf("failed to remove auth file: %v", err)
 	}
 
-	fmt.Printf("%s %s\n", au.BrightGreen("✓"), "Successfully logged out.")
+	fmt.Printf("%s %s\n", au.BrightGreen("✱"), "Successfully logged out.")
 	return nil
 }
 
 func handleAuthStatus(ctx context.Context, cmd *cli.Command) error {
 	// Check for API key in environment variables first
 	if apiKey := os.Getenv("STAINLESS_API_KEY"); apiKey != "" {
-		fmt.Printf("%s %s\n", au.BrightGreen("✓"), "Authenticated via STAINLESS_API_KEY environment variable")
+		fmt.Printf("%s %s\n", au.BrightGreen("✱"), "Authenticated via STAINLESS_API_KEY environment variable")
 		return nil
 	}
 
@@ -160,12 +160,12 @@ func handleAuthStatus(ctx context.Context, cmd *cli.Command) error {
 	}
 
 	if config == nil {
-		fmt.Printf("%s %s\n", au.BrightYellow("!"), "Not logged in.")
+		fmt.Printf("%s %s\n", au.BrightYellow("✱"), "Not logged in.")
 		return nil
 	}
 
 	// If we have a config file with a token
-	fmt.Printf("%s %s\n", au.BrightGreen("✓"), "Authenticated via saved credentials")
+	fmt.Printf("%s %s\n", au.BrightGreen("✱"), "Authenticated via saved credentials")
 
 	// Show a truncated version of the token for verification
 	if len(config.AccessToken) > 10 {
@@ -318,5 +318,11 @@ func getClientOptions(ctx context.Context, cmd *cli.Command) []option.RequestOpt
 		options = append(options, option.WithAPIKey(config.AccessToken))
 	}
 
+	// Add default project from workspace config if available
+	projectName := GetProjectNameFromConfig()
+	if projectName != "" {
+		options = append(options, option.WithProject(projectName))
+	}
+
 	return options
 }

build.go

@@ -3,6 +3,7 @@
 package main
 
 import (
+	"bytes"
 	"context"
 	"fmt"
 	"io"
@@ -117,7 +118,8 @@ var buildsCreate = cli.Command{
 			Action: getAPIFlagAction[bool]("body", "allow_empty"),
 		},
 		&cli.BoolFlag{
-			Name: "wait",
+			Name:  "wait",
+			Value: true,
 		},
 		&cli.BoolFlag{
 			Name:  "pull",
@@ -236,24 +238,23 @@ var buildsCompare = cli.Command{
 
 func handleBuildsCreate(ctx context.Context, cmd *cli.Command) error {
 	cc := getAPICommandContext(ctx, cmd)
-	// Log to stderr that we're creating a build (using white text)
-	fmt.Fprintf(os.Stderr, "Creating build...\n")
+	// Log to stderr that we're creating a build
+	fmt.Fprintf(os.Stderr, "%s Creating build...\n", au.BrightCyan("✱"))
 	params := stainlessv0.BuildNewParams{}
 	res, err := cc.client.Builds.New(
 		context.TODO(),
 		params,
 		option.WithMiddleware(cc.AsMiddleware()),
-		option.WithRequestBody("application/json", cc.body),
 	)
 	if err != nil {
 		return err
 	}
 
-	// Print the build ID to stderr (using white text)
-	fmt.Fprintf(os.Stderr, "Build created: %s\n", res.ID)
+	// Print the build ID to stderr
+	fmt.Fprintf(os.Stderr, "  %s Build created: %s\n", au.BrightGreen("•"), au.Bold(res.ID))
 
 	if cmd.Bool("wait") {
-		fmt.Fprintf(os.Stderr, "Waiting for build to complete...\n")
+		fmt.Fprintf(os.Stderr, "%s Waiting for build to complete...\n", au.BrightCyan("✱"))
 
 		ticker := time.NewTicker(3 * time.Second)
 		defer ticker.Stop()
@@ -289,14 +290,15 @@ func handleBuildsCreate(ctx context.Context, cmd *cli.Command) error {
 
 						// Only print completed statuses with a green checkmark
 						if isTargetCompleted(target.status) {
-							fmt.Fprintf(os.Stderr, "%s Target %s: %s\n",
-								au.BrightGreen("✓"),
+							fmt.Fprintf(os.Stderr, "  %s Target %s: %s\n",
+								au.BrightGreen("•"),
 								target.name,
 								string(target.status))
 							anyCompleted = true
 						} else if target.status == "failed" {
 							// For failures, use red text
-							fmt.Fprintf(os.Stderr, "Target %s: %s\n",
+							fmt.Fprintf(os.Stderr, "  %s Target %s: %s\n",
+								au.BrightRed("•"),
 								target.name,
 								au.BrightRed(string(target.status)))
 						}
@@ -310,7 +312,7 @@ func handleBuildsCreate(ctx context.Context, cmd *cli.Command) error {
 
 				if (allCompleted || anyCompleted) && len(targets) > 0 {
 					if allCompleted {
-						fmt.Fprintf(os.Stderr, "%s Build completed successfully\n", au.BrightGreen("✓"))
+						fmt.Fprintf(os.Stderr, "  %s Build completed successfully\n", au.BrightGreen("✱"))
 						break loop
 					}
 				}
@@ -321,11 +323,11 @@ func handleBuildsCreate(ctx context.Context, cmd *cli.Command) error {
 		}
 
 		if cmd.Bool("pull") {
-			fmt.Fprintf(os.Stderr, "Pulling build outputs...\n")
+			fmt.Fprintf(os.Stderr, "%s Pulling build outputs...\n", au.BrightCyan("✱"))
 			if err := pullBuildOutputs(context.TODO(), cc.client, *res); err != nil {
-				fmt.Fprintf(os.Stderr, "%s Failed to pull outputs: %v\n", au.BrightRed("!"), err)
+				fmt.Fprintf(os.Stderr, "%s Failed to pull outputs: %v\n", au.BrightRed("✱"), err)
 			} else {
-				fmt.Fprintf(os.Stderr, "%s Successfully pulled all outputs\n", au.BrightGreen("✓"))
+				fmt.Fprintf(os.Stderr, "%s Successfully pulled all outputs\n", au.BrightGreen("✱"))
 			}
 		}
 	}
@@ -367,11 +369,12 @@ func pullBuildOutputs(ctx context.Context, client stainlessv0.Client, res stainl
 		return fmt.Errorf("no completed targets found in build %s", res.ID)
 	}
 
-	fmt.Fprintf(os.Stderr, "Found completed targets: %v\n", targets)
-
 	// Pull each target
-	for _, target := range targets {
-		fmt.Fprintf(os.Stderr, "Pulling output for target: %s\n", target)
+	for i, target := range targets {
+		targetDir := fmt.Sprintf("%s-%s", res.Project, target)
+
+		fmt.Fprintf(os.Stderr, "%s [%d/%d] Pulling %s → %s\n",
+			au.BrightCyan("✱"), i+1, len(targets), au.Bold(target), au.Cyan(targetDir))
 
 		// Get the output details
 		outputRes, err := client.BuildTargetOutputs.Get(
@@ -384,57 +387,72 @@ func pullBuildOutputs(ctx context.Context, client stainlessv0.Client, res stainl
 			},
 		)
 		if err != nil {
-			fmt.Fprintf(os.Stderr, "%s Error getting output for target %s: %v\n", au.BrightRed("!"), target, err)
+			fmt.Fprintf(os.Stderr, "%s Failed to get output details for %s: %v\n",
+				au.BrightRed("✱"), target, err)
 			continue
 		}
 
-		targetDir := fmt.Sprintf("%s-sdk", target)
-
 		// Handle based on output type
-		err = pullOutput(outputRes.Output, outputRes.URL, outputRes.Ref, targetDir)
+		err = pullOutput(outputRes.Output, outputRes.URL, outputRes.Ref, targetDir, target)
 		if err != nil {
-			fmt.Fprintf(os.Stderr, "%s Error pulling output for target %s: %v\n", au.BrightRed("!"), target, err)
+			fmt.Fprintf(os.Stderr, "%s Failed to pull %s: %v\n",
+				au.BrightRed("✱"), target, err)
 			continue
-		} else {
-			fmt.Fprintf(os.Stderr, "%s Successfully pulled output for target %s\n", au.BrightGreen("✓"), target)
+		}
+
+		fmt.Fprintf(os.Stderr, "  %s Successfully pulled to %s\n",
+			au.BrightBlack("•"), au.Cyan(targetDir))
+
+		if i < len(targets)-1 {
+			fmt.Fprintf(os.Stderr, "\n")
 		}
 	}
 
 	return nil
 }
 
 // pullOutput handles downloading or cloning a build target output
-func pullOutput(output, url, ref, targetDir string) error {
-	// Create a directory for the output
+func pullOutput(output, url, ref, targetDir, target string) error {
+	// Remove existing directory if it exists
+	if _, err := os.Stat(targetDir); err == nil {
+		fmt.Fprintf(os.Stderr, "  %s Removing existing directory %s\n", au.BrightBlack("•"), targetDir)
+		if err := os.RemoveAll(targetDir); err != nil {
+			return fmt.Errorf("failed to remove existing directory %s: %v", targetDir, err)
+		}
+	}
+
+	// Create a fresh directory for the output
 	if err := os.MkdirAll(targetDir, 0755); err != nil {
 		return fmt.Errorf("failed to create directory %s: %v", targetDir, err)
 	}
 
 	switch output {
 	case "git":
 		// Clone the repository
-		fmt.Fprintf(os.Stderr, "Cloning repository %s (ref: %s) to %s\n", url, ref, targetDir)
+		fmt.Fprintf(os.Stderr, "  %s Cloning repository\n", au.BrightBlack("•"))
+		fmt.Fprintf(os.Stderr, "  %s Checking out ref %s\n", au.BrightBlack("•"), au.Bold(ref))
 
 		cmd := exec.Command("git", "clone", url, targetDir)
-		cmd.Stdout = os.Stderr
-		cmd.Stderr = os.Stderr
+		var stderr bytes.Buffer
+		cmd.Stdout = nil // Suppress git clone output
+		cmd.Stderr = &stderr
 		if err := cmd.Run(); err != nil {
-			return fmt.Errorf("git clone failed: %v", err)
+			return fmt.Errorf("git clone failed: %v\nGit error: %s", err, stderr.String())
 		}
 
 		// Checkout the specific ref
 		cmd = exec.Command("git", "-C", targetDir, "checkout", ref)
-		cmd.Stdout = os.Stderr
-		cmd.Stderr = os.Stderr
+		stderr.Reset()
+		cmd.Stdout = nil // Suppress git checkout output
+		cmd.Stderr = &stderr
 		if err := cmd.Run(); err != nil {
-			return fmt.Errorf("git checkout failed: %v", err)
+			return fmt.Errorf("git checkout failed: %v\nGit error: %s", err, stderr.String())
 		}
 
-		fmt.Fprintf(os.Stderr, "%s Successfully cloned repository to %s\n", au.BrightGreen("✓"), targetDir)
-
 	case "url":
 		// Download the tar file
-		fmt.Fprintf(os.Stderr, "Downloading from %s to %s\n", url, targetDir)
+		fmt.Fprintf(os.Stderr, "  %s Downloading archive %s\n", au.BrightBlack("•"), au.Underline(url))
+		fmt.Fprintf(os.Stderr, "  %s Extracting to %s\n", au.BrightBlack("•"), targetDir)
 
 		// Create a temporary file for the tar download
 		tmpFile, err := os.CreateTemp("", "stainless-*.tar.gz")
@@ -464,14 +482,12 @@ func pullOutput(output, url, ref, targetDir string) error {
 
 		// Extract the tar file
 		cmd := exec.Command("tar", "-xzf", tmpFile.Name(), "-C", targetDir)
-		cmd.Stdout = os.Stderr
-		cmd.Stderr = os.Stderr
+		cmd.Stdout = nil // Suppress tar output
+		cmd.Stderr = nil
 		if err := cmd.Run(); err != nil {
 			return fmt.Errorf("tar extraction failed: %v", err)
 		}
 
-		fmt.Fprintf(os.Stderr, "%s Successfully downloaded and extracted to %s\n", au.BrightGreen("✓"), targetDir)
-
 	default:
 		return fmt.Errorf("unsupported output type: %s. Supported types are 'git' and 'url'", output)
 	}
@@ -502,7 +518,6 @@ func handleBuildsCompare(ctx context.Context, cmd *cli.Command) error {
 		context.TODO(),
 		params,
 		option.WithMiddleware(cc.AsMiddleware()),
-		option.WithRequestBody("application/json", cc.body),
 	)
 	if err != nil {
 		return err

buildtargetoutput.go

@@ -95,5 +95,5 @@ func handleBuildTargetOutputsPull(ctx context.Context, cmd *cli.Command) error {
 	targetDir := fmt.Sprintf("%s-%s", "tmp", "target")
 
 	// Use the shared pullOutput function
-	return pullOutput(res.Output, res.URL, res.Ref, targetDir)
+	return pullOutput(res.Output, res.URL, res.Ref, targetDir, "target")
 }

go.mod

@@ -5,9 +5,12 @@ go 1.23.0
 toolchain go1.23.9
 
 require (
-	github.com/stainless-api/stainless-api-go v0.5.1
+	github.com/charmbracelet/bubbles v0.21.0
+	github.com/charmbracelet/huh v0.7.0
+	github.com/charmbracelet/lipgloss v1.1.0
 	github.com/logrusorgru/aurora/v4 v4.0.0
 	github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c
+	github.com/stainless-api/stainless-api-go v0.5.1
 	github.com/tidwall/gjson v1.17.0
 	github.com/tidwall/pretty v1.2.1
 	github.com/tidwall/sjson v1.2.5
@@ -16,6 +19,29 @@ require (
 )
 
 require (
+	github.com/atotto/clipboard v0.1.4 // indirect
+	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
+	github.com/catppuccin/go v0.3.0 // indirect
+	github.com/charmbracelet/bubbletea v1.3.4 // indirect
+	github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc // indirect
+	github.com/charmbracelet/x/ansi v0.8.0 // indirect
+	github.com/charmbracelet/x/cellbuf v0.0.13 // indirect
+	github.com/charmbracelet/x/exp/strings v0.0.0-20240722160745-212f7b056ed0 // indirect
+	github.com/charmbracelet/x/term v0.2.1 // indirect
+	github.com/dustin/go-humanize v1.0.1 // indirect
+	github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
+	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/mattn/go-localereader v0.0.1 // indirect
+	github.com/mattn/go-runewidth v0.0.16 // indirect
+	github.com/mitchellh/hashstructure/v2 v2.0.2 // indirect
+	github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
+	github.com/muesli/cancelreader v0.2.2 // indirect
+	github.com/muesli/termenv v0.16.0 // indirect
+	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/tidwall/match v1.1.1 // indirect
+	github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
+	golang.org/x/sync v0.12.0 // indirect
 	golang.org/x/sys v0.33.0 // indirect
+	golang.org/x/text v0.23.0 // indirect
 )