feat: enhance prompt management with column selection and JSON output

This commit introduces enhancements to the prompt management commands, allowing users to select specific columns for display and output results in JSON format.

The following changes were made:

- Added column selection functionality to `src prompts list` and `src prompts tags list` commands using the `-c` flag. Users can now specify a comma-separated list of columns to display.
- Implemented JSON output functionality for `src prompts list` and `src prompts tags list` commands using the `-json` flag.
- Updated `src prompts update`, `src prompts tags update`, `src prompts delete`, and `src prompts tags delete` commands to accept the ID as a positional argument instead of a flag.
- Updated `src prompts export` to fetch all tags with pagination.

Author: trly

cmd/src/prompts.go

@@ -16,14 +16,14 @@ Usage:
 
 The commands are:
 
-	list		lists prompts
-	get		    get a prompt by ID
-	create		create a prompt
-	update		update a prompt
-	delete		delete a prompt
-	export		export prompts to a JSON file
-	import		import prompts from a JSON file
-	tags		manage prompt tags (use "src prompts tags [command] -h" for more info)
+	list        lists prompts
+	get         get a prompt by ID
+	create      create a prompt
+	update      update a prompt
+	delete      delete a prompt
+	export      export prompts to a JSON file
+	import      import prompts from a JSON file
+	tags        manage prompt tags (use "src prompts tags [command] -h" for more info)
 
 Use "src prompts [command] -h" for more information about a command.
 `

cmd/src/prompts_delete.go

@@ -16,7 +16,7 @@ Examples:
 
   Delete a prompt by ID:
 
-    	$ src prompts delete -id=<prompt-id>
+    	$ src prompts delete <prompt-id>
 
 `
 
@@ -27,7 +27,6 @@ Examples:
 		fmt.Println(usage)
 	}
 	var (
-		idFlag   = flagSet.String("id", "", "The ID of the prompt to delete")
 		apiFlags = api.NewFlags(flagSet)
 	)
 
@@ -36,9 +35,11 @@ Examples:
 			return err
 		}
 
-		if *idFlag == "" {
-			return errors.New("provide the ID of the prompt to delete")
+		// Check for prompt ID as positional argument
+		if len(flagSet.Args()) != 1 {
+			return errors.New("provide exactly one prompt ID as an argument")
 		}
+		promptID := flagSet.Arg(0)
 
 		client := cfg.apiClient(apiFlags, flagSet.Output())
 
@@ -56,7 +57,7 @@ Examples:
 		}
 
 		if ok, err := client.NewRequest(query, map[string]interface{}{
-			"id": *idFlag,
+			"id": promptID,
 		}).Do(context.Background(), &result); err != nil || !ok {
 			return err
 		}

cmd/src/prompts_export.go

