integrations.lua

--- Tree integration module for OpenCode.nvim
--- Handles detection and selection of files from nvim-tree, neo-tree, mini.files, and oil.nvim
---@module 'opencode.integrations'
local M = {}
local logger = require("opencode.logger")

---Get selected files from the current tree explorer
---@return table|nil files List of file paths, or nil if error
---@return string|nil error Error message if operation failed
function M.get_selected_files_from_tree()
  local current_ft = vim.bo.filetype

  if current_ft == "NvimTree" then
    return M._get_nvim_tree_selection()
  elseif current_ft == "neo-tree" then
    return M._get_neotree_selection()
  elseif current_ft == "oil" then
    return M._get_oil_selection()
  elseif current_ft == "minifiles" then
    return M._get_mini_files_selection()
  elseif current_ft == "netrw" then
    return M._get_netrw_selection()
  else
    return nil, "Not in a supported tree buffer (current filetype: " .. current_ft .. ")"
  end
end

---Get selected files from nvim-tree
---Supports both multi-selection (marks) and single file under cursor
---@return table files List of file paths
---@return string|nil error Error message if operation failed
function M._get_nvim_tree_selection()
  local success, nvim_tree_api = pcall(require, "nvim-tree.api")
  if not success then
    return {}, "nvim-tree not available"
  end

  local files = {}

  local marks = nvim_tree_api.marks.list()

  if marks and #marks > 0 then
    for _, mark in ipairs(marks) do
      if mark.type == "file" and mark.absolute_path and mark.absolute_path ~= "" then
        -- Check if it's not a root-level file (basic protection)
        if not string.match(mark.absolute_path, "^/[^/]*$") then
          table.insert(files, mark.absolute_path)
        end
      end
    end

    if #files > 0 then
      return files, nil
    end
  end

  local node = nvim_tree_api.tree.get_node_under_cursor()
  if node then
    if node.type == "file" and node.absolute_path and node.absolute_path ~= "" then
      -- Check if it's not a root-level file (basic protection)
      if not string.match(node.absolute_path, "^/[^/]*$") then
        return { node.absolute_path }, nil
      else
        return {}, "Cannot add root-level file. Please select a file in a subdirectory."
      end
    elseif node.type == "directory" and node.absolute_path and node.absolute_path ~= "" then
      return { node.absolute_path }, nil
    end
  end

  return {}, "No file found under cursor"
end

