"
" original version from Steve Losh's gist: https://gist.github.com/1038710
-func! s:is_mkdCode(lnum)
+function! s:is_mkdCode(lnum)
return synIDattr(synID(a:lnum, 1, 0), 'name') == 'mkdCode'
-endfunc
+endfunction
-func! s:effective_line(lnum)
- return s:is_mkdCode(a:lnum) ? '' : getline(a:lnum)
-endfunc
+if get(g:, "vim_markdown_folding_style_pythonic", 0)
+ function! Foldexpr_markdown(lnum)
+ let l2 = getline(a:lnum+1)
+ if l2 =~ '^==\+\s*' && !s:is_mkdCode(a:lnum+1)
+ " next line is underlined (level 1)
+ return '>0'
+ elseif l2 =~ '^--\+\s*' && !s:is_mkdCode(a:lnum+1)
+ " next line is underlined (level 2)
+ return '>1'
+ endif
-func! Foldexpr_markdown(lnum)
- if (a:lnum == 1)
- let l0 = ''
- else
- let l0 = s:effective_line(a:lnum-1)
- endif
-
- let l1 = s:effective_line(a:lnum)
+ let l1 = getline(a:lnum)
+ if l1 =~ '^#' && !s:is_mkdCode(a:lnum)
+ " current line starts with hashes
+ return '>'.(matchend(l1, '^#\+') - 1)
+ elseif a:lnum == 1
+ " fold any 'preamble'
+ return '>1'
+ else
+ " keep previous foldlevel
+ return '='
+ endif
+ endfunction
- let l2 = s:effective_line(a:lnum+1)
+ function! Foldtext_markdown()
+ let line = getline(v:foldstart)
+ let has_numbers = &number || &relativenumber
+ let nucolwidth = &fdc + has_numbers * &numberwidth
+ let windowwidth = winwidth(0) - nucolwidth - 6
+ let foldedlinecount = v:foldend - v:foldstart
+ let line = strpart(line, 0, windowwidth - 2 -len(foldedlinecount))
+ let line = substitute(line, '\%("""\|''''''\)', '', '')
+ let fillcharcount = windowwidth - len(line) - len(foldedlinecount) + 1
+ return line . ' ' . repeat("-", fillcharcount) . ' ' . foldedlinecount
+ endfunction
+else
+ function! Foldexpr_markdown(lnum)
+ let l2 = getline(a:lnum+1)
+ if l2 =~ '^==\+\s*' && !s:is_mkdCode(a:lnum+1)
+ " next line is underlined (level 1)
+ return '>1'
+ elseif l2 =~ '^--\+\s*' && !s:is_mkdCode(a:lnum+1)
+ " next line is underlined (level 2)
+ return '>2'
+ endif
- if l2 =~ '^==\+\s*'
- " next line is underlined (level 1)
- return '>1'
- elseif l2 =~ '^--\+\s*'
- " next line is underlined (level 2)
- return '>2'
- elseif l1 =~ '^#'
- " don't include the section title in the fold
- return '-1'
- elseif l0 =~ '^#'
- " current line starts with hashes
- return '>'.matchend(l0, '^#\+')
- else
- " keep previous foldlevel
- return '='
- endif
-endfunc
+ let l1 = getline(a:lnum)
+ if l1 =~ '^#' && !s:is_mkdCode(a:lnum)
+ " don't include the section title in the fold
+ return '-1'
+ endif
+ if (a:lnum == 1)
+ let l0 = ''
+ else
+ let l0 = getline(a:lnum-1)
+ endif
+ if l0 =~ '^#' && !s:is_mkdCode(a:lnum-1)
+ " current line starts with hashes
+ return '>'.matchend(l0, '^#\+')
+ else
+ " keep previous foldlevel
+ return '='
+ endif
+ endfunction
+endif
if !get(g:, "vim_markdown_folding_disabled", 0)
- setlocal foldexpr=Foldexpr_markdown(v:lnum)
- setlocal foldmethod=expr
+ setlocal foldexpr=Foldexpr_markdown(v:lnum)
+ setlocal foldmethod=expr
+ if get(g:, "vim_markdown_folding_style_pythonic", 0)
+ setlocal foldtext=Foldtext_markdown()
+ endif
endif