@@ -225,9 +225,9 @@ func fetchAllPrompts(client api.Client, tagIDs []string) ([]Prompt, error) {
 
 // resolveTagNamesToIDs resolves tag names to their IDs
 func resolveTagNamesToIDs(client api.Client, tagNames []string) ([]string, error) {
-	// Query to get all tags
-	query := `query PromptTags {
-	promptTags {
+	// Query to get all tags with required pagination parameter
+	query := `query PromptTags($first: Int!) {
+	promptTags(first: $first) {
 		nodes {
 			...PromptTagFields
 		}
@@ -240,7 +240,11 @@ func resolveTagNamesToIDs(client api.Client, tagNames []string) ([]string, error
 		}
 	}
 
-	if ok, err := client.NewRequest(query, nil).Do(context.Background(), &result); err != nil || !ok {
+	vars := map[string]interface{}{
+		"first": 1000, // Get enough tags to cover all possible names
+	}
+
+	if ok, err := client.NewRequest(query, vars).Do(context.Background(), &result); err != nil || !ok {
 		return nil, err
 	}
 

cmd/src/prompts_list.go

@@ -9,6 +9,126 @@ import (
 	"github.com/sourcegraph/src-cli/internal/api"
 )
 
+// availablePromptColumns defines the available column names for output
+var availablePromptColumns = map[string]bool{
+	"id":          true,
+	"name":        true,
+	"description": true,
+	"draft":       true,
+	"visibility":  true,
+	"mode":        true,
+	"tags":        true,
+}
+
+// defaultPromptColumns defines the default columns to display
+var defaultPromptColumns = []string{"id", "name", "visibility", "tags"}
+
+// displayPrompts formats and outputs multiple prompts
+func displayPrompts(prompts []Prompt, columns []string, asJSON bool) error {
+	if asJSON {
+		return outputAsJSON(prompts)
+	}
+
+	// Collect all data first to calculate column widths
+	allRows := make([][]string, 0, len(prompts)+1)
+
+	// Add header row
+	headers := make([]string, 0, len(columns))
+	for _, col := range columns {
+		headers = append(headers, strings.ToUpper(col))
+	}
+	allRows = append(allRows, headers)
+
+	// Collect all data rows
+	for _, p := range prompts {
+		row := make([]string, 0, len(columns))
+
+		// Prepare tag names for display
+		tagNames := []string{}
+		for _, tag := range p.Tags.Nodes {
+			tagNames = append(tagNames, tag.Name)
+		}
+		tagsStr := joinStrings(tagNames, ", ")
+
+		for _, col := range columns {
+			switch col {
+			case "id":
+				row = append(row, p.ID)
+			case "name":
+				row = append(row, p.Name)
+			case "description":
+				row = append(row, p.Description)
+			case "draft":
+				row = append(row, fmt.Sprintf("%t", p.Draft))
+			case "visibility":
+				row = append(row, p.Visibility)
+			case "mode":
+				row = append(row, p.Mode)
+			case "tags":
+				row = append(row, tagsStr)
+			}
+		}
+		allRows = append(allRows, row)
+	}
+
+	// Calculate max width for each column
+	colWidths := make([]int, len(columns))
+	for _, row := range allRows {
+		for i, cell := range row {
+			if len(cell) > colWidths[i] {
+				colWidths[i] = len(cell)
+			}
+		}
+	}
+
+	// Print all rows with proper padding
+	for i, row := range allRows {
+		for j, cell := range row {
+			fmt.Print(cell)
+			// Add padding (at least 2 spaces between columns)
+			padding := colWidths[j] - len(cell) + 2
+			fmt.Print(strings.Repeat(" ", padding))
+		}
+		fmt.Println()
+
+		// Add separator line after headers
+		if i == 0 {
+			for j, width := range colWidths {
+				fmt.Print(strings.Repeat("-", width))
+				if j < len(colWidths)-1 {
+					fmt.Print("  ")
+				}
+			}
+			fmt.Println()
+		}
+	}
+
+	return nil
+}
+
+// parsePromptColumns parses and validates the columns flag
+func parsePromptColumns(columnsFlag string) []string {
+	if columnsFlag == "" {
+		return defaultPromptColumns
+	}
+
+	columns := strings.Split(columnsFlag, ",")
+	var validColumns []string
+
+	for _, col := range columns {
+		col = strings.ToLower(strings.TrimSpace(col))
+		if availablePromptColumns[col] {
+			validColumns = append(validColumns, col)
+		}
+	}
+
+	if len(validColumns) == 0 {
+		return defaultPromptColumns
+	}
+
+	return validColumns
+}
+
 func init() {
 	usage := `
 Examples:
@@ -32,6 +152,14 @@ Examples:
   Paginate through results:
 
     	$ src prompts list -after=<cursor>
+    	
+  Select specific columns to display:
+  
+    	$ src prompts list -c id,name,visibility,tags
+    	
+  Output results as JSON:
+  
+    	$ src prompts list -json
 
 `
 
@@ -52,6 +180,8 @@ Examples:
 		includeBuiltinFlag  = flagSet.Bool("include-builtin", false, "Whether to include builtin prompts")
 		limitFlag           = flagSet.Int("limit", 100, "Maximum number of prompts to list")
 		afterFlag           = flagSet.String("after", "", "Cursor for pagination (from previous page's endCursor)")
+		columnsFlag         = flagSet.String("c", strings.Join(defaultPromptColumns, ","), "Comma-separated list of columns to display. Available: id,name,description,draft,visibility,mode,tags")
+		jsonFlag            = flagSet.Bool("json", false, "Output results as JSON for programmatic access")
 		apiFlags            = api.NewFlags(flagSet)
 	)
 
@@ -171,24 +301,18 @@ Examples:
 			return err
 		}
 
-		fmt.Printf("Showing %d of %d prompts\n\n", len(result.Prompts.Nodes), result.Prompts.TotalCount)
+		// Parse columns
+		columns := parsePromptColumns(*columnsFlag)
 
-		for _, p := range result.Prompts.Nodes {
-			tagNames := []string{}
-			for _, tag := range p.Tags.Nodes {
-				tagNames = append(tagNames, tag.Name)
-			}
+		fmt.Printf("Showing %d of %d prompts\n\n", len(result.Prompts.Nodes), result.Prompts.TotalCount)
 
-			fmt.Printf("ID: %s\nName: %s\nDescription: %s\n", p.ID, p.Name, p.Description)
-			fmt.Printf("Draft: %t | Visibility: %s | Mode: %s\n", p.Draft, p.Visibility, p.Mode)
-			if len(tagNames) > 0 {
-				fmt.Printf("Tags: %s\n", joinStrings(tagNames, ", "))
-			}
-			fmt.Println()
+		// Display prompts in tabular format
+		if err := displayPrompts(result.Prompts.Nodes, columns, *jsonFlag); err != nil {
+			return err
 		}
 
 		if result.Prompts.PageInfo.HasNextPage {
-			fmt.Printf("More results available. Use -after=%s to fetch the next page.\n", result.Prompts.PageInfo.EndCursor)
+			fmt.Printf("\nMore results available. Use -after=%s to fetch the next page.\n", result.Prompts.PageInfo.EndCursor)
 		}
 
 		return nil

cmd/src/prompts_tags_delete.go

@@ -16,7 +16,7 @@ Examples:
 
   Delete a prompt tag by ID:
 
-    	$ src prompts tags delete -id=<tag-id>
+    	$ src prompts tags delete <tag-id>
 
 `
 
@@ -27,7 +27,6 @@ Examples:
 		fmt.Println(usage)
 	}
 	var (
-		idFlag   = flagSet.String("id", "", "The ID of the tag to delete")
 		apiFlags = api.NewFlags(flagSet)
 	)
 
@@ -36,9 +35,11 @@ Examples:
 			return err
 		}
 
-		if *idFlag == "" {
-			return errors.New("provide the ID of the tag to delete")
+		// Check for tag ID as positional argument
+		if len(flagSet.Args()) != 1 {
+			return errors.New("provide exactly one tag ID as an argument")
 		}
+		tagID := flagSet.Arg(0)
 
 		client := cfg.apiClient(apiFlags, flagSet.Output())
 
@@ -56,7 +57,7 @@ Examples:
 		}
 
 		if ok, err := client.NewRequest(query, map[string]interface{}{
-			"id": *idFlag,
+			"id": tagID,
 		}).Do(context.Background(), &result); err != nil || !ok {
 			return err
 		}