feat: add -s --skip flag

ycd · Jan 21, 2021 · 0937464 · 0937464
1 parent 9c3bccd
commit 0937464
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/toc/toc.go b/pkg/toc/toc.go
@@ -69,7 +69,11 @@ func (t *toc) logic() {
 }
 
 func (t *toc) String() (s string) {
-	for _, v := range t.Content {
+	if t.Options.Skip >= len(t.Content) {
+		color.Red("ERROR: skip value is bigger than the length of table of contents")
+		os.Exit(1)
+	}
+	for _, v := range t.Content[t.Options.Skip:] {
 		s += v
 	}