294 lines
8.4 KiB
Lua
294 lines
8.4 KiB
Lua
local actions = require('telescope.actions')
|
|
local action_state = require('telescope.actions.state')
|
|
local finders = require('telescope.finders')
|
|
local make_entry = require('telescope.make_entry')
|
|
local pickers = require('telescope.pickers')
|
|
local previewers = require('telescope.previewers')
|
|
local utils = require('telescope.utils')
|
|
local entry_display = require('telescope.pickers.entry_display')
|
|
local strings = require('plenary.strings')
|
|
|
|
local conf = require('telescope.config').values
|
|
|
|
local git = {}
|
|
|
|
git.files = function(opts)
|
|
local show_untracked = utils.get_default(opts.show_untracked, true)
|
|
local recurse_submodules = utils.get_default(opts.recurse_submodules, false)
|
|
if show_untracked and recurse_submodules then
|
|
error("Git does not support both --others and --recurse-submodules")
|
|
end
|
|
|
|
-- By creating the entry maker after the cwd options,
|
|
-- we ensure the maker uses the cwd options when being created.
|
|
opts.entry_maker = opts.entry_maker or make_entry.gen_from_file(opts)
|
|
|
|
pickers.new(opts, {
|
|
prompt_title = 'Git Files',
|
|
finder = finders.new_oneshot_job(
|
|
vim.tbl_flatten( {
|
|
"git", "ls-files", "--exclude-standard", "--cached",
|
|
show_untracked and "--others" or nil,
|
|
recurse_submodules and "--recurse-submodules" or nil
|
|
} ),
|
|
opts
|
|
),
|
|
previewer = conf.file_previewer(opts),
|
|
sorter = conf.file_sorter(opts),
|
|
}):find()
|
|
end
|
|
|
|
git.commits = function(opts)
|
|
local results = utils.get_os_command_output({
|
|
'git', 'log', '--pretty=oneline', '--abbrev-commit', '--', '.'
|
|
}, opts.cwd)
|
|
|
|
pickers.new(opts, {
|
|
prompt_title = 'Git Commits',
|
|
finder = finders.new_table {
|
|
results = results,
|
|
entry_maker = opts.entry_maker or make_entry.gen_from_git_commits(opts),
|
|
},
|
|
previewer = previewers.git_commit_diff.new(opts),
|
|
sorter = conf.file_sorter(opts),
|
|
attach_mappings = function()
|
|
actions.select_default:replace(actions.git_checkout)
|
|
return true
|
|
end
|
|
}):find()
|
|
end
|
|
|
|
git.stash = function(opts)
|
|
local results = utils.get_os_command_output({
|
|
'git', '--no-pager', 'stash', 'list',
|
|
}, opts.cwd)
|
|
|
|
pickers.new(opts, {
|
|
prompt_title = 'Git Stash',
|
|
finder = finders.new_table {
|
|
results = results,
|
|
entry_maker = opts.entry_maker or make_entry.gen_from_git_stash(),
|
|
},
|
|
previewer = previewers.git_stash_diff.new(opts),
|
|
sorter = conf.file_sorter(opts),
|
|
attach_mappings = function()
|
|
actions.select_default:replace(actions.git_apply_stash)
|
|
return true
|
|
end
|
|
}):find()
|
|
end
|
|
git.bcommits = function(opts)
|
|
local results = utils.get_os_command_output({
|
|
'git', 'log', '--pretty=oneline', '--abbrev-commit', vim.fn.expand('%')
|
|
}, opts.cwd)
|
|
|
|
pickers.new(opts, {
|
|
prompt_title = 'Git BCommits',
|
|
finder = finders.new_table {
|
|
results = results,
|
|
entry_maker = opts.entry_maker or make_entry.gen_from_git_commits(opts),
|
|
},
|
|
previewer = previewers.git_commit_diff.new(opts),
|
|
sorter = conf.file_sorter(opts),
|
|
attach_mappings = function()
|
|
actions.select_default:replace(actions.git_checkout)
|
|
return true
|
|
end
|
|
}):find()
|
|
end
|
|
|
|
git.branches = function(opts)
|
|
local format = '%(HEAD)'
|
|
.. '%(refname)'
|
|
.. '%(authorname)'
|
|
.. '%(upstream:lstrip=2)'
|
|
.. '%(committerdate:format-local:%Y/%m/%d%H:%M:%S)'
|
|
local output = utils.get_os_command_output({ 'git', 'for-each-ref', '--perl', '--format', format }, opts.cwd)
|
|
|
|
local results = {}
|
|
local widths = {
|
|
name = 0,
|
|
authorname = 0,
|
|
upstream = 0,
|
|
committerdate = 0,
|
|
}
|
|
local unescape_single_quote = function(v)
|
|
return string.gsub(v, "\\([\\'])", "%1")
|
|
end
|
|
local parse_line = function(line)
|
|
local fields = vim.split(string.sub(line, 2, -2), "''", true)
|
|
local entry = {
|
|
head = fields[1],
|
|
refname = unescape_single_quote(fields[2]),
|
|
authorname = unescape_single_quote(fields[3]),
|
|
upstream = unescape_single_quote(fields[4]),
|
|
committerdate = fields[5],
|
|
}
|
|
local prefix
|
|
if vim.startswith(entry.refname, 'refs/remotes/') then
|
|
prefix = 'refs/remotes/'
|
|
elseif vim.startswith(entry.refname, 'refs/heads/') then
|
|
prefix = 'refs/heads/'
|
|
else
|
|
return
|
|
end
|
|
local index = 1
|
|
if entry.head ~= '*' then
|
|
index = #results + 1
|
|
end
|
|
|
|
entry.name = string.sub(entry.refname, string.len(prefix)+1)
|
|
for key, value in pairs(widths) do
|
|
widths[key] = math.max(value, strings.strdisplaywidth(entry[key] or ''))
|
|
end
|
|
if string.len(entry.upstream) > 0 then
|
|
widths.upstream_indicator = 2
|
|
end
|
|
table.insert(results, index, entry)
|
|
end
|
|
for _, line in ipairs(output) do
|
|
parse_line(line)
|
|
end
|
|
if #results == 0 then
|
|
return
|
|
end
|
|
|
|
local displayer = entry_display.create {
|
|
separator = " ",
|
|
items = {
|
|
{ width = 1 },
|
|
{ width = widths.name },
|
|
{ width = widths.authorname },
|
|
{ width = widths.upstream_indicator },
|
|
{ width = widths.upstream },
|
|
{ width = widths.committerdate },
|
|
}
|
|
}
|
|
|
|
local make_display = function(entry)
|
|
return displayer {
|
|
{entry.head},
|
|
{entry.name, 'TelescopeResultsIdentifier'},
|
|
{entry.authorname},
|
|
{string.len(entry.upstream) > 0 and '=>' or ''},
|
|
{entry.upstream, 'TelescopeResultsIdentifier'},
|
|
{entry.committerdate}
|
|
}
|
|
end
|
|
|
|
pickers.new(opts, {
|
|
prompt_title = 'Git Branches',
|
|
finder = finders.new_table {
|
|
results = results,
|
|
entry_maker = function(entry)
|
|
entry.value = entry.name
|
|
entry.ordinal = entry.name
|
|
entry.display = make_display
|
|
return entry
|
|
end
|
|
},
|
|
previewer = previewers.git_branch_log.new(opts),
|
|
sorter = conf.file_sorter(opts),
|
|
attach_mappings = function(_, map)
|
|
actions.select_default:replace(actions.git_checkout)
|
|
map('i', '<c-t>', actions.git_track_branch)
|
|
map('n', '<c-t>', actions.git_track_branch)
|
|
|
|
map('i', '<c-r>', actions.git_rebase_branch)
|
|
map('n', '<c-r>', actions.git_rebase_branch)
|
|
|
|
map('i', '<c-a>', actions.git_create_branch)
|
|
map('n', '<c-a>', actions.git_create_branch)
|
|
|
|
map('i', '<c-s>', actions.git_switch_branch)
|
|
map('n', '<c-s>', actions.git_switch_branch)
|
|
|
|
map('i', '<c-d>', actions.git_delete_branch)
|
|
map('n', '<c-d>', actions.git_delete_branch)
|
|
return true
|
|
end
|
|
}):find()
|
|
end
|
|
|
|
git.status = function(opts)
|
|
local gen_new_finder = function()
|
|
local expand_dir = utils.if_nil(opts.expand_dir, true, opts.expand_dir)
|
|
local git_cmd = {'git', 'status', '-s', '--', '.'}
|
|
|
|
if expand_dir then
|
|
table.insert(git_cmd, table.getn(git_cmd) - 1, '-u')
|
|
end
|
|
|
|
local output = utils.get_os_command_output(git_cmd, opts.cwd)
|
|
|
|
if table.getn(output) == 0 then
|
|
print('No changes found')
|
|
return
|
|
end
|
|
|
|
return finders.new_table {
|
|
results = output,
|
|
entry_maker = opts.entry_maker or make_entry.gen_from_git_status(opts)
|
|
}
|
|
end
|
|
|
|
local initial_finder = gen_new_finder()
|
|
if not initial_finder then return end
|
|
|
|
pickers.new(opts, {
|
|
prompt_title = 'Git Status',
|
|
finder = initial_finder,
|
|
previewer = previewers.git_file_diff.new(opts),
|
|
sorter = conf.file_sorter(opts),
|
|
attach_mappings = function(prompt_bufnr, map)
|
|
actions.git_staging_toggle:enhance {
|
|
post = function()
|
|
action_state.get_current_picker(prompt_bufnr):refresh(gen_new_finder(), { reset_prompt = true })
|
|
end,
|
|
}
|
|
|
|
map('i', '<tab>', actions.git_staging_toggle)
|
|
map('n', '<tab>', actions.git_staging_toggle)
|
|
return true
|
|
end
|
|
}):find()
|
|
end
|
|
|
|
local set_opts_cwd = function(opts)
|
|
if opts.cwd then
|
|
opts.cwd = vim.fn.expand(opts.cwd)
|
|
else
|
|
opts.cwd = vim.loop.cwd()
|
|
end
|
|
|
|
-- Find root of git directory and remove trailing newline characters
|
|
local git_root, ret = utils.get_os_command_output({ "git", "rev-parse", "--show-toplevel" }, opts.cwd)
|
|
local use_git_root = utils.get_default(opts.use_git_root, true)
|
|
|
|
if ret ~= 0 then
|
|
local output = utils.get_os_command_output({ "git", "rev-parse", "--is-inside-work-tree" }, opts.cwd)
|
|
if output[1] ~= "true" then
|
|
error(opts.cwd .. ' is not a git directory')
|
|
end
|
|
else
|
|
if use_git_root then
|
|
opts.cwd = git_root[1]
|
|
end
|
|
end
|
|
end
|
|
|
|
local function apply_checks(mod)
|
|
for k, v in pairs(mod) do
|
|
mod[k] = function(opts)
|
|
opts = opts or {}
|
|
|
|
set_opts_cwd(opts)
|
|
v(opts)
|
|
end
|
|
end
|
|
|
|
return mod
|
|
end
|
|
|
|
return apply_checks(git)
|