fix: be able to fold all regions (#1494)

Author: scnwwu

server/src/sas/CompletionProvider.ts

@@ -5,6 +5,7 @@ import {
   CompletionItem,
   CompletionItemKind,
   CompletionList,
+  CompletionParams,
   Hover,
   MarkupKind,
   SignatureHelp,
@@ -611,12 +612,25 @@ export class CompletionProvider {
     });
   }
 
-  getCompleteItems(position: Position): Promise<CompletionList | undefined> {
+  getCompleteItems(
+    params: CompletionParams,
+  ): Promise<CompletionList | undefined> {
     return new Promise((resolve) => {
-      this._getZone(position);
+      this._getZone(params.position);
       const prefix = this.popupContext.prefix;
 
       this._loadAutoCompleteItems(this.popupContext.zone, (data) => {
+        if (
+          params.context?.triggerCharacter === "." &&
+          data &&
+          data[0] &&
+          (typeof data[0] === "string" ||
+            (data[0].type !== "DATA" && data[0].type !== "VIEW"))
+        ) {
+          // only data list can be triggered by "."
+          resolve(undefined);
+          return;
+        }
         const items = data?.map((item) => ({
           label: processLabelCase(
             typeof item === "string" ? item : item.name,

server/src/sas/LanguageServiceProvider.ts

@@ -10,7 +10,7 @@ import { Range, TextDocument } from "vscode-languageserver-textdocument";
 import { CodeZoneManager } from "./CodeZoneManager";
 import { CompletionProvider } from "./CompletionProvider";
 import { FormatOnTypeProvider } from "./FormatOnTypeProvider";
-import { FoldingBlock } from "./LexerEx";
+import { FoldingBlock, LexerEx } from "./LexerEx";
 import { Model } from "./Model";
 import type { LibService } from "./SyntaxDataProvider";
 import { SyntaxProvider } from "./SyntaxProvider";
@@ -203,6 +203,7 @@ export class LanguageServiceProvider {
           result.push({
             startLine: block.startLine,
             endLine: block.endFoldingLine,
+            kind: block.type === LexerEx.SEC_TYPE.CUSTOM ? "region" : undefined,
           });
         }
         i = rootBlock.endFoldingLine;

server/src/sas/SyntaxDataProvider.ts

@@ -2559,9 +2559,10 @@ export class SyntaxDataProvider {
     return _getKeywords(/*'statisticsKeywords'*/ "stat-kw", cb);
   }
   getDSStatements(cb?: (data: string[]) => void) {
+    const globalProcStatements = this.getGlobalProcedureStatements();
     return _tryToLoadStatementsFromPubs("datastep", cb, function () {
       const data = _procStmtObj("datastep");
-      return data[ID_STMTS];
+      return data[ID_STMTS].concat(globalProcStatements);
     });
   }
   getDSOptions(cb?: (data: string[]) => void) {

server/src/server.ts

@@ -144,14 +144,13 @@ export const runServer = (
       async sas(languageService) {
         if (
           params.context?.triggerCharacter &&
-          params.context.triggerCharacter !== " "
+          params.context.triggerCharacter !== " " &&
+          params.context.triggerCharacter !== "."
         ) {
           return undefined;
         }
         const completionList =
-          await languageService.completionProvider.getCompleteItems(
-            params.position,
-          );
+          await languageService.completionProvider.getCompleteItems(params);
         if (completionList) {
           for (const item of completionList.items) {
             if (!item.data) {