update api

Author: rebornix

src/extension/chatSessions/vscode-node/chatSessions.ts

@@ -22,6 +22,7 @@ import { ClaudeChatSessionItemProvider } from './claudeChatSessionItemProvider';
 import { ClaudeChatSessionParticipant } from './claudeChatSessionParticipant';
 import { OpenCodeChatSessionContentProvider } from './opencodeChatSessionContentProvider';
 import { OpenCodeChatSessionItemProvider, OpenCodeSessionDataStore } from './opencodeChatSessionItemProvider';
+import { OpenCodeChatSessionParticipant } from './opencodeChatSessionParticipant';
 
 export class ChatSessionsContrib extends Disposable implements IExtensionContribution {
 	readonly id = 'chatSessions';
@@ -77,6 +78,13 @@ export class ChatSessionsContrib extends Disposable implements IExtensionContrib
 			opencodeAgentManager,
 			opencodeSessionStore
 		);
-		this._register(vscode.chat.registerChatSessionContentProvider('opencode', opencodeChatSessionContentProvider));
+		const opencodeChatSessionParticipant = opencodeInstaService.createInstance(
+			OpenCodeChatSessionParticipant,
+			'opencode',
+			opencodeAgentManager,
+			opencodeSessionItemProvider
+		);
+		const opencodeParticipant = vscode.chat.createChatParticipant('opencode', opencodeChatSessionParticipant.createHandler());
+		this._register(vscode.chat.registerChatSessionContentProvider('opencode', opencodeChatSessionContentProvider, opencodeParticipant));
 	}
 }

src/extension/chatSessions/vscode-node/opencodeChatSessionItemProvider.ts

@@ -146,6 +146,9 @@ export class OpenCodeChatSessionItemProvider extends Disposable implements vscod
 	private readonly _onDidChangeChatSessionItems = this._register(new Emitter<void>());
 	public readonly onDidChangeChatSessionItems = this._onDidChangeChatSessionItems.event;
 
+	private readonly _onDidCommitChatSessionItem = this._register(new Emitter<{ original: vscode.ChatSessionItem; modified: vscode.ChatSessionItem }>());
+	public readonly onDidCommitChatSessionItem = this._onDidCommitChatSessionItem.event;
+
 	constructor(
 		private readonly sessionStore: OpenCodeSessionDataStore,
 		@IOpenCodeSessionService private readonly opencodeSessionService: IOpenCodeSessionService,
@@ -162,6 +165,13 @@ export class OpenCodeChatSessionItemProvider extends Disposable implements vscod
 		this._onDidChangeChatSessionItems.fire();
 	}
 
+	/**
+	 * Swaps an original session item with a modified one
+	 */
+	public swap(original: vscode.ChatSessionItem, modified: vscode.ChatSessionItem): void {
+		this._onDidCommitChatSessionItem.fire({ original, modified });
+	}
+
 	/**
 	 * Provides the list of chat session items for the VS Code UI
 	 */

src/extension/chatSessions/vscode-node/opencodeChatSessionParticipant.ts

@@ -0,0 +1,58 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import { ChatExtendedRequestHandler } from 'vscode';
+import { localize } from '../../../util/vs/nls';
+import { OpenCodeAgentManager } from '../../agents/opencode/node/opencodeAgentManager';
+import { OpenCodeChatSessionItemProvider } from './opencodeChatSessionItemProvider';
+
+export class OpenCodeChatSessionParticipant {
+	constructor(
+		private readonly sessionType: string,
+		private readonly opencodeAgentManager: OpenCodeAgentManager,
+		private readonly sessionItemProvider: OpenCodeChatSessionItemProvider,
+	) { }
+
+	createHandler(): ChatExtendedRequestHandler {
+		return this.handleRequest.bind(this);
+	}
+
+	private async handleRequest(request: vscode.ChatRequest, context: vscode.ChatContext, stream: vscode.ChatResponseStream, token: vscode.CancellationToken): Promise<vscode.ChatResult | void> {
+		const create = async () => {
+			const result = await this.opencodeAgentManager.handleRequest(undefined, request, context, stream, token);
+			const opencodeSessionId = result.sessionId;
+			if (!opencodeSessionId) {
+				stream.warning(localize('opencode.failedToCreateSession', "Failed to create a new OpenCode session."));
+				return undefined;
+			}
+			return opencodeSessionId;
+		};
+
+		const { chatSessionContext } = context;
+		if (chatSessionContext) {
+			if (chatSessionContext.isUntitled) {
+				/* New, empty session */
+				const opencodeSessionId = await create();
+				if (opencodeSessionId) {
+					// Tell UI to replace with opencode-backed session
+					this.sessionItemProvider.swap(chatSessionContext.chatSessionItem, { id: opencodeSessionId, label: request.prompt ?? 'OpenCode' });
+				}
+				return {};
+			}
+
+			/* Existing session */
+			const { id } = chatSessionContext.chatSessionItem;
+			await this.opencodeAgentManager.handleRequest(id, request, context, stream, token);
+			return {};
+		}
+
+		/* Via @opencode */
+		// TODO: Think about how this should work
+		stream.markdown(localize('opencode.viaAtOpencode', "Start a new OpenCode session"));
+		stream.button({ command: `workbench.action.chat.openNewSessionEditor.${this.sessionType}`, title: localize('opencode.startNewSession', "Start Session") });
+		return {};
+	}
+}