Merge pull request #1 from yangxgkem/master

LGTM! THX
KEINOS · Mar 23, 2018 · 097a0e5 · 097a0e5
2 parents 7651672 + 0ecb435
commit 097a0e5
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/Extension.php b/Extension.php
@@ -50,11 +50,23 @@ protected function setContentsList($Content)
         $text   = $this->fetchText($Content['text']);
         $link   = "[${text}](#${id})";
         $level  = (integer) trim($Content['level'],'h');
+
+        if ($this->firstHeadLevel === 0) {
+            $this->firstHeadLevel = $level;
+        }
+        $cutIndent = $this->firstHeadLevel - 1;
+        if ($cutIndent > $level) {
+            $level = 1;
+        } else {
+            $level = $level - $cutIndent;
+        }
+
         $indent = str_repeat('  ', $level);
 
         $this->contentsListString .= "${indent}- ${link}\n";
     }
     protected $contentsListString = '';
+    protected $firstHeadLevel = 0;
 
     #
     # Header