nvimConfig/plugin/04_treesitter.lua

local Config = require('config')

local M = {}

-- Default parsers list moved from startup config
M.default_parsers = { 
  "bash", "bibtex", "c", "caddy", "cmake", "comment", "commonlisp", "cpp", "css", "csv",
  "cuda", "desktop", "diff", "dockerfile", "doxygen", "editorconfig", "fortran", "git_config", "git_rebase",
  "gitattributes", "gitcommit", "gitignore", "gnuplot", "go", "gpg", "html", "javascript", "jq", "json", "json5",
  "julia", "just", "latex", "ledger", "lua", "luadoc", "luap", "luau", "make", "markdown", "markdown_inline",
  "matlab", "meson", "muttrc", "nix", "nu", "passwd", "powershell", "prql", "python", "r", "query", "readline", "regex",
  "requirements", "rnoweb", "rust", "sql", "ssh_config", "swift", "tmux", "toml", "tsv", "tsx", "typescript", "typst",
  "vala", "vim", "vimdoc", "yaml", "zig", 
}

-- Cache treesitter utils to avoid repeated requires
local smart_send = require('nix_smart_send')

-- Helper function to check if value exists in list (optimized with early return)
local function is_in_list(list, value)
  if not list or not value then
    return false
  end

  for _, v in ipairs(list) do
    if v == value then
      return true
    end
  end
  return false
end


function M.add_global_node(nodes)
  if not nodes then
    return nil
  end

  local node_type = M.get_type()
  if not node_type then
    return nodes
  end

  -- Create a copy to avoid modifying the original
  local global_nodes = vim.deepcopy(nodes)

  -- Check if node type already exists to avoid duplicates
  if not is_in_list(global_nodes, node_type) then
    table.insert(global_nodes, node_type)
  end

  return global_nodes
end

function M.remove_global_node(nodes)
  if not nodes then
    return nil
  end

  local node_type = M.get_type()
  if not node_type then
    return nodes
  end

  local global_nodes = vim.deepcopy(nodes)

  -- Remove all occurrences (iterate backwards to avoid index issues)
  for i = #global_nodes, 1, -1 do
    if global_nodes[i] == node_type then
      table.remove(global_nodes, i)
    end
  end

  return global_nodes
end

function M.set_global_nodes()
  local input = vim.fn.input("Enter root nodes: ")
  if input == "" then
    return {}
  end

  local nodes_in = {}
  -- Trim whitespace from each node name
  for node in string.gmatch(input, '([^,]+)') do
    local trimmed = vim.trim(node)
    if trimmed ~= "" then
      table.insert(nodes_in, trimmed)
    end
  end

  return nodes_in
end

function M.get_type()
  local cur_node = smart_send.get_current_node()
  if not cur_node then
    vim.notify("No Tree-sitter node under cursor", vim.log.levels.WARN)
    return nil
  end

  local node_type = cur_node:type()
  vim.notify("Node type: " .. node_type, vim.log.levels.INFO)
  return node_type
end


function M.setup_keybindings(global_nodes)
  local current_global_nodes = global_nodes

  vim.keymap.set({ 'n' }, '<localleader>r', function()
      current_global_nodes = M.set_global_nodes()
    end,
    { noremap = true, silent = true, desc = "set global_nodes", buffer = true })

  vim.keymap.set({ 'n', 'v' }, '<localleader>v', function()
      smart_send.move_to_next_non_empty_line(); smart_send.select_until_global(current_global_nodes)
    end,
    { noremap = true, silent = true, desc = "Visual select next node after WS", buffer = true })

  vim.keymap.set('n', '<localleader>a', function() smart_send.send_repl(current_global_nodes) end,
    { noremap = true, silent = true, desc = "Send node to REPL", buffer = true })

  -- Both `<CR>` and `<S-CR>` were removed from `M.setup_keybindings`. They
  -- were hard, implicit overrides that clobbered Vim/filetype defaults and
  -- the user's snippet + insert-mode workflows (see C2 + H5 in the PR
  -- review). To opt back in for a specific filetype, override per-buffer
  -- from a `ftplugin/<lang>.lua`:
  --
  --   -- e.g. ftplugin/r.lua or ftplugin/quarto.lua
  --   vim.keymap.set('n', '<CR>', function()
  --     require('config').treesitter_helpers.setup_keybindings(global_nodes)
  --     require('nix_smart_send').send_repl(global_nodes)
  --   end, { buffer = true, desc = 'Send node to REPL' })

  vim.keymap.set('n', '<localleader>n',
    function() current_global_nodes = M.add_global_node(current_global_nodes) end,
    { noremap = true, silent = true, desc = "Add node under cursor to globals", buffer = true })

  vim.keymap.set('n', '<localleader>x',
    function() current_global_nodes = M.remove_global_node(current_global_nodes) end,
    { noremap = true, silent = true, desc = "Remove node under cursor from globals", buffer = true })

  vim.keymap.set('n', '<localleader>o', function()
    local pout = table.concat(global_nodes, ', ')
    vim.notify("global_nodes: " .. pout, vim.log.levels.INFO)
  end, { noremap = true, silent = true, desc = "Print globals", buffer = true })

  vim.keymap.set('n', '<localleader>p', function() M.get_type() end,
    { noremap = true, silent = true, desc = "Print node type", buffer = true })
end

Config.treesitter_helpers = M

-- Tree-sitter text objects: functions, calls, and assignments are especially
-- useful when editing R/tidyverse pipelines (e.g. `df |> mutate(...)`).
local ts_ok, treesitter = pcall(require, "nvim-treesitter.configs")
if ts_ok then
  treesitter.setup({
    textobjects = {
      select = {
        enable = true,
        lookahead = true,
        keymaps = {
          ["af"] = "@function.outer",
          ["if"] = "@function.inner",
          ["ac"] = "@call.outer",
          ["ic"] = "@call.inner",
          ["aa"] = "@assignment.outer",
          ["ia"] = "@assignment.inner",
        },
        selection_modes = {
          ["@function.outer"] = "V",
          ["@function.inner"] = "V",
          ["@call.outer"] = "v",
          ["@call.inner"] = "v",
          ["@assignment.outer"] = "v",
          ["@assignment.inner"] = "v",
        },
      },
      move = {
        enable = true,
        set_jumps = true,
        goto_next_start = {
          ["]f"] = "@function.outer",
          ["]c"] = "@call.outer",
        },
        goto_previous_start = {
          ["[f"] = "@function.outer",
          ["[c"] = "@call.outer",
        },
      },
    },
  })
end

return M