feat: add checkhealth

1 files changed, 49 insertions, 0 deletions

diff --git a/lua/nvim-treesitter/health.lua b/lua/nvim-treesitter/health.lua
new file mode 100644
index 00000000..b5e86980
--- /dev/null
+++ b/lua/nvim-treesitter/health.lua
@@ -0,0 +1,49 @@
+local api = vim.api
+
+local install = require'nvim-treesitter.install'
+local queries = require'nvim-treesitter.query'
+local locals = require'nvim-treesitter.locals'
+
+local health_start = vim.fn["health#report_start"]
+local health_ok = vim.fn['health#report_ok']
+local health_info = vim.fn['health#report_info']
+local health_warn = vim.fn['health#report_warn']
+local health_error = vim.fn['health#report_error']
+
+local M = {}
+
+-- TODO(vigoux): Maybe we should move each check to be perform in its own module
+function M.checkhealth()
+  -- Installation dependency checks
+  health_start('Installation')
+  install.checkhealth()
+
+  local missing_parsers = {}
+  -- Parser installation checks
+  for parser_name, repo in pairs(install.repositories) do
+    local installed = #api.nvim_get_runtime_file('parser/'..parser_name..'.so', false)
+
+    -- Only print informations about installed parsers
+    if installed == 1 then
+      health_start(parser_name .. " parser healthcheck")
+      health_ok(parser_name .. " parser found.")
+
+      locals.checkhealth(parser_name)
+    elseif installed > 1 then
+      health_warn(string.format("Multiple parsers found for %s, only %s will be used.", parser_name, installed[1]))
+    else
+      table.insert(missing_parsers, parser_name)
+    end
+  end
+
+  -- Add informations on parsers we dont find
+  if #missing_parsers > 0 then
+    health_start('Missing parsers')
+
+    -- TODO(vigoux): The installation command should be changed so that its easier to find
+    health_warn('Some parsers are not installed:\n' .. table.concat(missing_parsers, '\n'), {
+      "Install them using `:lua require'nvim-treesitter'.install_parser('language')`"})
+  end
+end
+
+return M