-
Notifications
You must be signed in to change notification settings - Fork 16
/
combinator.go
179 lines (157 loc) · 4.08 KB
/
combinator.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
package goparsify
import (
"bytes"
)
// Seq matches all of the given parsers in order and returns their result as .Child[n]
func Seq(parsers ...Parserish) Parser {
parserfied := ParsifyAll(parsers...)
return NewParser("Seq()", func(ps *State, node *Result) {
node.Child = make([]Result, len(parserfied))
startpos := ps.Pos
for i, parser := range parserfied {
parser(ps, &node.Child[i])
if ps.Errored() {
ps.Pos = startpos
return
}
}
})
}
// NoAutoWS disables automatically ignoring whitespace between tokens for all parsers underneath
func NoAutoWS(parser Parserish) Parser {
parserfied := Parsify(parser)
return func(ps *State, node *Result) {
oldWS := ps.WS
ps.WS = NoWhitespace
parserfied(ps, node)
ps.WS = oldWS
}
}
// Any matches the first successful parser and returns its result
func Any(parsers ...Parserish) Parser {
parserfied := ParsifyAll(parsers...)
// Records which parser was successful for each byte, and will use it first next time.
return NewParser("Any()", func(ps *State, node *Result) {
ps.WS(ps)
if ps.Pos >= len(ps.Input) {
ps.ErrorHere("!EOF")
return
}
startpos := ps.Pos
longestError := ps.Error
if ps.Cut <= startpos {
ps.Recover()
} else {
return
}
for _, parser := range parserfied {
parser(ps, node)
if ps.Errored() {
if ps.Error.pos >= longestError.pos {
longestError = ps.Error
}
if ps.Cut > startpos {
break
}
ps.Recover()
continue
}
return
}
ps.Error = longestError
ps.Pos = startpos
})
}
// Some matches one or more parsers and returns the value as .Child[n]
// an optional separator can be provided and that value will be consumed
// but not returned. Only one separator can be provided.
func Some(parser Parserish, separator ...Parserish) Parser {
return NewParser("Some()", manyImpl(0, parser, separator...))
}
// Many matches zero or more parsers and returns the value as .Child[n]
// an optional separator can be provided and that value will be consumed
// but not returned. Only one separator can be provided.
func Many(parser Parserish, separator ...Parserish) Parser {
return NewParser("Many()", manyImpl(1, parser, separator...))
}
func manyImpl(min int, op Parserish, sep ...Parserish) Parser {
var opParser = Parsify(op)
var sepParser Parser
if len(sep) > 0 {
sepParser = Parsify(sep[0])
}
return func(ps *State, node *Result) {
node.Child = make([]Result, 0, 5)
startpos := ps.Pos
for {
node.Child = append(node.Child, Result{})
opParser(ps, &node.Child[len(node.Child)-1])
if ps.Errored() {
if len(node.Child)-1 < min || ps.Cut > ps.Pos {
ps.Pos = startpos
return
}
ps.Recover()
node.Child = node.Child[0 : len(node.Child)-1]
return
}
if sepParser != nil {
sepParser(ps, TrashResult)
if ps.Errored() {
ps.Recover()
return
}
}
}
}
}
// Maybe will 0 or 1 of the parser
func Maybe(parser Parserish) Parser {
parserfied := Parsify(parser)
return NewParser("Maybe()", func(ps *State, node *Result) {
startpos := ps.Pos
parserfied(ps, node)
if ps.Errored() && ps.Cut <= startpos {
ps.Recover()
}
})
}
// Bind will set the node .Result when the given parser matches
// This is useful for giving a value to keywords and constant literals
// like true and false. See the json parser for an example.
func Bind(parser Parserish, val interface{}) Parser {
p := Parsify(parser)
return func(ps *State, node *Result) {
p(ps, node)
if ps.Errored() {
return
}
node.Result = val
}
}
// Map applies the callback if the parser matches. This is used to set the Result
// based on the matched result.
func Map(parser Parserish, f func(n *Result)) Parser {
p := Parsify(parser)
return func(ps *State, node *Result) {
p(ps, node)
if ps.Errored() {
return
}
f(node)
}
}
func flatten(n *Result) {
if len(n.Child) > 0 {
sbuf := &bytes.Buffer{}
for _, child := range n.Child {
flatten(&child)
sbuf.WriteString(child.Token)
}
n.Token = sbuf.String()
}
}
// Merge all child Tokens together recursively
func Merge(parser Parserish) Parser {
return Map(parser, flatten)
}