fix: g?v for comments in lua

andrewferrier · Apr 2, 2024 · f9b9b87 · f9b9b87
1 parent bb6d1c9
commit f9b9b87
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/lua/debugprint/utils.lua b/lua/debugprint/utils.lua
@@ -127,7 +127,13 @@ M.find_treesitter_variable = function()
         return nil
     else
         local node_type = node:type()
-        local parent_node_type = node:parent():type()
+        local parent_node_type
+
+        if node:parent() ~= nil then
+            -- This check is necessary; it triggers for example in comments in
+            -- lua code
+            parent_node_type = node:parent():type()
+        end
 
         local variable_name
 

diff --git a/tests/debugprint.lua b/tests/debugprint.lua
@@ -1637,6 +1637,25 @@ if vim.fn.has("nvim-0.9.0") == 1 then
                 "</html>",
             })
         end)
+
+        it("comment in lua", function()
+            local filename = init_file({
+                "x = 3",
+                "-- abc",
+                "a = 2",
+            }, "lua", 2, 4)
+
+            feedkeys("g?v<CR>")
+
+            check_lines({
+                "x = 3",
+                "-- abc",
+                "print('DEBUGPRINT[1]: "
+                    .. filename
+                    .. ":2: abc=' .. vim.inspect(abc))",
+                "a = 2",
+            })
+        end)
     end)
 end