fix: dynamic menu for python

Author: jingjiajie

client/src/commands/run.ts

@@ -90,8 +90,7 @@ async function getSelectedRegions(
       "sas/getFoldingBlock",
       {
         textDocument: { uri: window.activeTextEditor.document.uri.toString() },
-        line,
-        col,
+        position: { line, col },
       },
     );
     if (block) {

server/src/browser/server.ts

@@ -107,8 +107,8 @@ connection.onFoldingRanges((params) => {
 connection.onRequest("sas/getFoldingBlock", (params) => {
   const languageService = getLanguageService(params.textDocument.uri);
   const block = languageService.getFoldingBlock(
-    params.line,
-    params.col,
+    params.position.line,
+    params.position.col,
     params.strict ?? true,
     params.ignoreCustomBlock,
     params.ignoreGlobalBlock,

server/src/node/server.ts

@@ -14,11 +14,15 @@ import {
   PrepareRenameParams,
   Range,
   ReferenceParams,
+  Registration,
+  RegistrationRequest,
   RenameParams,
   ResultProgressReporter,
   SemanticTokensRequest,
   TextDocumentPositionParams,
   TextDocumentSyncKind,
+  Unregistration,
+  UnregistrationRequest,
   WorkDoneProgressReporter,
   WorkspaceSymbolParams,
 } from "vscode-languageserver";
@@ -48,6 +52,7 @@ interface DocumentInfo {
 const documentPool: Record<string, DocumentInfo> = {};
 
 let supportSASGetLibList = false;
+let registeredAdvancedCapabilities = false;
 
 const connection: Connection = createConnection(ProposedFeatures.all);
 
@@ -81,15 +86,9 @@ connection.onInitialize((params) => {
         firstTriggerCharacter: "\n",
         moreTriggerCharacter: [";"],
       },
-      definitionProvider: { workDoneProgress: true },
-      declarationProvider: { workDoneProgress: true },
-      typeDefinitionProvider: { workDoneProgress: true },
-      referencesProvider: { workDoneProgress: true },
       documentSymbolProvider: { workDoneProgress: true },
       workspaceSymbolProvider: { workDoneProgress: true },
       hoverProvider: { workDoneProgress: true },
-      documentHighlightProvider: { workDoneProgress: true },
-      renameProvider: { prepareProvider: true, workDoneProgress: true },
       completionProvider: {
         triggerCharacters: _pyrightLanguageProvider.getClientCapabilities()
           .hasVisualStudioExtensionsCapability
@@ -105,7 +104,6 @@ connection.onInitialize((params) => {
         triggerCharacters: ["(", ",", ")"],
         workDoneProgress: true,
       },
-      callHierarchyProvider: true,
       workspace: {
         workspaceFolders: {
           supported: true,
@@ -196,8 +194,10 @@ connection.onDocumentSymbol(async (params, token) => {
   const sasSymbols = languageService.getDocumentSymbols();
   const pythonSymbols =
     (await _pyrightLanguageProvider.onDocumentSymbol(params, token)) ?? [];
+  let hasPythonCode = false;
   for (const sasSymbol of sasSymbols) {
     if (sasSymbol.name?.toUpperCase() === "PROC PYTHON") {
+      hasPythonCode = true;
       for (const pythonSymbol of pythonSymbols) {
         if (!("range" in pythonSymbol)) {
           continue;
@@ -208,6 +208,15 @@ connection.onDocumentSymbol(async (params, token) => {
       }
     }
   }
+  if (registeredAdvancedCapabilities) {
+    if (!hasPythonCode) {
+      unregisterAdvancedCapabilities(connection);
+    }
+  } else {
+    if (hasPythonCode) {
+      registerAdvancedCapabilities(connection);
+    }
+  }
   return sasSymbols;
 });
 
@@ -574,3 +583,42 @@ const syncAllChangedDoc = () => {
     syncIfDocChange(uri);
   }
 };
+
+const advancedCapabilities = [
+  "textDocument/declaration",
+  "textDocument/definition",
+  "textDocument/typeDefinition",
+  "textDocument/references",
+  "textDocument/rename",
+  "textDocument/documentHighlight",
+  "textDocument/prepareCallHierarchy",
+];
+
+const registerAdvancedCapabilities = async (conn: Connection) => {
+  if (registeredAdvancedCapabilities) {
+    return;
+  }
+  registeredAdvancedCapabilities = true;
+  const registerOptions = {
+    documentSelector: [{ language: "sas" }],
+  };
+  const registrations: Registration[] = [];
+  for (const capability of advancedCapabilities) {
+    registrations.push({ id: capability, method: capability, registerOptions });
+  }
+  await conn.sendRequest(RegistrationRequest.type, { registrations });
+};
+
+const unregisterAdvancedCapabilities = (conn: Connection) => {
+  if (!registeredAdvancedCapabilities) {
+    return;
+  }
+  registeredAdvancedCapabilities = false;
+  const unregisterations: Unregistration[] = [];
+  for (const capability of advancedCapabilities) {
+    unregisterations.push({ id: capability, method: capability });
+  }
+  conn.sendRequest(UnregistrationRequest.type, {
+    unregisterations: unregisterations,
+  });
+};