1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
-- nvim-treesitter `main` branch: required on Neovim 0.12+ (`master` is frozen at 0.11).
-- Modules are gone: highlight/indent/folds are Neovim's, keymaps are plain keymaps.
local langs = {
'c',
'cpp',
'bash',
'ini',
'go',
'lua',
'python',
'typescript',
'vimdoc',
'vim',
'css',
'html',
'markdown',
'markdown_inline',
'php',
'sql',
'gdscript',
'godot_resource',
'rust',
}
local no_indent = { cpp = true, python = true, css = true }
local function move(fn, query)
return function()
require('nvim-treesitter-textobjects.move')[fn](query, 'textobjects')
end
end
local function select(query)
return function()
require('nvim-treesitter-textobjects.select').select_textobject(query, 'textobjects')
end
end
-- Replaces the `incremental_selection` module, dropped with the rest of them on `main`.
local nodes = {}
local function visual_select(node)
local sr, sc, er, ec = node:range()
if ec == 0 then -- node ends at col 0 of the next line; back up to the real last char
er = er - 1
ec = #(vim.api.nvim_buf_get_lines(0, er, er + 1, false)[1] or '')
end
vim.fn.setpos("'<", { 0, sr + 1, sc + 1, 0 })
vim.fn.setpos("'>", { 0, er + 1, ec, 0 })
vim.cmd('normal! gv')
end
local function node_init()
local node = vim.treesitter.get_node()
if not node then
return
end
nodes = { node }
visual_select(node)
end
local function node_incremental()
local node = nodes[#nodes]
if not node then
return node_init()
end
local parent = node:parent()
while parent and vim.deep_equal({ parent:range() }, { node:range() }) do
parent = parent:parent()
end
if parent then
nodes[#nodes + 1] = parent
end
visual_select(nodes[#nodes])
end
local function node_decremental()
if #nodes > 1 then
table.remove(nodes)
end
if nodes[#nodes] then
visual_select(nodes[#nodes])
end
end
return {
{
'nvim-treesitter/nvim-treesitter',
branch = 'main',
lazy = false,
build = ':TSUpdate',
config = function()
require('nvim-treesitter').install(langs)
vim.api.nvim_create_autocmd('FileType', {
callback = function(ev)
if not pcall(vim.treesitter.start, ev.buf) then
return
end
if not no_indent[ev.match] then
vim.bo[ev.buf].indentexpr = "v:lua.require'nvim-treesitter'.indentexpr()"
end
end,
})
vim.keymap.set('n', '<cr>', node_init, { desc = 'Init selection' })
vim.keymap.set('x', '<tab>', node_incremental, { desc = 'Increment selection' })
vim.keymap.set('x', '<s-tab>', node_decremental, { desc = 'Decrement selection' })
end,
},
{
'nvim-treesitter/nvim-treesitter-textobjects',
branch = 'main',
dependencies = { 'nvim-treesitter/nvim-treesitter' },
opts = {
select = {
lookahead = true,
include_surrounding_whitespace = false,
},
move = { set_jumps = true },
},
keys = {
{ 'aa', select('@parameter.outer'), mode = { 'x', 'o' }, desc = 'a parameter' },
{ 'ia', select('@parameter.inner'), mode = { 'x', 'o' }, desc = 'inner parameter' },
{ 'af', select('@function.outer'), mode = { 'x', 'o' }, desc = 'a function' },
{ 'if', select('@function.inner'), mode = { 'x', 'o' }, desc = 'inner function' },
{ 'ac', select('@class.outer'), mode = { 'x', 'o' }, desc = 'a class' },
{ 'ic', select('@class.inner'), mode = { 'x', 'o' }, desc = 'inner class' },
{ ']m', move('goto_next_start', '@function.outer'), mode = { 'n', 'x', 'o' }, desc = 'Next function start' },
{ ']]', move('goto_next_start', '@class.outer'), mode = { 'n', 'x', 'o' }, desc = 'Next class start' },
{ ']M', move('goto_next_end', '@function.outer'), mode = { 'n', 'x', 'o' }, desc = 'Next function end' },
{ '][', move('goto_next_end', '@class.outer'), mode = { 'n', 'x', 'o' }, desc = 'Next class end' },
{ '[m', move('goto_previous_start', '@function.outer'), mode = { 'n', 'x', 'o' }, desc = 'Previous function start' },
{ '[[', move('goto_previous_start', '@class.outer'), mode = { 'n', 'x', 'o' }, desc = 'Previous class start' },
{ '[M', move('goto_previous_end', '@function.outer'), mode = { 'n', 'x', 'o' }, desc = 'Previous function end' },
{ '[]', move('goto_previous_end', '@class.outer'), mode = { 'n', 'x', 'o' }, desc = 'Previous class end' },
{
'<leader>a',
function()
require('nvim-treesitter-textobjects.swap').swap_next('@parameter.inner')
end,
desc = 'Swap with next [a]rgument',
},
{
'<leader>A',
function()
require('nvim-treesitter-textobjects.swap').swap_previous('@parameter.inner')
end,
desc = 'Swap with previous [A]rgument',
},
},
},
}
|
