- let l1 = s:effective_line(a:lnum)
-
- let l2 = s:effective_line(a:lnum+1)
-
- 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 = s:effective_line(a:lnum)
+ if l1 =~ '^#'
+ " current line starts with hashes
+ return '>'.(matchend(l1, '^#\+') - 1)
+ elseif a:lnum == 1
+ " fold any 'preamble'
+ return '>1'
+ else
+ " keep previous foldlevel
+ return '='
+ endif
+ endfunc
+
+ fun! 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
+ endfunc
+else
+ func! Foldexpr_markdown(lnum)
+ let l2 = s:effective_line(a:lnum+1)
+ if l2 =~ '^==\+\s*'
+ " next line is underlined (level 1)
+ return '>1'
+ elseif l2 =~ '^--\+\s*'
+ " next line is underlined (level 2)
+ return '>2'
+ endif