Generate cell for all chat users. (microsoft#210487)

Author: rebornix

src/vs/workbench/contrib/notebook/browser/controller/chat/cellChatActions.ts

@@ -9,6 +9,8 @@ import { EditorContextKeys } from 'vs/editor/common/editorContextKeys';
 import { localize, localize2 } from 'vs/nls';
 import { CONTEXT_ACCESSIBILITY_MODE_ENABLED } from 'vs/platform/accessibility/common/accessibility';
 import { MenuId, MenuRegistry, registerAction2 } from 'vs/platform/actions/common/actions';
+import { ICommandService } from 'vs/platform/commands/common/commands';
+import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
 import { ContextKeyExpr } from 'vs/platform/contextkey/common/contextkey';
 import { InputFocusedContextKey } from 'vs/platform/contextkey/common/contextkeys';
 import { ServicesAccessor } from 'vs/platform/instantiation/common/instantiation';
@@ -17,6 +19,7 @@ import { CTX_INLINE_CHAT_FOCUSED, CTX_INLINE_CHAT_HAS_PROVIDER, CTX_INLINE_CHAT_
 import { CTX_NOTEBOOK_CELL_CHAT_FOCUSED, CTX_NOTEBOOK_CHAT_HAS_ACTIVE_REQUEST, CTX_NOTEBOOK_CHAT_OUTER_FOCUS_POSITION, CTX_NOTEBOOK_CHAT_USER_DID_EDIT, MENU_CELL_CHAT_INPUT, MENU_CELL_CHAT_WIDGET, MENU_CELL_CHAT_WIDGET_FEEDBACK, MENU_CELL_CHAT_WIDGET_STATUS } from 'vs/workbench/contrib/notebook/browser/controller/chat/notebookChatContext';
 import { NotebookChatController } from 'vs/workbench/contrib/notebook/browser/controller/chat/notebookChatController';
 import { CELL_TITLE_CELL_GROUP_ID, INotebookActionContext, INotebookCellActionContext, NotebookAction, NotebookCellAction, getEditorFromArgsOrActivePane } from 'vs/workbench/contrib/notebook/browser/controller/coreActions';
+import { insertNewCell } from 'vs/workbench/contrib/notebook/browser/controller/insertCellActions';
 import { CellEditState } from 'vs/workbench/contrib/notebook/browser/notebookBrowser';
 import { CellKind, NOTEBOOK_EDITOR_CURSOR_BOUNDARY, NotebookSetting } from 'vs/workbench/contrib/notebook/common/notebookCommon';
 import { NOTEBOOK_CELL_EDITOR_FOCUSED, NOTEBOOK_CELL_GENERATED_BY_CHAT, NOTEBOOK_EDITOR_EDITABLE, NOTEBOOK_EDITOR_FOCUSED } from 'vs/workbench/contrib/notebook/common/notebookContextKeys';
@@ -364,6 +367,26 @@ interface IInsertCellWithChatArgs extends INotebookActionContext {
 	autoSend?: boolean;
 }
 
+async function startChat(accessor: ServicesAccessor, context: INotebookActionContext, index: number, input?: string, autoSend?: boolean) {
+	const configurationService = accessor.get(IConfigurationService);
+	const commandService = accessor.get(ICommandService);
+
+	if (configurationService.getValue<boolean>(NotebookSetting.cellChat)) {
+		context.notebookEditor.focusContainer();
+		NotebookChatController.get(context.notebookEditor)?.run(index, input, autoSend);
+	} else {
+		const newCell = await insertNewCell(accessor, context, CellKind.Code, 'below', true);
+		if (newCell) {
+			await context.notebookEditor.revealFirstLineIfOutsideViewport(newCell);
+			const codeEditor = context.notebookEditor.codeEditors.find(ce => ce[0] === newCell)?.[1];
+			if (codeEditor) {
+				codeEditor.focus();
+				commandService.executeCommand('inlineChat.start');
+			}
+		}
+	}
+}
+
 registerAction2(class extends NotebookAction {
 	constructor() {
 		super(
@@ -404,7 +427,6 @@ registerAction2(class extends NotebookAction {
 						NOTEBOOK_EDITOR_EDITABLE.isEqualTo(true),
 						ContextKeyExpr.not(InputFocusedContextKey),
 						CTX_INLINE_CHAT_HAS_PROVIDER,
-						ContextKeyExpr.equals(`config.${NotebookSetting.cellChat}`, true)
 					),
 					weight: KeybindingWeight.WorkbenchContrib,
 					primary: KeyMod.CtrlCmd | KeyCode.KeyI,
@@ -418,7 +440,6 @@ registerAction2(class extends NotebookAction {
 						when: ContextKeyExpr.and(
 							NOTEBOOK_EDITOR_EDITABLE.isEqualTo(true),
 							CTX_INLINE_CHAT_HAS_PROVIDER,
-							ContextKeyExpr.equals(`config.${NotebookSetting.cellChat}`, true)
 						)
 					}
 				]
@@ -467,8 +488,7 @@ registerAction2(class extends NotebookAction {
 
 	async runWithContext(accessor: ServicesAccessor, context: IInsertCellWithChatArgs) {
 		const index = Math.max(0, context.cell ? context.notebookEditor.getCellIndex(context.cell) + 1 : 0);
-		context.notebookEditor.focusContainer();
-		NotebookChatController.get(context.notebookEditor)?.run(index, context.input, context.autoSend);
+		await startChat(accessor, context, index, context.input, context.autoSend);
 	}
 });
 
@@ -490,17 +510,15 @@ registerAction2(class extends NotebookAction {
 						order: -1,
 						when: ContextKeyExpr.and(
 							NOTEBOOK_EDITOR_EDITABLE.isEqualTo(true),
-							CTX_INLINE_CHAT_HAS_PROVIDER,
-							ContextKeyExpr.equals(`config.${NotebookSetting.cellChat}`, true)
+							CTX_INLINE_CHAT_HAS_PROVIDER
 						)
 					},
 				]
 			});
 	}
 
 	async runWithContext(accessor: ServicesAccessor, context: INotebookActionContext) {
-		context.notebookEditor.focusContainer();
-		NotebookChatController.get(context.notebookEditor)?.run(0, '', false);
+		await startChat(accessor, context, 0, '', false);
 	}
 });
 
@@ -518,7 +536,6 @@ MenuRegistry.appendMenuItem(MenuId.NotebookToolbar, {
 		ContextKeyExpr.notEquals('config.notebook.insertToolbarLocation', 'betweenCells'),
 		ContextKeyExpr.notEquals('config.notebook.insertToolbarLocation', 'hidden'),
 		CTX_INLINE_CHAT_HAS_PROVIDER,
-		ContextKeyExpr.equals(`config.${NotebookSetting.cellChat}`, true)
 	)
 });