123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114 |
- package org
-
- import (
- "fmt"
- "regexp"
- "strings"
- "unicode"
- )
-
- type List struct {
- Kind string
- Items []Node
- }
-
- type ListItem struct {
- Bullet string
- Status string
- Children []Node
- }
-
- type DescriptiveListItem struct {
- Bullet string
- Status string
- Term []Node
- Details []Node
- }
-
- var unorderedListRegexp = regexp.MustCompile(`^(\s*)([+*-])(\s+(.*)|$)`)
- var orderedListRegexp = regexp.MustCompile(`^(\s*)(([0-9]+|[a-zA-Z])[.)])(\s+(.*)|$)`)
- var descriptiveListItemRegexp = regexp.MustCompile(`\s::(\s|$)`)
- var listItemStatusRegexp = regexp.MustCompile(`\[( |X|-)\]\s`)
-
- func lexList(line string) (token, bool) {
- if m := unorderedListRegexp.FindStringSubmatch(line); m != nil {
- return token{"unorderedList", len(m[1]), m[4], m}, true
- } else if m := orderedListRegexp.FindStringSubmatch(line); m != nil {
- return token{"orderedList", len(m[1]), m[5], m}, true
- }
- return nilToken, false
- }
-
- func isListToken(t token) bool {
- return t.kind == "unorderedList" || t.kind == "orderedList"
- }
-
- func listKind(t token) (string, string) {
- kind := ""
- switch bullet := t.matches[2]; {
- case bullet == "*" || bullet == "+" || bullet == "-":
- kind = "unordered"
- case unicode.IsLetter(rune(bullet[0])), unicode.IsDigit(rune(bullet[0])):
- kind = "ordered"
- default:
- panic(fmt.Sprintf("bad list bullet '%s': %#v", bullet, t))
- }
- if descriptiveListItemRegexp.MatchString(t.content) {
- return kind, "descriptive"
- }
- return kind, kind
- }
-
- func (d *Document) parseList(i int, parentStop stopFn) (int, Node) {
- start, lvl := i, d.tokens[i].lvl
- listMainKind, kind := listKind(d.tokens[i])
- list := List{Kind: kind}
- stop := func(*Document, int) bool {
- if parentStop(d, i) || d.tokens[i].lvl != lvl || !isListToken(d.tokens[i]) {
- return true
- }
- itemMainKind, _ := listKind(d.tokens[i])
- return itemMainKind != listMainKind
- }
- for !stop(d, i) {
- consumed, node := d.parseListItem(list, i, parentStop)
- i += consumed
- list.Items = append(list.Items, node)
- }
- return i - start, list
- }
-
- func (d *Document) parseListItem(l List, i int, parentStop stopFn) (int, Node) {
- start, nodes, bullet := i, []Node{}, d.tokens[i].matches[2]
- minIndent, dterm, content, status := d.tokens[i].lvl+len(bullet), "", d.tokens[i].content, ""
- if m := listItemStatusRegexp.FindStringSubmatch(content); m != nil {
- status, content = m[1], content[len("[ ] "):]
- }
- if l.Kind == "descriptive" {
- if m := descriptiveListItemRegexp.FindStringIndex(content); m != nil {
- dterm, content = content[:m[0]], content[m[1]:]
- }
- }
-
- d.tokens[i] = tokenize(strings.Repeat(" ", minIndent) + content)
- stop := func(d *Document, i int) bool {
- if parentStop(d, i) {
- return true
- }
- t := d.tokens[i]
- return t.lvl < minIndent && !(t.kind == "text" && t.content == "")
- }
- for !stop(d, i) && (i <= start+1 || !isSecondBlankLine(d, i)) {
- consumed, node := d.parseOne(i, stop)
- i += consumed
- nodes = append(nodes, node)
- }
- if l.Kind == "descriptive" {
- return i - start, DescriptiveListItem{bullet, status, d.parseInline(dterm), nodes}
- }
- return i - start, ListItem{bullet, status, nodes}
- }
-
- func (n List) String() string { return orgWriter.WriteNodesAsString(n) }
- func (n ListItem) String() string { return orgWriter.WriteNodesAsString(n) }
- func (n DescriptiveListItem) String() string { return orgWriter.WriteNodesAsString(n) }
|