file_create.lua

local telescope = require("telescope")
local actions = require("telescope.actions")
local finders = require("telescope.finders")
local pickers = require("telescope.pickers")
local action_state = require("telescope.actions.state")
local Path = require("plenary.path")
local os_sep = Path.path.sep
local scan = require("plenary.scandir")
local utils = require("telescope.utils")
-- local action_set = require("telescope.actions.set")
local make_entry = require("telescope.make_entry")
local conf = require("telescope.config").values

local folder_list = function()
  local list = {}
  local p = io.popen("find . -type d")
  for file in p:lines() do
    table.insert(list, file:sub(3))
  end
  return list
end

function _G.dump(...)
  local objects = vim.tbl_map(vim.inspect, { ... })
  print(unpack(objects))
end

local function clear_prompt()
  vim.api.nvim_command("normal :esc<CR>")
end

local function get_user_input()
  return vim.fn.nr2char(vim.fn.getchar())
end

local function remove_dir(cwd)
  return vim.loop.fs_rmdir(cwd .. "/")
end

local file_create = function(opts)
  opts = opts or {}
  local results = folder_list()
  local is_dir = function(value)
    return value:sub(-1, -1) == os_sep
  end
  opts.new_finder = opts.new_finder
    or function(o)
      opts.cwd = o.path
      opts.hidden = o.hidden
      local data = {}

      if not vim.loop.fs_access(o.path, "X") then
        print("You don't have access to this directory")
        return nil
      end

      scan.scan_dir(o.path, {
        hidden = opts.hidden or false,
        add_dirs = true,
        depth = opts.depth,
        on_insert = function(entry, typ)
          table.insert(data, typ == "directory" and (entry .. os_sep) or entry)
        end,
      })
      table.insert(data, 1, ".." .. os_sep)

      local maker = function()
        local mt = {}
        mt.cwd = opts.cwd
        mt.display = function(entry)
          local hl_group
          local display = utils.transform_path(opts, entry.value)
          if is_dir(entry.value) then
            display = display .. os_sep
            if not opts.disable_devicons then
              display = (opts.dir_icon or "") .. " " .. display
              hl_group = "Default"
            end
          else
            display, hl_group = utils.transform_devicons(
              entry.value,
              display,
              opts.disable_devicons
            )
          end

          if hl_group then
            return display, { { { 1, 3 }, hl_group } }
          else
            return display
          end
        end

        mt.__index = function(t, k)
          local raw = rawget(mt, k)
          if raw then
            return raw
          end

          if k == "path" then
            local retpath = Path:new({ t.cwd, t.value }):absolute()
            if not vim.loop.fs_access(retpath, "R", nil) then
              retpath = t.value
            end
            if is_dir(t.value) then
              retpath = retpath .. os_sep
            end
            return retpath
          end

          return rawget(t, rawget({ value = 1 }, k))
        end

        return function(line)
          local tbl = { line }
          tbl.ordinal = Path:new(line):make_relative(opts.cwd)
          return setmetatable(tbl, mt)
        end
      end

      return finders.new_table({ results = data, entry_maker = maker() })
    end

  pickers.new(opts, {
    prompt_title = "Create file in",
    results_title = "File Creation",
    finder = finders.new_table({
      results = results,
      entry_maker = make_entry.gen_from_file(opts),
    }),
    previewer = conf.file_previewer(opts),
    sorter = conf.file_sorter(opts),
    attach_mappings = function(prompt_bufnr, map)
      local get_marked_files = function()
        local current_picker = action_state.get_current_picker(prompt_bufnr)
        local multi_selected = current_picker:get_multi_selection()
        local entries

        if vim.tbl_isempty(multi_selected) then
          entries = { action_state.get_selected_entry() }
        else
          entries = multi_selected
        end

        local selected = vim.tbl_map(function(entry)
          return Path:new(entry[1])
        end, entries)

        return selected
      end
      local create_new_file = function(bufnr)
        local new_cwd = vim.fn.expand(action_state.get_selected_entry().path)
        local fileName = vim.fn.input("File Name: ")
        if fileName == "" and not new_cwd then
          print("To create a new file name or directory")
          return
        end
        local result = new_cwd .. "/" .. fileName
        actions.close(bufnr)
        Path:new(result):touch({ parents = true })
        vim.cmd(string.format(":e %s", result))
      end

      local remove_file = function()
        local current_picker = action_state.get_current_picker(prompt_bufnr)
        local marked_files = get_marked_files()

        print("These files are going to be deleted:")
        for _, file in ipairs(marked_files) do
          print(file.filename)
        end

        local confirm = vim.fn.confirm(
          "You're about to perform a destructive action." .. " Proceed? [y/N]: ",
          "&Yes\n&No",
          "No"
        )

        if confirm == 1 then
          current_picker:delete_selection(function(entry)
            local p = Path:new(entry[1])
            p:rm({ recursive = p:is_dir() })
          end)
          print("\nThe file has been removed!")
          current_picker:reset_multi_selection()
        end
      end

      local enter = function()
        local current_picker = action_state.get_current_picker(prompt_bufnr)
        local new_cwd = vim.fn.expand(action_state.get_selected_entry().path)
        current_picker:refresh(
          opts.new_finder({
            path = new_cwd,
          }),
          { reset_prompt = true }
        )
      end

      map("i", "<CR>", enter)
      map("n", "<CR>", enter)
      map("i", "<C-e>", create_new_file)
      map("n", "<C-e>", create_new_file)
      map("i", "<C-d>", remove_file)
      map("n", "<C-d>", remove_file)
      return true
    end,
  }):find()
end

return telescope.register_extension({ exports = { file_create = file_create } })