1
0
mirror of https://github.com/SpaceVim/SpaceVim.git synced 2025-02-09 16:20:07 +08:00
SpaceVim/bundle/gina.vim/autoload/gina/command/grep.vim

322 lines
9.0 KiB
VimL
Vendored

let s:String = vital#gina#import('Data.String')
let s:SCHEME = gina#command#scheme(expand('<sfile>'))
function! gina#command#grep#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#grep#complete(arglead, cmdline, cursorpos) abort
let args = gina#core#args#new(matchstr(a:cmdline, '^.*\ze .*'))
if a:cmdline =~# '\s--\s'
return gina#complete#filename#any(a:arglead, a:cmdline, a:cursorpos)
elseif a:arglead[0] ==# '-' || !empty(args.get(2))
let options = s:get_options()
return options.complete(a:arglead, a:cmdline, a:cursorpos)
endif
return gina#complete#commit#any(a:arglead, a:cmdline, a:cursorpos)
endfunction
function! gina#command#grep#parse_record(...) abort
return call('s:parse_record', a:000)
endfunction
function! gina#command#grep#_is_column_supported(version) abort
return s:is_column_supported(a:version)
endfunction
" Private --------------------------------------------------------------------
function! s:is_column_supported(version) abort
" https://github.com/git/git/blob/master/Documentation/RelNotes/2.19.0.txt#L18-L19
return a:version =~# '\%(^[^012]\|^2\.[^01]\|^2\.19\)'
endfunction
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.',
\)
call options.define(
\ '--cached',
\ 'Search in index instead of in the work tree',
\)
call options.define(
\ '--no-index',
\ 'Find in contents not managed by git',
\)
call options.define(
\ '--untracked',
\ 'Search in both tracked and untracked files',
\)
call options.define(
\ '--exclude-standard',
\ 'Ignore files specified via .gitignore',
\)
call options.define(
\ '-v|--invert-match',
\ 'Show non-matching lines',
\)
call options.define(
\ '-i|--ignore-case',
\ 'Case insensitive matching',
\)
call options.define(
\ '-w|--word-regexp',
\ 'Match patterns only at word boundaries',
\)
call options.define(
\ '-a|--text',
\ 'Process binary files as text',
\)
call options.define(
\ '-I',
\ 'Don''t match patterns in binary files',
\)
call options.define(
\ '--textconv',
\ 'Process binary files with textconv filters',
\)
call options.define(
\ '--max-depth=',
\ 'Descend at most <depth> levels',
\)
call options.define(
\ '-E|--extended-regexp',
\ 'Use extended POSIC regular expression',
\)
call options.define(
\ '-G|--basic-regexp',
\ 'Use basic POSIX regular expression',
\)
call options.define(
\ '-F|--fixed-string',
\ 'Interpret patterns as fixed strings',
\)
call options.define(
\ '-P|--perl-regexp',
\ 'Use Perl-compatible regular expression',
\)
call options.define(
\ '--break',
\ 'Print empty line between matches from different files',
\)
call options.define(
\ '-C|--context=',
\ 'Show <n> context lines before and after matches',
\)
call options.define(
\ '-B|--before-context=',
\ 'Show <n> context lines before matches',
\)
call options.define(
\ '-A|--after-context=',
\ 'Show <n> context lines after matches',
\)
call options.define(
\ '--threads=',
\ 'Use <n> worker threads',
\)
call options.define(
\ '-p|--show-function',
\ 'Show a line with the function name before matches',
\)
call options.define(
\ '-W|--function-context',
\ 'Show the surrounding function',
\)
call options.define(
\ '-f',
\ 'Read patterns from file',
\)
call options.define(
\ '-e',
\ 'Match <pattern>',
\)
call options.define(
\ '--and|--or|--not',
\ 'Combine patterns specified with -e',
\)
call options.define(
\ '--all-match',
\ 'Show only matches from files that match all patterns',
\)
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())
" Ask pattern if no option has specified.
if !s:is_pattern_given(args)
let pattern = gina#core#console#ask('Pattern: ')
if empty(pattern)
throw gina#core#revelator#info('Cancel')
endif
call args.set('-e', pattern)
endif
" Remove unsupported options
call args.pop('-h')
call args.pop('-H')
call args.pop('-l|--files-with-matches')
call args.pop('--name-only')
call args.pop('-L|--files-without-match')
call args.pop('-z|--null')
call args.pop('-c|--count')
call args.pop('--heading')
" Force required options
if !args.has('--no-column') && s:is_column_supported(gina#core#git_version())
call insert(args.raw, '--no-column', 1)
endif
if !args.has('--line-number')
call insert(args.raw, '--line-number', 1)
endif
if !args.has('--full-name')
call insert(args.raw, '--full-name', 1)
endif
if !args.has('--color')
call insert(args.raw, '--color=always', 1)
else
call args.set('--color', 'always')
endif
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
" Attach modules
call gina#core#locator#attach()
call gina#action#attach(function('s:get_candidates'), {
\ 'markable': 1,
\})
augroup gina_command_grep_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-grep
endfunction
function! s:get_candidates(fline, lline) abort
let args = gina#core#meta#get_or_fail('args')
let residual = args.residual()
let candidates = map(
\ getline(a:fline, a:lline),
\ 's:parse_record(v:val, residual)'
\)
return filter(candidates, '!empty(v:val)')
endfunction
function! s:parse_record(record, residual) abort
let record = s:String.remove_ansi_sequences(a:record)
let m = matchlist(record, '^\([^:]\+:\)\?\(.*\):\(\d\+\):\(.*\)$')
if empty(m)
return {}
endif
let matched = matchstr(a:record, '\e\[1;31m\zs.\{-}\ze\e\[m')
let line = str2nr(m[3])
let col = stridx(m[4], matched) + 1
let candidate = {
\ 'word': m[4],
\ 'abbr': a:record,
\ 'line': line,
\ 'col': col,
\ 'path': m[2],
\ 'rev': m[1],
\ 'residual': a:residual,
\}
return candidate
endfunction
function! s:is_pattern_given(args) abort
let cmdline = join(a:args.raw[1:])
let value_options = [
\ '--max-depth',
\ '-C', '--context',
\ '-A', '--after-context',
\ '-B', '--before-context',
\ '--threads',
\]
for value_option in value_options
let cmdline = substitute(
\ cmdline,
\ value_option . '\s\+\d\+',
\ '', 'g',
\)
endfor
return cmdline =~# '\<\%(-e\|-f\|[^-].\{-}\)\>'
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>'), {
\ 'send_to_quickfix': 1,
\ 'use_default_aliases': 1,
\ 'use_default_mappings': 1,
\})