nix/config/nvim/lua/plugins/treesitter.lua

138 lines
4.6 KiB
Lua

vim.pack.add({
'https://github.com/nvim-treesitter/nvim-treesitter',
'https://github.com/nvim-treesitter/nvim-treesitter-textobjects',
'https://github.com/Wansmer/treesj',
})
vim.api.nvim_create_autocmd('PackChanged', {
callback = function(ev)
local name, kind = ev.data.spec.name, ev.data.kind
if kind == 'delete' then
return
end
if name == 'nvim-treesitter' then
vim.schedule(function()
vim.cmd('TSUpdate all')
end)
end
end,
})
return {
{
'nvim-treesitter/nvim-treesitter',
after = function()
require('nvim-treesitter').setup({ auto_install = true })
end,
},
{
'nvim-treesitter/nvim-treesitter-textobjects',
before = function()
vim.g.no_plugin_maps = true
end,
after = function()
require('nvim-treesitter-textobjects').setup({
select = {
enable = true,
lookahead = true,
},
move = {
enable = true,
set_jumps = true,
},
})
local select = require('nvim-treesitter-textobjects.select')
local select_maps = {
{ 'aa', '@parameter.outer' },
{ 'ia', '@parameter.inner' },
{ 'as', '@class.outer' },
{ 'is', '@class.inner' },
{ 'aC', '@call.outer' },
{ 'iC', '@call.inner' },
{ 'af', '@function.outer' },
{ 'if', '@function.inner' },
{ 'ai', '@conditional.outer' },
{ 'ii', '@conditional.inner' },
{ 'aL', '@loop.outer' },
{ 'iL', '@loop.inner' },
}
for _, m in ipairs({ 'x', 'o' }) do
for _, t in ipairs(select_maps) do
vim.keymap.set(m, t[1], function()
select.select_textobject(t[2], 'textobjects', m)
end)
end
end
local move = require('nvim-treesitter-textobjects.move')
local move_textobjects = {
{ 'a', '@parameter.inner' },
{ 's', '@class.outer' },
{ 'f', '@function.outer' },
{ 'i', '@conditional.outer' },
{ '/', '@comment.outer' },
}
for _, m in ipairs({ 'n', 'x', 'o' }) do
for _, t in ipairs(move_textobjects) do
local key, capture = t[1], t[2]
vim.keymap.set(m, ']' .. key, function()
move.goto_next_start(capture, 'textobjects')
end)
vim.keymap.set(m, '[' .. key, function()
move.goto_previous_start(capture, 'textobjects')
end)
local upper = key:upper()
if upper ~= key then
vim.keymap.set(m, ']' .. upper, function()
move.goto_next_end(capture, 'textobjects')
end)
vim.keymap.set(m, '[' .. upper, function()
move.goto_previous_end(capture, 'textobjects')
end)
end
end
end
local ts_repeat =
require('nvim-treesitter-textobjects.repeatable_move')
for _, m in ipairs({ 'n', 'x', 'o' }) do
vim.keymap.set(m, ';', ts_repeat.repeat_last_move_next)
vim.keymap.set(m, ',', ts_repeat.repeat_last_move_previous)
vim.keymap.set(
m,
'f',
ts_repeat.builtin_f_expr,
{ expr = true }
)
vim.keymap.set(
m,
'F',
ts_repeat.builtin_F_expr,
{ expr = true }
)
vim.keymap.set(
m,
't',
ts_repeat.builtin_t_expr,
{ expr = true }
)
vim.keymap.set(
m,
'T',
ts_repeat.builtin_T_expr,
{ expr = true }
)
end
end,
},
{
'Wansmer/treesj',
after = function()
require('treesj').setup()
end,
keys = {
{ 'gt', '<cmd>lua require("treesj").toggle()<cr>' },
},
},
}