fix TOC generation for h2 h3 h2 heading sequence

sunaku · May 28, 2015 · fd89685 · fd89685
1 parent a3080cc
commit fd89685
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/Rakefile b/Rakefile
@@ -56,13 +56,9 @@ begin
       # keep track of the heading to generate the table_of_contents() later on
       heading = Heading.new(level, id, text, [], nil)
       if parent = @headings.last
-        if parent.level == level
-          parent = parent.parent
-        end
-        if parent and parent.level < level
-          parent.children << heading
-          heading.parent = parent
-        end
+        parent = parent.parent while parent and parent.level >= level
+        parent.children << heading if parent
+        heading.parent = parent
       end
       @headings << heading