use preview

Author: sumneko

script/provider/diagnostic.lua

@@ -489,8 +489,7 @@ function m.diagnosticsScope(uri, force)
 end
 
 ---@async
-function m.pullDiagnosticScope()
-    local results = {}
+function m.pullDiagnosticScope(callback)
     local processing = 0
 
     for _, scp in ipairs(scope.folders) do
@@ -515,7 +514,7 @@ function m.pullDiagnosticScope()
                 m.awaitDiagnosticsScope(scp.uri, function (fileUri)
                     local suc, result, unchanged = xpcall(m.pullDiagnostic, log.error, fileUri, true)
                     if suc then
-                        results[#results+1] = {
+                        callback {
                             uri       = fileUri,
                             result    = result,
                             unchanged = unchanged,
@@ -527,18 +526,21 @@ function m.pullDiagnosticScope()
         end
     end
 
-    while processing > 0 do
-        await.sleep(0.1)
+    -- sleep for ever
+    while true do
+        await.sleep(1.0)
     end
-
-    return results
 end
 
 ---@param uri uri
 ---@return 'server' | 'client'
 function m.getOwner(uri)
-    --TODO
-    return 'client'
+    -- TODO
+    if PREVIEW then
+        return 'client'
+    else
+        return 'server'
+    end
 end
 
 function m.refreshClient()

script/provider/provider.lua

@@ -1213,6 +1213,7 @@ m.register 'inlayHint/resolve' {
 }
 
 m.register 'textDocument/diagnostic' {
+    preview = true,
     capability = {
         diagnosticProvider = {
             identifier            = 'identifier',
@@ -1245,9 +1246,10 @@ m.register 'textDocument/diagnostic' {
 }
 
 m.register 'workspace/diagnostic' {
+    preview = true,
     capability = {
         diagnosticProvider = {
-            workspaceDiagnostics  = true,
+            workspaceDiagnostics  = false,
         }
     },
     ---@async
@@ -1258,18 +1260,16 @@ m.register 'workspace/diagnostic' {
             excepts[#excepts+1] = id.value
         end
         core.clearCacheExcept(excepts)
-        local results = core.pullDiagnosticScope()
-        local items = {}
-        for i, result in ipairs(results) do
+        local function convertItem(result)
             if result.unchanged then
-                items[i] = {
+                return {
                     kind     = 'unchanged',
                     resultId = result.uri,
                     uri      = result.uri,
                     version  = result.version,
                 }
             else
-                items[i] = {
+                return {
                     kind     = 'full',
                     resultId = result.uri,
                     items    = result.result or {},
@@ -1278,9 +1278,17 @@ m.register 'workspace/diagnostic' {
                 }
             end
         end
-        return {
-            items = items,
-        }
+        core.pullDiagnosticScope(function (result)
+            proto.notify('$/progress', {
+                token = params.partialResultToken,
+                value = {
+                    items = {
+                        convertItem(result)
+                    }
+                }
+            })
+        end)
+        return { items = {} }
     end
 }