summaryrefslogtreecommitdiffstats
path: root/ftplugin/cryptol/folding.vim
blob: 210d0573caf7dcb70def072189868e6672327ca0 (plain) (blame)
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
if !exists('g:polyglot_disabled') || index(g:polyglot_disabled, 'cryptol') == -1

" Copyright © 2013 Edward O'Callaghan. All Rights Reserved.

"setlocal foldmethod=indent
"setlocal foldignore=

setlocal foldmethod=expr
setlocal foldexpr=GetPotionFold(v:lnum)

" Helper function: To tackle non-blank lines,
" wish to know their indentation level
function! IndentLevel(lnum)
    return indent(a:lnum) / &shiftwidth
endfunction

" Helper function: .
function! NextNonBlankLine(lnum)
    let numlines = line('$')
    let current = a:lnum + 1

    while current <= numlines
        if getline(current) =~? '\v\S'
            return current
        endif

        let current += 1
    endwhile

    return -2
endfunction

function! GetPotionFold(lnum)
    if getline(a:lnum) =~? '\v^\s*$'
        return '-1'
    endif

    let this_indent = IndentLevel(a:lnum)
    let next_indent = IndentLevel(NextNonBlankLine(a:lnum))

    if next_indent == this_indent
        return this_indent
    elseif next_indent < this_indent
        return this_indent
    elseif next_indent > this_indent
        return '>' . next_indent
    endif
endfunction

endif