63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
package org
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
type Table struct {
|
|
Header Node
|
|
Rows []Node
|
|
}
|
|
|
|
type TableSeparator struct{ Content string }
|
|
|
|
type TableHeader struct {
|
|
Columns [][]Node
|
|
Separator TableSeparator
|
|
}
|
|
|
|
type TableRow struct{ Columns [][]Node }
|
|
|
|
var tableSeparatorRegexp = regexp.MustCompile(`^(\s*)(\|[+-|]*)\s*$`)
|
|
var tableRowRegexp = regexp.MustCompile(`^(\s*)(\|.*)`)
|
|
|
|
func lexTable(line string) (token, bool) {
|
|
if m := tableSeparatorRegexp.FindStringSubmatch(line); m != nil {
|
|
return token{"tableSeparator", len(m[1]), m[2], m}, true
|
|
} else if m := tableRowRegexp.FindStringSubmatch(line); m != nil {
|
|
return token{"tableRow", len(m[1]), m[2], m}, true
|
|
}
|
|
return nilToken, false
|
|
}
|
|
|
|
func (d *Document) parseTable(i int, parentStop stopFn) (int, Node) {
|
|
rows, start := []Node{}, i
|
|
for !parentStop(d, i) && (d.tokens[i].kind == "tableRow" || d.tokens[i].kind == "tableSeparator") {
|
|
consumed, row := d.parseTableRowOrSeparator(i, parentStop)
|
|
i += consumed
|
|
rows = append(rows, row)
|
|
}
|
|
|
|
consumed := i - start
|
|
if len(rows) >= 2 {
|
|
if row, ok := rows[0].(TableRow); ok {
|
|
if separator, ok := rows[1].(TableSeparator); ok {
|
|
return consumed, Table{TableHeader{row.Columns, separator}, rows[2:]}
|
|
}
|
|
}
|
|
}
|
|
return consumed, Table{nil, rows}
|
|
}
|
|
|
|
func (d *Document) parseTableRowOrSeparator(i int, _ stopFn) (int, Node) {
|
|
if d.tokens[i].kind == "tableSeparator" {
|
|
return 1, TableSeparator{d.tokens[i].content}
|
|
}
|
|
fields := strings.FieldsFunc(d.tokens[i].content, func(r rune) bool { return r == '|' })
|
|
row := TableRow{}
|
|
for _, field := range fields {
|
|
row.Columns = append(row.Columns, d.parseInline(strings.TrimSpace(field)))
|
|
}
|
|
return 1, row
|
|
}
|