11package attributes
22
33import (
4+ "github.com/gohugoio/hugo/markup/goldmark/goldmark_config"
5+ "github.com/gohugoio/hugo/markup/goldmark/internal/render"
46 "github.com/yuin/goldmark"
57 "github.com/yuin/goldmark/ast"
8+ east "github.com/yuin/goldmark/extension/ast"
69 "github.com/yuin/goldmark/parser"
710 "github.com/yuin/goldmark/text"
811 "github.com/yuin/goldmark/util"
@@ -14,24 +17,29 @@ import (
1417
1518var (
1619 kindAttributesBlock = ast .NewNodeKind ("AttributesBlock" )
20+ attrNameID = []byte ("id" )
1721
18- defaultParser = new (attrParser )
19- defaultTransformer = new (transformer )
20- attributes goldmark.Extender = new (attrExtension )
22+ defaultParser = new (attrParser )
2123)
2224
23- func New () goldmark.Extender {
24- return attributes
25+ func New (cfg goldmark_config. Parser ) goldmark.Extender {
26+ return & attrExtension { cfg : cfg }
2527}
2628
27- type attrExtension struct {}
29+ type attrExtension struct {
30+ cfg goldmark_config.Parser
31+ }
2832
2933func (a * attrExtension ) Extend (m goldmark.Markdown ) {
34+ if a .cfg .Attribute .Block {
35+ m .Parser ().AddOptions (
36+ parser .WithBlockParsers (
37+ util .Prioritized (defaultParser , 100 )),
38+ )
39+ }
3040 m .Parser ().AddOptions (
31- parser .WithBlockParsers (
32- util .Prioritized (defaultParser , 100 )),
3341 parser .WithASTTransformers (
34- util .Prioritized (defaultTransformer , 100 ),
42+ util .Prioritized (& transformer { cfg : a . cfg } , 100 ),
3543 ),
3644 )
3745}
@@ -92,18 +100,47 @@ func (a *attributesBlock) Kind() ast.NodeKind {
92100 return kindAttributesBlock
93101}
94102
95- type transformer struct {}
103+ type transformer struct {
104+ cfg goldmark_config.Parser
105+ }
106+
107+ func (a * transformer ) isFragmentNode (n ast.Node ) bool {
108+ switch n .Kind () {
109+ case east .KindDefinitionTerm , ast .KindHeading :
110+ return true
111+ default :
112+ return false
113+ }
114+ }
96115
97116func (a * transformer ) Transform (node * ast.Document , reader text.Reader , pc parser.Context ) {
98- attributes := make ([]ast.Node , 0 , 500 )
117+ var attributes []ast.Node
118+ if a .cfg .Attribute .Block {
119+ attributes = make ([]ast.Node , 0 , 500 )
120+ }
99121 ast .Walk (node , func (node ast.Node , entering bool ) (ast.WalkStatus , error ) {
100- if entering && node .Kind () == kindAttributesBlock {
122+ if ! entering {
123+ return ast .WalkContinue , nil
124+ }
125+
126+ if a .isFragmentNode (node ) {
127+ if id , found := node .Attribute (attrNameID ); ! found {
128+ a .generateAutoID (node , reader , pc )
129+ } else {
130+ pc .IDs ().Put (id .([]byte ))
131+ }
132+ }
133+
134+ if a .cfg .Attribute .Block && node .Kind () == kindAttributesBlock {
101135 // Attributes for fenced code blocks are handled in their own extension,
102136 // but note that we currently only support code block attributes when
103137 // CodeFences=true.
104138 if node .PreviousSibling () != nil && node .PreviousSibling ().Kind () != ast .KindFencedCodeBlock && ! node .HasBlankPreviousLines () {
105139 attributes = append (attributes , node )
106140 return ast .WalkSkipChildren , nil
141+ } else {
142+ // remove attributes node
143+ node .Parent ().RemoveChild (node .Parent (), node )
107144 }
108145 }
109146
@@ -123,3 +160,33 @@ func (a *transformer) Transform(node *ast.Document, reader text.Reader, pc parse
123160 attr .Parent ().RemoveChild (attr .Parent (), attr )
124161 }
125162}
163+
164+ func (a * transformer ) generateAutoID (n ast.Node , reader text.Reader , pc parser.Context ) {
165+ var text []byte
166+ switch n := n .(type ) {
167+ case * ast.Heading :
168+ if a .cfg .AutoHeadingID {
169+ text = textHeadingID (n , reader )
170+ }
171+ case * east.DefinitionTerm :
172+ if a .cfg .AutoDefinitionTermID {
173+ text = []byte (render .TextPlain (n , reader .Source ()))
174+ }
175+ }
176+
177+ if len (text ) > 0 {
178+ headingID := pc .IDs ().Generate (text , n .Kind ())
179+ n .SetAttribute (attrNameID , headingID )
180+ }
181+ }
182+
183+ // Markdown settext headers can have multiple lines, use the last line for the ID.
184+ func textHeadingID (node * ast.Heading , reader text.Reader ) []byte {
185+ var line []byte
186+ lastIndex := node .Lines ().Len () - 1
187+ if lastIndex > - 1 {
188+ lastLine := node .Lines ().At (lastIndex )
189+ line = lastLine .Value (reader .Source ())
190+ }
191+ return line
192+ }
0 commit comments