Merge pull request microsoft#257579 from microsoft/osortega/adding-chat-session-change-event

Adding event for chat session items changes

Author: osortega

src/vs/workbench/api/browser/mainThreadChatSessions.ts

@@ -33,6 +33,11 @@ export class MainThreadChatSessions extends Disposable implements MainThreadChat
 		this._registrations.set(handle, this._chatSessionsService.registerChatSessionItemProvider(provider));
 	}
 
+	$onDidChangeChatSessionItems(chatSessionType: string): void {
+		// Notify the provider that its chat session items have changed
+		this._chatSessionsService.notifySessionItemsChange(chatSessionType);
+	}
+
 	private async _provideChatSessionItems(handle: number, token: CancellationToken): Promise<IChatSessionItem[]> {
 		const proxy = this._extHostContext.getProxy(ExtHostContext.ExtHostChatSessions);
 

src/vs/workbench/api/common/extHost.protocol.ts

@@ -3113,6 +3113,7 @@ export interface MainThreadChatStatusShape {
 export interface MainThreadChatSessionsShape extends IDisposable {
 	$registerChatSessionItemProvider(handle: number, chatSessionType: string, label: string): void;
 	$unregisterChatSessionItemProvider(handle: number): void;
+	$onDidChangeChatSessionItems(chatSessionType: string): void;
 }
 
 export interface ExtHostChatSessionsShape {

src/vs/workbench/api/common/extHostChatSessions.ts

@@ -53,7 +53,11 @@ export class ExtHostChatSessions extends Disposable implements ExtHostChatSessio
 
 		this._statusProviders.set(handle, { provider, disposable: disposables });
 		this._proxy.$registerChatSessionItemProvider(handle, chatSessionType, provider.label);
-
+		if (provider.onDidChangeChatSessionItems) {
+			disposables.add(provider.onDidChangeChatSessionItems(() => {
+				this._proxy.$onDidChangeChatSessionItems(chatSessionType);
+			}));
+		}
 		return {
 			dispose: () => {
 				this._statusProviders.delete(handle);

src/vs/workbench/contrib/chat/browser/chatSessions.ts

@@ -367,6 +367,10 @@ class ChatSessionsViewPaneContainer extends ViewPaneContainer {
 			this.updateViewRegistration();
 		}));
 
+		// Listen for session items changes and refresh the appropriate provider tree
+		this._register(this.chatSessionsService.onDidChangeSessionItems((chatSessionType) => {
+			this.refreshProviderTree(chatSessionType);
+		}));
 	}
 
 	override getTitle(): string {
@@ -382,6 +386,21 @@ class ChatSessionsViewPaneContainer extends ViewPaneContainer {
 		}
 	}
 
+	private refreshProviderTree(chatSessionType: string): void {
+		// Find the provider with the matching chatSessionType
+		const providers = this.getAllChatSessionProviders();
+		const targetProvider = providers.find(provider => provider.chatSessionType === chatSessionType);
+
+		if (targetProvider) {
+			// Find the corresponding view and refresh its tree
+			const viewId = `${VIEWLET_ID}.${chatSessionType}`;
+			const view = this.getView(viewId) as SessionsViewPane | undefined;
+			if (view) {
+				view.refreshTree();
+			}
+		}
+	}
+
 	private updateViewRegistration(): void {
 		const currentProviders = this.getAllChatSessionProviders();
 		const currentProviderIds = new Set(currentProviders.map(p => p.chatSessionType));
@@ -601,6 +620,12 @@ class SessionsViewPane extends ViewPane {
 		return ('editor' in item && 'group' in item) || ('widget' in item && 'sessionType' in item);
 	}
 
+	public refreshTree(): void {
+		if (this.tree && this.isBodyVisible()) {
+			this.tree.updateChildren(this.provider);
+		}
+	}
+
 	protected override renderBody(container: HTMLElement): void {
 		super.renderBody(container);
 

src/vs/workbench/contrib/chat/common/chatSessionsService.ts

@@ -39,12 +39,14 @@ export interface IChatSessionItemProvider {
 export interface IChatSessionsService {
 	readonly _serviceBrand: undefined;
 	readonly onDidChangeItemsProviders: Event<IChatSessionItemProvider>;
+	readonly onDidChangeSessionItems: Event<string>;
 	registerContribution(contribution: IChatSessionsExtensionPoint): IDisposable;
 	getChatSessionProviders(): IChatSessionsExtensionPoint[];
 	getChatSessionItemProviders(): IChatSessionItemProvider[];
 	registerChatSessionItemProvider(provider: IChatSessionItemProvider): IDisposable;
 	hasChatSessionItemProviders: boolean;
 	provideChatSessionItems(chatSessionType: string, token: CancellationToken): Promise<IChatSessionItem[]>;
+	notifySessionItemsChange(chatSessionType: string): void;
 }
 
 export const IChatSessionsService = createDecorator<IChatSessionsService>('chatSessionsService');
@@ -55,6 +57,9 @@ export class ChatSessionsService extends Disposable implements IChatSessionsServ
 
 	private readonly _onDidChangeItemsProviders = this._register(new Emitter<IChatSessionItemProvider>());
 	readonly onDidChangeItemsProviders: Event<IChatSessionItemProvider> = this._onDidChangeItemsProviders.event;
+	private readonly _onDidChangeSessionItems = this._register(new Emitter<string>());
+	readonly onDidChangeSessionItems: Event<string> = this._onDidChangeSessionItems.event;
+
 	private _contributions: Map<string, IChatSessionsExtensionPoint> = new Map();
 
 	constructor(
@@ -98,6 +103,10 @@ export class ChatSessionsService extends Disposable implements IChatSessionsServ
 		return this._itemsProviders.has(chatViewType);
 	}
 
+	public notifySessionItemsChange(chatSessionType: string): void {
+		this._onDidChangeSessionItems.fire(chatSessionType);
+	}
+
 	public async provideChatSessionItems(chatSessionType: string, token: CancellationToken): Promise<IChatSessionItem[]> {
 		if (!(await this.canResolve(chatSessionType))) {
 			throw Error(`Can not find provider for ${chatSessionType}`);