Fix #33

Author: wbthomason

lua/lsp-status/diagnostics.lua

@@ -1,5 +1,5 @@
 -- Gather diagnostics
-local function get_all_diagnostics()
+local function get_all_diagnostics(bufnr)
   local result = {}
   local levels = {
     errors = 'Error',
@@ -9,7 +9,7 @@ local function get_all_diagnostics()
   }
 
   for k, level in pairs(levels) do
-    result[k] = vim.lsp.diagnostic.get_count(0, level)
+    result[k] = vim.lsp.diagnostic.get_count(bufnr, level)
   end
 
   return result

lua/lsp-status/statusline.lua

@@ -10,12 +10,13 @@ local aliases = {
   pyls_ms = 'MPLS',
 }
 
-local function statusline_lsp()
-  if #vim.lsp.buf_get_clients() == 0 then
+local function statusline_lsp(bufnr)
+  bufnr = bufnr or 0
+  if #vim.lsp.buf_get_clients(bufnr) == 0 then
     return ''
   end
 
-  local buf_diagnostics = diagnostics()
+  local buf_diagnostics = diagnostics(bufnr)
   local buf_messages = messages()
   local only_hint = true
   local some_diagnostics = false