提供范围格式化

2 files changed, 32 insertions, 1 deletions

diff --git a/script/provider/capability.lua b/script/provider/capability.lua
index 27950d7f..31baf690 100644
--- a/script/provider/capability.lua
+++ b/script/provider/capability.lua
@@ -112,7 +112,8 @@ function m.getIniter()
             range = true,
             full  = false,
         },
-        documentFormattingProvider = true
+        documentFormattingProvider = true,
+        documentRangeFormattingProvider = true
         --documentOnTypeFormattingProvider = {
         --    firstTriggerCharacter = '}',
         --},
diff --git a/script/provider/provider.lua b/script/provider/provider.lua
index 63450f59..a5e0a1ca 100644
--- a/script/provider/provider.lua
+++ b/script/provider/provider.lua
@@ -954,6 +954,36 @@ m.register 'textDocument/formatting' {
     end
 }
 
+m.register 'textDocument/rangeFormatting' {
+    abortByFileUpdate = true,
+    ---@async
+    function(params)
+        local uri = files.getRealUri(params.textDocument.uri)
+        workspace.awaitReady(uri)
+        local _<close> = progress.create(workspace.getScope(uri), lang.script.WINDOW_PROCESSING_TYPE_FORMATTING, 0.5)
+
+        if not files.exists(uri) then
+            return nil
+        end
+
+        local core = require 'core.rangeformatting'
+        local edits = core(uri, params.range)
+        if not edits or #edits == 0 then
+            return nil
+        end
+
+        local results = {}
+        for i, edit in ipairs(edits) do
+            results[i] = {
+                range   = converter.packRange(uri, edit.start, edit.finish),
+                newText = edit.text,
+            }
+        end
+
+        return results
+    end
+}
+
 m.register 'config/editorconfig/update' {
     abortByFileUpdate = true,
     ---@async