Adding a session-sharing-url param (#46)

Author: ianhodge

internal/worker/worker.go

@@ -37,6 +37,8 @@ type Config struct {
 	// IdleOnComplete is passed to the oz CLI's --idle-on-complete flag for every task.
 	// Empty string means use the oz CLI default (45m). Use "0s" to disable idle.
 	IdleOnComplete string
+	// SessionSharingServerURL, when non-empty, is forwarded to the oz CLI via --session-sharing-server-url.
+	SessionSharingServerURL string
 
 	// Backend-specific configs. Only the one matching BackendType should be set.
 	Docker     *DockerBackendConfig
@@ -376,8 +378,11 @@ func (w *Worker) prepareTaskParams(assignment *types.TaskAssignmentMessage) *Tas
 	baseArgs = common.AugmentArgsForTask(task, baseArgs, common.TaskAugmentOptions{
 		IdleOnComplete: w.config.IdleOnComplete,
 	})
+	if w.config.SessionSharingServerURL != "" {
+		baseArgs = append(baseArgs, "--session-sharing-server-url", w.config.SessionSharingServerURL)
+	}
 
-	// Build a unified sidecar list: the Warp agent sidecar (mounted at /agent, where
+	// Build a unified sidecar list:
 	// entrypoint.sh lives) comes first, followed by any additional sidecars.
 	var sidecars []types.SidecarMount
 	if assignment.SidecarImage != "" {

main.go

@@ -20,19 +20,20 @@ import (
 )
 
 var CLI struct {
-	ConfigFile         string   `help:"Path to YAML config file" type:"path"`
-	Backend            string   `help:"Backend type (docker, direct, or kubernetes)" enum:"docker,direct,kubernetes," default:""`
-	APIKey             string   `help:"API key for authentication" env:"WARP_API_KEY" required:""`
-	WorkerID           string   `help:"Worker host identifier (required via flag or config file)"`
-	WebSocketURL       string   `default:"wss://oz.warp.dev/api/v1/selfhosted/worker/ws" hidden:""`
-	ServerRootURL      string   `default:"https://app.warp.dev" hidden:""`
-	LogLevel           string   `help:"Log level (debug, info, warn, error)" default:"info" enum:"debug,info,warn,error"`
-	TargetDir          string   `help:"Run all tasks in this directory instead of creating per-task workspaces (direct backend only)"`
-	NoCleanup          bool     `help:"Do not remove containers after execution (for debugging)"`
-	Volumes            []string `help:"Volume mounts for task containers (format: HOST_PATH:CONTAINER_PATH or HOST_PATH:CONTAINER_PATH:MODE)" short:"v"`
-	Env                []string `help:"Environment variables for task containers (format: KEY=VALUE or KEY to pass through from host)" short:"e"`
-	MaxConcurrentTasks int      `help:"Maximum number of tasks to run concurrently (0 for unlimited)" default:"0"`
-	IdleOnComplete     string   `help:"How long to keep the oz agent alive after a task completes, for follow-ups (e.g. 45m, 10m, 0s). Defaults to 45m when not set."`
+	ConfigFile              string   `help:"Path to YAML config file" type:"path"`
+	Backend                 string   `help:"Backend type (docker, direct, or kubernetes)" enum:"docker,direct,kubernetes," default:""`
+	APIKey                  string   `help:"API key for authentication" env:"WARP_API_KEY" required:""`
+	WorkerID                string   `help:"Worker host identifier (required via flag or config file)"`
+	WebSocketURL            string   `default:"wss://oz.warp.dev/api/v1/selfhosted/worker/ws" hidden:""`
+	ServerRootURL           string   `default:"https://app.warp.dev" hidden:""`
+	LogLevel                string   `help:"Log level (debug, info, warn, error)" default:"info" enum:"debug,info,warn,error"`
+	TargetDir               string   `help:"Run all tasks in this directory instead of creating per-task workspaces (direct backend only)"`
+	NoCleanup               bool     `help:"Do not remove containers after execution (for debugging)"`
+	Volumes                 []string `help:"Volume mounts for task containers (format: HOST_PATH:CONTAINER_PATH or HOST_PATH:CONTAINER_PATH:MODE)" short:"v"`
+	Env                     []string `help:"Environment variables for task containers (format: KEY=VALUE or KEY to pass through from host)" short:"e"`
+	MaxConcurrentTasks      int      `help:"Maximum number of tasks to run concurrently (0 for unlimited)" default:"0"`
+	IdleOnComplete          string   `help:"How long to keep the oz agent alive after a task completes, for follow-ups (e.g. 45m, 10m, 0s). Defaults to 45m when not set."`
+	SessionSharingServerURL string   `help:"Session sharing server WebSocket URL to pass through to the oz CLI (e.g. ws://127.0.0.1:8081)" hidden:""`
 }
 
 func main() {
@@ -139,14 +140,15 @@ func mergeConfig(fileConfig *config.FileConfig) (worker.Config, error) {
 	}
 
 	wc := worker.Config{
-		APIKey:             CLI.APIKey,
-		WorkerID:           workerID,
-		WebSocketURL:       CLI.WebSocketURL,
-		ServerRootURL:      CLI.ServerRootURL,
-		LogLevel:           CLI.LogLevel,
-		BackendType:        backendType,
-		MaxConcurrentTasks: maxConcurrentTasks,
-		IdleOnComplete:     idleOnComplete,
+		APIKey:                  CLI.APIKey,
+		WorkerID:                workerID,
+		WebSocketURL:            CLI.WebSocketURL,
+		ServerRootURL:           CLI.ServerRootURL,
+		LogLevel:                CLI.LogLevel,
+		BackendType:             backendType,
+		MaxConcurrentTasks:      maxConcurrentTasks,
+		IdleOnComplete:          idleOnComplete,
+		SessionSharingServerURL: CLI.SessionSharingServerURL,
 	}
 
 	switch backendType {