feat(sources): treesitter: handle cursor pos in insert mode differently

Bekaboo · Jul 4, 2023 · c25bef8 · c25bef8
1 parent c72bd7f
commit c25bef8
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/lua/dropbar/sources/treesitter.lua b/lua/dropbar/sources/treesitter.lua
@@ -157,10 +157,16 @@ local function get_symbols(buf, win, cursor)
   local symbols = {}
   local prev_type_rank = math.huge
   local prev_row = math.huge
-  local current_node = vim.treesitter.get_node({
-    bufnr = buf,
-    pos = { cursor[1] - 1, cursor[2] },
-  })
+  local current_node =
+    vim.treesitter.get_node({
+      bufnr = buf,
+      pos = {
+        cursor[1] - 1,
+        cursor[2] - (cursor[2] >= 1 and vim.api
+          .nvim_get_mode().mode
+          :match('^i') and 1 or 0),
+      },
+    })
   while current_node do
     local name = get_node_short_name(current_node, buf)
     local type, type_rank = get_node_short_type(current_node)