add-plugin: nvim-keysitter, my plugin :)
This commit is contained in:
Abdulrahman Sheikho
parent
d73982f20d
commit
dbdbd789f3
|
|
@ -1,143 +1,68 @@
|
||||||
---@class TreesitterTextobjectGroup
|
|
||||||
---@field group_leader string
|
|
||||||
local TSTO_GROUP = {}
|
|
||||||
|
|
||||||
---@param query_group string
|
|
||||||
---@param opts table<string,string>
|
|
||||||
---@return any
|
|
||||||
function TSTO_GROUP:new(query_group, opts)
|
|
||||||
self.query_group = query_group
|
|
||||||
self.__index = self
|
|
||||||
return setmetatable(opts or {
|
|
||||||
group_leader = '',
|
|
||||||
}, self)
|
|
||||||
end
|
|
||||||
|
|
||||||
---@param key string
|
|
||||||
---@param object string
|
|
||||||
---@return TreesitterTextobjectGroup
|
|
||||||
function TSTO_GROUP:set(key, object)
|
|
||||||
self.key = key
|
|
||||||
self.object = object
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
|
|
||||||
---@param query_name string
|
|
||||||
---@return string|nil
|
|
||||||
function TSTO_GROUP:textobject_check(query_name)
|
|
||||||
local lang = vim.treesitter.language.get_lang(vim.bo.filetype) or ''
|
|
||||||
local query = vim.treesitter.query.get(lang, self.query_group) or {}
|
|
||||||
|
|
||||||
return vim.iter(query.captures or {}):any(function(val)
|
|
||||||
return val:match(query_name)
|
|
||||||
end) and query_name or nil
|
|
||||||
end
|
|
||||||
|
|
||||||
---@param attr string|nil
|
|
||||||
---@return TreesitterTextobjectGroup
|
|
||||||
function TSTO_GROUP:sel_outer(attr)
|
|
||||||
attr = attr or 'outer'
|
|
||||||
|
|
||||||
local obj = TSTO_GROUP:textobject_check(self.object .. '.' .. attr)
|
|
||||||
if not obj then
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
obj = '@' .. obj
|
|
||||||
|
|
||||||
vim.keymap.set({ 'x', 'o' }, 'a' .. self.group_leader .. self.key, function()
|
|
||||||
require('nvim-treesitter-textobjects.select').select_textobject(obj, self.query_group)
|
|
||||||
end, { buffer = true, desc = 'arround ' .. self.object })
|
|
||||||
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
|
|
||||||
---@param attr string|nil
|
|
||||||
---@return TreesitterTextobjectGroup
|
|
||||||
function TSTO_GROUP:sel_inner(attr)
|
|
||||||
attr = attr or 'inner'
|
|
||||||
|
|
||||||
local obj = TSTO_GROUP:textobject_check(self.object .. '.' .. attr)
|
|
||||||
if not obj then
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
obj = '@' .. obj
|
|
||||||
|
|
||||||
vim.keymap.set({ 'x', 'o' }, 'i' .. self.group_leader .. self.key, function()
|
|
||||||
require('nvim-treesitter-textobjects.select').select_textobject(obj, self.query_group)
|
|
||||||
end, { buffer = true, desc = 'inner ' .. self.object })
|
|
||||||
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
|
|
||||||
---@param opts table<string,string>
|
|
||||||
---@return TreesitterTextobjectGroup
|
|
||||||
function TSTO_GROUP:goto_start(opts)
|
|
||||||
---@class opts
|
|
||||||
---@field attribute? string
|
|
||||||
---@field key? string
|
|
||||||
---@field desc? string
|
|
||||||
opts = vim.tbl_extend('keep', opts or {}, {
|
|
||||||
attribute = 'outer',
|
|
||||||
key = self.key,
|
|
||||||
desc = 'start',
|
|
||||||
})
|
|
||||||
|
|
||||||
local obj = TSTO_GROUP:textobject_check(self.object .. '.' .. opts.attribute)
|
|
||||||
if not obj then
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
obj = '@' .. obj
|
|
||||||
|
|
||||||
vim.keymap.set({ 'n', 'x', 'o' }, ']' .. self.group_leader .. opts.key, function()
|
|
||||||
require('nvim-treesitter-textobjects.move').goto_next_start(obj, self.query_group)
|
|
||||||
end, { buffer = true, desc = 'next ' .. self.object .. ' ' .. opts.desc })
|
|
||||||
vim.keymap.set({ 'n', 'x', 'o' }, '[' .. self.group_leader .. opts.key, function()
|
|
||||||
require('nvim-treesitter-textobjects.move').goto_previous_start(obj, self.query_group)
|
|
||||||
end, { buffer = true, desc = 'previous ' .. self.object .. ' ' .. opts.desc })
|
|
||||||
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
|
|
||||||
---@param attr string|nil
|
|
||||||
---@return TreesitterTextobjectGroup
|
|
||||||
function TSTO_GROUP:goto_end(opts)
|
|
||||||
attr = attr or 'outer'
|
|
||||||
opts = vim.tbl_extend('keep', opts or {}, {
|
|
||||||
attribute = 'outer',
|
|
||||||
key = self.key:upper(),
|
|
||||||
desc = 'end',
|
|
||||||
})
|
|
||||||
|
|
||||||
local obj = TSTO_GROUP:textobject_check(self.object .. '.' .. opts.attribute)
|
|
||||||
if not obj then
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
obj = '@' .. obj
|
|
||||||
|
|
||||||
vim.keymap.set({ 'n', 'x', 'o' }, ']' .. self.group_leader .. opts.key, function()
|
|
||||||
require('nvim-treesitter-textobjects.move').goto_next_end(obj, self.query_group)
|
|
||||||
end, { buffer = true, desc = 'next ' .. self.object .. ' ' .. opts.desc })
|
|
||||||
vim.keymap.set({ 'n', 'x', 'o' }, '[' .. self.group_leader .. opts.key, function()
|
|
||||||
require('nvim-treesitter-textobjects.move').goto_previous_end(obj, self.query_group)
|
|
||||||
end, { buffer = true, desc = 'previous ' .. self.object .. ' ' .. opts.desc })
|
|
||||||
|
|
||||||
return self
|
|
||||||
end
|
|
||||||
|
|
||||||
return {
|
return {
|
||||||
'nvim-treesitter/nvim-treesitter-textobjects',
|
'nvim-treesitter/nvim-treesitter-textobjects',
|
||||||
branch = 'main',
|
branch = 'main',
|
||||||
-- init = function()
|
dependencies = {
|
||||||
-- -- Disable entire built-in ftplugin mappings to avoid conflicts.
|
{
|
||||||
-- -- See https://github.com/neovim/neovim/tree/master/runtime/ftplugin for built-in ftplugins.
|
'ABDsheikho/nvim-keysitter',
|
||||||
-- vim.g.no_plugin_maps = true
|
-- dir = '~/Repos/nvim-keysitter/',
|
||||||
--
|
-- name = 'keysitter',
|
||||||
-- -- Or, disable per filetype (add as you like)
|
config = function()
|
||||||
-- -- vim.g.no_python_maps = true
|
local keysitter = require 'keysitter'
|
||||||
-- -- vim.g.no_ruby_maps = true
|
local tsto = keysitter.new('treesitter-textobjects', { group_prefix = 'o' })
|
||||||
-- -- vim.g.no_rust_maps = true
|
|
||||||
-- -- vim.g.no_go_maps = true
|
-- a setup function
|
||||||
-- end,
|
tsto.setup({ 'FileType', 'BufEnter' }, 'keysitter', function()
|
||||||
|
-- tsto:set('f', 'function'):around():inner():next():prev()
|
||||||
|
|
||||||
|
for k, v in pairs {
|
||||||
|
-- ['b'] = 'block', -- or b for brace-less languages like python
|
||||||
|
['f'] = 'function',
|
||||||
|
['i'] = 'conditional',
|
||||||
|
['l'] = 'loop',
|
||||||
|
['r'] = 'return',
|
||||||
|
['t'] = 'attribute',
|
||||||
|
['x'] = 'regex',
|
||||||
|
} do
|
||||||
|
tsto:set(k, v):around():inner():next():prev()
|
||||||
|
end
|
||||||
|
|
||||||
|
tsto:set('/', 'comment'):around():inner():goto_start()
|
||||||
|
tsto:set('{', 'block'):around():inner():goto_start():goto_end { key = '}' }
|
||||||
|
tsto:set('(', 'call'):around():inner():goto_start():goto_end { key = ')' }
|
||||||
|
tsto:set(',', 'parameter'):around():inner():goto_start():goto_end { key = '.' }
|
||||||
|
tsto:set(';', 'statement'):around():inner({ attribute = 'outer' }):goto_start():goto_end { key = ':' }
|
||||||
|
|
||||||
|
tsto
|
||||||
|
:set('=', 'assignment')
|
||||||
|
:around()
|
||||||
|
:inner({ attribute = 'rhs' })
|
||||||
|
:goto_start({ attribute = 'lhs' })
|
||||||
|
:next_start({ attribute = 'rhs', key = '-' }, { desc = 'next assignment rhs' })
|
||||||
|
:previous_start({ attribute = 'rhs', key = '-' }, { desc = 'previous assignment rhs' })
|
||||||
|
|
||||||
|
tsto
|
||||||
|
:set('c', 'class')
|
||||||
|
:around()
|
||||||
|
:inner()
|
||||||
|
:next_start({ motion = ']', group_prefix = '', key = ']' })
|
||||||
|
:next_end({ motion = ']', group_prefix = '', key = '[' })
|
||||||
|
:previous_start({ motion = '[', group_prefix = '', key = '[' })
|
||||||
|
:previous_end { motion = '[', group_prefix = '', key = ']' }
|
||||||
|
end, { desc = 'Set keysitter keymaps for nvim-treesitter-textobjects' })
|
||||||
|
end,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
init = function()
|
||||||
|
-- Disable entire built-in ftplugin mappings to avoid conflicts.
|
||||||
|
-- See https://github.com/neovim/neovim/tree/master/runtime/ftplugin for built-in ftplugins.
|
||||||
|
vim.g.no_plugin_maps = true
|
||||||
|
|
||||||
|
-- Or, disable per filetype (add as you like)
|
||||||
|
-- vim.g.no_python_maps = true
|
||||||
|
-- vim.g.no_ruby_maps = true
|
||||||
|
-- vim.g.no_rust_maps = true
|
||||||
|
-- vim.g.no_go_maps = true
|
||||||
|
end,
|
||||||
config = function()
|
config = function()
|
||||||
-- put your config here
|
-- put your config here
|
||||||
-- configuration
|
-- configuration
|
||||||
|
|
@ -174,44 +99,5 @@ return {
|
||||||
set_jumps = true,
|
set_jumps = true,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
vim.api.nvim_create_autocmd({ 'FileType', 'BufEnter' }, {
|
|
||||||
desc = 'Set treesitter keymaps when treesitter is avaliable for the file',
|
|
||||||
group = vim.api.nvim_create_augroup('TSparser', { clear = true }),
|
|
||||||
callback = function()
|
|
||||||
local buf = vim.api.nvim_get_current_buf()
|
|
||||||
local has_parser, parser = pcall(vim.treesitter.get_parser, buf, nil, { error = false })
|
|
||||||
|
|
||||||
if not has_parser or parser == nil then
|
|
||||||
return
|
|
||||||
end
|
|
||||||
|
|
||||||
local tsto = TSTO_GROUP:new('textobjects', { group_leader = 'o' })
|
|
||||||
for k, v in pairs {
|
|
||||||
-- ['b'] = 'block', -- for brace-less languages like python
|
|
||||||
['c'] = 'class',
|
|
||||||
['f'] = 'function',
|
|
||||||
['i'] = 'conditional',
|
|
||||||
['l'] = 'loop',
|
|
||||||
['p'] = 'parameter',
|
|
||||||
['r'] = 'return',
|
|
||||||
['t'] = 'attribute',
|
|
||||||
['x'] = 'regex',
|
|
||||||
} do
|
|
||||||
tsto:set(k, v):sel_outer():sel_inner():goto_start():goto_end()
|
|
||||||
end
|
|
||||||
|
|
||||||
tsto:set('{', 'block'):sel_outer():sel_inner():goto_start():goto_end { key = '}' }
|
|
||||||
tsto:set('(', 'call'):sel_outer():sel_inner():goto_start():goto_end { key = ')' }
|
|
||||||
tsto:set('/', 'comment'):sel_outer():sel_inner():goto_start()
|
|
||||||
tsto:set(';', 'statement'):sel_outer():goto_start():goto_end { key = ':' }
|
|
||||||
tsto
|
|
||||||
:set('=', 'assignment')
|
|
||||||
:sel_outer()
|
|
||||||
:sel_inner('rhs')
|
|
||||||
:goto_start({ attribute = 'lhs' })
|
|
||||||
:goto_start { attribute = 'rhs', key = '-', desc = 'rhs' }
|
|
||||||
end,
|
|
||||||
})
|
|
||||||
end,
|
end,
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue