-
Notifications
You must be signed in to change notification settings - Fork 782
/
root_expression_parser.go
109 lines (101 loc) · 2.71 KB
/
root_expression_parser.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
package parser
import (
"github.com/benthosdev/benthos/v4/internal/bloblang/mapping"
"github.com/benthosdev/benthos/v4/internal/bloblang/query"
)
func rootLevelIfExpressionParser(pCtx Context) Func[mapping.Statement] {
return func(input []rune) Result[mapping.Statement] {
ifParser := Sequence(
FuncAsAny(Expect(Term("if"), "assignment")),
FuncAsAny(SpacesAndTabs),
FuncAsAny(MustBe(queryParser(pCtx))),
FuncAsAny(DiscardedWhitespaceNewlineComments),
FuncAsAny(DelimitedPattern(
Sequence(
charSquigOpen,
DiscardedWhitespaceNewlineComments,
),
mappingStatement(pCtx, true, nil),
Sequence(
Discard(SpacesAndTabs),
NewlineAllowComment,
DiscardedWhitespaceNewlineComments,
),
Sequence(
DiscardedWhitespaceNewlineComments,
charSquigClose,
),
)),
)
elseIfParser := Optional(Sequence(
FuncAsAny(DiscardedWhitespaceNewlineComments),
FuncAsAny(Term("else if")),
FuncAsAny(SpacesAndTabs),
FuncAsAny(MustBe(queryParser(pCtx))),
FuncAsAny(DiscardedWhitespaceNewlineComments),
FuncAsAny(DelimitedPattern(
Sequence(
charSquigOpen,
DiscardedWhitespaceNewlineComments,
),
mappingStatement(pCtx, true, nil),
Sequence(
Discard(SpacesAndTabs),
NewlineAllowComment,
DiscardedWhitespaceNewlineComments,
),
Sequence(
DiscardedWhitespaceNewlineComments,
charSquigClose,
),
)),
))
elseParser := Optional(Sequence(
FuncAsAny(DiscardedWhitespaceNewlineComments),
FuncAsAny(Term("else")),
FuncAsAny(DiscardedWhitespaceNewlineComments),
FuncAsAny(DelimitedPattern(
Sequence(
charSquigOpen,
DiscardedWhitespaceNewlineComments,
),
mappingStatement(pCtx, true, nil),
Sequence(
Discard(SpacesAndTabs),
NewlineAllowComment,
DiscardedWhitespaceNewlineComments,
),
Sequence(
DiscardedWhitespaceNewlineComments,
charSquigClose,
),
)),
))
res := ifParser(input)
if res.Err != nil {
return Fail[mapping.Statement](res.Err, input)
}
seqSlice := res.Payload
stmt := mapping.NewRootLevelIfStatement(input)
stmt.Add(seqSlice[2].(query.Function), seqSlice[4].([]mapping.Statement)...)
for {
res = elseIfParser(res.Remaining)
if res.Err != nil {
return Fail[mapping.Statement](res.Err, input)
}
if res.Payload == nil {
break
}
seqSlice = res.Payload
stmt.Add(seqSlice[3].(query.Function), seqSlice[5].([]mapping.Statement)...)
}
res = elseParser(res.Remaining)
if res.Err != nil {
return Fail[mapping.Statement](res.Err, input)
}
if seqSlice = res.Payload; seqSlice != nil {
stmt.Add(nil, seqSlice[3].([]mapping.Statement)...)
}
return Success[mapping.Statement](stmt, res.Remaining)
}
}