benbot/nvim-treesitter
1
0
Fork 0
mirror of https://github.com/nvim-treesitter/nvim-treesitter.git synced 2026-07-01 19:17:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(queries): allow for user overrides

Browse source
This commit is contained in:
Steven Sojka 2020-06-12 14:11:37 -05:00 • committed by Steven Sojka
parent ca438b7b2e
commit 44108fe03e
2 changed files with 18 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
lua/nvim-treesitter/query.lua
View file

@ -3,14 +3,21 @@ local ts = vim.treesitter
local M = {}
local function read_query_file(fname)
return table.concat(vim.fn.readfile(fname), '\n')
local function read_query_files(filenames)
local contents = {}
for _,filename in ipairs(filenames) do
vim.list_extend(contents, vim.fn.readfile(filename))
end
return table.concat(contents, '\n')
end
function M.get_query(ft, query_name)
local query_files = api.nvim_get_runtime_file(string.format('queries/%s/%s.scm', ft, query_name), false)
local query_files = api.nvim_get_runtime_file(string.format('queries/%s/%s.scm', ft, query_name), true)
if #query_files > 0 then
return ts.parse_query(ft, read_query_file(query_files[1]))
return ts.parse_query(ft, read_query_files(query_files))
end
end