fix: Support variables in sh

Author: Andrew Ferrier

lua/debugprint/utils.lua

@@ -21,10 +21,12 @@ M.find_treesitter_variable = function()
             local node_type = node:type()
             local variable_name = vim.treesitter.query.get_node_text(node, 0)
 
-            if node_type ~= "identifier" then
-                return nil
-            else
+            -- lua -> identifier
+            -- sh -> variable_name
+            if node_type == "identifier" or node_type == "variable_name" then
                 return variable_name
+            else
+                return nil
             end
         end
     end

tests/debugprint.lua

@@ -635,7 +635,7 @@ describe("can handle treesitter identifiers", function()
             "end",
         }, "lua", 2, 10)
 
-        feedkeys("g?v<CR>")
+        feedkeys("g?v")
 
         check_lines({
             "function x()",
@@ -646,7 +646,24 @@ describe("can handle treesitter identifiers", function()
             "end",
         })
 
-        assert.are.same(vim.api.nvim_win_get_cursor(0), { 3, 4 })
+        assert.are.same(vim.api.nvim_win_get_cursor(0), { 2, 10 })
+    end)
+
+    it("standard (bash)", function()
+        debugprint.setup({})
+
+        local filename = init_file({
+            "XYZ=123",
+        }, "sh", 1, 1)
+
+        feedkeys("g?v")
+
+        check_lines({
+            "XYZ=123",
+            'echo "DEBUGPRINT[1]: ' .. filename .. ':1: XYZ=${XYZ}"',
+        })
+
+        assert.are.same(vim.api.nvim_win_get_cursor(0), { 1, 1 })
     end)
 
     it("non-identifier", function()