From dd083d3b246b5db8e7a158585e0fc912c9dd03ce Mon Sep 17 00:00:00 2001
From: johnsoncodehk <johnsoncodehk@gmail.com>
Date: Thu, 9 Jun 2022 19:15:03 +0800
Subject: [PATCH] feat: implemented provideFoldingRanges

---
 src/monaco/code2monaco.ts | 10 ++++++++++
 src/monaco/ls.ts          | 11 +++++++++++
 2 files changed, 21 insertions(+)

diff --git a/src/monaco/code2monaco.ts b/src/monaco/code2monaco.ts
index be94d569..244c41e3 100644
--- a/src/monaco/code2monaco.ts
+++ b/src/monaco/code2monaco.ts
@@ -426,3 +426,13 @@ export function asColorPresentation(item: vscode.ColorPresentation): monaco.lang
         additionalTextEdits: item.additionalTextEdits ? item.additionalTextEdits.map(asTextEdit) : undefined,
     };
 }
+
+export function asFoldingRange(item: vscode.FoldingRange): monaco.languages.FoldingRange {
+    return {
+        start: item.startLine,
+        end: item.endLine,
+        kind: {
+            value: item.kind ?? '',
+        },
+    };
+}
diff --git a/src/monaco/ls.ts b/src/monaco/ls.ts
index 8200542e..2f08cfea 100644
--- a/src/monaco/ls.ts
+++ b/src/monaco/ls.ts
@@ -509,6 +509,17 @@ export async function setupLs(modelsMap: Ref<Map<string, monaco.editor.ITextMode
                 }
             },
         }),
+        monaco.languages.registerFoldingRangeProvider(lang, {
+            provideFoldingRanges: async (model) => {
+                const document = getTextDocument(model);
+                if (document) {
+                    const codeResult = await ds.getFoldingRanges(document);
+                    if (codeResult) {
+                        return codeResult.map(code2monaco.asFoldingRange);
+                    }
+                }
+            },
+        }),
     );
 
     return ls;