Picker: binding a key to swap between files() and grep()

[synic]

In Telescope, I've bound some keys to toggle between files() and grep(), it looks something like this:

-- switch to live_grep
["<m-r>"] = function(prompt_bufnr)
    local picker = state.get_current_picker(prompt_bufnr)
    local prompt = picker:_get_prompt()
    actions.close(prompt_bufnr)
    builtin.live_grep({ cwd = picker.cwd, default_text = prompt, debounce = 100 })
end

Is it possible to do this in snacks.picker?

To make it like what I have here, I would need:

1. Be able to retrieve the current picker
2. Be able to retrieve the prompt in the current picker
3. Be able to retrieve the opts (specifically cwd) in the current picker

[synic]

This is how I did this, again, not sure if I'm doing this correctly, but hey, it works.

sources = {
  files = {
    actions = {
      switch_to_grep = function(picker, _)
        local snacks = require("snacks")
        local pattern = picker.input.filter.pattern or picker.input.filter.search
        local cwd = picker.input.filter.cwd

        picker:close()

        ---@diagnostic disable-next-line: missing-fields
        snacks.picker.grep({ cwd = cwd, search = pattern })
      end,
    },
    win = {
      input = {
        keys = {
          ["<a-r>"] = { "switch_to_grep", desc = "Switch to grep", mode = { "i", "n" } },
        },
      },
    },
  },
  grep = {
    actions = {
      switch_to_files = function(picker, _)
        local snacks = require("snacks")
        local pattern = picker.input.filter.search or picker.input.filter.pattern
        local cwd = picker.input.filter.cwd

        picker:close()

        ---@diagnostic disable-next-line: missing-fields
        snacks.picker.files({ cwd = cwd, pattern = pattern })
      end,
    },
    win = {
      input = {
        keys = {
          ["<a-r>"] = { "switch_to_files", desc = "Switch to files", mode = { "i", "n" } },
        },
      },
    },
  },
}

[rbhanot4739]

@synic Thank you for sharing, this is pretty interesting. I just updated it a little bit to have a single key to alternate b/w grep and files pickers.

local M = {}
M.is_grep = nil

local function switch_grep_files(picker, _)
  -- switch b/w grep and files picker
  local snacks = require("snacks")
  local cwd = picker.input.filter.cwd

  picker:close()

  if M.is_grep then
    -- if we are inside grep picker then switch to files picker and set M.is_grep = false
    local pattern = picker.input.filter.search or picker.input.filter.pattern
    snacks.picker.files({ cwd = cwd, pattern = pattern })
    M.is_grep = false
    return
  else
    -- if we are inside files picker then switch to grep picker and set M.is_grep = true
    local pattern = picker.input.filter.pattern or picker.input.filter.search
    snacks.picker.grep({ cwd = cwd, search = pattern })
    M.is_grep = true
  end
end

And then

sources = {
        files = {
          actions = {
            switch_grep_files = function(picker, _)
              M.is_grep = false
              switch_grep_files(picker, _)
            end,
          },
          win = {
            input = {
              keys = {
                ["<c-k>"] = { "switch_grep_files", desc = "Switch to grep", mode = { "i", "v" } },
              },
            },
          },
        },
        grep = {
          actions = {
            switch_grep_files = function(picker, _)
              M.is_grep = true
              switch_grep_files(picker, _)
            end,
          },
          win = {
            input = {
              keys = {
                ["<c-k>"] = { "switch_grep_files", desc = "Switch to grep", mode = { "i", "v" } },
              },
            },
          },
        },
      }

Now with this whether I open grep or files picker pressing the <c-k> would alternate to other picker and pressing it second time would again bring me back to original picker.

[synic]

Nice, I am going to switch to your version.

This switching works great for me, because I have a keybind to open a project, and when opening a project, it automatically runs the files() picker on the project. Sometimes, though, I want to search for specific text, and the toggle comes in really handy

[rbhanot4739]

@synic I modified this a little to support this for other pickers as well.

M.switch_to_grep = nil
M.grep_alt_picker = nil
---@param picker snacks.Picker
local function switch_to_grep(picker, _)
  local picker_type = picker.opts.source
  local allowed_pickers = { "files", "buffers", "recent", "smart", "grep" }
  if not vim.tbl_contains(allowed_pickers, picker_type) then
    Snacks.notify.warn("Switching to grep is not supported for `" .. picker_type .. "`", { title = "Snacks Picker" })
    return
  end

  if picker_type == "grep" then
    M.switch_to_grep = false
    M.grep_alt_picker = M.grep_alt_picker or "files"
  else
    M.switch_to_grep = true
    if picker_type == "recent_files" then
      picker_type = "recent"
    end
    M.grep_alt_picker = picker_type
  end
  local snacks = require("snacks")
  local cwd = picker.input.filter.cwd

  picker:close()

  if M.switch_to_grep then
    local pattern = picker.input.filter.pattern or ""
    ---@diagnostic disable-next-line: missing-fields
    snacks.picker.grep({ cwd = cwd, search = pattern })
  else
    local pattern = picker.input.filter.search or ""
    ---@diagnostic disable-next-line: missing-fields
    snacks.picker.pick(M.grep_alt_picker, { cwd = cwd, pattern = pattern })
  end
end

And then you map it to supported pickers

{
        smart = {
          actions = {
            switch_grep_files = function(picker, _)
              switch_to_grep(picker, _)
            end,
          },
          win = {
            input = {
              keys = {
                ["<c-k>"] = { "switch_grep_files", desc = "Switch to grep", mode = { "i", "n" } },
              },
            },
          },
        },
        recent = {
          actions = {
            switch_grep_files = function(picker, _)
              switch_to_grep(picker, _)
            end,
          },
          win = {
            input = {
              keys = {
                ["<c-k>"] = { "switch_grep_files", desc = "Switch to grep", mode = { "i", "n" } },
              },
            },
          },
        },
        buffers = {
          actions = {
            switch_grep_files = function(picker, _)
              switch_to_grep(picker, _)
            end,
          },
          win = {
            input = {
              keys = {
                ["<c-k>"] = { "switch_grep_files", desc = "Switch to grep", mode = { "i", "n" } },
              },
            },
          },
        },
        files = {
          -- live = true,
          actions = {
            switch_grep_files = function(picker, _)
              switch_to_grep(picker, _)
            end,
            cd_up = function(picker, _)
              cd_up(picker, _)
            end,
          },
          win = {
            input = {
              keys = {
                ["<c-k>"] = { "switch_grep_files", desc = "Switch to grep", mode = { "i", "n" } },
                ["<c-u>"] = { "cd_up", desc = "cd_up", mode = { "i", "n" } },
              },
            },
          },
        },
        grep = {
          actions = {
            switch_grep_files = function(picker, _)
              switch_to_grep(picker, _)
            end,
            cd_up = function(picker, _)
              cd_up(picker, _)
            end,
          },
          win = {
            input = {
              keys = {
                ["<c-k>"] = { "switch_grep_files", desc = "Switch to grep", mode = { "i", "n" } },
                ["<c-u>"] = { "cd_up", desc = "cd_up", mode = { "i", "n" } },
              },
            },
          },
        },
      }

[synic]

Nice, kinda old CtrlP style