refactor: Migrate ft-specific treesitter logic into filetypes.lua

Keep this undocumented/unofficial for now until it's a bit more stable.

Author: andrewferrier

lua/debugprint/filetypes.lua

@@ -12,6 +12,13 @@ local shell = {
     right = '"',
     mid_var = "${",
     right_var = '}"',
+    find_treesitter_variable = function(opts)
+        if opts.node:type() == "variable_name" then
+            return opts.node_text
+        else
+            return nil
+        end
+    end,
 }
 
 local docker = vim.deepcopy(shell)
@@ -28,6 +35,16 @@ local js = {
     right = '")',
     mid_var = '", ',
     right_var = ")",
+    find_treesitter_variable = function(opts)
+        if
+            opts.node:type() == "identifier"
+            or opts.node:type() == "shorthand_property_identifier_pattern"
+        then
+            return opts.node_text
+        else
+            return nil
+        end
+    end,
 }
 
 local cs = {
@@ -133,6 +150,13 @@ return {
         right = "')",
         mid_var = "' .. vim.inspect(",
         right_var = "))",
+        find_treesitter_variable = function(opts)
+            if opts.node:type() == "identifier" then
+                return opts.node_text
+            else
+                return nil
+            end
+        end,
     },
     ["make"] = {
         left = '\t@echo >&2 "',

lua/debugprint/init.lua

@@ -166,7 +166,7 @@ M.debugprint_cache = function(opts)
     if opts and opts.prerepeat == true then
         if get_filetype_config() and opts.variable == true then
             opts.variable_name = utils.get_variable_name(
-                global_opts.ignore_treesitter or opts.ignore_treesitter
+                global_opts.ignore_treesitter or opts.ignore_treesitter, get_filetype_config()
             )
 
             if opts.variable_name == false then

lua/debugprint/utils.lua

@@ -29,6 +29,16 @@ local get_node_at_cursor = function()
     end
 end
 
+local get_node_text = function(node)
+    if vim.treesitter.get_node_text then
+        -- vim.treesitter.query.get_node_text deprecated as of NeoVim
+        -- 0.9
+        return vim.treesitter.get_node_text(node, 0)
+    else
+        return vim.treesitter.query.get_node_text(node, 0)
+    end
+end
+
 M.is_modifiable = function()
     if
         not vim.api.nvim_get_option_value(
@@ -43,15 +53,15 @@ M.is_modifiable = function()
     end
 end
 
-M.get_variable_name = function(ignore_treesitter)
+M.get_variable_name = function(ignore_treesitter, filetype_config)
     local variable_name = M.get_visual_selection()
 
     if variable_name == false then
         return false
     end
 
     if variable_name == nil and ignore_treesitter ~= true then
-        variable_name = M.find_treesitter_variable()
+        variable_name = M.find_treesitter_variable(filetype_config)
     end
 
     if variable_name == nil then
@@ -163,44 +173,27 @@ M.get_effective_filetypes = function()
     end
 end
 
-M.find_treesitter_variable = function()
-    local node = get_node_at_cursor()
+M.find_treesitter_variable = function(filetype_config)
+    local obj = {}
+
+    obj.node = get_node_at_cursor()
 
-    if node == nil then
+    if obj.node == nil then
         return nil
     else
-        local node_type = node:type()
-        local parent_node_type
+        obj.node_text = get_node_text(obj.node)
+        obj.parent_node = obj.node:parent()
 
-        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
-
-        if vim.treesitter.get_node_text then
-            -- vim.treesitter.query.get_node_text deprecated as of NeoVim
-            -- 0.9
-            variable_name = vim.treesitter.get_node_text(node, 0)
+        if obj.parent_node then
+            obj.parent_node_text = get_node_text(obj.parent_node)
         else
-            variable_name = vim.treesitter.query.get_node_text(node, 0)
+            obj.parent_node_text = nil
         end
 
-        -- lua, typescript -> identifier
-        -- sh              -> variable_name
-        -- typescript      -> shorthand_property_identifier_pattern (see issue #60)
-        -- Makefile        -> variable_reference
-        if
-            node_type == "identifier"
-            or node_type == "variable_name"
-            or node_type == "shorthand_property_identifier_pattern"
-            or parent_node_type == "variable_reference"
-        then
-            return variable_name
+        if vim.list_contains(vim.tbl_keys(filetype_config), "find_treesitter_variable") then
+            return filetype_config.find_treesitter_variable(obj)
         else
-            return nil
+            return obj.node_text
         end
     end
 end