summaryrefslogtreecommitdiff
path: root/lua/telescope/builtin/git.lua
blob: a5fdb825a5460bccf24e2be84dc51ea6a57eea4e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
local actions = require('telescope.actions')
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 conf = require('telescope.config').values

local git = {}

git.files = function(opts)
  local show_untracked = utils.get_default(opts.show_untracked, true)

  -- 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 File',
    finder = finders.new_oneshot_job(
      { "git", "ls-files", "--exclude-standard", "--cached", show_untracked and "--others" },
      opts
    ),
    previewer = previewers.cat.new(opts),
    sorter = conf.file_sorter(opts),
  }):find()
end

git.commits = function(opts)
  local cmd = 'git log --pretty=oneline --abbrev-commit'
  local results = vim.split(utils.get_os_command_output(cmd), '\n')

  pickers.new(opts, {
    prompt_title = 'Git Commits',
    finder = finders.new_table {
      results = results,
      entry_maker = make_entry.gen_from_git_commits(opts),
    },
    previewer = previewers.git_commit_diff.new(opts),
    sorter = conf.file_sorter(opts),
    attach_mappings = function()
      actions.goto_file_selection_edit:replace(actions.git_checkout)
      return true
    end
  }):find()
end

git.bcommits = function(opts)
  local cmd = 'git log --pretty=oneline --abbrev-commit ' .. vim.fn.expand('%')
  local results = vim.split(utils.get_os_command_output(cmd), '\n')

  pickers.new(opts, {
    prompt_title = 'Git BCommits',
    finder = finders.new_table {
      results = results,
      entry_maker = make_entry.gen_from_git_commits(opts),
    },
    previewer = previewers.git_commit_diff.new(opts),
    sorter = conf.file_sorter(opts),
    attach_mappings = function()
      actions.goto_file_selection_edit:replace(actions.git_checkout)
      return true
    end
  }):find()
end

git.branches = function(opts)
  -- Does this command in lua (hopefully):
  -- 'git branch --all | grep -v HEAD | sed "s/.* //;s#remotes/[^/]*/##" | sort -u'
  local output = vim.split(utils.get_os_command_output('git branch --all'), '\n')

  local tmp_results = {}
  for _, v in ipairs(output) do
    if not string.match(v, 'HEAD') and v ~= '' then
      v = string.gsub(v, '.* ', '')
      v = string.gsub(v, '^remotes/.*/', '')
      tmp_results[v] = true
    end
  end

  local results = {}
  for k, _ in pairs(tmp_results) do
    table.insert(results, k)
  end

  pickers.new(opts, {
    prompt_title = 'Git Branches',
    finder = finders.new_table {
      results = results,
      entry_maker = function(entry)
        return {
          value = entry,
          ordinal = entry,
          display = entry,
        }
      end
    },
    previewer = previewers.git_branch_log.new(opts),
    sorter = conf.file_sorter(opts),
    attach_mappings = function()
      actions.goto_file_selection_edit:replace(actions.git_checkout)
      return true
    end
  }):find()
end

git.status = function(opts)
  local output = vim.split(utils.get_os_command_output('git status -s'), '\n')
  local results = {}
  for _, v in ipairs(output) do
    if v ~= "" then
      local mod, fname = string.match(v, '(..)%s(.+)')
      if mod ~= 'A ' and mod ~= 'M ' and mod ~= 'R ' and mod ~= 'D ' then
        table.insert(results, { mod = mod, file = fname })
      end
    end
  end

  if vim.tbl_isempty(results) then
    print('No changes found')
    return
  end

  pickers.new(opts, {
    prompt_title = 'Git Status',
    finder = finders.new_table {
      results = results,
      entry_maker = function(entry)
        return {
          value = entry.file,
          ordinal = entry.mod .. ' ' .. entry.file,
          display = entry.mod .. ' ' .. entry.file,
        }
      end
    },
    previewer = previewers.git_file_diff.new(opts),
    sorter = conf.file_sorter(opts),
    attach_mappings = function()
      actions.goto_file_selection_edit:replace(actions.git_add)
      return true
    end
  }):find()
end

local set_opts_cwd = function(opts)
  local is_git_dir = function(path)
    vim.fn.system('cd ' .. path .. ' && git rev-parse HEAD >/dev/null 2>&1')
    if vim.v.shell_error ~= 0 then
      error(path .. ' is not a git directory')
    end
  end

  if opts.cwd then
    opts.cwd = vim.fn.expand(opts.cwd)
    is_git_dir(opts.cwd)
  else
    is_git_dir(vim.fn.expand('%:p:h'))
    --- Find root of git directory and remove trailing newline characters
    opts.cwd = vim.fn.systemlist("git rev-parse --show-toplevel")[1]
  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)