local core = require "nvim-tree.core"
local filters = require "nvim-tree.explorer.filters"
local find_file = require("nvim-tree.actions.finders.find-file").fn

local M = {}

---@param search_dir string|nil
---@param input_path string
---@return string|nil
local function search(search_dir, input_path)
  local realpaths_searched = {}

  if not search_dir then
    return
  end

  ---@param dir string
  ---@return string|nil
  local function iter(dir)
    local realpath, path, name, stat, handle, _

    local filter_status = filters.prepare()

    handle, _ = vim.loop.fs_scandir(dir)
    if not handle then
      return
    end

    realpath, _ = vim.loop.fs_realpath(dir)
    if not realpath or vim.tbl_contains(realpaths_searched, realpath) then
      return
    end
    table.insert(realpaths_searched, realpath)

    name, _ = vim.loop.fs_scandir_next(handle)
    while name do
      path = dir .. "/" .. name

      stat, _ = vim.loop.fs_stat(path)
      if not stat then
        break
      end

      if not filters.should_filter(path, filter_status) then
        if string.find(path, "/" .. input_path .. "$") then
          return path
        end

        if stat.type == "directory" then
          path = iter(path)
          if path then
            return path
          end
        end
      end

      name, _ = vim.loop.fs_scandir_next(handle)
    end
  end

  return iter(search_dir)
end

function M.fn()
  if not core.get_explorer() then
    return
  end

  -- temporarily set &path
  local bufnr = vim.api.nvim_get_current_buf()
  local path_existed, path_opt = pcall(vim.api.nvim_buf_get_option, bufnr, "path")
  vim.api.nvim_buf_set_option(bufnr, "path", core.get_cwd() .. "/**")

  vim.ui.input({ prompt = "Search: ", completion = "file_in_path" }, function(input_path)
    if not input_path or input_path == "" then
      return
    end
    -- reset &path
    if path_existed then
      vim.api.nvim_buf_set_option(bufnr, "path", path_opt)
    else
      vim.api.nvim_buf_set_option(bufnr, "path", nil)
    end

    -- strip trailing slash
    input_path = string.gsub(input_path, "/$", "")

    -- search under cwd
    local found = search(core.get_cwd(), input_path)
    if found then
      find_file(found)
    end
  end)
end

return M