Merge pull request luciopaiva#1 from dnchu/patch-1

Support setext-style headers

Author: luciopaiva

index.js

@@ -33,20 +33,65 @@ class MarkdownToc {
         const menus = ["# Table of contents", ""];
         let isCodeBlock = false;
         let topLevel = NaN;
+        let previous = null;
 
         for (let line of input.split("\n")) {
 
-            if (line.startsWith("```")) {
+            const trimmed = line.trim();
+    
+            if (trimmed.startsWith("```")) {
                 isCodeBlock = !isCodeBlock;
             }
 
             if (isCodeBlock) {
                 continue;
             }
 
-            if (line.startsWith("#")) {
-                const match = line.match(/(#+)\s*(.*?)#*\s*$/);
-                const level = match[1].length;
+            let level = NaN;
+            let title = null;
+
+            // Check for:
+            // 1. ATX-style headers: ## My Header
+            //
+            // 2. Setext-style headers:
+            //     a) Level 1 header: My Header
+            //                        =========
+            //
+            //     b) Level 2 header: My Header
+            //                        ---------
+            //
+            //    Edge cases that do not count as headers:
+            //     i) Horizontal rule ("Underline" preceded by empty line):
+            //
+            //           Some paragraph 1
+            //           <empty line>
+            //           -----
+            //           Some paragraph 2
+            //
+            //     ii) Two or more horizontal rules:
+            //
+            //           Some paragraph 1
+            //
+            //           -----
+            //           -----
+            //           -----
+            //           Some paragraph 2
+
+            if (trimmed.startsWith("#")) {
+                const match = trimmed.match(/(#+)\s*(.*?)#*\s*$/);
+                level = match[1].length;
+                title = match[2].trim();
+            } else if (previous != null && previous.length > 0 && trimmed.length > 0) {
+                if (trimmed.match(/[^=]/g) == null) {
+                    level = 1;
+                    title = previous;
+                } else if (trimmed.match(/[^-]/g) == null && previous.match(/[^-]/g) != null) {
+                    level = 2;
+                    title = previous;
+                }
+            }
+
+            if (!isNaN(level) && title != null) {
                 if (isNaN(topLevel)) {
                     topLevel = level;
                 }
@@ -55,12 +100,15 @@ class MarkdownToc {
                     continue;
                 }
 
-                const title = match[2].trim();
                 const link = title.toLocaleLowerCase()
                     .replace(/\s/g, "-")
                     .replace(/[^A-Za-z0-9-]/g, "");
                 const menu = `${"  ".repeat(level - topLevel)}- [${title}](#${link})`;
                 menus.push(menu);
+
+                previous = null;
+            } else {
+                previous = trimmed;
             }
         }