vim-signify/autoload/sy/fold.vim

" vim: et sw=2 sts=2

" Function: SignifyFoldExpr {{{1
function! SignifyFoldExpr(lnum)
  return s:levels[a:lnum]
endfunction

" Function: SignifyFoldText {{{1
function! SignifyFoldText()
  let linelen = &textwidth ? &textwidth : 80
  let marker  = &foldmarker[:stridx(&foldmarker, ',')-1]
  let range   = foldclosedend(v:foldstart) - foldclosed(v:foldstart) + 1

  let left    = substitute(getline(v:foldstart), marker, '', '')
  let leftlen = len(left)

  let right    = printf('%d [%d]', range, v:foldlevel)
  let rightlen = len(right)

  let tmp    = strpart(left, 0, linelen - rightlen)
  let tmplen = len(tmp)

  if leftlen > tmplen
    let left    = strpart(tmp, 0, tmplen - 4) . '... '
    let leftlen = tmplen
  endif

  let fill = repeat(' ', linelen - (leftlen + rightlen))

  " return left . fill . right . repeat(' ', 100)
  return left . fill . right
endfunction

" Function: #dispatch {{{1
function! sy#fold#dispatch(do_tab) abort
  if a:do_tab
    call sy#fold#enable(1)
  else
    call sy#fold#toggle()
  endif
endfunction

" Function: #enable {{{1
function! sy#fold#enable(do_tab) abort
  execute sy#util#return_if_no_changes()

  if a:do_tab
    tabedit %
  endif

  let [s:context0, s:context1] = get(g:, 'signify_fold_context', [3, 8])
  let s:levels = s:get_levels(s:get_lines())

  setlocal foldexpr=SignifyFoldExpr(v:lnum)
  setlocal foldtext=SignifyFoldText()
  setlocal foldmethod=expr
  setlocal foldlevel=0
endfunction

" Function: #disable {{{1
function! sy#fold#disable() abort
  let &l:foldmethod = b:sy_folded.method
  let &l:foldtext = b:sy_folded.text
  normal! zv
endfunction

" Function: #toggle {{{1
function! sy#fold#toggle() abort
  if exists('b:sy_folded')
    call sy#fold#disable()
    if b:sy_folded.method == 'manual'
      loadview
    endif
    unlet b:sy_folded
  else
    let b:sy_folded = { 'method': &foldmethod, 'text': &foldtext }
    if &foldmethod == 'manual'
      let old_vop = &viewoptions
      mkview
      let &viewoptions = old_vop
    endif
    call sy#fold#enable(0)
  endif

  redraw!
  call sy#start()
endfunction

" Function: s:get_lines {{{1
function! s:get_lines() abort
  let lang = v:lang
  language message C
  redir => signlist
    silent! execute 'sign place buffer='. b:sy.buffer
  redir END
  silent! execute 'language message' lang

  let lines = []
  for line in split(signlist, '\n')[2:]
    call insert(lines, matchlist(line, '\v^\s+line\=(\d+)')[1], 0)
  endfor

  return reverse(lines)
endfunction
" }}}

" Function: s:get_levels {{{1
function! s:get_levels(lines) abort
  let levels = {}

  for line in range(1, line('$'))
    let levels[line] = 2
  endfor

  for line in a:lines
    for l in range(line - s:context1, line + s:context1)
      if (l < 1) || (l > line('$'))
        continue
      endif
      if levels[l] == 2
        let levels[l] = 1
      endif
      for ll in range(line - s:context0, line + s:context0)
        let levels[ll] = 0
      endfor
    endfor
  endfor

  return levels
endfunction