---Get selected files from neo-tree
---Uses neo-tree's own visual selection method when in visual mode
---@return table files List of file paths
---@return string|nil error Error message if operation failed
function M._get_neotree_selection()
  local success, manager = pcall(require, "neo-tree.sources.manager")
  if not success then
    logger.debug("integrations/neotree", "neo-tree not available (require failed)")
    return {}, "neo-tree not available"
  end

  local state = manager.get_state("filesystem")
  if not state then
    logger.debug("integrations/neotree", "filesystem state not available from manager")
    return {}, "neo-tree filesystem state not available"
  end

  local files = {}

  -- Use neo-tree's own visual selection method (like their copy/paste feature)
  local mode = vim.fn.mode()
  local current_win = vim.api.nvim_get_current_win()
  logger.debug(
    "integrations/neotree",
    "begin selection",
    "mode=",
    mode,
    "current_win=",
    current_win,
    "state.winid=",
    tostring(state.winid)
  )

  if mode == "V" or mode == "v" or mode == "\22" then
    if state.winid and state.winid == current_win then
      -- Use neo-tree's exact method to get visual range (from their get_selected_nodes implementation)
      local start_pos = vim.fn.getpos("'<")[2]
      local end_pos = vim.fn.getpos("'>")[2]

      -- Fallback to current cursor and anchor if marks are not valid
      if start_pos == 0 or end_pos == 0 then
        local cursor_pos = vim.api.nvim_win_get_cursor(0)[1]
        local anchor_pos = vim.fn.getpos("v")[2]
        if anchor_pos > 0 then
          start_pos = math.min(cursor_pos, anchor_pos)
          end_pos = math.max(cursor_pos, anchor_pos)
        else
          start_pos = cursor_pos
          end_pos = cursor_pos
        end
      end

      if end_pos < start_pos then
        start_pos, end_pos = end_pos, start_pos
      end

      logger.debug("integrations/neotree", "visual selection range", start_pos, "to", end_pos)

      local selected_nodes = {}

      for line = start_pos, end_pos do
        local node = state.tree:get_node(line)
        if node then
          -- Add validation for node types before adding to selection
          if node.type and node.type ~= "message" then
            table.insert(selected_nodes, node)
            local depth = (node.get_depth and node:get_depth()) and node:get_depth() or 0
            logger.debug(
              "integrations/neotree",
              "line",
              line,
              "node type=",
              tostring(node.type),
              "depth=",
              depth,
              "path=",
              tostring(node.path)
            )
          else
            logger.debug("integrations/neotree", "line", line, "node rejected (type)", tostring(node and node.type))
          end
        else
          logger.debug("integrations/neotree", "line", line, "no node returned from state.tree:get_node")
        end
      end

      logger.debug("integrations/neotree", "selected_nodes count=", #selected_nodes)

      for _, node in ipairs(selected_nodes) do
        -- Enhanced validation: check for file type and valid path
        if node.type == "file" and node.path and node.path ~= "" then
          -- Additional check: ensure it's not a root node (depth protection)
          local depth = (node.get_depth and node:get_depth()) and node:get_depth() or 0
          if depth > 1 then
            table.insert(files, node.path)
            logger.debug("integrations/neotree", "accepted file", node.path)
          else
            logger.debug("integrations/neotree", "rejected file (depth<=1)", node.path)
          end
        elseif node.type == "directory" and node.path and node.path ~= "" then
          local depth = (node.get_depth and node:get_depth()) and node:get_depth() or 0
          if depth > 1 then
            table.insert(files, node.path)
            logger.debug("integrations/neotree", "accepted directory", node.path)
          else
            logger.debug("integrations/neotree", "rejected directory (depth<=1)", node.path)
          end
        else
          logger.debug(
            "integrations/neotree",
            "rejected node (missing path or unsupported type)",
            tostring(node and node.type),
            tostring(node and node.path)
          )
        end
      end

      if #files > 0 then
        logger.debug("integrations/neotree", "files from visual selection:", files)
        return files, nil
      end
    end
  end

  if state.tree then
    local selection = nil

    if state.tree.get_selection then
      selection = state.tree:get_selection()
    end

    if (not selection or #selection == 0) and state.selected_nodes then
      selection = state.selected_nodes
    end

    if selection and #selection > 0 then
      logger.debug("integrations/neotree", "using state selection count=", #selection)
      for _, node in ipairs(selection) do
        if node.type == "file" and node.path then
          table.insert(files, node.path)
          logger.debug("integrations/neotree", "accepted file from state selection", node.path)
        else
          logger.debug(
            "integrations/neotree",
            "ignored non-file in state selection",
            tostring(node and node.type),
            tostring(node and node.path)
          )
        end
      end

      if #files > 0 then
        logger.debug("integrations/neotree", "files from state selection:", files)
        return files, nil
      end
    end
  end

  if state.tree then
    local node = state.tree:get_node()

    if node then
      logger.debug(
        "integrations/neotree",
        "fallback single node",
        "type=",
        tostring(node.type),
        "path=",
        tostring(node.path)
      )
      if node.type == "file" and node.path then
        return { node.path }, nil
      elseif node.type == "directory" and node.path then
        return { node.path }, nil
      end
    end
  end

  logger.debug("integrations/neotree", "no file found under cursor/selection")
  return {}, "No file found under cursor"
end

---Get selected files from oil.nvim
---Supports both visual selection and single file under cursor
---@return table files List of file paths
---@return string|nil error Error message if operation failed
function M._get_oil_selection()
  local success, oil = pcall(require, "oil")
  if not success then
    return {}, "oil.nvim not available"
  end

  local bufnr = vim.api.nvim_get_current_buf() --[[@as number]]
  local files = {}

  -- Check if we're in visual mode
  local mode = vim.fn.mode()
  if mode == "V" or mode == "v" or mode == "\22" then
    -- Visual mode: use the common visual range function
    local visual_commands = require("opencode.visual_commands")
    local start_line, end_line = visual_commands.get_visual_range()

    -- Get current directory once
    local dir_ok, current_dir = pcall(oil.get_current_dir, bufnr)
    if not dir_ok or not current_dir then
      return {}, "Failed to get current directory"
    end

    -- Process each line in the visual selection
    for line = start_line, end_line do
      local entry_ok, entry = pcall(oil.get_entry_on_line, bufnr, line)
      if entry_ok and entry and entry.name then
        -- Skip parent directory entries
        if entry.name ~= ".." and entry.name ~= "." then
          local full_path = current_dir .. entry.name
          -- Handle various entry types
          if entry.type == "file" or entry.type == "link" then
            table.insert(files, full_path)
          elseif entry.type == "directory" then
            -- Ensure directory paths end with /
            table.insert(files, full_path:match("/$") and full_path or full_path .. "/")
          else
            -- For unknown types, return the path anyway
            table.insert(files, full_path)
          end
        end
      end
    end

    if #files > 0 then
      return files, nil
    end
  else
    -- Normal mode: get file under cursor with error handling
    local ok, entry = pcall(oil.get_cursor_entry)
    if not ok or not entry then
      return {}, "Failed to get cursor entry"
    end

    local dir_ok, current_dir = pcall(oil.get_current_dir, bufnr)
    if not dir_ok or not current_dir then
      return {}, "Failed to get current directory"
    end

    -- Process the entry
    if entry.name and entry.name ~= ".." and entry.name ~= "." then
      local full_path = current_dir .. entry.name
      -- Handle various entry types
      if entry.type == "file" or entry.type == "link" then
        return { full_path }, nil
      elseif entry.type == "directory" then
        -- Ensure directory paths end with /
        return { full_path:match("/$") and full_path or full_path .. "/" }, nil
      else
        -- For unknown types, return the path anyway
        return { full_path }, nil
      end
    end
  end

  return {}, "No file found under cursor"
end

-- Helper function to get mini.files selection using explicit range
function M._get_mini_files_selection_with_range(start_line, end_line)
  local success, mini_files = pcall(require, "mini.files")
  if not success then
    return {}, "mini.files not available"
  end

  local files = {}
  local bufnr = vim.api.nvim_get_current_buf()

  -- Process each line in the range
  for line = start_line, end_line do
    local entry_ok, entry = pcall(mini_files.get_fs_entry, bufnr, line)

    if entry_ok and entry and entry.path and entry.path ~= "" then
      -- Extract real filesystem path from mini.files buffer path
      local real_path = entry.path
      -- Remove mini.files buffer protocol prefix if present
      if real_path:match("^minifiles://") then
        real_path = real_path:gsub("^minifiles://[^/]*/", "")
      end

      -- Validate that the path exists
      if vim.fn.filereadable(real_path) == 1 or vim.fn.isdirectory(real_path) == 1 then
        table.insert(files, real_path)
      end
    end
  end

  if #files > 0 then
    return files, nil
  else
    return {}, "No files found in range"
  end
end

---Get selected files from mini.files
---Supports both visual selection and single file under cursor
---Reference: mini.files API MiniFiles.get_fs_entry()
---@return table files List of file paths
---@return string|nil error Error message if operation failed
function M._get_mini_files_selection()
  local success, mini_files = pcall(require, "mini.files")
  if not success then
    return {}, "mini.files not available"
  end

  local bufnr = vim.api.nvim_get_current_buf()

  -- Normal mode: get file under cursor
  local entry_ok, entry = pcall(mini_files.get_fs_entry, bufnr)
  if not entry_ok or not entry then
    return {}, "Failed to get entry from mini.files"
  end

  if entry.path and entry.path ~= "" then
    -- Extract real filesystem path from mini.files buffer path
    local real_path = entry.path
    -- Remove mini.files buffer protocol prefix if present
    if real_path:match("^minifiles://") then
      real_path = real_path:gsub("^minifiles://[^/]*/", "")
    end

    -- Validate that the path exists
    if vim.fn.filereadable(real_path) == 1 or vim.fn.isdirectory(real_path) == 1 then
      return { real_path }, nil
    else
      return {}, "Invalid file or directory path: " .. real_path
    end
  end

  return {}, "No file found under cursor"
end

--- Get selected files from netrw
--- Supports both marked files and single file under cursor
--- Reference: :help netrw-mf, :help markfilelist
--- @return table files List of file paths
--- @return string|nil error Error message if operation failed
function M._get_netrw_selection()
  local has_call = (vim.fn.exists("*netrw#Call") == 1)
  local has_expose = (vim.fn.exists("*netrw#Expose") == 1)
  if not (has_call and has_expose) then
    return {}, "netrw not available"
  end

  -- function to resolve a 'word' (filename in netrw listing) to an absolute path using b:netrw_curdir
  local function resolve_word_to_path(word)
    if type(word) ~= "string" or word == "" then
      return nil
    end
    if word == "." or word == ".." or word == "../" then
      return nil
    end
    local curdir = vim.b.netrw_curdir or vim.fn.getcwd()
    local joined = curdir .. "/" .. word
    return vim.fn.fnamemodify(joined, ":p")
  end

  -- 1. Check for marked files
  do
    local mf_ok, mf_result = pcall(function()
      if has_expose then
        return vim.fn.call("netrw#Expose", { "netrwmarkfilelist" })
      end
      return nil
    end)

    local marked_files = {}
    if mf_ok and type(mf_result) == "table" and #mf_result > 0 then
      for _, file_path in ipairs(mf_result) do
        if vim.fn.filereadable(file_path) == 1 or vim.fn.isdirectory(file_path) == 1 then
          table.insert(marked_files, vim.fn.fnamemodify(file_path, ":p"))
        end
      end
    end

    if #marked_files > 0 then
      return marked_files, nil
    end
  end

  -- 2. No marked files. Check for a file or dir under cursor
  local path_ok, path_result = pcall(function()
    if has_call then
      local word = vim.fn.call("netrw#Call", { "NetrwGetWord" })
      local p = resolve_word_to_path(word)
      return p
    end
    return nil
  end)

  if not path_ok or not path_result or path_result == "" then
    return {}, "Failed to get path from netrw"
  end

  if vim.fn.filereadable(path_result) == 1 or vim.fn.isdirectory(path_result) == 1 then
    return { path_result }, nil
  end

  return {}, "Invalid file or directory path: " .. path_result
end

return M