vim: fix builtin diagnostics

1 files changed, 8 insertions, 8 deletions

diff --git a/.vim/init.lua b/.vim/init.lua
index 2f37ad2..07b9fb2 100644
--- a/.vim/init.lua
+++ b/.vim/init.lua
@@ -254,17 +254,17 @@ function _G.statusline_color(mode)
 end
 function _G.statusline_lsp_diagnostics(bufnr)
   -- Error, Warning, Information, Hint
-  local error_num = vim.lsp.diagnostic.get_count(bufnr, 'Error')
-  local warn_num = vim.lsp.diagnostic.get_count(bufnr, 'Warning')
+  local error_num = vim.diagnostic.get(bufnr, {severity = vim.diagnostic.severity.ERROR})
+  local warn_num = vim.diagnostic.get(bufnr, {severity = vim.diagnostic.severity.WARN})
 
   local s = ''
-  if error_num > 0 then
-    s = s .. '%#LspDiagnosticsError#'
-    s = s .. vim.fn.sign_getdefined('LspDiagnosticsSignError')[1].text .. error_num
+  if #error_num > 0 then
+    s = s .. '%#DiagnosticError#'
+    s = s .. vim.fn.sign_getdefined('DiagnosticSignError')[1].text .. #error_num
   end
-  if warn_num > 0 then
-    s = s .. '%#LspDiagnosticsWarning#'
-    s = s .. vim.fn.sign_getdefined('LspDiagnosticsSignWarning')[1].text .. warn_num
+  if #warn_num > 0 then
+    s = s .. '%#DiagnosticWarn#'
+    s = s .. vim.fn.sign_getdefined('DiagnosticSignWarn')[1].text .. #warn_num
   end
   return s
 end