From 324721fa0e397a71891f5efcb7fe21b1b6b00f07 Mon Sep 17 00:00:00 2001
From: Steve Kemp <steve@steve.org.uk>
Date: Fri, 9 Apr 2021 17:10:59 +0300
Subject: [PATCH] [markdown-toc] Allow capping the level of headers printed
 (#31)

---
 cmd_markdown_toc.go | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/cmd_markdown_toc.go b/cmd_markdown_toc.go
index 473705f..e2b8a62 100644
--- a/cmd_markdown_toc.go
+++ b/cmd_markdown_toc.go
@@ -2,18 +2,17 @@ package main
 
 import (
 	"bufio"
+	"flag"
 	"fmt"
 	"os"
 	"strings"
-
-	"github.com/skx/subcommands"
 )
 
 // Structure for our options and state.
 type markdownTOCCommand struct {
 
-	// We embed the NoFlags option, because we accept no command-line flags.
-	subcommands.NoFlags
+	// Maximum level to include.
+	max int
 }
 
 // tocItem holds state for a single entry
@@ -58,6 +57,12 @@ func (t tocItem) String() string {
 		link)
 }
 
+// Arguments adds per-command args to the object.
+func (m *markdownTOCCommand) Arguments(f *flag.FlagSet) {
+	f.IntVar(&m.max, "max", 100, "The maximum nesting level to generate.")
+
+}
+
 // Info returns the name of this subcommand.
 func (m *markdownTOCCommand) Info() (string, string) {
 	return "markdown-toc", `Create a table-of-contents for a markdown file.
@@ -82,7 +87,9 @@ func (m *markdownTOCCommand) process(reader *bufio.Reader) error {
 	for fileScanner.Scan() {
 		line := fileScanner.Text()
 
-		if headerCount := m.countHashes(line); headerCount > 0 {
+		headerCount := m.countHashes(line)
+
+		if headerCount >= 1 && headerCount < m.max {
 
 			// Create an item for this header
 			item := tocItem{