update chat lib extraction with new path and add context setup for lib

Author: jeffreyhunter77

chat-lib/package.json

@@ -33,7 +33,6 @@
 		"outdent": "^0.8.0",
 		"rimraf": "^6.0.1",
 		"typescript": "^5.8.3",
-		"vite-tsconfig-paths": "^5.1.4",
 		"vitest": "^3.0.5"
 	},
 	"keywords": [

chat-lib/test/getInlineCompletions.spec.ts

@@ -0,0 +1,40 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+// Load env
+import * as dotenv from 'dotenv';
+dotenv.config({ path: '../.env' });
+
+import { createTextDocument } from '#lib/test/textDocument';
+import { assert, describe, it } from 'vitest';
+import { createInlineCompletionsProvider } from '../src/main';
+
+describe('getInlineCompletions', () => {
+	it('should return completions for a document and position', async () => {
+		const provider = createInlineCompletionsProvider({
+			fetcher: undefined as any,
+			authService: undefined as any,
+			telemetrySender: undefined as any,
+			isRunningInTest: true,
+			contextProviderMatch: undefined as any,
+			statusHandler: undefined as any,
+			documentManager: undefined as any,
+			workspace: undefined as any,
+			urlOpener: undefined as any,
+			editorInfo: undefined as any,
+			editorPluginInfo: undefined as any,
+			relatedPluginInfo: undefined as any,
+			editorSession: undefined as any,
+			notificationSender: undefined as any,
+			endpointProvider: undefined as any,
+			capiClientService: undefined as any,
+		});
+		const doc = createTextDocument('file:///test.txt', 'javascript', 1, 'function main() {\n\n\n}\n');
+
+		const result = await provider.getInlineCompletions(doc, { line: 1, character: 0 });
+
+		assert(result);
+	});
+});

chat-lib/tsconfig.json

@@ -13,16 +13,16 @@
 		],
 		"paths": {
 			"#lib/*": [
-				"./src/_internal/extension/completions-core/lib/src/*"
+				"./src/_internal/extension/completions-core/vscode-node/lib/src/*"
 			],
 			"#prompt/*": [
-				"./src/_internal/extension/completions-core/prompt/src/*"
+				"./src/_internal/extension/completions-core/vscode-node/prompt/src/*"
 			],
 			"#bridge/*": [
-				"./src/_internal/extension/completions-core/bridge/src/*"
+				"./src/_internal/extension/completions-core/vscode-node/bridge/src/*"
 			],
 			"#types": [
-				"./src/_internal/extension/completions-core/types/src"
+				"./src/_internal/extension/completions-core/vscode-node/types/src"
 			]
 		}
 	},

eslint.config.mjs

@@ -366,5 +366,11 @@ export default tseslint.config(
 			'local/no-unlayered-files': 'off',
 			'no-restricted-imports': 'off'
 		}
+	},
+	{
+		files: ['./src/lib/node/chatLibMain.ts'],
+		rules: {
+			'import/no-restricted-paths': 'off'
+		}
 	}
 );

script/build/extractChatLib.ts

@@ -27,6 +27,7 @@ const entryPoints = [
 	'src/platform/tokenizer/node/tikTokenizerWorker.ts',
 	// For tests:
 	'src/platform/authentication/test/node/simulationTestCopilotTokenManager.ts',
+	'src/extension/completions-core/vscode-node/lib/src/test/textDocument.ts',
 ];
 
 interface FileInfo {

src/extension/completions-core/vscode-node/completionsServiceBridges.ts

@@ -143,7 +143,7 @@ export function createContext(serviceAccessor: ServicesAccessor): Context {
 		}
 	});
 
-	ctx.set(NotificationSender, new ExtensionNotificationSender());
+	ctx.set(NotificationSender, instaService.createInstance(ExtensionNotificationSender));
 	ctx.set(EditorAndPluginInfo, new VSCodeEditorInfo());
 	ctx.set(EditorSession, new EditorSession(env.sessionId, env.machineId, env.remoteName, uiKindToString(env.uiKind)));
 	ctx.set(CopilotExtensionStatus, new CopilotExtensionStatus());

src/extension/completions-core/vscode-node/lib/src/config.ts

@@ -373,14 +373,14 @@ type NameAndVersion = {
 	version: string;
 };
 
-type EditorInfo = NameAndVersion & {
+export type EditorInfo = NameAndVersion & {
 	// The root directory of the installation, currently only used to simplify stack traces.
 	root?: string;
 	// A programmatic name, used for error reporting.
 	devName?: string;
 };
 
-type EditorPluginInfo = NameAndVersion;
+export type EditorPluginInfo = NameAndVersion;
 
 export type EditorPluginFilter = { filter: Filter; value: string; isVersion?: boolean };
 

src/extension/completions-core/vscode-node/lib/src/notificationSender.ts

@@ -2,7 +2,7 @@
  *  Copyright (c) Microsoft Corporation. All rights reserved.
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
-import { window } from 'vscode';
+import { INotificationService } from '../../../../../platform/notification/common/notificationService';
 
 export interface ActionItem {
 	title: string;
@@ -13,8 +13,12 @@ export abstract class NotificationSender {
 }
 
 export class ExtensionNotificationSender extends NotificationSender {
+	constructor(@INotificationService private readonly notificationService: INotificationService) {
+		super();
+	}
+
 	async showWarningMessage(message: string, ...actions: ActionItem[]): Promise<ActionItem | undefined> {
-		const response = await window.showWarningMessage(message, ...actions.map(action => action.title));
+		const response = await this.notificationService.showWarningMessage(message, ...actions.map(action => action.title));
 		if (response === undefined) { return; }
 		return { title: response };
 	}