Merge pull request #8270 from continuedev/tomasz/con-4212

Build a local takeover MVP

Author: tomasz-stefaniak

core/control-plane/client.ts

@@ -54,6 +54,39 @@ export interface RemoteSessionMetadata extends BaseSessionMetadata {
   remoteId: string;
 }
 
+export interface AgentSessionMetadata {
+  createdBy: string;
+  github_repo: string;
+  organizationId?: string;
+  idempotencyKey?: string;
+  source?: string;
+  continueApiKeyId?: string;
+  s3Url?: string;
+  prompt?: string | null;
+  createdBySlug?: string;
+}
+
+export interface AgentSessionView {
+  id: string;
+  devboxId: string | null;
+  name: string | null;
+  icon: string | null;
+  status: string;
+  agentStatus: string | null;
+  unread: boolean;
+  state: string;
+  metadata: AgentSessionMetadata;
+  repoUrl: string;
+  branch: string | null;
+  pullRequestUrl: string | null;
+  pullRequestStatus: string | null;
+  tunnelUrl: string | null;
+  createdAt: string;
+  updatedAt: string;
+  create_time_ms: string;
+  end_time_ms: string;
+}
+
 export class ControlPlaneClient {
   constructor(
     readonly sessionInfoPromise: Promise<ControlPlaneSessionInfo | undefined>,
@@ -545,23 +578,19 @@ export class ControlPlaneClient {
       });
 
       const result = (await resp.json()) as {
-        agents: any[];
+        agents: AgentSessionView[];
         totalCount: number;
       };
 
       return {
-        agents: result.agents.map((agent: any) => ({
+        agents: result.agents.map((agent) => ({
           id: agent.id,
-          name: agent.name || agent.metadata?.name || null,
+          name: agent.name,
           status: agent.status,
-          repoUrl: agent.metadata?.repo_url || agent.repo_url || "",
-          createdAt:
-            agent.created_at || agent.create_time_ms
-              ? new Date(agent.created_at || agent.create_time_ms).toISOString()
-              : new Date().toISOString(),
+          repoUrl: agent.repoUrl,
+          createdAt: agent.createdAt,
           metadata: {
-            github_repo:
-              agent.metadata?.github_repo || agent.metadata?.repo_url,
+            github_repo: agent.metadata.github_repo,
           },
         })),
         totalCount: result.totalCount,
@@ -573,4 +602,73 @@ export class ControlPlaneClient {
       return { agents: [], totalCount: 0 };
     }
   }
+
+  /**
+   * Get the full agent session information
+   * @param agentSessionId - The ID of the agent session
+   * @returns The agent session view including metadata and status
+   */
+  public async getAgentSession(
+    agentSessionId: string,
+  ): Promise<AgentSessionView | null> {
+    if (!(await this.isSignedIn())) {
+      return null;
+    }
+
+    try {
+      const resp = await this.requestAndHandleError(
+        `agents/${agentSessionId}`,
+        {
+          method: "GET",
+        },
+      );
+
+      return (await resp.json()) as AgentSessionView;
+    } catch (e) {
+      Logger.error(e, {
+        context: "control_plane_get_agent_session",
+        agentSessionId,
+      });
+      return null;
+    }
+  }
+
+  /**
+   * Get the state of a specific background agent
+   * @param agentSessionId - The ID of the agent session
+   * @returns The agent's session state including history, workspace, and branch
+   */
+  public async getAgentState(agentSessionId: string): Promise<{
+    session: Session;
+    isProcessing: boolean;
+    messageQueueLength: number;
+    pendingPermission: any;
+  } | null> {
+    if (!(await this.isSignedIn())) {
+      return null;
+    }
+
+    try {
+      const resp = await this.requestAndHandleError(
+        `agents/${agentSessionId}/state`,
+        {
+          method: "GET",
+        },
+      );
+
+      const result = (await resp.json()) as {
+        session: Session;
+        isProcessing: boolean;
+        messageQueueLength: number;
+        pendingPermission: any;
+      };
+      return result;
+    } catch (e) {
+      Logger.error(e, {
+        context: "control_plane_get_agent_state",
+        agentSessionId,
+      });
+      return null;
+    }
+  }
 }

core/protocol/ideWebview.ts

@@ -78,6 +78,12 @@ export type ToIdeFromWebviewProtocol = ToIdeFromWebviewOrCoreProtocol & {
       totalCount: number;
     },
   ];
+  openAgentLocally: [
+    {
+      agentSessionId: string;
+    },
+    void,
+  ];
 };
 
 export type ToWebviewFromIdeProtocol = ToWebviewFromIdeOrCoreProtocol & {
@@ -94,6 +100,7 @@ export type ToWebviewFromIdeProtocol = ToWebviewFromIdeOrCoreProtocol & {
 
   focusContinueSessionId: [{ sessionId: string | undefined }, void];
   newSession: [undefined, void];
+  loadAgentSession: [{ session: any }, void];
   setTheme: [{ theme: any }, void];
   setColors: [{ [key: string]: string }, void];
   "jetbrains/editorInsetRefresh": [undefined, void];

core/util/repoUrl.ts

@@ -0,0 +1,70 @@
+/**
+ * Utility functions for normalizing and handling repository URLs.
+ */
+
+/**
+ * Normalizes a repository URL to a consistent format.
+ *
+ * Handles various Git URL formats and converts them to a standard HTTPS GitHub URL:
+ * - SSH format: `[email protected]:owner/repo.git` → `https://github.com/owner/repo`
+ * - SSH protocol: `ssh://[email protected]/owner/repo.git` → `https://github.com/owner/repo`
+ * - Shorthand: `owner/repo` → `https://github.com/owner/repo`
+ * - Removes `.git` suffix
+ * - Removes trailing slashes
+ * - Normalizes to lowercase
+ *
+ * @param url - The repository URL to normalize
+ * @returns The normalized repository URL in lowercase HTTPS format
+ *
+ * @example
+ * ```typescript
+ * normalizeRepoUrl("[email protected]:owner/repo.git")
+ * // Returns: "https://github.com/owner/repo"
+ *
+ * normalizeRepoUrl("owner/repo")
+ * // Returns: "https://github.com/owner/repo"
+ *
+ * normalizeRepoUrl("https://github.com/Owner/Repo.git")
+ * // Returns: "https://github.com/owner/repo"
+ * ```
+ */
+export function normalizeRepoUrl(url: string): string {
+  if (!url) return "";
+
+  let normalized = url.trim();
+
+  // Convert SSH to HTTPS: git@github.com:owner/repo.git -> https://github.com/owner/repo
+  if (normalized.startsWith("[email protected]:")) {
+    normalized = normalized.replace("[email protected]:", "https://github.com/");
+  }
+
+  // Convert SSH protocol to HTTPS: ssh://git@github.com/owner/repo.git -> https://github.com/owner/repo
+  // Also handles: ssh://git@github.com:owner/repo.git (less common)
+  if (normalized.startsWith("ssh://[email protected]")) {
+    normalized = normalized
+      .replace("ssh://[email protected]/", "https://github.com/")
+      .replace("ssh://[email protected]:", "https://github.com/");
+  }
+
+  // Convert shorthand owner/repo to full URL
+  if (
+    normalized.includes("/") &&
+    !/^[a-z]+:\/\//i.test(normalized) &&
+    !normalized.startsWith("git@")
+  ) {
+    normalized = `https://github.com/${normalized}`;
+  }
+
+  // Remove trailing slash before removing .git suffix
+  if (normalized.endsWith("/")) {
+    normalized = normalized.slice(0, -1);
+  }
+
+  // Remove .git suffix
+  if (normalized.endsWith(".git")) {
+    normalized = normalized.slice(0, -4);
+  }
+
+  // Normalize to lowercase
+  return normalized.toLowerCase();
+}