src abc command

Author: bahrmichael

cmd/src/abc.go

@@ -0,0 +1,37 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+)
+
+var abcCommands commander
+
+func init() {
+	usage := `'src abc' is a tool that manages agentic batch changes on a Sourcegraph instance.
+
+Usage:
+
+	src abc command [command options]
+
+The commands are:
+
+	variables	manage workflow instance variables
+
+Use "src abc [command] -h" for more information about a command.
+`
+
+	flagSet := flag.NewFlagSet("abc", flag.ExitOnError)
+	handler := func(args []string) error {
+		abcCommands.run(flagSet, "src abc", usage, args)
+		return nil
+	}
+
+	commands = append(commands, &command{
+		flagSet: flagSet,
+		handler: handler,
+		usageFunc: func() {
+			fmt.Println(usage)
+		},
+	})
+}

cmd/src/abc_variables.go

@@ -0,0 +1,37 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+)
+
+var abcVariablesCommands commander
+
+func init() {
+	usage := `'src abc variables' is a tool that manages workflow instance variables on a Sourcegraph instance.
+
+Usage:
+
+	src abc variables command [command options]
+
+The commands are:
+
+	set	set or remove a workflow instance variable
+
+Use "src abc variables [command] -h" for more information about a command.
+`
+
+	flagSet := flag.NewFlagSet("variables", flag.ExitOnError)
+	handler := func(args []string) error {
+		abcVariablesCommands.run(flagSet, "src abc variables", usage, args)
+		return nil
+	}
+
+	abcCommands = append(abcCommands, &command{
+		flagSet: flagSet,
+		handler: handler,
+		usageFunc: func() {
+			fmt.Println(usage)
+		},
+	})
+}

cmd/src/abc_variables_set.go

@@ -0,0 +1,113 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"flag"
+	"fmt"
+
+	"github.com/sourcegraph/src-cli/internal/api"
+	"github.com/sourcegraph/src-cli/internal/cmderrors"
+)
+
+const updateABCWorkflowInstanceVariablesMutation = `mutation UpdateAgenticWorkflowInstanceVariables(
+	$instanceID: ID!,
+	$variables: [AgenticWorkflowInstanceVariableInput!]!,
+) {
+	updateAgenticWorkflowInstanceVariables(instanceID: $instanceID, variables: $variables) {
+		id
+	}
+}`
+
+func init() {
+	usage := `
+Examples:
+
+  Set a string variable on a workflow instance:
+
+    	$ src abc variables set QWdlbnRpY1dvcmtmbG93SW5zdGFuY2U6MQ== prompt "tighten the review criteria"
+
+  Remove a variable by setting it to null:
+
+    	$ src abc variables set QWdlbnRpY1dvcmtmbG93SW5zdGFuY2U6MQ== approval null
+
+  Set a structured JSON value:
+
+    	$ src abc variables set QWdlbnRpY1dvcmtmbG93SW5zdGFuY2U6MQ== checkpoints '[1,2,3]'
+
+Values are interpreted as JSON literals when valid. Otherwise they are sent as plain strings.
+	`
+
+	flagSet := flag.NewFlagSet("set", flag.ExitOnError)
+	usageFunc := func() {
+		fmt.Fprintf(flag.CommandLine.Output(), "Usage of 'src abc variables %s':\n", flagSet.Name())
+		flagSet.PrintDefaults()
+		fmt.Println(usage)
+	}
+	apiFlags := api.NewFlags(flagSet)
+
+	handler := func(args []string) error {
+		if err := flagSet.Parse(args); err != nil {
+			return err
+		}
+
+		if flagSet.NArg() != 3 {
+			return cmderrors.Usage("must provide an instance ID, variable name, and variable value")
+		}
+
+		instanceID := flagSet.Arg(0)
+		key := flagSet.Arg(1)
+		value, remove, err := marshalABCVariableValue(flagSet.Arg(2))
+		if err != nil {
+			return err
+		}
+
+		client := cfg.apiClient(apiFlags, flagSet.Output())
+		var result struct {
+			UpdateAgenticWorkflowInstanceVariables struct {
+				ID string `json:"id"`
+			} `json:"updateAgenticWorkflowInstanceVariables"`
+		}
+		if ok, err := client.NewRequest(updateABCWorkflowInstanceVariablesMutation, map[string]any{
+			"instanceID": instanceID,
+			"variables": []map[string]string{{
+				"key":   key,
+				"value": value,
+			}},
+		}).Do(context.Background(), &result); err != nil || !ok {
+			return err
+		}
+
+		if apiFlags.GetCurl() {
+			return nil
+		}
+
+		if remove {
+			fmt.Printf("Removed variable %q from workflow instance %q.\n", key, instanceID)
+			return nil
+		}
+
+		fmt.Printf("Set variable %q on workflow instance %q.\n", key, instanceID)
+		return nil
+	}
+
+	abcVariablesCommands = append(abcVariablesCommands, &command{
+		flagSet:   flagSet,
+		handler:   handler,
+		usageFunc: usageFunc,
+	})
+}
+
+func marshalABCVariableValue(raw string) (value string, remove bool, err error) {
+	var parsed any
+	if err := json.Unmarshal([]byte(raw), &parsed); err != nil {
+		parsed = raw
+	}
+
+	encoded, err := json.Marshal(parsed)
+	if err != nil {
+		return "", false, err
+	}
+
+	return string(encoded), parsed == nil, nil
+}

cmd/src/abc_variables_set_test.go

@@ -0,0 +1,64 @@
+package main
+
+import "testing"
+
+func TestMarshalABCVariableValue(t *testing.T) {
+	tests := []struct {
+		name       string
+		raw        string
+		wantValue  string
+		wantRemove bool
+	}{
+		{
+			name:       "plain string",
+			raw:        "hello",
+			wantValue:  `"hello"`,
+			wantRemove: false,
+		},
+		{
+			name:       "number literal",
+			raw:        "42",
+			wantValue:  `42`,
+			wantRemove: false,
+		},
+		{
+			name:       "boolean literal",
+			raw:        "true",
+			wantValue:  `true`,
+			wantRemove: false,
+		},
+		{
+			name:       "null removes variable",
+			raw:        "null",
+			wantValue:  `null`,
+			wantRemove: true,
+		},
+		{
+			name:       "quoted null stays a string",
+			raw:        "\"null\"",
+			wantValue:  "\"null\"",
+			wantRemove: false,
+		},
+		{
+			name:       "object literal",
+			raw:        "{\"retries\":3,\"notify\":true}",
+			wantValue:  "{\"notify\":true,\"retries\":3}",
+			wantRemove: false,
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			gotValue, gotRemove, err := marshalABCVariableValue(test.raw)
+			if err != nil {
+				t.Fatalf("marshalABCVariableValue returned error: %s", err)
+			}
+			if gotValue != test.wantValue {
+				t.Fatalf("value = %q, want %q", gotValue, test.wantValue)
+			}
+			if gotRemove != test.wantRemove {
+				t.Fatalf("remove = %v, want %v", gotRemove, test.wantRemove)
+			}
+		})
+	}
+}

cmd/src/main.go

@@ -50,6 +50,7 @@ The options are:
 
 The commands are:
 
+	abc             manages agentic batch changes
 	auth            authentication helper commands
 	api             interacts with the Sourcegraph GraphQL API
 	batch           manages batch changes