Support new context resolve handler

Author: genlu

src/lsptoolshost/copilot/contextProviders.ts

@@ -30,10 +30,16 @@ export interface ContextResolveParam {
     activeExperiments: ActiveExperiments;
 }
 
-const resolveContextMethodName = 'roslyn/resolveContext';
-const resolveContextMethodSupportedVersion = '1';
-const resolveContextRequest = new lsp.RequestType<ContextResolveParam, SupportedContextItem[], void>(
-    resolveContextMethodName,
+const oldResolveContextMethodName = 'roslyn/resolveContext';
+const oldresolveContextMethodSupportedVersion = '1';
+const newResolveContextMethodName = 'roslyn/resolveContext@2';
+const newResolveContextMethodSupportedVersion = '1';
+const oldResolveContextRequest = new lsp.RequestType<ContextResolveParam, SupportedContextItem[], void>(
+    oldResolveContextMethodName,
+    lsp.ParameterStructures.auto
+);
+const newResolveContextRequest = new lsp.RequestType<ContextResolveParam, SupportedContextItem[], void>(
+    newResolveContextMethodName,
     lsp.ParameterStructures.auto
 );
 
@@ -81,27 +87,35 @@ export function registerCopilotContextProviders(
 
     devkit.activate().then(async (devKitExports) => {
         try {
-            // Check if the Copilot Language Server extension is installed and has the correct capabilities
-            let hasCapabilities = false;
+            let resolveMethod: lsp.RequestType<ContextResolveParam, SupportedContextItem[], void> | undefined =
+                undefined;
             const copilotServerExtensionfolder = devKitExports.components[copilotLanguageServerExtensionComponentName];
             if (copilotServerExtensionfolder) {
                 const capabilitiesFilePath = path.join(
                     copilotServerExtensionfolder,
                     copilotLanguageServerExtensionCapabilitiesFileName
                 );
                 const capabilitiesContent = await readJsonSync(capabilitiesFilePath);
-                if (
-                    capabilitiesContent?.capabilities?.find(
-                        (capability: any) =>
-                            capability?.method === resolveContextMethodName &&
-                            capability?.version === resolveContextMethodSupportedVersion
-                    )
-                ) {
-                    hasCapabilities = true;
+                for (const capability of capabilitiesContent?.capabilities ?? []) {
+                    if (
+                        capability.method === oldResolveContextMethodName &&
+                        capability.version === oldresolveContextMethodSupportedVersion
+                    ) {
+                        resolveMethod = oldResolveContextRequest;
+                        channel.debug(`supported 'roslyn/resolveContext' method found in capabilities.json`);
+                        break;
+                    } else if (
+                        capability.method === newResolveContextMethodName &&
+                        capability.version === newResolveContextMethodSupportedVersion
+                    ) {
+                        resolveMethod = newResolveContextRequest;
+                        channel.debug(`supported 'roslyn/resolveContext@2' method found in capabilities.json`);
+                        break;
+                    }
                 }
             }
 
-            if (!hasCapabilities) {
+            if (!resolveMethod) {
                 channel.debug(
                     `Failed to find compatible version of context provider from installed version of ${csharpDevkitExtensionId}.`
                 );
@@ -136,11 +150,7 @@ export function registerCopilotContextProviders(
                             if (!contextResolveParam) {
                                 return [];
                             }
-                            const items = await languageServer.sendRequest(
-                                resolveContextRequest,
-                                contextResolveParam,
-                                token
-                            );
+                            const items = await languageServer.sendRequest(resolveMethod, contextResolveParam, token);
                             channel.trace(`Copilot context provider resolved ${items.length} items`);
                             return items;
                         },