2 files changed, 44 insertions, 22 deletions
diff --git a/script/client.lua b/script/client.lua
index f0f7aa44..d9399ddd 100644
--- a/script/client.lua
+++ b/script/client.lua
@@ -12,6 +12,7 @@ local scope     = require 'workspace.scope'
 local inspect   = require 'inspect'
 
 local m = {}
+m._eventList = {}
 
 function m.client(newClient)
     if newClient then
@@ -389,8 +390,22 @@ function m.editText(uri, edits)
     })
 end
 
+---@param callback async fun()
+function m.event(callback)
+    m._eventList[#m._eventList+1] = callback
+end
+
+function m._callEvent(ev)
+    for _, callback in ipairs(m._eventList) do
+        await.call(function ()
+            callback(ev)
+        end)
+    end
+end
+
 function m.setReady()
     m._ready = true
+    m._callEvent('ready')
 end
 
 function m.isReady()
@@ -415,6 +430,7 @@ function m.init(t)
     lang(LOCALE or t.locale)
     converter.setOffsetEncoding(m.getOffsetEncoding())
     hookPrint()
+    m._callEvent('init')
 end
 
 return m
diff --git a/script/provider/provider.lua b/script/provider/provider.lua
index 44805130..ac952715 100644
--- a/script/provider/provider.lua
+++ b/script/provider/provider.lua
@@ -909,29 +909,35 @@ local function toArray(map)
     return array
 end
 
-m.register 'textDocument/semanticTokens/full' {
-    capability = {
-        semanticTokensProvider = {
-            legend = {
-                tokenTypes     = toArray(define.TokenTypes),
-                tokenModifiers = toArray(define.TokenModifiers),
-            },
-            full  = true,
-        },
-    },
-    ---@async
-    function (params)
-        log.debug('textDocument/semanticTokens/full')
-        local uri = files.getRealUri(params.textDocument.uri)
-        workspace.awaitReady(uri)
-        local _ <close> = progress.create(uri, lang.script.WINDOW_PROCESSING_SEMANTIC_FULL, 0.5)
-        local core = require 'core.semantic-tokens'
-        local results = core(uri, 0, math.huge)
-        return {
-            data = results
-        }
+client.event(function (ev)
+    if ev == 'init' then
+        if not client.isVSCode() then
+            m.register 'textDocument/semanticTokens/full' {
+                capability = {
+                    semanticTokensProvider = {
+                        legend = {
+                            tokenTypes     = toArray(define.TokenTypes),
+                            tokenModifiers = toArray(define.TokenModifiers),
+                        },
+                        full  = true,
+                    },
+                },
+                ---@async
+                function (params)
+                    log.debug('textDocument/semanticTokens/full')
+                    local uri = files.getRealUri(params.textDocument.uri)
+                    workspace.awaitReady(uri)
+                    local _ <close> = progress.create(uri, lang.script.WINDOW_PROCESSING_SEMANTIC_FULL, 0.5)
+                    local core = require 'core.semantic-tokens'
+                    local results = core(uri, 0, math.huge)
+                    return {
+                        data = results
+                    }
+                end
+            }
+        end
     end
-}
+end)
 
 m.register 'textDocument/semanticTokens/range' {
     capability = {