SpaceVim/bundle/gina.vim/autoload/gina/command/status.vim

let s:Path = vital#gina#import('System.Filepath')
let s:String = vital#gina#import('Data.String')

let s:SCHEME = gina#command#scheme(expand('<sfile>'))


function! gina#command#status#call(range, args, mods) abort
  call gina#core#options#help_if_necessary(a:args, s:get_options())
  let git = gina#core#get_or_fail()
  let args = s:build_args(git, a:args)
  let bufname = gina#core#buffer#bufname(git, s:SCHEME, {
        \ 'params': [
        \   args.params.partial ? '--' : '',
        \ ],
        \})
  call gina#core#buffer#open(bufname, {
        \ 'mods': a:mods,
        \ 'group': args.params.group,
        \ 'opener': args.params.opener,
        \ 'cmdarg': args.params.cmdarg,
        \ 'callback': {
        \   'fn': function('s:init'),
        \   'args': [args],
        \ }
        \})
endfunction

function! gina#command#status#complete(arglead, cmdline, cursorpos) abort
  let args = gina#core#args#new(matchstr(a:cmdline, '^.*\ze .*'))
  if a:arglead[0] ==# '-' || !empty(args.get(1))
    let options = s:get_options()
    return options.complete(a:arglead, a:cmdline, a:cursorpos)
  endif
  return gina#complete#filename#any(a:arglead, a:cmdline, a:cursorpos)
endfunction


" Private --------------------------------------------------------------------
function! s:get_options() abort
  let options = gina#core#options#new()
  call options.define(
        \ '-h|--help',
        \ 'Show this help.',
        \)
  call options.define(
        \ '--opener=',
        \ 'A Vim command to open a new buffer.',
        \ ['edit', 'split', 'vsplit', 'tabedit', 'pedit'],
        \)
  call options.define(
        \ '--group=',
        \ 'A window group name used for the buffer.',
        \)
  return options
endfunction

function! s:build_args(git, args) abort
  let args = a:args.clone()
  let args.params.group = args.pop('--group', '')
  let args.params.opener = args.pop('--opener', '')
  let args.params.partial = !empty(args.residual())
  return args.lock()
endfunction

function! s:init(args) abort
  call gina#core#meta#set('args', a:args)

  if exists('b:gina_initialized')
    return
  endif
  let b:gina_initialized = 1

  setlocal buftype=nofile
  setlocal bufhidden=hide
  setlocal noswapfile
  setlocal nomodifiable
  setlocal autoread

  " Attach modules
  call gina#core#locator#attach()
  call gina#action#attach(function('s:get_candidates'), {
        \ 'markable': 1,
        \})

  augroup gina_command_status_internal
    autocmd! * <buffer>
    autocmd BufReadCmd <buffer>
          \ call gina#core#revelator#call(function('s:BufReadCmd'), [])
  augroup END
endfunction

function! s:BufReadCmd() abort
  let git = gina#core#get_or_fail()
  let args = gina#core#meta#get_or_fail('args')
  let pipe = gina#process#pipe#stream(s:writer)
  call gina#core#buffer#assign_cmdarg()
  call gina#process#open(git, args, pipe)
  setlocal filetype=gina-status
endfunction

function! s:compare_record(a, b) abort
  let a = matchstr(a:a, '...\zs.*')
  let b = matchstr(a:b, '...\zs.*')
  return a ==# b ? 0 : a > b ? 1 : -1
endfunction

function! s:get_candidates(fline, lline) abort
  let git = gina#core#get_or_fail()
  let args = gina#core#meta#get_or_fail('args')
  let conf = gina#core#repo#config(git)
  let residual = args.residual()
  if args.get('-s|--short') || get(conf, 'status.short', 'false') ==? 'true'
    let candidates = map(
          \ getline(a:fline, a:lline),
          \ 's:parse_record_short(v:val, residual)'
          \)
  else
    let candidates = map(
          \ getline(a:fline, a:lline),
          \ 's:parse_record_normal(v:val, residual)'
          \)
  endif
  return filter(candidates, '!empty(v:val)')
