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

Merge pull request #80 from steelsojka/feat/user-query-overrides

Browse source 
feat(queries): allow user query overrides
This commit is contained in:
Kiyan Yazdani 2020-06-15 11:43:42 +02:00 • committed by GitHub
commit 6b26674ba0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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