76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
|
package parser
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/yuin/goldmark/ast"
|
||
|
|
"github.com/yuin/goldmark/text"
|
||
|
|
"github.com/yuin/goldmark/util"
|
||
|
|
)
|
||
|
|
|
||
|
|
type thematicBreakPraser struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
var defaultThematicBreakPraser = &thematicBreakPraser{}
|
||
|
|
|
||
|
|
// NewThematicBreakParser returns a new BlockParser that
|
||
|
|
// parses thematic breaks.
|
||
|
|
func NewThematicBreakParser() BlockParser {
|
||
|
|
return defaultThematicBreakPraser
|
||
|
|
}
|
||
|
|
|
||
|
|
func isThematicBreak(line []byte, offset int) bool {
|
||
|
|
w, pos := util.IndentWidth(line, offset)
|
||
|
|
if w > 3 {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
mark := byte(0)
|
||
|
|
count := 0
|
||
|
|
for i := pos; i < len(line); i++ {
|
||
|
|
c := line[i]
|
||
|
|
if util.IsSpace(c) {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if mark == 0 {
|
||
|
|
mark = c
|
||
|
|
count = 1
|
||
|
|
if mark == '*' || mark == '-' || mark == '_' {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
if c != mark {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
count++
|
||
|
|
}
|
||
|
|
return count > 2
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *thematicBreakPraser) Trigger() []byte {
|
||
|
|
return []byte{'-', '*', '_'}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *thematicBreakPraser) Open(parent ast.Node, reader text.Reader, pc Context) (ast.Node, State) {
|
||
|
|
line, segment := reader.PeekLine()
|
||
|
|
if isThematicBreak(line, reader.LineOffset()) {
|
||
|
|
reader.Advance(segment.Len() - 1)
|
||
|
|
return ast.NewThematicBreak(), NoChildren
|
||
|
|
}
|
||
|
|
return nil, NoChildren
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *thematicBreakPraser) Continue(node ast.Node, reader text.Reader, pc Context) State {
|
||
|
|
return Close
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *thematicBreakPraser) Close(node ast.Node, reader text.Reader, pc Context) {
|
||
|
|
// nothing to do
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *thematicBreakPraser) CanInterruptParagraph() bool {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *thematicBreakPraser) CanAcceptIndentedLine() bool {
|
||
|
|
return false
|
||
|
|
}
|