perf(status): use vim.diagnostic.count if available

Author: mehalter

lua/astroui/status/condition.lua

@@ -122,7 +122,12 @@ end
 function M.has_diagnostics(bufnr)
   if type(bufnr) == "table" then bufnr = bufnr.bufnr end
   if package.loaded["astrolsp"] and require("astrolsp").config.features.diagnostics_mode == 0 then return false end
-  return #vim.diagnostic.get(bufnr or 0) > 0
+  -- TODO: remove when dropping support for neovim 0.9
+  if vim.diagnostic.count then
+    return vim.tbl_contains(vim.diagnostic.count(bufnr or 0), function(v) return v > 0 end, { predicate = true })
+  else
+    return #vim.diagnostic.get(bufnr or 0) > 0
+  end
 end
 
 --- A condition function if there is a defined filetype

lua/astroui/status/provider.lua

@@ -452,7 +452,13 @@ function M.diagnostics(opts)
   if not opts or not opts.severity then return end
   return function(self)
     local bufnr = self and self.bufnr or 0
-    local count = #vim.diagnostic.get(bufnr, opts.severity and { severity = vim.diagnostic.severity[opts.severity] })
+    local count
+    -- TODO: remove when dropping support for neovim 0.9
+    if vim.diagnostic.count then
+      count = vim.diagnostic.count(bufnr)[vim.diagnostic.severity[opts.severity]] or 0
+    else
+      count = #vim.diagnostic.get(bufnr, opts.severity and { severity = vim.diagnostic.severity[opts.severity] })
+    end
     return status_utils.stylize(count ~= 0 and tostring(count) or "", opts)
   end
 end