115 lines
2.6 KiB
VimL
115 lines
2.6 KiB
VimL
function! gitgutter#fold#enable()
|
|
call s:save_fold_state()
|
|
|
|
call s:set_fold_levels()
|
|
setlocal foldexpr=gitgutter#fold#level(v:lnum)
|
|
setlocal foldmethod=expr
|
|
setlocal foldlevel=0
|
|
setlocal foldenable
|
|
|
|
call gitgutter#utility#setbufvar(bufnr(''), 'folded', 1)
|
|
endfunction
|
|
|
|
|
|
function! gitgutter#fold#disable()
|
|
call s:restore_fold_state()
|
|
call gitgutter#utility#setbufvar(bufnr(''), 'folded', 0)
|
|
endfunction
|
|
|
|
|
|
function! gitgutter#fold#toggle()
|
|
if s:folded()
|
|
call gitgutter#fold#disable()
|
|
else
|
|
call gitgutter#fold#enable()
|
|
endif
|
|
endfunction
|
|
|
|
|
|
function! gitgutter#fold#level(lnum)
|
|
return gitgutter#utility#getbufvar(bufnr(''), 'fold_levels')[a:lnum]
|
|
endfunction
|
|
|
|
|
|
function! gitgutter#fold#foldtext()
|
|
if !gitgutter#fold#is_changed()
|
|
return foldtext()
|
|
endif
|
|
|
|
return substitute(foldtext(), ':', ' (*):', '')
|
|
endfunction
|
|
|
|
|
|
" Returns 1 if any of the folded lines have been changed
|
|
" (added, removed, or modified), 0 otherwise.
|
|
function! gitgutter#fold#is_changed()
|
|
for hunk in gitgutter#hunk#hunks(bufnr(''))
|
|
let hunk_begin = hunk[2]
|
|
let hunk_end = hunk[2] + (hunk[3] == 0 ? 1 : hunk[3])
|
|
|
|
if hunk_end < v:foldstart
|
|
continue
|
|
endif
|
|
|
|
if hunk_begin > v:foldend
|
|
break
|
|
endif
|
|
|
|
return 1
|
|
endfor
|
|
|
|
return 0
|
|
endfunction
|
|
|
|
|
|
" A line in a hunk has a fold level of 0.
|
|
" A line within 3 lines of a hunk has a fold level of 1.
|
|
" All other lines have a fold level of 2.
|
|
function! s:set_fold_levels()
|
|
let fold_levels = ['']
|
|
|
|
for lnum in range(1, line('$'))
|
|
let in_hunk = gitgutter#hunk#in_hunk(lnum)
|
|
call add(fold_levels, (in_hunk ? 0 : 2))
|
|
endfor
|
|
|
|
let lines_of_context = 3
|
|
|
|
for lnum in range(1, line('$'))
|
|
if fold_levels[lnum] == 2
|
|
let pre = lnum >= 3 ? lnum - lines_of_context : 0
|
|
let post = lnum + lines_of_context
|
|
if index(fold_levels[pre:post], 0) != -1
|
|
let fold_levels[lnum] = 1
|
|
endif
|
|
endif
|
|
endfor
|
|
|
|
call gitgutter#utility#setbufvar(bufnr(''), 'fold_levels', fold_levels)
|
|
endfunction
|
|
|
|
|
|
function! s:save_fold_state()
|
|
let bufnr = bufnr('')
|
|
call gitgutter#utility#setbufvar(bufnr, 'foldlevel', &foldlevel)
|
|
call gitgutter#utility#setbufvar(bufnr, 'foldmethod', &foldmethod)
|
|
if &foldmethod ==# 'manual'
|
|
mkview
|
|
endif
|
|
endfunction
|
|
|
|
function! s:restore_fold_state()
|
|
let bufnr = bufnr('')
|
|
let &foldlevel = gitgutter#utility#getbufvar(bufnr, 'foldlevel')
|
|
let &foldmethod = gitgutter#utility#getbufvar(bufnr, 'foldmethod')
|
|
if &foldmethod ==# 'manual'
|
|
loadview
|
|
else
|
|
normal! zx
|
|
endif
|
|
endfunction
|
|
|
|
function! s:folded()
|
|
return gitgutter#utility#getbufvar(bufnr(''), 'folded')
|
|
endfunction
|
|
|