Fix panic on empty TOC

kubernetes-sigs · Jun 27, 2019 · 4dc3d6f · 4dc3d6f
1 parent bfd0b5a
commit 4dc3d6f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/mdtoc.go b/mdtoc.go
@@ -135,9 +135,9 @@ func run(file string, opts options) (string, error) {
 		return toc, err
 	}
 
-	realStart := start + len(startTOC) + len("\n")
+	realStart := start + len(startTOC)
 	oldTOC := string(raw[realStart:end])
-	if oldTOC == toc {
+	if strings.TrimSpace(oldTOC) == strings.TrimSpace(toc) {
 		// No changes required.
 		return toc, nil
 	} else if opts.dryrun {

diff --git a/testdata/empty_toc.md b/testdata/empty_toc.md
@@ -5,7 +5,6 @@ description: we should ignore it
 
 # TOC
 
-<!-- toc -->
-<!-- /toc -->
+<!-- toc --><!-- /toc -->
 
 # Only Heading