66 lines
2.8 KiB
TypeScript
66 lines
2.8 KiB
TypeScript
|
/*---------------------------------------------------------------------------------------------
|
||
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
||
|
|
*--------------------------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
import { Range, commands, window, type Disposable } from 'vscode';
|
||
|
|
import { CopilotNamedAnnotationList } from '../../../../../../platform/completions-core/common/openai/copilotAnnotations';
|
||
|
|
import { DisposableStore, IDisposable } from '../../../../../../util/vs/base/common/lifecycle';
|
||
|
|
import { IInstantiationService, type ServicesAccessor } from '../../../../../../util/vs/platform/instantiation/common/instantiation';
|
||
|
|
import * as constants from '../constants';
|
||
|
|
import { registerCommand } from '../telemetry';
|
||
|
|
import { wrapDoc } from '../textDocumentManager';
|
||
|
|
import { CopilotSuggestionsPanelManager } from './copilotSuggestionsPanelManager';
|
||
|
|
|
||
|
|
// Exported for testing
|
||
|
|
export enum PanelNavigationType {
|
||
|
|
Previous = 'previous',
|
||
|
|
Next = 'next',
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* This interface contains data associated to a completion displayed in the panel.
|
||
|
|
*/
|
||
|
|
export interface PanelCompletion {
|
||
|
|
insertText: string;
|
||
|
|
range: Range;
|
||
|
|
copilotAnnotations?: CopilotNamedAnnotationList;
|
||
|
|
postInsertionCallback: () => PromiseLike<void> | void;
|
||
|
|
}
|
||
|
|
|
||
|
|
export function registerPanelSupport(accessor: ServicesAccessor): Disposable {
|
||
|
|
const instantiationService = accessor.get(IInstantiationService);
|
||
|
|
const suggestionsPanelManager = instantiationService.createInstance(CopilotSuggestionsPanelManager);
|
||
|
|
|
||
|
|
const disposableStore = new DisposableStore();
|
||
|
|
|
||
|
|
function registerOpenPanelCommand(id: string): IDisposable {
|
||
|
|
return registerCommand(accessor, id, async () => {
|
||
|
|
// hide ghost text while opening the generation ui
|
||
|
|
await commands.executeCommand('editor.action.inlineSuggest.hide');
|
||
|
|
await instantiationService.invokeFunction(commandOpenPanel, suggestionsPanelManager);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
// Register both commands to also support command palette
|
||
|
|
disposableStore.add(registerOpenPanelCommand(constants.CMDOpenPanelChat));
|
||
|
|
disposableStore.add(registerOpenPanelCommand(constants.CMDOpenPanelClient));
|
||
|
|
|
||
|
|
// No command palette support needed for these commands
|
||
|
|
disposableStore.add(suggestionsPanelManager.registerCommands());
|
||
|
|
|
||
|
|
return disposableStore;
|
||
|
|
}
|
||
|
|
|
||
|
|
function commandOpenPanel(accessor: ServicesAccessor, suggestionsPanelManager: CopilotSuggestionsPanelManager) {
|
||
|
|
const editor = window.activeTextEditor;
|
||
|
|
if (!editor) { return; }
|
||
|
|
const wrapped = wrapDoc(editor.document);
|
||
|
|
if (!wrapped) { return; }
|
||
|
|
|
||
|
|
const { line, character } = editor.selection.active;
|
||
|
|
|
||
|
|
suggestionsPanelManager.renderPanel(editor.document, { line, character }, wrapped);
|
||
|
|
return commands.executeCommand('setContext', constants.CopilotPanelVisible, true);
|
||
|
|
}
|