feat(language-server): notification-based tsserver request forwarding (#5395)

Author: johnsoncodehk

extensions/vscode/src/languageClient.ts

@@ -106,7 +106,7 @@ async function getInitializationOptions(context: vscode.ExtensionContext): Promi
 	return {
 		typescript: {
 			tsdk: (await lsp.getTsdk(context))!.tsdk,
-			tsserverRequestCommand: 'tsserverRequest',
+			tsserverRequestCommand: 'typescript.tsserverRequest',
 		},
 	};
 }

extensions/vscode/src/nodeClientMain.ts

@@ -85,7 +85,7 @@ export const { activate, deactivate } = defineExtension(async () => {
 
 			updateProviders(client);
 
-			client.onRequest('tsserverRequest', async ([command, args]) => {
+			client.onRequest('typescript.tsserverRequest', async ([command, args]) => {
 				const tsserver = (globalThis as any).__TSSERVER__?.semantic;
 				if (!tsserver) {
 					return;

packages/language-server/lib/types.ts

@@ -1,7 +1,7 @@
 export type VueInitializationOptions = {
 	typescript: {
 		tsdk: string;
-		tsserverRequestCommand?: string;
+		tsserverRequestCommand?: string | [request: string, response: string];
 	};
 };
 

packages/language-server/node.ts

@@ -38,6 +38,16 @@ connection.onInitialize(params => {
 	});
 
 	let simpleLs: LanguageService | undefined;
+	let tsserverRequestId = 0;
+
+	const tsserverRequestHandlers = new Map<number, (res: any) => void>();
+
+	if (Array.isArray(options.typescript.tsserverRequestCommand)) {
+		connection.onNotification(options.typescript.tsserverRequestCommand[1], ([id, res]) => {
+			tsserverRequestHandlers.get(id)?.(res);
+			tsserverRequestHandlers.delete(id);
+		});
+	}
 
 	return server.initialize(
 		params,
@@ -139,8 +149,17 @@ connection.onInitialize(params => {
 		} : undefined)
 	);
 
-	function sendTsRequest<T>(command: string, args: any): Promise<T | null> {
-		return connection.sendRequest<T>(options.typescript.tsserverRequestCommand!, [command, args]);
+	async function sendTsRequest<T>(command: string, args: any): Promise<T | null> {
+		if (typeof options.typescript.tsserverRequestCommand === 'string') {
+			return await connection.sendRequest<T>(options.typescript.tsserverRequestCommand, [command, args]);
+		} else {
+			const [requestCommand] = options.typescript.tsserverRequestCommand!;
+			return await new Promise<T | null>(resolve => {
+				const requestId = ++tsserverRequestId;
+				tsserverRequestHandlers.set(requestId, resolve);
+				connection.sendNotification(requestCommand, [command, args, requestId]);
+			});
+		}
 	}
 
 	function createLs(server: LanguageServer, tsconfig: string | undefined) {
@@ -185,3 +204,4 @@ connection.onInitialize(params => {
 connection.onInitialized(server.initialized);
 
 connection.onShutdown(server.shutdown);
+