fix diagnostic

3 files changed, 29 insertions, 26 deletions

diff --git a/script/provider/diagnostic.lua b/script/provider/diagnostic.lua
index 0b607bd4..a9132e70 100644
--- a/script/provider/diagnostic.lua
+++ b/script/provider/diagnostic.lua
@@ -13,7 +13,6 @@ local converter = require 'proto.converter'
 
 ---@class diagnosticProvider
 local m = {}
-m._start = false
 m.cache = {}
 m.sleepRest = 0.0
 m.coroutineUri = setmetatable({}, { __mode = 'k' })
@@ -170,11 +169,7 @@ function m.syntaxErrors(uri, ast)
 end
 
 function m.diagnostics(uri, diags)
-    if not m._start then
-        return
-    end
-
-    if not ws.isReady() then
+    if not ws.isReady(uri) then
         return
     end
 
@@ -272,7 +267,7 @@ function m.doDiagnostic(uri)
 end
 
 function m.refresh(uri)
-    if not m._start then
+    if not ws.isReady(uri) then
         return
     end
     await.close('diag:' .. uri)
@@ -333,18 +328,12 @@ local function askForDisable()
     end
 end
 
-function m.diagnosticsAll(force)
-    for _, scp in ipairs(ws.folders) do
-        m.diagnosticsScope(scp.uri, force)
-    end
-end
-
 function m.diagnosticsScope(uri, force)
-    if not force and not config.get(uri, 'Lua.diagnostics.enable') then
-        m.clearAll()
+    if not ws.isReady(uri) then
         return
     end
-    if not m._start then
+    if not force and not config.get(uri, 'Lua.diagnostics.enable') then
+        m.clearAll()
         return
     end
     local delay = config.get(uri, 'Lua.diagnostics.workspaceDelay') / 1000
@@ -380,16 +369,6 @@ function m.diagnosticsScope(uri, force)
     end, 'files.version', 'diagnosticsScope')
 end
 
-function m.start()
-    m._start = true
-    m.diagnosticsAll()
-end
-
-function m.pause()
-    m._start = false
-    await.close 'diagnosticsScope'
-end
-
 function m.checkStepResult(uri)
     if await.hasID 'diagnosticsScope' then
         return
@@ -427,6 +406,12 @@ function m.checkWorkspaceDiag(uri)
     return false
 end
 
+ws.watch(function (ev, uri)
+    if ev == 'reload' then
+        m.diagnosticsScope(uri)
+    end
+end)
+
 files.watch(function (ev, uri) ---@async
     if ev == 'remove' then
         m.clear(uri)
diff --git a/script/provider/init.lua b/script/provider/init.lua
index a69fbab3..dfba351a 100644
--- a/script/provider/init.lua
+++ b/script/provider/init.lua
@@ -1 +1,2 @@
+require 'provider.diagnostic'
 return require 'provider.provider'
diff --git a/script/workspace/workspace.lua b/script/workspace/workspace.lua
index 92aca131..2ae5f322 100644
--- a/script/workspace/workspace.lua
+++ b/script/workspace/workspace.lua
@@ -18,6 +18,21 @@ local m = {}
 m.type = 'workspace'
 ---@type scope[]
 m.folders = {}
+m.watchList = {}
+
+--- 注册事件
+---@param callback async fun(ev: string, uri: uri)
+function m.watch(callback)
+    m.watchList[#m.watchList+1] = callback
+end
+
+function m.onWatch(ev, uri)
+    for _, callback in ipairs(m.watchList) do
+        await.call(function ()
+            callback(ev, uri)
+        end)
+    end
+end
 
 function m.initRoot(uri)
     m.rootUri  = uri
@@ -433,6 +448,8 @@ function m.awaitReload(scp)
             waker()
         end
     end
+
+    m.onWatch('reload', scp.uri)
 end
 
 ---@param uri uri