vim-clap/autoload/clap/provider/blines.vim at master · gniw/vim-clap · GitHub

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
" Author: liuchengxu <xuliuchengxlc@gmail.com>
" Description: List the buffer lines.

let s:save_cpo = &cpoptions
set cpoptions&vim

let s:blines = {}

function! s:blines.sink(selected) abort
  let lnum = matchstr(a:selected, '^\s*\(\d\+\) ')
  let lnum = str2nr(trim(lnum))
  " Push the current position to the jumplist
  normal! m'
  silent call cursor(lnum, 1)
  normal! ^zvzz
endfunction

function! clap#provider#blines#format(lines) abort
  let linefmt = '%4d %s'
  return map(a:lines, 'printf(linefmt, v:key + 1, v:val)')
endfunction

function! s:blines.source() abort
  return clap#provider#blines#format(g:clap.start.get_lines())
endfunction

function! s:blines.on_move() abort
  let items = split(g:clap.display.getcurline())
  if empty(items)
    return
  endif
  if items[0] !~# '^\s*\d\+$'
    return
  endif
  let lnum = str2nr(items[0])
  call clap#preview#buffer(lnum, s:origin_syntax)
endfunction

function! s:blines.on_enter() abort
  let s:origin_syntax = getbufvar(g:clap.start.bufnr, '&syntax')
  call g:clap.display.setbufvar('&syntax', 'clap_blines')
endfunction

function! s:blines.init() abort
  let line_count = g:clap.start.line_count()
  let g:clap.display.initial_size = line_count

  if line_count > 0 && line_count < 100000
    let lines = getbufline(g:clap.start.bufnr, 1, g:clap.display.preload_capacity)
    call g:clap.display.set_lines_lazy(clap#provider#blines#format(lines))
    call g:clap#display_win.shrink_if_undersize()
    call clap#indicator#set_matches_number(line_count)
    call clap#sign#toggle_cursorline()
  endif
endfunction

function! s:into_qf_entry(line) abort
  if a:line =~# '^\s*\d\+ '
    let items = matchlist(a:line, '^\s*\(\d\+\) \(.*\)')
    return { 'bufnr': g:clap.start.bufnr, 'lnum': str2nr(trim(items[1])), 'text': clap#util#trim_leading(items[2]) }
  else
    return { 'bufnr': g:clap.start.bufnr, 'text': a:line }
  endif
endfunction

function! s:blines_sink_star(lines) abort
  call clap#util#open_quickfix(map(a:lines, 's:into_qf_entry(v:val)'))
endfunction

" if Source() is 1,000,000+ lines, it could be very slow, e.g.,
" `blines` provider, so we did a hard code for blines provider here.
let s:blines.source_type = g:__t_func_list
let s:blines['sink*'] = function('s:blines_sink_star')
let s:blines.on_move_async = function('clap#impl#on_move#async')
let g:clap#provider#blines# = s:blines

let &cpoptions = s:save_cpo
unlet s:save_cpo