feat: persist agentId across remote-authority reload

commands.open() triggers vscode.openFolder with a remote authority,
which reloads the window and wipes in-memory state.  To survive this,
handleOpen now saves the agentId to globalState before the reload.
After the reload, activate() reads and clears it once the deployment
is configured, then opens the chat panel.

This follows the same set-and-clear pattern used by firstConnect.

Author: ThomasK33

src/core/mementoManager.ts

@@ -57,4 +57,28 @@ export class MementoManager {
 		}
 		return isFirst === true;
 	}
+
+	/**
+	 * Store a chat agent ID to open after a window reload.
+	 * Used by the /open deep link handler: it must call
+	 * commands.open() which triggers a remote-authority
+	 * reload, wiping in-memory state.  The agent ID is
+	 * persisted here so the extension can pick it up on
+	 * the other side of the reload.
+	 */
+	public async setPendingChatAgentId(agentId: string): Promise<void> {
+		await this.memento.update("pendingChatAgentId", agentId);
+	}
+
+	/**
+	 * Read and clear the pending chat agent ID.  Returns
+	 * undefined if none was stored.
+	 */
+	public async getAndClearPendingChatAgentId(): Promise<string | undefined> {
+		const agentId = this.memento.get<string>("pendingChatAgentId");
+		if (agentId !== undefined) {
+			await this.memento.update("pendingChatAgentId", undefined);
+		}
+		return agentId;
+	}
 }

src/extension.ts

@@ -336,6 +336,15 @@ export async function activate(ctx: vscode.ExtensionContext): Promise<void> {
 					url: details.url,
 					token: details.token,
 				});
+
+				// If a deep link stored a chat agent ID before the
+				// remote-authority reload, open it now that the
+				// deployment is configured.
+				const pendingAgentId =
+					await mementoManager.getAndClearPendingChatAgentId();
+				if (pendingAgentId) {
+					chatPanelProvider.openChat(pendingAgentId);
+				}
 			}
 		} catch (ex) {
 			if (ex instanceof CertificateError) {

src/uri/uriHandler.ts

@@ -91,13 +91,7 @@ function getRequiredParam(params: URLSearchParams, name: string): string {
 }
 
 async function handleOpen(ctx: UriRouteContext): Promise<void> {
-	const {
-		params,
-		serviceContainer,
-		deploymentManager,
-		commands,
-		chatPanelProvider,
-	} = ctx;
+	const { params, serviceContainer, deploymentManager, commands } = ctx;
 
 	const owner = getRequiredParam(params, "owner");
 	const workspace = getRequiredParam(params, "workspace");
@@ -107,10 +101,14 @@ async function handleOpen(ctx: UriRouteContext): Promise<void> {
 		params.has("openRecent") &&
 		(!params.get("openRecent") || params.get("openRecent") === "true");
 
-	// Optional: if agentId is present, also open the embedded chat
-	// panel. Old extensions silently ignore this unknown param,
-	// giving backwards compatibility.
+	// Persist the chat agent ID before commands.open() triggers
+	// a remote-authority reload that wipes in-memory state.
+	// The extension picks this up after the reload in activate().
 	const agentId = params.get("agentId");
+	if (agentId) {
+		const mementoManager = serviceContainer.getMementoManager();
+		await mementoManager.setPendingChatAgentId(agentId);
+	}
 
 	await setupDeployment(params, serviceContainer, deploymentManager);
 
@@ -121,10 +119,6 @@ async function handleOpen(ctx: UriRouteContext): Promise<void> {
 		folder ?? undefined,
 		openRecent,
 	);
-
-	if (agentId) {
-		chatPanelProvider.openChat(agentId);
-	}
 }
 
 async function handleOpenDevContainer(ctx: UriRouteContext): Promise<void> {