Opening multiple files at once in telescope pickers and codecompanion

[jmmarotta]

I modified a function to enable opening multiple files in Telescope, making it compatible with CodeCompanion. It's hacky, but it works.

In your telescope config:

config = function()
  local select_one_or_multi = function(prompt_bufnr)
    local picker = require("telescope.actions.state").get_current_picker(prompt_bufnr)
    local multi = picker:get_multi_selection()

    local picker_type = picker.prompt_title or ""

    local is_codecompanion = string.find(picker_type, "Select file%(s%)")
      or string.find(picker_type, "Select buffer%(s%)")

    if is_codecompanion then
      require("telescope.actions").select_default(prompt_bufnr)
      return
    end

    if not vim.tbl_isempty(multi) then
      require("telescope.actions").close(prompt_bufnr)
      for _, j in pairs(multi) do
        if j.path ~= nil then
          if j.lnum ~= nil then
            vim.cmd(string.format("%s %s:%s", "edit", j.path, j.lnum))
          else
            vim.cmd(string.format("%s %s", "edit", j.path))
          end
        end
      end
    else
      require("telescope.actions").select_default(prompt_bufnr)
    end
  end
  
  local telescope = require("telescope")
  telescope.setup({
    -- You can put your default mappings / updates / etc. in here
    --  All the info you're looking for is in `:help telescope.setup()`
    --
    defaults = {
      mappings = {
        i = {
          ["<CR>"] = select_one_or_multi,
        },
      },
    },
  })
end

[bdogabriel]

Hi, thanks for sharing your solution. I was struggling with this and you helped a lot! I made a version based on yours, using map_selections. Sharing here for anyone interested:

local select_one_or_multi = function(prompt_bufnr)
	local files = {}

	require("telescope.actions.utils").map_selections(prompt_bufnr, function(entry, _)
		table.insert(files, entry.path)
	end)

	if #files == 0 then
		local selection = require("telescope.actions.state").get_selected_entry()
		if selection and selection.path then
			table.insert(files, selection.path)
		end
	end

	local picker_type = require("telescope.actions.state").get_current_picker(prompt_bufnr).prompt_title or ""

	local is_not_find_and_edit = string.find(picker_type, "Select file%(s%)")
		or string.find(picker_type, "Select buffer%(s%)")

	if is_not_find_and_edit then
		require("telescope.actions").select_default(prompt_bufnr)
		return
	end

	actions.close(prompt_bufnr)

	for _, file in ipairs(files) do
		if file ~= nil then
			vim.cmd(string.format("edit %s", file))
		end
	end
end