feat: workspace bash background parameter

hugodutka · hugodutka · commit da40c5103929 · 2025-07-24T13:00:33.000Z
diff --git a/codersdk/toolsdk/bash.go b/codersdk/toolsdk/bash.go
@@ -2,6 +2,8 @@ package toolsdk
 
 import (
 	"context"
+	_ "embed"
+	"encoding/base64"
 	"errors"
 	"fmt"
 	"io"
@@ -15,18 +17,23 @@ import (
 	"github.com/coder/coder/v2/cli/cliui"
 	"github.com/coder/coder/v2/codersdk"
 	"github.com/coder/coder/v2/codersdk/workspacesdk"
+	"github.com/coder/coder/v2/cryptorand"
 )
 
 type WorkspaceBashArgs struct {
-	Workspace string `json:"workspace"`
-	Command   string `json:"command"`
+	Workspace  string `json:"workspace"`
+	Command    string `json:"command"`
+	Background bool   `json:"background"`
 }
 
 type WorkspaceBashResult struct {
 	Output   string `json:"output"`
 	ExitCode int    `json:"exit_code"`
 }
 
+//go:embed resources/background.sh
+var backgroundScript string
+
 var WorkspaceBash = Tool[WorkspaceBashArgs, WorkspaceBashResult]{
 	Tool: aisdk.Tool{
 		Name: ToolNameWorkspaceBash,
@@ -45,6 +52,7 @@ The workspace parameter supports various formats:
 
 Examples:
 - workspace: "my-workspace", command: "ls -la"
+- workspace: "my-workspace", command: "npm run dev", background: true
 - workspace: "john/dev-env", command: "git status"
 - workspace: "my-workspace.main", command: "docker ps"`,
 		Schema: aisdk.Schema{
@@ -57,6 +65,10 @@ Examples:
 					"type":        "string",
 					"description": "The bash command to execute in the workspace.",
 				},
+				"background": map[string]any{
+					"type":        "boolean",
+					"description": "Whether to run the command in the background.",
+				},
 			},
 			Required: []string{"workspace", "command"},
 		},
@@ -119,8 +131,24 @@ Examples:
 		}
 		defer session.Close()
 
+		command := args.Command
+		if args.Background {
+			encodedCommand := base64.StdEncoding.EncodeToString([]byte(args.Command))
+			encodedScript := base64.StdEncoding.EncodeToString([]byte(backgroundScript))
+			commandID, err := cryptorand.StringCharset(cryptorand.Human, 8)
+			if err != nil {
+				return WorkspaceBashResult{}, xerrors.Errorf("failed to generate command ID: %w", err)
+			}
+			command = fmt.Sprintf(
+				"ARG_COMMAND=\"$(echo -n %s | base64 -d)\" ARG_COMMAND_ID=%s bash -c \"$(echo -n %s | base64 -d)\"",
+				encodedCommand,
+				commandID,
+				encodedScript,
+			)
+		}
+
 		// Execute command and capture output
-		output, err := session.CombinedOutput(args.Command)
+		output, err := session.CombinedOutput(command)
 		outputStr := strings.TrimSpace(string(output))
 
 		if err != nil {
diff --git a/codersdk/toolsdk/resources/background.sh b/codersdk/toolsdk/resources/background.sh
@@ -0,0 +1,23 @@
+#!/bin/bash
+
+# This script is used to run a command in the background.
+
+set -o errexit
+set -o pipefail
+
+set -o nounset
+
+COMMAND="$ARG_COMMAND"
+COMMAND_ID="$ARG_COMMAND_ID"
+
+set +o nounset
+
+LOG_DIR="/tmp/mcp-bg"
+LOG_PATH="$LOG_DIR/$COMMAND_ID.log"
+mkdir -p "$LOG_DIR"
+
+nohup bash -c "$COMMAND" > "$LOG_PATH" 2>&1 &
+COMMAND_PID="$!"
+
+echo "Command started with PID: $COMMAND_PID"
+echo "Log path: $LOG_PATH"
