From 27f424260283e1d3f87c144f1756ef2228c3f009 Mon Sep 17 00:00:00 2001
From: Kaleb Elwert <belak@coded.io>
Date: Mon, 9 Dec 2024 23:37:58 -0800
Subject: [PATCH] Allow configuring the max parsed heading level

This allows for implementing parsers for services like Discord which use
a somewhat broken form of markdown only supporting H1-H3.
---
 parser/atx_heading.go | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/parser/atx_heading.go b/parser/atx_heading.go
index dae5e84..3c937fe 100644
--- a/parser/atx_heading.go
+++ b/parser/atx_heading.go
@@ -10,6 +10,7 @@ import (
 type HeadingConfig struct {
 	AutoHeadingID bool
 	Attribute     bool
+	MaxLevel      int
 }
 
 // SetOption implements SetOptioner.
@@ -56,6 +57,22 @@ func (o *withHeadingAttribute) SetHeadingOption(p *HeadingConfig) {
 	p.Attribute = true
 }
 
+// WithMaxHeadingLevel is a functional option that enables limiting heading
+// parsing a custom max level.
+func WithMaxHeadingLevel(maxLevel int) HeadingOption {
+	return &withMaxHeadingLevel{maxLevel: maxLevel}
+}
+
+type withMaxHeadingLevel struct {
+	Option
+
+	maxLevel int
+}
+
+func (o *withMaxHeadingLevel) SetHeadingOption(p *HeadingConfig) {
+	p.MaxLevel = o.maxLevel
+}
+
 // WithHeadingAttribute is a functional option that enables custom heading attributes.
 func WithHeadingAttribute() HeadingOption {
 	return &withHeadingAttribute{WithAttribute()}
@@ -67,7 +84,11 @@ type atxHeadingParser struct {
 
 // NewATXHeadingParser return a new BlockParser that can parse ATX headings.
 func NewATXHeadingParser(opts ...HeadingOption) BlockParser {
-	p := &atxHeadingParser{}
+	p := &atxHeadingParser{
+		HeadingConfig: HeadingConfig{
+			MaxLevel: 6,
+		},
+	}
 	for _, o := range opts {
 		o.SetHeadingOption(&p.HeadingConfig)
 	}
@@ -88,7 +109,7 @@ func (b *atxHeadingParser) Open(parent ast.Node, reader text.Reader, pc Context)
 	for ; i < len(line) && line[i] == '#'; i++ {
 	}
 	level := i - pos
-	if i == pos || level > 6 {
+	if i == pos || level > b.HeadingConfig.MaxLevel {
 		return nil, NoChildren
 	}
 	if i == len(line) { // alone '#' (without a new line character)