32 lines
778 B
Go
32 lines
778 B
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"regexp"
|
|
)
|
|
|
|
var sectionHeadingRe = regexp.MustCompile(`(?m)^#{1,6} `)
|
|
|
|
// splitSections splits raw markdown into sections.
|
|
// Section 0 is any content before the first heading.
|
|
// Each subsequent section begins at a heading line and runs to the next.
|
|
func splitSections(raw []byte) [][]byte {
|
|
locs := sectionHeadingRe.FindAllIndex(raw, -1)
|
|
if len(locs) == 0 {
|
|
return [][]byte{raw}
|
|
}
|
|
sections := make([][]byte, 0, len(locs)+1)
|
|
prev := 0
|
|
for _, loc := range locs {
|
|
sections = append(sections, raw[prev:loc[0]])
|
|
prev = loc[0]
|
|
}
|
|
sections = append(sections, raw[prev:])
|
|
return sections
|
|
}
|
|
|
|
// joinSections reassembles sections produced by splitSections.
|
|
func joinSections(sections [][]byte) []byte {
|
|
return bytes.Join(sections, nil)
|
|
}
|