From 8b67005a9d765ff39e7f923514a6bf75eddac98e Mon Sep 17 00:00:00 2001
From: David JULIEN <swytch@mailo.com>
Date: Sat, 7 Jan 2023 19:17:00 +0100
Subject: [PATCH] [nvim] feat: use textobjects with treesitter

---
 .config/nvim/lua/plugin/packer.lua     |   5 ++
 .config/nvim/lua/plugin/treesitter.lua | 104 ++++++++++++++++++++++++-
 2 files changed, 107 insertions(+), 2 deletions(-)

diff --git a/.config/nvim/lua/plugin/packer.lua b/.config/nvim/lua/plugin/packer.lua
index 8c53feb..781d35e 100644
--- a/.config/nvim/lua/plugin/packer.lua
+++ b/.config/nvim/lua/plugin/packer.lua
@@ -54,6 +54,11 @@ return require("packer").startup(function()
         config = function() require("plugin.treesitter") end
     }
 
+    use { -- Additional text objects via treesitter
+        'nvim-treesitter/nvim-treesitter-textobjects',
+        after = 'nvim-treesitter',
+    }
+
     -- telescope
     use {
         "nvim-telescope/telescope.nvim",
diff --git a/.config/nvim/lua/plugin/treesitter.lua b/.config/nvim/lua/plugin/treesitter.lua
index 2b7f3a3..314c6f8 100644
--- a/.config/nvim/lua/plugin/treesitter.lua
+++ b/.config/nvim/lua/plugin/treesitter.lua
@@ -4,12 +4,112 @@
 -- Description   : treesitter config file
 
 require("nvim-treesitter.configs").setup {
-    ensure_installed = { "c", "lua", "python", "rust", "bash" },
-    ignore_install = { "javascript" },
+    ensure_installed = { "c", "lua", "rust", "bash", "vim" },
     highlight = {
         enable = true,
     },
     indent = {
         enable = true,
     },
+    incremental_selection = {
+        enable = true,
+        keymaps = {
+            init_selection = "<C-Space>",
+            node_incremental = "<C-Space>",
+            scope_incremental = "<Tab>",
+            node_decremental = "<BS>",
+        },
+    },
+    textobjects = {
+        select = {
+            enable = true,
+            lookahead = true, -- Automatically jump forward to textobj, similar to targets.vim
+            keymaps = {
+                -- use the capture groups defined in textobjects.scm
+                ["ap"] = {
+                    query = "@parameter.outer",
+                    desc = "Select parameter region"
+                },
+                ["ip"] = {
+                    query = "@parameter.inner",
+                    desc = "Select inner part of a parameter region"
+                },
+                ["af"] = {
+                    query = "@function.outer",
+                    desc = "Select a function block"
+                },
+                ["if"] = {
+                    query = "@function.inner",
+                    desc = "Select inner part of a function"
+                },
+                ["ac"] = {
+                    query = "@class.outer",
+                    desc = "Select a class block"
+                },
+                ["ic"] = {
+                    query = "@class.inner",
+                    desc = "Select inner part of a class"
+                },
+            },
+        },
+        move = {
+            enable = true,
+            set_jumps = true, -- whether to set jumps in the jumplist
+            goto_next_start = {
+                ["]m"] = {
+                    query = "@function.outer",
+                    desc = "Jump to next function",
+                },
+                ["])"] = {
+                    query = "@class.outer",
+                    desc = "Jump to next class",
+                },
+            },
+            goto_next_end = {
+                ["]M"] = {
+                    query = "@function.outer",
+                    desc = "Jump after next function",
+                },
+                ["]]"] = {
+                    query = "@class.outer",
+                    desc = "Jump after next class",
+                },
+            },
+            goto_previous_start = {
+                ["[m"] = {
+                    query = "@function.outer",
+                    desc = "Jump to previous function",
+                },
+                ["[)"] = {
+                    query = "@class.outer",
+                    desc = "Jump to previous class",
+                },
+            },
+            goto_previous_end = {
+                ["[M"] = {
+                    query = "@function.outer",
+                    desc = "Jump after previous function",
+                },
+                ["[]"] = {
+                    query = "@class.outer",
+                    desc = "Jump after previous class",
+                },
+            },
+        },
+        swap = {
+            enable = true,
+            swap_next = {
+                ["<leader>a"] = {
+                    query = "@parameter.inner",
+                    desc = "Swap with next parameter",
+                },
+            },
+            swap_previous = {
+                ["<leader>A"] = {
+                    query = "@parameter.inner",
+                    desc = "Swap with previous parameter",
+                },
+            },
+        },
+    },
 }