Set up persisting of terminal chat model

Author: Tyriar

src/vs/workbench/contrib/terminalContrib/chat/browser/terminalChatWidget.ts

@@ -26,6 +26,7 @@ import { ChatModel, IChatResponseModel } from '../../../chat/common/chatModel.js
 import { IChatService, IChatProgress } from '../../../chat/common/chatService.js';
 import { CancellationTokenSource } from '../../../../../base/common/cancellation.js';
 import { MenuId } from '../../../../../platform/actions/common/actions.js';
+import type { IChatViewState } from '../../../chat/browser/chatWidget.js';
 
 const enum Constants {
 	HorizontalMargin = 10,
@@ -65,7 +66,8 @@ export class TerminalChatWidget extends Disposable {
 
 	private _messages = this._store.add(new Emitter<Message>());
 
-	private _storageKey = 'terminal-inline-chat-history';
+	private _historyStorageKey = 'terminal-inline-chat-history';
+	private _viewStateStorageKey = 'terminal-inline-chat-view-state';
 	private _promptHistory: string[] = [];
 
 	private _lastResponseContent: string | undefined;
@@ -137,6 +139,7 @@ export class TerminalChatWidget extends Disposable {
 				}
 			},
 		);
+		this._register(this._inlineChatWidget.chatWidget.onDidChangeViewModel(() => this._saveViewState()));
 		this._register(Event.any(
 			this._inlineChatWidget.onDidChangeHeight,
 			this._instance.onDimensionsChanged,
@@ -161,7 +164,7 @@ export class TerminalChatWidget extends Disposable {
 		this._responseContainsCodeBlockContextKey = TerminalChatContextKeys.responseContainsCodeBlock.bindTo(this._contextKeyService);
 		this._responseContainsMulitpleCodeBlocksContextKey = TerminalChatContextKeys.responseContainsMultipleCodeBlocks.bindTo(this._contextKeyService);
 
-		this._promptHistory = JSON.parse(this._storageService.get(this._storageKey, StorageScope.PROFILE, '[]'));
+		this._promptHistory = JSON.parse(this._storageService.get(this._historyStorageKey, StorageScope.PROFILE, '[]'));
 		this._historyUpdate = (prompt: string) => {
 			const idx = this._promptHistory.indexOf(prompt);
 			if (idx >= 0) {
@@ -170,7 +173,7 @@ export class TerminalChatWidget extends Disposable {
 			this._promptHistory.unshift(prompt);
 			this._historyOffset = -1;
 			this._historyCandidate = '';
-			this._storageService.store(this._storageKey, JSON.stringify(this._promptHistory), StorageScope.PROFILE, StorageTarget.USER);
+			this._storageService.store(this._historyStorageKey, JSON.stringify(this._promptHistory), StorageScope.PROFILE, StorageTarget.USER);
 		};
 	}
 
@@ -217,8 +220,8 @@ export class TerminalChatWidget extends Disposable {
 		this._inlineChatWidget.updateInfo(localize('welcome.1', "AI-generated commands may be incorrect"));
 	}
 
-	async reveal(): Promise<void> {
-		await this._createSession();
+	async reveal(viewState?: IChatViewState): Promise<void> {
+		await this._createSession(viewState);
 		this._doLayout(this._inlineChatWidget.contentHeight);
 		this._container.classList.remove('hide');
 		this._visibleContextKey.set(true);
@@ -310,13 +313,13 @@ export class TerminalChatWidget extends Disposable {
 		return this._focusTracker;
 	}
 
-	private async _createSession(): Promise<void> {
+	private async _createSession(viewState?: IChatViewState): Promise<void> {
 		this._sessionCtor = createCancelablePromise<void>(async token => {
 			if (!this._model.value) {
 				this._model.value = this._chatService.startSession(ChatAgentLocation.Terminal, token);
 				const model = this._model.value;
 				if (model) {
-					this._inlineChatWidget.setChatModel(model);
+					this._inlineChatWidget.setChatModel(model, this._loadViewState());
 				}
 				if (!this._model.value) {
 					throw new Error('Failed to start chat session');
@@ -326,6 +329,23 @@ export class TerminalChatWidget extends Disposable {
 		this._register(toDisposable(() => this._sessionCtor?.cancel()));
 	}
 
+	private _loadViewState() {
+		const rawViewState = this._storageService.get(this._viewStateStorageKey, StorageScope.PROFILE, undefined);
+		let viewState: IChatViewState | undefined;
+		if (rawViewState) {
+			try {
+				viewState = JSON.parse(rawViewState);
+			} catch {
+				viewState = undefined;
+			}
+		}
+		return viewState;
+	}
+
+	private _saveViewState() {
+		this._storageService.store(this._viewStateStorageKey, JSON.stringify(this._inlineChatWidget.chatWidget.getViewState()), StorageScope.PROFILE, StorageTarget.USER);
+	}
+
 	private _forcedPlaceholder: string | undefined = undefined;
 
 	private _updatePlaceholder(): void {