mirror of
https://github.com/LazyVim/LazyVim.git
synced 2026-03-10 09:27:25 +00:00
102 lines
2.8 KiB
Lua
102 lines
2.8 KiB
Lua
---@class lazyvim.util.treesitter
|
|
local M = {}
|
|
|
|
M._installed = nil ---@type table<string,boolean>?
|
|
M._queries = {} ---@type table<string,boolean>
|
|
|
|
---@param update boolean?
|
|
function M.get_installed(update)
|
|
if update then
|
|
M._installed, M._queries = {}, {}
|
|
for _, lang in ipairs(require("nvim-treesitter").get_installed("parsers")) do
|
|
M._installed[lang] = true
|
|
end
|
|
end
|
|
return M._installed or {}
|
|
end
|
|
|
|
---@param lang string
|
|
---@param query string
|
|
function M.have_query(lang, query)
|
|
local key = lang .. ":" .. query
|
|
if M._queries[key] == nil then
|
|
M._queries[key] = vim.treesitter.query.get(lang, query) ~= nil
|
|
end
|
|
return M._queries[key]
|
|
end
|
|
|
|
---@param what string|number|nil
|
|
---@param query? string
|
|
---@overload fun(buf?:number):boolean
|
|
---@overload fun(ft:string):boolean
|
|
---@return boolean
|
|
function M.have(what, query)
|
|
what = what or vim.api.nvim_get_current_buf()
|
|
what = type(what) == "number" and vim.bo[what].filetype or what --[[@as string]]
|
|
local lang = vim.treesitter.language.get_lang(what)
|
|
if lang == nil or M.get_installed()[lang] == nil then
|
|
return false
|
|
end
|
|
if query and not M.have_query(lang, query) then
|
|
return false
|
|
end
|
|
return true
|
|
end
|
|
|
|
function M.foldexpr()
|
|
return M.have(nil, "folds") and vim.treesitter.foldexpr() or "0"
|
|
end
|
|
|
|
function M.indentexpr()
|
|
return M.have(nil, "indents") and require("nvim-treesitter").indentexpr() or -1
|
|
end
|
|
|
|
---@param cb fun()
|
|
function M.ensure_treesitter_cli(cb)
|
|
if vim.fn.executable("tree-sitter") == 1 then
|
|
return cb()
|
|
end
|
|
|
|
---@param msg? string
|
|
local function fail(msg)
|
|
return LazyVim.error({
|
|
"**treesitter-nvim** `main` requires the `tree-sitter` CLI executable to be installed.",
|
|
"Please install it manually from https://github.com/tree-sitter/tree-sitter/tree/master/crates/cli or",
|
|
"use your system package manager.",
|
|
"Run `:checkhealth nvim-treesitter` for more information.",
|
|
msg,
|
|
})
|
|
end
|
|
|
|
-- try installing with mason
|
|
if not pcall(require, "mason") then
|
|
return fail("`mason.nvim` is disabled in your config, so we cannot install it automatically.")
|
|
end
|
|
|
|
-- check again since we might have installed it already
|
|
if vim.fn.executable("tree-sitter") == 1 then
|
|
return cb()
|
|
end
|
|
|
|
local mr = require("mason-registry")
|
|
mr.refresh(function()
|
|
local p = mr.get_package("tree-sitter-cli")
|
|
if not p:is_installed() then
|
|
LazyVim.info("Installing `tree-sitter-cli` with `mason.nvim`...")
|
|
p:install(
|
|
nil,
|
|
vim.schedule_wrap(function(success)
|
|
if success then
|
|
LazyVim.info("Installed `tree-sitter-cli` with `mason.nvim`.")
|
|
cb()
|
|
else
|
|
fail("Failed to install `tree-sitter-cli` with `mason.nvim`.")
|
|
end
|
|
end)
|
|
)
|
|
end
|
|
end)
|
|
end
|
|
|
|
return M
|