endfunction

function! s:parse_record_short(record, residual) abort
  let record = s:String.remove_ansi_sequences(a:record)
  let m = matchlist(
        \ record,
        \ '^\(..\) \("[^"]\{-}"\|.\{-}\)\%( -> \("[^"]\{-}"\|[^ ]\+\)\)\?$'
        \)
  if empty(m) || m[1] ==# '##'
    return {}
  endif
  let candidate = {
        \ 'word': record,
        \ 'abbr': a:record,
        \ 'sign': m[1],
        \ 'residual': a:residual,
        \}
  if len(m) && !empty(m[3])
    return extend(candidate, {
          \ 'path': s:strip_quotes(m[3]),
          \ 'path1': s:strip_quotes(m[2]),
          \ 'path2': s:strip_quotes(m[3]),
          \})
  else
    return extend(candidate, {
          \ 'path': s:strip_quotes(m[2]),
          \ 'path1': s:strip_quotes(m[2]),
          \ 'path2': '',
          \})
  endif
endfunction

function! s:parse_record_normal(record, residual) abort
  let signs = {
        \ 'modified': 'M',
        \ 'new file': 'A',
        \ 'deleted': 'D',
        \ 'renamed': 'R',
        \ 'copied': 'C',
        \ 'both added': 'AA',
        \ 'both deleted': 'DD',
        \ 'both modified': 'UU',
        \ 'added by us': 'AU',
        \ 'added by them': 'UA',
        \ 'deleted by us': 'DU',
        \ 'deleted by them': 'UD',
        \}
  let record = s:String.remove_ansi_sequences(a:record)
  if record !~# '^.\?\t'
    return {}
  endif
  let m = matchlist(record, printf(
        \ '^.\?\s\+\(%s\):\s\+\("[^"]\{-}"\|.\{-}\)\%%( -> \("[^"]\{-}"\|[^ ]\+\)\)\?$',
        \ join(keys(signs), '\|')
        \))
  if empty(m)
    " Untracked files
    let path = s:strip_quotes(substitute(record, '^\t', '', ''))
    return {
          \ 'word': record,
          \ 'abbr': a:record,
          \ 'sign': '??',
          \ 'residual': a:residual,
          \ 'path': path,
          \ 'path1': path,
          \ 'path2': '',
          \}
  endif
  if search('^Unmerged paths:', 'bnW') != 0
    " Conflict
    let sign = signs[m[1]]
  elseif search('^Changes not staged for commit:', 'bnW') != 0
    " Unstaged
    let sign = ' ' . signs[m[1]]
  else
    " Staged
    let sign = signs[m[1]] . ' '
  endif
  let candidate = {
        \ 'word': record,
        \ 'abbr': a:record,
        \ 'sign': sign,
        \ 'residual': a:residual,
        \}
  if len(m) && !empty(m[3])
    return extend(candidate, {
          \ 'path': s:strip_quotes(m[3]),
          \ 'path1': s:strip_quotes(m[2]),
          \ 'path2': s:strip_quotes(m[3]),
          \})
  else
    return extend(candidate, {
          \ 'path': s:strip_quotes(m[2]),
          \ 'path1': s:strip_quotes(m[2]),
          \ 'path2': '',
          \})
  endif
endfunction

function! s:strip_quotes(str) abort
  return a:str =~# '^\%(".*"\|''.*''\)$' ? a:str[1:-2] : a:str
endfunction


" Writer ---------------------------------------------------------------------
function! s:_writer_on_exit() abort dict
  call call(s:original_writer.on_exit, [], self)
  call gina#core#emitter#emit('command:called', s:SCHEME)
endfunction

let s:original_writer = gina#process#pipe#stream_writer()
let s:writer = extend(deepcopy(s:original_writer), {
      \ 'on_exit': function('s:_writer_on_exit'),
      \})


" Config ---------------------------------------------------------------------
call gina#config(expand('<sfile>'), {
      \ 'use_default_aliases': 1,
      \ 'use_default_mappings': 1,
      \})