feat: session detection for all supported clients

Signed-off-by: Danny Kopping <danny@coder.com>

Author: dannykopping

session.go

@@ -49,6 +49,25 @@ func guessSessionID(client Client, r *http.Request) string {
 		return strings.TrimSpace(headers.Get("session_id"))
 	case ClientMux:
 		return strings.TrimSpace(headers.Get("X-Mux-Workspace-Id"))
+	case ClientZed:
+		return "" // Zed does not send a session ID from Zed Agent or Text Thread.
+	case ClientCopilotVSC:
+		// This does not map precisely to what we consider a session, but it's close enough.
+		// Most other providers' equivalent of this would persist for the duration of a
+		// conversation; it does seem to persist across an agentic loop though, which is
+		// all we really need.
+		//
+		// There's also `vscode-sessionid` but that's persistent for the duration of the
+		// VS Code window.
+		return strings.TrimSpace(headers.Get("x-interaction-id"))
+	case ClientCopilotCLI:
+		return strings.TrimSpace(headers.Get("X-Client-Session-Id"))
+	case ClientKilo:
+		return strings.TrimSpace(headers.Get("X-KILOCODE-TASKID"))
+	case ClientRoo:
+		return "" // RooCode doesn't send a session ID.
+	case ClientCursor:
+		return "" // Cursor is not currently supported.
 	default:
 		return ""
 	}

session_test.go

@@ -88,6 +88,66 @@ func TestGuessSessionID(t *testing.T) {
 			body:      `{"metadata":{"user_id":"user_abc_account_456_session_some-id"}}`,
 			sessionID: "",
 		},
+		// Mux.
+		{
+			name:      "mux_with_workspace_header",
+			client:    ClientMux,
+			headers:   map[string]string{"X-Mux-Workspace-Id": "ws-abc-123"},
+			sessionID: "ws-abc-123",
+		},
+		{
+			name:      "mux_without_workspace_header",
+			client:    ClientMux,
+			sessionID: "",
+		},
+		// Copilot VS Code.
+		{
+			name:      "copilot_vsc_with_interaction_id",
+			client:    ClientCopilotVSC,
+			headers:   map[string]string{"x-interaction-id": "interaction-xyz"},
+			sessionID: "interaction-xyz",
+		},
+		{
+			name:      "copilot_vsc_without_interaction_id",
+			client:    ClientCopilotVSC,
+			sessionID: "",
+		},
+		// Copilot CLI.
+		{
+			name:      "copilot_cli_with_session_header",
+			client:    ClientCopilotCLI,
+			headers:   map[string]string{"X-Client-Session-Id": "cli-sess-456"},
+			sessionID: "cli-sess-456",
+		},
+		{
+			name:      "copilot_cli_without_session_header",
+			client:    ClientCopilotCLI,
+			sessionID: "",
+		},
+		// Kilo.
+		{
+			name:      "kilo_with_task_id",
+			client:    ClientKilo,
+			headers:   map[string]string{"X-KILOCODE-TASKID": "task-789"},
+			sessionID: "task-789",
+		},
+		{
+			name:      "kilo_without_task_id",
+			client:    ClientKilo,
+			sessionID: "",
+		},
+		// Roo.
+		{
+			name:      "roo_returns_empty",
+			client:    ClientRoo,
+			sessionID: "",
+		},
+		// Cursor.
+		{
+			name:      "cursor_returns_empty",
+			client:    ClientCursor,
+			sessionID: "",
+		},
 	}
 
 	for _, tc := range cases {