put ID to generator

This commit is contained in:
Dmitry Sedykh 2019-12-22 09:46:40 +03:00
parent 2aab93edb4
commit e0a5ff07c2
2 changed files with 6 additions and 2 deletions

View file

@ -171,9 +171,11 @@ func (b *atxHeadingParser) Close(node ast.Node, reader text.Reader, pc Context)
}
if b.AutoHeadingID {
_, ok := node.AttributeString("id")
id, ok := node.AttributeString("id")
if !ok {
generateAutoHeadingID(node.(*ast.Heading), reader, pc)
} else {
pc.IDs().Put(id.([]byte))
}
}
}

View file

@ -108,9 +108,11 @@ func (b *setextHeadingParser) Close(node ast.Node, reader text.Reader, pc Contex
}
if b.AutoHeadingID {
_, ok := node.AttributeString("id")
id, ok := node.AttributeString("id")
if !ok {
generateAutoHeadingID(heading, reader, pc)
} else {
pc.IDs().Put(id.([]byte))
}
}
}