update semantic-tokens

1 files changed, 20 insertions, 1 deletions

diff --git a/script/provider/capability.lua b/script/provider/capability.lua
index 439978d6..8a1424ca 100644
--- a/script/provider/capability.lua
+++ b/script/provider/capability.lua
@@ -1,11 +1,22 @@
-local sp         = require 'bee.subprocess'
 local nonil      = require 'without-check-nil'
 local client     = require 'client'
 local platform   = require 'bee.platform'
 local completion = require 'provider.completion'
+local define     = require 'proto.define'
 
 require 'provider.semantic-tokens'
 
+local function toArray(map)
+    local array = {}
+    for k in pairs(map) do
+        array[#array+1] = k
+    end
+    table.sort(array, function (a, b)
+        return map[a] < map[b]
+    end)
+    return array
+end
+
 local m = {}
 
 local function testFileEvents(initer)
@@ -93,6 +104,14 @@ function m.getIniter()
             firstTriggerCharacter = '\n',
             moreTriggerCharacter  = nil, -- string[]
         },
+        semanticTokensProvider = {
+            legend = {
+                tokenTypes     = toArray(define.TokenTypes),
+                tokenModifiers = toArray(define.TokenModifiers),
+            },
+            range = true,
+            full  = false,
+        },
         --documentOnTypeFormattingProvider = {
         --    firstTriggerCharacter = '}',
         --},