X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/2eaed44e965cf5a9a84a0cfbbf654c3b7b8f46b2..bd875ab511fafa4504f36852a42927ef9c0e1fd1:/after/ftplugin/markdown.vim

diff --git a/after/ftplugin/markdown.vim b/after/ftplugin/markdown.vim
index 997ab9d..a140c25 100644
--- a/after/ftplugin/markdown.vim
+++ b/after/ftplugin/markdown.vim
@@ -12,30 +12,31 @@ endfunc
 
 func! s:effective_line(lnum)
     let line = getline(a:lnum)
-    return (line !~ '^=\|^#' || s:is_mkdCode(a:lnum)) ? '' : line
+    return (line !~ '^[=-#]' || s:is_mkdCode(a:lnum)) ? '' : line
 endfunc
 
 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 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 =~ '^#'
+    endif
+
+    let l1 = s:effective_line(a:lnum)
+    if l1 =~ '^#'
         " don't include the section title in the fold
         return '-1'
-    elseif l0 =~ '^#'
+    endif
+
+    if (a:lnum == 1)
+        let l0 = ''
+    else
+        let l0 = s:effective_line(a:lnum-1)
+    endif
+    if l0 =~ '^#'
         " current line starts with hashes
         return '>'.matchend(l0, '^#\+')
     else