X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/1ec89a223c39af3eb4dbf4aff2698878cb03aa9c..e380b38bc4d60f6cf0274cf6fb0af57feac49967:/after/ftplugin/markdown.vim diff --git a/after/ftplugin/markdown.vim b/after/ftplugin/markdown.vim index 06a1583..9eee3c3 100644 --- a/after/ftplugin/markdown.vim +++ b/after/ftplugin/markdown.vim @@ -1,3 +1,4 @@ +" vim: ts=4: " folding for Markdown headers, both styles (atx- and setex-) " http://daringfireball.net/projects/markdown/syntax#header " @@ -13,19 +14,67 @@ endfunction if get(g:, "vim_markdown_folding_style_pythonic", 0) function! Foldexpr_markdown(lnum) + let l1 = getline(a:lnum) + "~~~~~ keep track of fenced code blocks ~~~~~ + "If we hit a code block fence + if l1 =~ '````*' || l1 =~ '\~\~\~\~*' + " toggle the variable that says if we're in a code block + if b:fenced_block == 0 + let b:fenced_block = 1 + elseif b:fenced_block == 1 + let b:fenced_block = 0 + endif + " else, if we're caring about front matter + elseif g:vim_markdown_frontmatter == 1 + " if we're in front matter and not on line 1 + if b:front_matter == 1 && a:lnum > 2 + let l0 = getline(a:lnum-1) + " if the previous line fenced front matter + if l0 == '---' + " we must not be in front matter + let b:front_matter = 0 + endif + " else, if we're on line one + elseif a:lnum == 1 + " if we hit a front matter fence + if l1 == '---' + " we're in the front matter + let b:front_matter = 1 + endif + endif + endif + + " if we're in a code block or front matter + if b:fenced_block == 1 || b:front_matter == 1 + if a:lnum == 1 + " fold any 'preamble' + return '>1' + else + " keep previous foldlevel + return '=' + endif + endif + let l2 = getline(a:lnum+1) - if l2 =~ '^==\+\s*' && !s:is_mkdCode(a:lnum+1) + " if the next line starts with two or more '=' + " and is not code + if l2 =~ '^==\+\s*' && !s:is_mkdCode(a:lnum+1) " next line is underlined (level 1) return '>0' + " else, if the nex line starts with two or more '-' + " and is not code elseif l2 =~ '^--\+\s*' && !s:is_mkdCode(a:lnum+1) " next line is underlined (level 2) return '>1' endif - let l1 = getline(a:lnum) + "if we're on a non-code line starting with a pound sign if l1 =~ '^#' && !s:is_mkdCode(a:lnum) - " current line starts with hashes - return '>'.(matchend(l1, '^#\+') - 1) + " set the fold level to the number of hashes -1 + " return '>'.(matchend(l1, '^#\+') - 1) + " set the fold level to the number of hashes + return '>'.(matchend(l1, '^#\+')) + " else, if we're on line 1 elseif a:lnum == 1 " fold any 'preamble' return '>1' @@ -46,13 +95,14 @@ if get(g:, "vim_markdown_folding_style_pythonic", 0) let fillcharcount = windowwidth - len(line) - len(foldedlinecount) + 1 return line . ' ' . repeat("-", fillcharcount) . ' ' . foldedlinecount endfunction -else +else " vim_markdown_folding_style_pythonic == 1 function! Foldexpr_markdown(lnum) if (a:lnum == 1) let l0 = '' else let l0 = getline(a:lnum-1) endif + " keep track of fenced code blocks if l0 =~ '````*' || l0 =~ '\~\~\~\~*' if b:fenced_block == 0 @@ -60,6 +110,21 @@ else elseif b:fenced_block == 1 let b:fenced_block = 0 endif + elseif g:vim_markdown_frontmatter == 1 + if b:front_matter == 1 + if l0 == '---' + let b:front_matter = 0 + endif + elseif a:lnum == 2 + if l0 == '---' + let b:front_matter = 1 + endif + endif + endif + + if b:fenced_block == 1 || b:front_matter == 1 + " keep previous foldlevel + return '=' endif let l2 = getline(a:lnum+1) @@ -68,7 +133,7 @@ else return '>1' elseif l2 =~ '^--\+\s*' && !s:is_mkdCode(a:lnum+1) " next line is underlined (level 2) - if g:vim_markdown_folding_level == 2 + if s:vim_markdown_folding_level >= 2 return '>1' else return '>2' @@ -78,24 +143,23 @@ else let l1 = getline(a:lnum) if l1 =~ '^#' && !s:is_mkdCode(a:lnum) " fold level according to option - let l:level = matchend(l1, '^#\+') - if g:vim_markdown_folding_level == 1 || l:level > g:vim_markdown_folding_level - return -1 + if s:vim_markdown_folding_level == 1 || matchend(l1, '^#\+') > s:vim_markdown_folding_level + if a:lnum == line('$') + return matchend(l1, '^#\+') - 1 + else + return -1 + endif else - " code blocks are always folded - return b:fenced_block + " headers are not folded + return 0 endif endif if l0 =~ '^#' && !s:is_mkdCode(a:lnum-1) - " collapse comments in fenced code blocks into a single fold - if b:fenced_block == 1 - return 1 - endif - " current line starts with hashes + " previous line starts with hashes return '>'.matchend(l0, '^#\+') else - " fold here because of setext headers + " keep previous foldlevel return '=' endif endfunction @@ -103,12 +167,23 @@ endif let b:fenced_block = 0 -let g:vim_markdown_folding_level = get(g:, "vim_markdown_folding_level", 1) +let b:front_matter = 0 +let s:vim_markdown_folding_level = get(g:, "vim_markdown_folding_level", 1) -if !get(g:, "vim_markdown_folding_disabled", 0) - setlocal foldexpr=Foldexpr_markdown(v:lnum) - setlocal foldmethod=expr - if get(g:, "vim_markdown_folding_style_pythonic", 0) - setlocal foldtext=Foldtext_markdown() - endif -endif +function! s:MarkdownSetupFolding() + if !get(g:, "vim_markdown_folding_disabled", 0) + setlocal foldexpr=Foldexpr_markdown(v:lnum) + setlocal foldmethod=expr + if get(g:, "vim_markdown_folding_style_pythonic", 0) && get(g:, "vim_markdown_override_foldtext", 1) + setlocal foldtext=Foldtext_markdown() + endif + endif +endfunction +call s:MarkdownSetupFolding() +augroup Mkd + " These autocmds need to be kept in sync with the autocmds calling + " s:MarkdownRefreshSyntax in ftplugin/markdown.vim. + autocmd BufWinEnter,BufWritePost call s:MarkdownSetupFolding() + autocmd InsertEnter,InsertLeave call s:MarkdownSetupFolding() + autocmd CursorHold,CursorHoldI call s:MarkdownSetupFolding() +augroup END