mini.nvim/lua/mini/extra.lua at main · yurihbm/mini.nvim

Repository navigation

--- *mini.extra* Extra 'mini.nvim' functionality
--- MIT License Copyright (c) 2023 Evgeni Chasnovski
--- Extra useful functionality which is not essential enough for other 'mini.nvim'
--- modules to include directly.
--- Features:
--- - Various pickers for |mini.pick|:
---     - Built-in diagnostic (|MiniExtra.pickers.diagnostic()|).
---     - File explorer (|MiniExtra.pickers.explorer()|).
---     - Git branches/commits/files/hunks (|MiniExtra.pickers.git_hunks()|, etc.).
---     - Command/search/input history (|MiniExtra.pickers.history()|).
---     - LSP references/symbols/etc. (|MiniExtra.pickers.lsp()|).
---     - Tree-sitter nodes (|MiniExtra.pickers.treesitter()|).
---     - And much more.
---   See |MiniExtra.pickers| for more.
--- - Various textobject specifications for |mini.ai|. See |MiniExtra.gen_ai_spec|.
--- - Various highlighters for |mini.hipatterns|. See |MiniExtra.gen_highlighter|.
--- - This module requires only those 'mini.nvim' modules which are needed for
---   a particular functionality: 'mini.pick' for pickers, etc.
--- # Setup ~
--- This module needs a setup with `require('mini.extra').setup({})` (replace
--- `{}` with your `config` table). It will create global Lua table `MiniExtra`
--- which you can use for scripting or manually (with `:lua MiniExtra.*`).
--- See |MiniExtra.config| for `config` structure and default values.
--- This module doesn't have runtime options, so using `vim.b.miniextra_config`
--- will have no effect here.
--- # Comparisons ~
--- - [nvim-telescope/telescope.nvim](https://github.com/nvim-telescope/telescope.nvim):
---     - With |MiniExtra.pickers|, 'mini.pick' is reasonably on par when it comes
---       to built-in pickers.
--- - [ibhagwan/fzf-lua](https://github.com/ibhagwan/fzf-lua):
---     - Same as 'nvim-telescope/telescope.nvim'.
---@tag MiniExtra
---@diagnostic disable:undefined-field
---@diagnostic disable:discard-returns
---@diagnostic disable:unused-local
---@diagnostic disable:cast-local-type
---@diagnostic disable:undefined-doc-name
---@diagnostic disable:luadoc-miss-type-name
---@alias __extra_ai_spec_return function Function implementing |MiniAi-textobject-specification|.
---@alias __extra_pickers_local_opts table|nil Options defining behavior of this particular picker.
---@alias __extra_pickers_opts table|nil Options forwarded to |MiniPick.start()|.
---@alias __extra_pickers_return any Output of the called picker.
---@alias __extra_pickers_git_notes Notes:
--- - Requires executable `git`.
--- - Requires target path to be part of git repository.
--- - Present for exploration and navigation purposes. Doing any Git operations
---   is suggested to be done in a dedicated Git client and is not planned.
---@alias __extra_pickers_preserve_order - <preserve_order> `(boolean)` - whether to preserve original order
---     during query. Default: `false`.
---@alias __extra_pickers_git_path - <path> `(string|nil)` - target path for Git operation (if required). Also
---     used to find Git repository inside which to construct items.
---     Default: `nil` for root of Git repository containing |current-directory|.
-- Module definition ==========================================================
local MiniExtra = {}
local H = {}
--- Module setup
---@param config table|nil Module config table. See |MiniExtra.config|.
---@usage >lua
---   require('mini.extra').setup() -- use default config
---   -- OR
---   require('mini.extra').setup({}) -- replace {} with your config table
MiniExtra.setup = function(config)
  -- Export module
  _G.MiniExtra = MiniExtra
  -- Setup config
  config = H.setup_config(config)
  -- Apply config
  H.apply_config(config)
--stylua: ignore
--- Defaults ~
---@eval return MiniDoc.afterlines_to_code(MiniDoc.current.eval_section)
MiniExtra.config = {}
--minidoc_afterlines_end
--- |mini.ai| textobject specification generators
--- This is a table with function elements. Call to actually get specification.
--- Assumed to be used as part of |MiniAi.setup()|. Example: >lua
---   local gen_ai_spec = require('mini.extra').gen_ai_spec
---   require('mini.ai').setup({
---     custom_textobjects = {
---       B = gen_ai_spec.buffer(),
---       D = gen_ai_spec.diagnostic(),
---       I = gen_ai_spec.indent(),
---       L = gen_ai_spec.line(),
---       N = gen_ai_spec.number(),
MiniExtra.gen_ai_spec = {}
--- Current buffer textobject
--- - `a` textobject selects all lines in a buffer.
--- - `i` textobject selects all lines except blank lines at start and end.
---@return __extra_ai_spec_return
MiniExtra.gen_ai_spec.buffer = function()
  return function(ai_type)
    local start_line, end_line = 1, vim.fn.line('$')
    if ai_type == 'i' then
      -- Skip first and last blank lines for `i` textobject
      local first_nonblank, last_nonblank = vim.fn.nextnonblank(start_line), vim.fn.prevnonblank(end_line)
      -- Do nothing for buffer with all blanks
      if first_nonblank == 0 or last_nonblank == 0 then return { from = { line = start_line, col = 1 } } end
      start_line, end_line = first_nonblank, last_nonblank
    local to_col = math.max(vim.fn.getline(end_line):len(), 1)
    return { from = { line = start_line, col = 1 }, to = { line = end_line, col = to_col } }
--- Current buffer diagnostic textobject
--- - Both `a` and `i` textobjects return |vim.diagnostic.get()| output for the
---   current buffer. It is modified to fit |MiniAi-textobject-specification|.
---@param severity any Which severity to use. Forwarded to |vim.diagnostic.get()|.
---   Default: `nil` to use all diagnostic entries.
---@return __extra_ai_spec_return
MiniExtra.gen_ai_spec.diagnostic = function(severity)
  return function(ai_type)
    local cur_diag = vim.diagnostic.get(0, { severity = severity })
    local regions = {}
    for _, diag in ipairs(cur_diag) do
      local from = { line = diag.lnum + 1, col = diag.col + 1 }
      local to = { line = diag.end_lnum + 1, col = diag.end_col }
      if to.line == nil or to.col == nil then to = { line = diag.lnum + 1, col = diag.col + 1 } end
      table.insert(regions, { from = from, to = to })
    return regions
--- Current buffer indent scopes textobject
--- Indent scope is a set of consecutive lines with the following properties:
--- - Lines above first and below last are non-blank. They are called borders.
--- - There is at least one non-blank line in a set.
--- - All non-blank lines between borders have strictly greater indent
---   (perceived leading space respecting |'tabstop'|) than either of borders.
--- - `a` textobject selects scope including borders.
--- - `i` textobject selects the scope charwise.
--- - Differences with |MiniIndentscope.textobject()|:
---     - This textobject always treats blank lines on top and bottom of `i`
---       textobject as part of it, while 'mini.indentscope' can configure that.
---     - This textobject can select non-covering scopes, while 'mini.indentscope'
---       can not (by design).
---     - In this textobject scope computation is done only by "casting rays" from
---       top to bottom and not in both ways as in 'mini.indentscope'.
---       This works in most common scenarios and doesn't work only if indent of
---       of the bottom border is expected to be larger than the top.
---@return function Function implementing |MiniAi-textobject-specification|.
---   It returns array of regions representing all indent scopes in the buffer
---   ordered increasingly by the start line.
MiniExtra.gen_ai_spec.indent = function() return H.ai_indent_spec end
--- Current line textobject
--- - `a` textobject selects whole line.
--- - `i` textobject selects line after initial indent.
---@return __extra_ai_spec_return
MiniExtra.gen_ai_spec.line = function()
  return function(ai_type)
    local line_num = vim.fn.line('.')
    local line = vim.fn.getline(line_num)
    -- Ignore indentation for `i` textobject
    local from_col = ai_type == 'a' and 1 or (line:match('^(%s*)'):len() + 1)
    -- Don't select `\n` past the line to operate within a line
    local to_col = line:len()
    return { from = { line = line_num, col = from_col }, to = { line = line_num, col = to_col } }
--- Number textobject
--- - `a` textobject selects a whole number possibly preceded with "-" and
---   possibly followed by decimal part (dot and digits).
--- - `i` textobject selects consecutive digits.
---@return __extra_ai_spec_return
MiniExtra.gen_ai_spec.number = function()
  local digits_pattern = '%f[%d]%d+%f[%D]'
  local find_a_number = function(line, init)
    -- First find consecutive digits
    local from, to = line:find(digits_pattern, init)
    if from == nil then return nil, nil end
    -- Make sure that these digits were not processed before. This can happen
    -- because 'miin.ai' does next with `init = from + 1`, meaning that
    -- "-12.34" was already matched, then it would try to match starting from
    -- "1": we want to avoid matching that right away and avoid matching "34"
    -- from this number.
    if from == init and line:sub(from - 1, from - 1) == '-' then
      init = to + 1
      from, to = line:find(digits_pattern, init)
    if from == nil then return nil, nil end
    if line:sub(from - 2):find('^%d%.') ~= nil then
      init = to + 1
      from, to = line:find(digits_pattern, init)
    if from == nil then return nil, nil end
    -- Match the whole number with minus and decimal part
    if line:sub(from - 1, from - 1) == '-' then from = from - 1 end
    local dec_part = line:sub(to + 1):match('^%.%d+()')
    if dec_part ~= nil then to = to + dec_part - 1 end
    return from, to
  return function(ai_type)
    if ai_type == 'i' then return { digits_pattern } end
    return { find_a_number, { '^().*()$' } }
--- |mini.hipatterns| highlighter generators
--- This is a table with function elements. Call to actually get specification.
--- Assumed to be used as part of |MiniHipatterns.setup()|. Example: >lua
---   local hi_words = require('mini.extra').gen_highlighter.words
---   require('mini.hipatterns').setup({
---     highlighters = {
---       todo = hi_words({ 'TODO', 'Todo', 'todo' }, 'MiniHipatternsTodo'),
MiniExtra.gen_highlighter = {}
--- Highlight words
--- - Words should start and end with alphanumeric symbol (latin letter or digit).
--- - Words will be highlighted only in full and not if part bigger word, i.e.
---   there should not be alphanumeric symbol before and after it.
---@param words table Array of words to highlight. Will be matched as is, not
---   as Lua pattern.
---@param group string|function Proper `group` field for `highlighter`.
---   See |MiniHipatterns.config|.
---@param extmark_opts any Proper `extmark_opts` field for `highlighter`.
---   See |MiniHipatterns.config|.
MiniExtra.gen_highlighter.words = function(words, group, extmark_opts)
  if not H.islist(words) then H.error('`words` should be an array.') end
  if not (type(group) == 'string' or vim.is_callable(group)) then H.error('`group` should be string or callable.') end
  local pattern = vim.tbl_map(function(x)
    if type(x) ~= 'string' then H.error('All elements of `words` should be strings.') end
    return '%f[%w]()' .. vim.pesc(x) .. '()%f[%W]'
  end, words)
  return { pattern = pattern, group = group, extmark_opts = extmark_opts }
--- |mini.pick| pickers
--- A table with 'mini.pick' pickers (which is a hard dependency).
--- - All have the same signature:
---     - <local_opts> - optional table with options local to picker.
---     - <opts> - optional table with options forwarded to |MiniPick.start()|.
--- - All of them are automatically registered in |MiniPick.registry| inside
---   both |MiniExtra.setup()| or |MiniPick.setup()| (only one is enough).
--- - All use default versions of |MiniPick-source.preview|, |MiniPick-source.choose|,
---   and |MiniPick-source.choose_marked| if not stated otherwise.
---   Shown text and |MiniPick-source.show| are targeted to the picked items.
--- Examples of usage:
--- - As Lua code: `MiniExtra.pickers.buf_lines()`.
--- - With |:Pick| command: `:Pick buf_lines scope='current'`
---   Note: this requires calling |MiniExtra.setup()|.
MiniExtra.pickers = {}
--- Buffer lines picker
--- Pick from buffer lines. Notes:
--- - Loads all target buffers which are currently unloaded.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   - <scope> `(string)` - one of "all" (normal listed buffers) or "current".
---     Default: "all".
---   __extra_pickers_preserve_order
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.buf_lines = function(local_opts, opts)
  local pick = H.validate_pick('buf_lines')
  local_opts = vim.tbl_deep_extend('force', { scope = 'all', preserve_order = false }, local_opts or {})
  local scope = H.pick_validate_scope(local_opts, { 'all', 'current' }, 'buf_lines')
  local is_scope_all = scope == 'all'
  -- Define non-blocking callable `items` because getting all lines from all
  -- buffers (plus loading them) may take visibly long time
  local buffers = {}
  if is_scope_all then
    for _, buf_id in ipairs(vim.api.nvim_list_bufs()) do
      if vim.bo[buf_id].buflisted and vim.bo[buf_id].buftype == '' then table.insert(buffers, buf_id) end
    buffers = { vim.api.nvim_get_current_buf() }
  local poke_picker = pick.poke_is_picker_active
  local f = function()
    local items = {}
    for _, buf_id in ipairs(buffers) do
      if not poke_picker() then return end
      H.buf_ensure_loaded(buf_id)
      local buf_name = H.buf_get_name(buf_id) or ''
      local n_digits = math.floor(math.log10(vim.api.nvim_buf_line_count(buf_id))) + 1
      local format_pattern = '%s%' .. n_digits .. 'd\0%s'
      for lnum, l in ipairs(vim.api.nvim_buf_get_lines(buf_id, 0, -1, false)) do
        local prefix = is_scope_all and (buf_name .. '\0') or ''
        table.insert(items, { text = format_pattern:format(prefix, lnum, l), bufnr = buf_id, lnum = lnum })
    pick.set_picker_items(items)
  local items = vim.schedule_wrap(coroutine.wrap(f))
  local show = H.pick_get_config().source.show
  if is_scope_all and show == nil then show = H.show_with_icons end
  local match_opts = { preserve_order = local_opts.preserve_order }
  local match = function(stritems, inds, query) pick.default_match(stritems, inds, query, match_opts) end
  local default_source = { name = string.format('Buffer lines (%s)', scope), show = show, match = match }
  return H.pick_start(items, { source = default_source }, opts)
--- Color scheme picker
--- Pick and apply color scheme. Preview temporarily applies item's color scheme
--- and shows how selected highlight groups look.
--- Canceling reverts to color scheme before picker start:
--- - With |MiniColors-colorscheme:apply()| if |mini.colors| was available.
--- - With |:colorscheme| if |g:colors_name| was available.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   - <names> `(table)` - array of color scheme names to pick from.
---     Default: all available color schemes.
---   - <preview_hl_groups> `(table)` - array of highlight groups to show in preview
---     window. Default: all defined highlight groups in alphabetical order.
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.colorschemes = function(local_opts, opts)
  local pick = H.validate_pick('colorschemes')
  local_opts = local_opts or {}
  -- Infer data to show
  local all_cs = vim.fn.getcompletion('', 'color')
  local items = local_opts.names or all_cs
  if not H.islist(items) then H.error('`names` should be array of color scheme names') end
  for _, item in ipairs(items) do
    if not vim.tbl_contains(all_cs, item) then H.error(vim.inspect(item) .. ' is not a color scheme name') end
  local hl_groups = local_opts.preview_hl_groups
  if hl_groups ~= nil and not H.islist(hl_groups) then H.error('`preview_hl_groups` should be array') end
  -- Compute original color scheme to restore
  local bg_orig = vim.o.background
  local has_minicolors, minicolors = pcall(require, 'mini.colors')
  local cs_orig = has_minicolors and minicolors.get_colorscheme() or vim.g.colors_name
  local restore = function()
    vim.o.background = bg_orig
    if cs_orig == nil then return end
    if type(cs_orig) == 'string' then return vim.cmd('colorscheme ' .. cs_orig) end
    cs_orig:apply()
    -- Trigger to indicate actual color scheme application
    vim.api.nvim_exec_autocmds('ColorScheme', {})
  -- Define source
  local choose = function(item) vim.cmd('colorscheme ' .. item) end
  local choose_marked = function(items_to_mark) choose(items_to_mark[1] or '') end
  local needs_restore = false
  local preview = function(buf_id, item)
    needs_restore = true
    -- Ensure that previewed color scheme tries to use the initial background.
    -- This can be not the case if previous preview forced new background.
    vim.o.background = bg_orig
    vim.cmd('colorscheme ' .. item)
    H.preview_cs_hl_groups(buf_id, hl_groups)
  local default_source = { name = 'Colorschemes', preview = preview, choose = choose, choose_marked = choose_marked }
  local result = H.pick_start(items, { source = default_source }, opts)
  if result == nil and needs_restore then restore() end
  return result
--- Neovim commands picker
--- Pick from Neovim built-in (|Ex-commands|) and |user-commands|.
--- - Preview shows information about the command (if available).
--- - Choosing either executes command (if reliably known that it doesn't need
---   arguments) or populates Command line with the command.
---@param local_opts __extra_pickers_local_opts
---   Not used at the moment.
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.commands = function(local_opts, opts)
  local pick = H.validate_pick('commands')
  local commands = vim.tbl_deep_extend('force', vim.api.nvim_get_commands({}), vim.api.nvim_buf_get_commands(0, {}))
  local preview = function(buf_id, item)
    local data = commands[item]
    local lines = data == nil and { string.format('No command data for `%s` is yet available.', item) }
      or vim.split(vim.inspect(data), '\n')
    H.set_buflines(buf_id, lines)
  local choose = function(item)
    local data = commands[item] or {}
    -- If no arguments needed, execute immediately
    local keys = string.format(':%s%s', item, data.nargs == '0' and '\r' or ' ')
    vim.schedule(function() vim.fn.feedkeys(keys) end)
  local items = vim.fn.getcompletion('', 'command')
  local default_opts = { source = { name = 'Commands', preview = preview, choose = choose } }
  return H.pick_start(items, default_opts, opts)
--- Built-in diagnostic picker
--- Pick from |vim.diagnostic| using |vim.diagnostic.get()|.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   - <get_opts> `(table)` - options for |vim.diagnostic.get()|. Can be used
---     to limit severity or namespace. Default: `{}`.
---   - <scope> `(string)` - one of "all" (available) or "current" (buffer).
---     Default: "all".
---   - <sort_by> `(string)` - sort priority. One of "severity", "path", "none".
---     Default: "severity".
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.diagnostic = function(local_opts, opts)
  local pick = H.validate_pick('diagnostic')
  local_opts = vim.tbl_deep_extend('force', { get_opts = {}, scope = 'all', sort_by = 'severity' }, local_opts or {})
  local scope = H.pick_valida
10BC8
te_scope(local_opts, { 'all', 'current' }, 'diagnostic')
  local sort_by = H.pick_validate_one_of('sort_by', local_opts, { 'severity', 'path', 'none' }, 'diagnostic')
  local plus_one = function(x)
    if x == nil then return nil end
    return x + 1
  local diag_buf_id
  if scope == 'current' then diag_buf_id = vim.api.nvim_get_current_buf() end
  local items = vim.deepcopy(vim.diagnostic.get(diag_buf_id, local_opts.get_opts))
  -- Compute final path width
  local path_width = 0
  for _, item in ipairs(items) do
    item.path = H.buf_get_name(item.bufnr) or ''
    item.severity = item.severity or 0
    path_width = math.max(path_width, vim.fn.strchars(item.path))
  local compare = H.diagnostic_make_compare(sort_by)
  if vim.is_callable(compare) then table.sort(items, compare) end
  -- Update items
  for _, item in ipairs(items) do
    local severity = vim.diagnostic.severity[item.severity] or ' '
    local text = item.message:gsub('\n', ' ')
    item.text = string.format('%s │ %s │ %s', severity:sub(1, 1), H.ensure_text_width(item.path, path_width), text)
    item.lnum, item.col, item.end_lnum, item.end_col =
      plus_one(item.lnum), plus_one(item.col), plus_one(item.end_lnum), plus_one(item.end_col)
  local hl_groups_ref = {
    [vim.diagnostic.severity.ERROR] = 'DiagnosticFloatingError',
    [vim.diagnostic.severity.WARN] = 'DiagnosticFloatingWarn',
    [vim.diagnostic.severity.INFO] = 'DiagnosticFloatingInfo',
    [vim.diagnostic.severity.HINT] = 'DiagnosticFloatingHint',
  -- Define source
  local show = function(buf_id, items_to_show, query)
    pick.default_show(buf_id, items_to_show, query)
    H.pick_clear_namespace(buf_id, H.ns_id.pickers)
    for i, item in ipairs(items_to_show) do
      H.pick_highlight_line(buf_id, i, hl_groups_ref[item.severity], 199)
  local name = string.format('Diagnostic (%s)', scope)
  return H.pick_start(items, { source = { name = name, choose = H.choose_with_buflisted, show = show } }, opts)
--- File explorer picker
--- Explore file system and open file.
--- - Choosing a directory navigates inside it, changing picker's items and
---   current working directory.
--- - Query and preview work as usual (not only `move_next`/`move_prev` can be used).
--- - Preview works for any item.
--- Examples:
--- - `MiniExtra.pickers.explorer()`
--- - `:Pick explorer cwd='..'` - open explorer in parent directory.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   - <cwd> `(string)` - initial directory to explore. Should be a valid
---     directory path. Default: `nil` for |current-directory|.
---   - <filter> `(function)` - callable predicate to filter items to show.
---     Will be called for every item and should return `true` if it should be
---     shown. Each item is a table with the following fields:
---       - <fs_type> `(string)` - path type. One of "directory" or "file".
---       - <path> `(string)` - item path.
---       - <text> `(string)` - shown text (path's basename).
---   - <sort> `(function)` - callable item sorter. Will be called with array
---     of items (each element with structure as described above) and should
---     return sorted array of items.
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.explorer = function(local_opts, opts)
  local pick = H.validate_pick('explorer')
  local_opts = vim.tbl_deep_extend('force', { cwd = nil, filter = nil, sort = nil }, local_opts or {})
  local cwd = local_opts.cwd or vim.fn.getcwd()
  if vim.fn.isdirectory(cwd) == 0 then H.error('`local_opts.cwd` should be valid directory path.') end
  -- - Call twice "full path" to make sure that possible '..' are collapsed
  cwd = H.full_path(vim.fn.fnamemodify(cwd, ':p'))
  local filter = local_opts.filter or function() return true end
  if not vim.is_callable(filter) then H.error('`local_opts.filter` should be callable.') end
  local sort = local_opts.sort or H.explorer_default_sort
  if not vim.is_callable(sort) then H.error('`local_opts.sort` should be callable.') end
  -- Define source
  local choose = function(item)
    local path = item.path
    if vim.fn.filereadable(path) == 1 then return pick.default_choose(path) end
    if vim.fn.isdirectory(path) == 0 then return false end
    pick.set_picker_items(H.explorer_make_items(path, filter, sort))
    pick.set_picker_opts({ source = { cwd = path } })
    pick.set_picker_query({})
    return true
  local show = H.pick_get_config().source.show or H.show_with_icons
  local items = H.explorer_make_items(cwd, filter, sort)
  local source = { items = items, name = 'File explorer', cwd = cwd, show = show, choose = choose }
  opts = vim.tbl_deep_extend('force', { source = source }, opts or {})
  return pick.start(opts)
--- Git branches picker
--- Pick from Git branches using `git branch`.
--- __extra_pickers_git_notes
--- - On choose opens scratch buffer with branch's history.
--- Examples:
--- - `MiniExtra.pickers.git_branches({ scope = 'local' })` - local branches of
---   the |current-directory| parent Git repository.
--- - `:Pick git_branches path='%'` - all branches of the current file parent
---   Git repository.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   __extra_pickers_git_path
---   - <scope> `(string)` - branch scope to show. One of "all", "local", "remotes".
---     Default: "all".
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.git_branches = function(local_opts, opts)
  local pick = H.validate_pick('git_branches')
  H.validate_git('git_branches')
  local_opts = vim.tbl_deep_extend('force', { path = nil, scope = 'all' }, local_opts or {})
  local scope = H.pick_validate_scope(local_opts, { 'all', 'local', 'remotes' }, 'git_branches')
  -- Compute path to repo with target path (as it might differ from current)
  local path, path_type = H.git_normalize_path(local_opts.path, 'git_branches')
  local repo_dir = H.git_get_repo_dir(path, path_type, 'git_branches')
  -- Define source
  local preview = function(buf_id, item)
    local branch = item:match('^%*?%s*(%S+)')
    H.cli_show_output(buf_id, { 'git', '-C', repo_dir, 'log', branch, '--format=format:%h %s' })
  local choose = function(item)
    local win_target = (pick.get_picker_state().windows or {}).target
    if win_target == nil or not H.is_valid_win(win_target) then return end
    local buf_id = vim.api.nvim_create_buf(true, true)
    H.set_buf_name(buf_id, item:match('^%*?%s*(%S+)'))
    preview(buf_id, item)
    vim.api.nvim_win_set_buf(win_target, buf_id)
  local command = { 'git', 'branch', '-v', '--no-color', '--list' }
  if scope == 'all' or scope == 'remotes' then table.insert(command, 3, '--' .. scope) end
  local name = string.format('Git branches (%s)', scope)
  local default_source = { name = name, cwd = repo_dir, preview = preview, choose = choose }
  opts = vim.tbl_deep_extend('force', { source = default_source }, opts or {})
  return pick.builtin.cli({ command = command }, opts)
--- Git commits picker
--- Pick from Git commits using `git log`.
--- __extra_pickers_git_notes
--- - On choose opens scratch buffer with commit's diff.
--- Examples:
--- - `MiniExtra.pickers.git_commits()` - all commits from parent Git
---   repository of |current-directory|.
--- - `MiniExtra.pickers.git_commits({ path = 'subdir' })` - commits affecting
---   files from 'subdir' subdirectory.
--- - `:Pick git_commits path='%'` commits affecting current file.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   __extra_pickers_git_path
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.git_commits = function(local_opts, opts)
  local pick = H.validate_pick('git_commits')
  H.validate_git('git_commits')
  local_opts = vim.tbl_deep_extend('force', { path = nil }, local_opts or {})
  -- Compute path to repo with target path (as it might differ from current)
  local path, path_type = H.git_normalize_path(local_opts.path, 'git_commits')
  local repo_dir = H.git_get_repo_dir(path, path_type, 'git_commits')
  if local_opts.path == nil then path = repo_dir end
  -- Define source
  local preview = function(buf_id, item)
    if type(item) ~= 'string' then return end
    local has_parser, parser = pcall(vim.treesitter.get_parser, buf_id, 'git', { error = false })
    has_parser = has_parser and parser ~= nil
    if has_parser then has_parser = pcall(vim.treesitter.start, buf_id, 'git') end
    if not has_parser then vim.bo[buf_id].syntax = 'git' end
    H.cli_show_output(buf_id, { 'git', '-C', repo_dir, '--no-pager', 'show', item:match('^(%S+)') })
  local choose = function(item)
    local win_target = (pick.get_picker_state().windows or {}).target
    if win_target == nil or not H.is_valid_win(win_target) then return end
    local buf_id = vim.api.nvim_create_buf(true, true)
    H.set_buf_name(buf_id, item:match('^(%S+)'))
    preview(buf_id, item)
    -- Set filetype on opened buffer to trigger appropriate `FileType` event
    vim.bo[buf_id].filetype = 'git'
    vim.api.nvim_win_set_buf(win_target, buf_id)
  local command = { 'git', 'log', [[--format=format:%h %s]], '--', path }
  local name = string.format('Git commits (%s)', local_opts.path == nil and 'all' or 'for path')
  local default_source = { name = name, cwd = repo_dir, preview = preview, choose = choose }
  opts = vim.tbl_deep_extend('force', { source = default_source }, opts or {})
  return pick.builtin.cli({ command = command }, opts)
--- Git files picker
--- Pick from Git files using `git ls-files`.
--- __extra_pickers_git_notes
--- Examples:
--- - `MiniExtra.pickers.git_files({ scope = 'ignored' })` - ignored files from
---   parent Git repository of |current-directory|.
--- - `:Pick git_files path='subdir' scope='modified'` - files from 'subdir'
---   subdirectory which are ignored by Git.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   __extra_pickers_git_path
---   - <scope> `(string)` - files scope to show. One of
---       - "tracked"   (`--cached`   Git flag).
---       - "modified"  (`--modified` Git flag).
---       - "untracked" (`--others`   Git flag).
---       - "ignored"   (`--ignored`  Git flag).
---       - "deleted"   (`--deleted`  Git flag).
---     Default: "tracked".
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.git_files = function(local_opts, opts)
  local pick = H.validate_pick('git_files')
  H.validate_git('git_files')
  local_opts = vim.tbl_deep_extend('force', { path = nil, scope = 'tracked' }, local_opts or {})
  local allowed_scope = { 'tracked', 'modified', 'untracked', 'ignored', 'deleted' }
  local scope = H.pick_validate_scope(local_opts, allowed_scope, 'git_files')
  -- Compute path to repo with target path (as it might differ from current)
  local path, path_type = H.git_normalize_path(local_opts.path, 'git_files')
  H.git_get_repo_dir(path, path_type, 'git_files')
  local path_dir = path_type == 'directory' and path or vim.fn.fnamemodify(path, ':h')
  -- Define source
  local show = H.pick_get_config().source.show or H.show_with_icons
  local args = ({
    tracked = { '--cached' },
    modified = { '--modified' },
    untracked = { '--others' },
    ignored = { '--others', '--ignored', '--exclude-standard' },
    deleted = { '--deleted' },
  })[local_opts.scope]
  local command = vim.list_extend({ 'git', '-C', path_dir, '-c', 'core.quotepath=false', 'ls-files' }, args)
  local name = string.format('Git files (%s)', local_opts.scope)
  local default_source = { name = name, cwd = path_dir, show = show }
  opts = vim.tbl_deep_extend('force', { source = default_source }, opts or {})
  return pick.builtin.cli({ command = command }, opts)
--- Git hunks picker
--- Pick from Git hunks using `git diff`.
--- __extra_pickers_git_notes
--- - On choose navigates to hunk's first change.
--- Examples:
--- - `MiniExtra.pickers.git_hunks({ scope = 'staged' })` - staged hunks from
---   parent Git repository of |current-directory|.
--- - `:Pick git_hunks path='%' n_context=0` - hunks from current file computed
---   with no context.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   - <n_context> `(number)` - number of context lines to show in hunk's preview.
---     Default: 3.
---   __extra_pickers_git_path
---   - <scope> `(string)` - hunks scope to show. One of "unstaged" or "staged".
---     Default: "unstaged".
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.git_hunks = function(local_opts, opts)
  local pick = H.validate_pick('git_hunks')
  H.validate_git('git_hunks')
  local default_local_opts = { n_context = 3, path = nil, scope = 'unstaged' }
  local_opts = vim.tbl_deep_extend('force', default_local_opts, local_opts or {})
  if not (type(local_opts.n_context) == 'number' and local_opts.n_context >= 0) then
    H.error('`n_context` option in `pickers.git_hunks` picker should be non-negative number.')
  local n_context = math.floor(local_opts.n_context)
  local scope = H.pick_validate_scope(local_opts, { 'unstaged', 'staged' }, 'git_hunks')
  -- Compute path to repo with target path (as it might differ from current)
  local path, path_type = H.git_normalize_path(local_opts.path, 'git_hunks')
  local repo_dir = H.git_get_repo_dir(path, path_type, 'git_hunks')
  if local_opts.path == nil then path = repo_dir end
  -- Define source
  local show = H.pick_get_config().source.show or H.show_with_icons
  local preview = function(buf_id, item)
    local has_parser, parser = pcall(vim.treesitter.get_parser, buf_id, 'diff', { error = false })
    has_parser = has_parser and parser ~= nil
    if has_parser then has_parser = pcall(vim.treesitter.start, buf_id, 'diff') end
    if not has_parser then vim.bo[buf_id].syntax = 'diff' end
    local lines = vim.deepcopy(item.header)
    vim.list_extend(lines, item.hunk)
    H.set_buflines(buf_id, lines)
  local command = { 'git', 'diff', '--patch', '--unified=' .. n_context, '--color=never', '--', path }
  if scope == 'staged' then table.insert(command, 4, '--cached') end
  local postprocess = function(lines) return H.git_difflines_to_hunkitems(lines, n_context) end
  local name = string.format('Git hunks (%s %s)', scope, local_opts.path == nil and 'all' or 'for path')
  local default_source = { name = name, cwd = repo_dir, show = show, preview = preview }
  opts = vim.tbl_deep_extend('force', { source = default_source }, opts or {})
  return pick.builtin.cli({ command = command, postprocess = postprocess }, opts)
--- Matches from 'mini.hipatterns' picker
--- Pick from |mini.hipatterns| matches using |MiniHipatterns.get_matches()|.
--- - Requires 'mini.hipatterns'.
--- - Highlighter identifier is highlighted with its highlight group.
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   - <scope> `(string)` - one of "all" (buffers with enabled 'mini.hipatterns')
---     or "current" (buffer). Default: "all".
---   - <highlighters> `(table|nil)` - highlighters for which to find matches.
---     Forwarded to |MiniHipatterns.get_matches()|. Default: `nil`.
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.hipatterns = function(local_opts, opts)
  local pick = H.validate_pick('hipatterns')
  local has_hipatterns, hipatterns = pcall(require, 'mini.hipatterns')
  if not has_hipatterns then H.error([[`pickers.hipatterns` requires 'mini.hipatterns' which can not be found.]]) end
  local_opts = vim.tbl_deep_extend('force', { highlighters = nil, scope = 'all' }, local_opts or {})
  if local_opts.highlighters ~= nil and not H.islist(local_opts.highlighters) then
    H.error('`local_opts.highlighters` should be an array of highlighter identifiers.')
  local highlighters = local_opts.highlighters
  local scope = H.pick_validate_scope(local_opts, { 'all', 'current' }, 'hipatterns')
  -- Get items
  local buffers = scope == 'all' and hipatterns.get_enabled_buffers() or { vim.api.nvim_get_current_buf() }
  local items, highlighter_width = {}, 0
  for _, buf_id in ipairs(buffers) do
    local lines = vim.api.nvim_buf_get_lines(buf_id, 0, -1, false)
    local buf_name = H.buf_get_name(buf_id)
    if buf_name == '' then buf_name = 'Buffer_' .. buf_id end
    for _, match in ipairs(hipatterns.get_matches(buf_id, highlighters)) do
      match.highlighter = tostring(match.highlighter)
      match.buf_name, match.line = buf_name, lines[match.lnum]
      table.insert(items, match)
      highlighter_width = math.max(highlighter_width, vim.fn.strchars(match.highlighter))
  for _, item in ipairs(items) do
    --stylua: ignore
    item.text = string.format(
      '%s │ %s│%d│%d│%s',
      H.ensure_text_width(item.highlighter, highlighter_width),
      item.buf_name, item.lnum, item.col, item.line
    item.buf_name, item.line = nil, nil
  local show = function(buf_id, items_to_show, query)
    pick.default_show(buf_id, items_to_show, query)
    H.pick_clear_namespace(buf_id, H.ns_id.pickers)
    for i, item in ipairs(items_to_show) do
      local end_col = string.len(item.highlighter)
      local extmark_opts = { hl_group = item.hl_group, end_row = i - 1, end_col = end_col, priority = 1 }
      vim.api.nvim_buf_set_extmark(buf_id, H.ns_id.pickers, i - 1, 0, extmark_opts)
  local name = string.format('Mini.hipatterns matches (%s)', scope)
  return H.pick_start(items, { source = { name = name, show = show } }, opts)
--- Neovim history picker
--- Pick from output of |:history|. Use `<C-e>` to edit current match in
--- Command line.
--- - Has no preview.
--- - Choosing action depends on scope:
---     - For "cmd" / ":" scopes, the command is executed.
---     - For "search" / "/" / "?" scopes, search is redone.
---     - For other scopes nothing is done (but chosen item is still returned).
--- - `<C-e>` only works for "cmd" / ":" / "search" / "/" / "?" scopes.
--- Examples:
--- - Command history: `MiniExtra.pickers.history({ scope = ':' })`
--- - Search history: `:Pick history scope='/'`
---@param local_opts __extra_pickers_local_opts
---   Possible fields:
---   - <scope> `(string)` - any allowed {name} flag of |:history| command.
---     Note: word abbreviations are not allowed. Default: "all".
---@param opts __extra_pickers_opts
---@return __extra_pickers_return
MiniExtra.pickers.history = function(local_opts, opts)
  local pick = H.validate_pick('history')
  local_opts = vim.tbl_deep_extend('force', { scope = 'all' }, local_opts or {})
  local allowed_scope = { 'all', 'cmd', 'search', 'expr', 'input', 'debug', ':', '/', '?', '=', '@', '>' }
  local scope = H.pick_validate_scope(local_opts, allowed_scope, 'history')
  --stylua: ignore
  local type_ids = {
    cmd = ':',   search = '/', expr  = '=', input = '@', debug = '>',
    [':'] = ':', ['/']  = '/', ['='] = '=', ['@'] = '@', ['>'] = '>',
    ['?'] = '?',
  -- Construct items
  local items = {}
  local names = scope == 'all' and { 'cmd', 'search', 'expr', 'input', 'debug' } or { scope }
  for _, cur_name in ipairs(names) do
    local id = type_ids[cur_name]
    for i = 1, vim.fn.histnr(cur_name) do
      local hist_entry = vim.fn.histget(cur_name, -i)
      if hist_entry == '' then break end
      table.insert(items, string.format('%s %s', id, hist_entry))
  -- Define source
  local preview = H.pick_make_no_preview('history')
  local choose = function(item)
    if not (type(item) == 'string' and vim.fn.mode() == 'n') then return end
    local id, entry = item:match('^(.) (.*)$')
    if id == ':' or id == '/' or id == '?' then
      vim.schedule(function() vim.fn.feedkeys(id .. entry .. '\r', 'nxt') end)
  local edit_command = function()
    local cur_match = MiniPick.get_picker_matches().current
    local cur_scope, cur_item = cur_match:match('^(.) (.*)$')
    if not (cur_scope == ':' or cur_scope == '/' or cur_scope == '?') then return end
    vim.schedule(function() vim.api.nvim_input(cur_scope .. cur_item) end)
    return true
  local mappings = { edit_command = { char = '<C-e>', func = edit_command } }
  opts = vim.tbl_deep_extend('force', opts or {}, { mappings = mappings })
  local default_source = { name = string.format('History (%s)', scope), preview = preview, choose = choose }
  return H.pick_start(items, { source = default_source }, opts)
--- Highlight groups picker
--- Pick and preview highlight groups.
--- - Item line is colored with same highlight group it represents.
--- - Preview shows highlight's definition (as in |:highlight| with {group-name}).
--- - Choosing places highlight definition in Command line to update and apply.

View remainder of file in raw view

FilesExpand file tree

extra.lua

Latest commit

History

extra.lua

File metadata and controls