From 2526baf4cc5ea540188345c856adb162ce102674 Mon Sep 17 00:00:00 2001 From: Thomas Vigouroux Date: Sun, 19 Apr 2020 09:45:54 +0200 Subject: feat: add parsers module and better match iter The `parsers` module manages parsers for us, for now only in a really basic way. iter_prepared_mathes iters on an enhanced versions of the matches, where captures are directly accessible via their names to allow things like : ((itentifier) @def.first (identifier) @def.last) To be handled like this in lua: match.def.first match.def.last Also adds a `set!` predicate to allow setting data within the prepared match (see queries/lua/locals.scm) for examples. --- lua/nvim-treesitter/parsers.lua | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 lua/nvim-treesitter/parsers.lua (limited to 'lua/nvim-treesitter/parsers.lua') diff --git a/lua/nvim-treesitter/parsers.lua b/lua/nvim-treesitter/parsers.lua new file mode 100644 index 00000000..c8517773 --- /dev/null +++ b/lua/nvim-treesitter/parsers.lua @@ -0,0 +1,21 @@ +local api = vim.api +local ts = vim.treesitter + +local M = {} + +function M.has_parser(lang) + local lang = lang or api.nvim_buf_get_option(0, 'filetype') + return #api.nvim_get_runtime_file('parser/' .. lang .. '.*', false) > 0 +end + +function M.get_parser(bufnr) + if M.has_parser() then + local buf = bufnr or api.nvim_get_current_buf() + if not M[buf] then + M[buf] = ts.get_parser(buf) + end + return M[buf] + end +end + +return M -- cgit v1.2.3