add-plugin: nvim-keysitter, my plugin :)

Author: ABDsheikho

lua/plugins/nvim-treesitter-textobjects.lua

@@ -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 {
   'nvim-treesitter/nvim-treesitter-textobjects',
   branch = 'main',
-  -- 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,
+  dependencies = {
+    {
+      'ABDsheikho/nvim-keysitter',
+      -- dir = '~/Repos/nvim-keysitter/',
+      -- name = 'keysitter',
+      config = function()
+        local keysitter = require 'keysitter'
+        local tsto = keysitter.new('treesitter-textobjects', { group_prefix = 'o' })
+
+        -- a setup function
+        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()
     -- put your config here
     -- configuration
@@ -174,44 +99,5 @@ return {
         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,
 }