Implement HeadingDepth() for underlined headings.

masukomi · Oct 24, 2012 · a80b1b3 · a80b1b3
1 parent 06ff407
commit a80b1b3
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 1 deletion.
diff --git a/after/ftplugin/markdown/folding.vim b/after/ftplugin/markdown/folding.vim
@@ -5,6 +5,15 @@ function! HeadingDepth(lnum)
   let hashCount = len(matchstr(thisline, '^#\{1,6}'))
   if hashCount > 0
     let level = hashCount
+  else
+    if thisline != ''
+      let nextline = getline(a:lnum + 1)
+      if nextline =~ '^==='
+        let level = 1
+      elseif nextline =~ '^---'
+        let level = 2
+      endif
+    endif
   endif
   return level
 endfunction

diff --git a/test/folding.vim b/test/folding.vim
@@ -68,7 +68,7 @@ describe 'HeadingDepth'
     Expect HeadingDepth(2)  ==# 0
   end
 
-  it 'returns 0 for content lines'
+  it 'returns 0 for lines of content'
     call PopulateBuffer([
           \ 'lorem ipsum dolor sit amet',
           \ 'consequeteur blah blah'
@@ -92,4 +92,41 @@ describe 'HeadingDepth'
     Expect HeadingDepth(1)  ==# 3
   end
 
+  it 'returns 1 for === underscored headings'
+    call PopulateBuffer([
+          \ 'Level one heading',
+          \ '=================',
+          \ ])
+    Expect HeadingDepth(1)  ==# 1
+    Expect HeadingDepth(2)  ==# 0
+  end
+
+  it 'returns 2 for --- underscored headings'
+    call PopulateBuffer([
+          \ 'Level two heading',
+          \ '-----------------',
+          \ ])
+    Expect HeadingDepth(1)  ==# 2
+    Expect HeadingDepth(2)  ==# 0
+  end
+
+  it 'ignores --- and === when preceded by a blank line'
+    call PopulateBuffer([
+          \ '=================',
+          \ '',
+          \ '=================',
+          \ '',
+          \ '-----------------',
+          \ '',
+          \ '-----------------',
+          \ ])
+    Expect HeadingDepth(1)  ==# 0
+    Expect HeadingDepth(2)  ==# 0
+    Expect HeadingDepth(3)  ==# 0
+    Expect HeadingDepth(4)  ==# 0
+    Expect HeadingDepth(5)  ==# 0
+    Expect HeadingDepth(6)  ==# 0
+    Expect HeadingDepth(7)  ==# 0
+  end
+
 end