diff options
Diffstat (limited to 'src/parser.rs')
-rw-r--r-- | src/parser.rs | 67 |
1 files changed, 30 insertions, 37 deletions
diff --git a/src/parser.rs b/src/parser.rs index 8b04307..0719f93 100644 --- a/src/parser.rs +++ b/src/parser.rs | |||
@@ -13,20 +13,21 @@ use nom::{ | |||
13 | use crate::ast::*; | 13 | use crate::ast::*; |
14 | use crate::string::parse_string; | 14 | use crate::string::parse_string; |
15 | 15 | ||
16 | fn ws<'a, F: 'a, O, E>(inner: F) -> impl FnMut(&'a str) -> IResult<&'a str, O, E> | 16 | fn ws<'a, F, O, E>(inner: F) -> impl FnMut(&'a str) -> IResult<&'a str, O, E> |
17 | where | 17 | where |
18 | F: FnMut(&'a str) -> IResult<&'a str, O, E>, | 18 | F: FnMut(&'a str) -> IResult<&'a str, O, E>, |
19 | F: 'a, | ||
19 | E: ParseError<&'a str>, | 20 | E: ParseError<&'a str>, |
20 | { | 21 | { |
21 | delimited(multispace0, inner, multispace0) | 22 | delimited(multispace0, inner, multispace0) |
22 | } | 23 | } |
23 | 24 | ||
24 | // TODO use this | 25 | // TODO use this |
25 | fn _parse_comment<'a>(i: &'a str) -> IResult<&'a str, ()> { | 26 | fn _parse_comment(i: &str) -> IResult<&str, ()> { |
26 | value((), pair(tag("//"), is_not("\n\r")))(i) | 27 | value((), pair(tag("//"), is_not("\n\r")))(i) |
27 | } | 28 | } |
28 | 29 | ||
29 | fn parse_unit<'a>(i: &'a str) -> IResult<&'a str, ()> { | 30 | fn parse_unit(i: &str) -> IResult<&str, ()> { |
30 | let open = char('('); | 31 | let open = char('('); |
31 | let close = char(')'); | 32 | let close = char(')'); |
32 | let unit = tuple((open, close)); | 33 | let unit = tuple((open, close)); |
@@ -39,7 +40,7 @@ fn parse_bool(i: &str) -> IResult<&str, bool> { | |||
39 | alt((t, f)).parse(i) | 40 | alt((t, f)).parse(i) |
40 | } | 41 | } |
41 | 42 | ||
42 | fn parse_int<'a>(i: &'a str) -> IResult<&'a str, i128> { | 43 | fn parse_int(i: &str) -> IResult<&str, i128> { |
43 | map(recognize(many1(one_of("0123456789"))), |s: &str| { | 44 | map(recognize(many1(one_of("0123456789"))), |s: &str| { |
44 | s.parse::<i128>().unwrap() | 45 | s.parse::<i128>().unwrap() |
45 | })(i) | 46 | })(i) |
@@ -57,7 +58,7 @@ fn parse_ident(i: &str) -> IResult<&str, Identifier> { | |||
57 | map(parse_name, str::to_owned)(i) | 58 | map(parse_name, str::to_owned)(i) |
58 | } | 59 | } |
59 | 60 | ||
60 | fn parse_lit<'a>(i: &'a str) -> IResult<&'a str, Literal> { | 61 | fn parse_lit(i: &str) -> IResult<&str, Literal> { |
61 | alt(( | 62 | alt(( |
62 | map(parse_string, Literal::Str), | 63 | map(parse_string, Literal::Str), |
63 | map(parse_int, Literal::Int), | 64 | map(parse_int, Literal::Int), |
@@ -118,28 +119,28 @@ where | |||
118 | }) | 119 | }) |
119 | } | 120 | } |
120 | 121 | ||
121 | fn parse_assign<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 122 | fn parse_assign(i: &str) -> IResult<&str, Expr> { |
122 | let op = map(parse_assign_op, BinOp::Assign); | 123 | let op = map(parse_assign_op, BinOp::Assign); |
123 | let recursive = parse_binary(parse_atom, op, parse_assign); | 124 | let recursive = parse_binary(parse_atom, op, parse_assign); |
124 | let base = parse_union; | 125 | let base = parse_union; |
125 | alt((recursive, base)).parse(i) | 126 | alt((recursive, base)).parse(i) |
126 | } | 127 | } |
127 | 128 | ||
128 | fn parse_union<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 129 | fn parse_union(i: &str) -> IResult<&str, Expr> { |
129 | let op = parse_op("||", BinOp::Logic(LogicOp::Or)); | 130 | let op = parse_op("||", BinOp::Logic(LogicOp::Or)); |
130 | let recursive = parse_binary(parse_intersection, op, parse_union); | 131 | let recursive = parse_binary(parse_intersection, op, parse_union); |
131 | let base = parse_intersection; | 132 | let base = parse_intersection; |
132 | alt((recursive, base)).parse(i) | 133 | alt((recursive, base)).parse(i) |
133 | } | 134 | } |
134 | 135 | ||
135 | fn parse_intersection<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 136 | fn parse_intersection(i: &str) -> IResult<&str, Expr> { |
136 | let op = parse_op("&&", BinOp::Logic(LogicOp::And)); | 137 | let op = parse_op("&&", BinOp::Logic(LogicOp::And)); |
137 | let recursive = parse_binary(parse_negated, op, parse_intersection); | 138 | let recursive = parse_binary(parse_negated, op, parse_intersection); |
138 | let base = parse_negated; | 139 | let base = parse_negated; |
139 | alt((recursive, base)).parse(i) | 140 | alt((recursive, base)).parse(i) |
140 | } | 141 | } |
141 | 142 | ||
142 | fn parse_negated<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 143 | fn parse_negated(i: &str) -> IResult<&str, Expr> { |
143 | let op = parse_op("!", UnaryOp::Not); | 144 | let op = parse_op("!", UnaryOp::Not); |
144 | let recursive = map(tuple((op, parse_rel)), |(op, expr)| { | 145 | let recursive = map(tuple((op, parse_rel)), |(op, expr)| { |
145 | Expr::Unary(expr.boxed(), op) | 146 | Expr::Unary(expr.boxed(), op) |
@@ -148,14 +149,14 @@ fn parse_negated<'a>(i: &'a str) -> IResult<&'a str, Expr> { | |||
148 | alt((recursive, base)).parse(i) | 149 | alt((recursive, base)).parse(i) |
149 | } | 150 | } |
150 | 151 | ||
151 | fn parse_rel<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 152 | fn parse_rel(i: &str) -> IResult<&str, Expr> { |
152 | let op = map(parse_cmp_op, BinOp::Cmp); | 153 | let op = map(parse_cmp_op, BinOp::Cmp); |
153 | let recursive = parse_binary(parse_sum, op, parse_rel); | 154 | let recursive = parse_binary(parse_sum, op, parse_rel); |
154 | let base = parse_sum; | 155 | let base = parse_sum; |
155 | alt((recursive, base)).parse(i) | 156 | alt((recursive, base)).parse(i) |
156 | } | 157 | } |
157 | 158 | ||
158 | fn parse_sum<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 159 | fn parse_sum(i: &str) -> IResult<&str, Expr> { |
159 | let add = parse_op("+", BinOp::Arith(ArithOp::Add)); | 160 | let add = parse_op("+", BinOp::Arith(ArithOp::Add)); |
160 | let sub = parse_op("-", BinOp::Arith(ArithOp::Sub)); | 161 | let sub = parse_op("-", BinOp::Arith(ArithOp::Sub)); |
161 | let op = alt((add, sub)); | 162 | let op = alt((add, sub)); |
@@ -164,7 +165,7 @@ fn parse_sum<'a>(i: &'a str) -> IResult<&'a str, Expr> { | |||
164 | alt((recursive, base)).parse(i) | 165 | alt((recursive, base)).parse(i) |
165 | } | 166 | } |
166 | 167 | ||
167 | fn parse_mul<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 168 | fn parse_mul(i: &str) -> IResult<&str, Expr> { |
168 | let mul = parse_op("*", BinOp::Arith(ArithOp::Mul)); | 169 | let mul = parse_op("*", BinOp::Arith(ArithOp::Mul)); |
169 | let div = parse_op("/", BinOp::Arith(ArithOp::Div)); | 170 | let div = parse_op("/", BinOp::Arith(ArithOp::Div)); |
170 | let mod_ = parse_op("%", BinOp::Arith(ArithOp::Mod)); | 171 | let mod_ = parse_op("%", BinOp::Arith(ArithOp::Mod)); |
@@ -174,7 +175,7 @@ fn parse_mul<'a>(i: &'a str) -> IResult<&'a str, Expr> { | |||
174 | alt((recursive, base)).parse(i) | 175 | alt((recursive, base)).parse(i) |
175 | } | 176 | } |
176 | 177 | ||
177 | fn parse_field_or_index<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 178 | fn parse_field_or_index(i: &str) -> IResult<&str, Expr> { |
178 | enum FieldOrIndex { | 179 | enum FieldOrIndex { |
179 | Field(String), | 180 | Field(String), |
180 | Index(Expr), | 181 | Index(Expr), |
@@ -200,14 +201,14 @@ fn parse_field_or_index<'a>(i: &'a str) -> IResult<&'a str, Expr> { | |||
200 | )(i) | 201 | )(i) |
201 | } | 202 | } |
202 | 203 | ||
203 | fn parse_list<'a>(i: &'a str) -> IResult<&'a str, List> { | 204 | fn parse_list(i: &str) -> IResult<&str, List> { |
204 | let open = ws(char('[')); | 205 | let open = ws(char('[')); |
205 | let items = separated_list0(char(','), parse_expr); | 206 | let items = separated_list0(char(','), parse_expr); |
206 | let close = ws(char(']')); | 207 | let close = ws(char(']')); |
207 | map(tuple((open, items, close)), |(_, items, _)| List { items }).parse(i) | 208 | map(tuple((open, items, close)), |(_, items, _)| List { items }).parse(i) |
208 | } | 209 | } |
209 | 210 | ||
210 | fn parse_atom<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 211 | fn parse_atom(i: &str) -> IResult<&str, Expr> { |
211 | let inner = alt(( | 212 | let inner = alt(( |
212 | map(tag("node"), |_| Expr::Node), | 213 | map(tag("node"), |_| Expr::Node), |
213 | map(parse_block, Expr::Block), | 214 | map(parse_block, Expr::Block), |
@@ -221,7 +222,7 @@ fn parse_atom<'a>(i: &'a str) -> IResult<&'a str, Expr> { | |||
221 | ws(inner).parse(i) | 222 | ws(inner).parse(i) |
222 | } | 223 | } |
223 | 224 | ||
224 | fn parse_call<'a>(i: &'a str) -> IResult<&'a str, Call> { | 225 | fn parse_call(i: &str) -> IResult<&str, Call> { |
225 | let ident = parse_ident; | 226 | let ident = parse_ident; |
226 | let open = ws(char('(')); | 227 | let open = ws(char('(')); |
227 | let args = separated_list0(char(','), parse_expr); | 228 | let args = separated_list0(char(','), parse_expr); |
@@ -236,14 +237,14 @@ fn parse_call<'a>(i: &'a str) -> IResult<&'a str, Call> { | |||
236 | .parse(i) | 237 | .parse(i) |
237 | } | 238 | } |
238 | 239 | ||
239 | fn parse_block<'a>(i: &'a str) -> IResult<&'a str, Block> { | 240 | fn parse_block(i: &str) -> IResult<&str, Block> { |
240 | let open = ws(char('{')); | 241 | let open = ws(char('{')); |
241 | let statements = map(many0(parse_statement), |body| Block { body }); | 242 | let statements = map(many0(parse_statement), |body| Block { body }); |
242 | let close = ws(char('}')); | 243 | let close = ws(char('}')); |
243 | delimited(open, statements, close).parse(i) | 244 | delimited(open, statements, close).parse(i) |
244 | } | 245 | } |
245 | 246 | ||
246 | fn parse_if<'a>(i: &'a str) -> IResult<&'a str, IfExpr> { | 247 | fn parse_if(i: &str) -> IResult<&str, IfExpr> { |
247 | let if_ = delimited(multispace0, tag("if"), multispace1); | 248 | let if_ = delimited(multispace0, tag("if"), multispace1); |
248 | 249 | ||
249 | let open = char('('); | 250 | let open = char('('); |
@@ -265,15 +266,15 @@ fn parse_if<'a>(i: &'a str) -> IResult<&'a str, IfExpr> { | |||
265 | )(i) | 266 | )(i) |
266 | } | 267 | } |
267 | 268 | ||
268 | fn parse_expr<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 269 | fn parse_expr(i: &str) -> IResult<&str, Expr> { |
269 | parse_assign.parse(i) | 270 | parse_assign.parse(i) |
270 | } | 271 | } |
271 | 272 | ||
272 | fn parse_bare<'a>(i: &'a str) -> IResult<&'a str, Expr> { | 273 | fn parse_bare(i: &str) -> IResult<&str, Expr> { |
273 | parse_expr(i) | 274 | parse_expr(i) |
274 | } | 275 | } |
275 | 276 | ||
276 | fn parse_type<'a>(i: &'a str) -> IResult<&'a str, Type> { | 277 | fn parse_type(i: &str) -> IResult<&str, Type> { |
277 | let int = value(Type::Integer, tag("int")); | 278 | let int = value(Type::Integer, tag("int")); |
278 | let string = value(Type::String, tag("string")); | 279 | let string = value(Type::String, tag("string")); |
279 | let bool_ = value(Type::Boolean, tag("bool")); | 280 | let bool_ = value(Type::Boolean, tag("bool")); |
@@ -281,7 +282,7 @@ fn parse_type<'a>(i: &'a str) -> IResult<&'a str, Type> { | |||
281 | alt((int, string, bool_, list)).parse(i) | 282 | alt((int, string, bool_, list)).parse(i) |
282 | } | 283 | } |
283 | 284 | ||
284 | fn parse_declaration<'a>(i: &'a str) -> IResult<&'a str, Declaration> { | 285 | fn parse_declaration(i: &str) -> IResult<&str, Declaration> { |
285 | let ty = parse_type; | 286 | let ty = parse_type; |
286 | let name = parse_ident; | 287 | let name = parse_ident; |
287 | let op = ws(char('=')); | 288 | let op = ws(char('=')); |
@@ -292,7 +293,7 @@ fn parse_declaration<'a>(i: &'a str) -> IResult<&'a str, Declaration> { | |||
292 | )(i) | 293 | )(i) |
293 | } | 294 | } |
294 | 295 | ||
295 | fn parse_statement<'a>(i: &'a str) -> IResult<&'a str, Statement> { | 296 | fn parse_statement(i: &str) -> IResult<&str, Statement> { |
296 | let semicolon = ws(char(';')); | 297 | let semicolon = ws(char(';')); |
297 | let inner = alt(( | 298 | let inner = alt(( |
298 | map(parse_declaration, Statement::Declaration), | 299 | map(parse_declaration, Statement::Declaration), |
@@ -364,38 +365,30 @@ fn parse_statement<'a>(i: &'a str) -> IResult<&'a str, Statement> { | |||
364 | // }; | 365 | // }; |
365 | // } | 366 | // } |
366 | 367 | ||
367 | fn parse_modifier<'a>(i: &str) -> IResult<&str, Modifier> { | 368 | fn parse_modifier(i: &str) -> IResult<&str, Modifier> { |
368 | let pre = value(Modifier::Enter, tag("enter")); | 369 | let pre = value(Modifier::Enter, tag("enter")); |
369 | let post = value(Modifier::Leave, tag("leave")); | 370 | let post = value(Modifier::Leave, tag("leave")); |
370 | map(opt(alt((pre, post))), Option::unwrap_or_default)(i) | 371 | map(opt(alt((pre, post))), Option::unwrap_or_default)(i) |
371 | } | 372 | } |
372 | 373 | ||
373 | fn parse_pattern<'a>(i: &str) -> IResult<&str, Pattern> { | 374 | fn parse_pattern(i: &str) -> IResult<&str, Pattern> { |
374 | let begin = value(Pattern::Begin, ws(tag("BEGIN"))); | 375 | let begin = value(Pattern::Begin, ws(tag("BEGIN"))); |
375 | let end = value(Pattern::End, ws(tag("END"))); | 376 | let end = value(Pattern::End, ws(tag("END"))); |
376 | ws(alt((begin, end, parse_tree_pattern))).parse(i) | 377 | ws(alt((begin, end, parse_tree_pattern))).parse(i) |
377 | } | 378 | } |
378 | 379 | ||
379 | // fn parse_node_pattern<'a>(i: &str) -> IResult<&str, Pattern> { | 380 | fn parse_tree_pattern(i: &str) -> IResult<&str, Pattern> { |
380 | // map( | ||
381 | // tuple((parse_modifier, multispace0, parse_ident)), | ||
382 | // |(modifier, _, kind)| Pattern::Node(NodePattern { modifier, kind }), | ||
383 | // ) | ||
384 | // .parse(i) | ||
385 | // } | ||
386 | |||
387 | fn parse_tree_pattern<'a>(i: &str) -> IResult<&str, Pattern> { | ||
388 | let parse_matcher = alt((parse_tree_atom, parse_tree_list)); | 381 | let parse_matcher = alt((parse_tree_atom, parse_tree_list)); |
389 | tuple((parse_modifier, multispace0, parse_matcher)) | 382 | tuple((parse_modifier, multispace0, parse_matcher)) |
390 | .map(|(modifier, _, matcher)| Pattern::Tree { modifier, matcher }) | 383 | .map(|(modifier, _, matcher)| Pattern::Tree { modifier, matcher }) |
391 | .parse(i) | 384 | .parse(i) |
392 | } | 385 | } |
393 | 386 | ||
394 | fn parse_tree_atom<'a>(i: &str) -> IResult<&str, TreePattern> { | 387 | fn parse_tree_atom(i: &str) -> IResult<&str, TreePattern> { |
395 | parse_ident.map(TreePattern::Atom).parse(i) | 388 | parse_ident.map(TreePattern::Atom).parse(i) |
396 | } | 389 | } |
397 | 390 | ||
398 | fn parse_tree_list<'a>(i: &str) -> IResult<&str, TreePattern> { | 391 | fn parse_tree_list(i: &str) -> IResult<&str, TreePattern> { |
399 | let open = terminated(char('('), multispace0); | 392 | let open = terminated(char('('), multispace0); |
400 | let close = preceded(multispace0, char(')')); | 393 | let close = preceded(multispace0, char(')')); |
401 | let list = separated_list1(multispace1, alt((parse_tree_atom, parse_tree_list))); | 394 | let list = separated_list1(multispace1, alt((parse_tree_atom, parse_tree_list))); |
@@ -404,7 +397,7 @@ fn parse_tree_list<'a>(i: &str) -> IResult<&str, TreePattern> { | |||
404 | .parse(i) | 397 | .parse(i) |
405 | } | 398 | } |
406 | 399 | ||
407 | pub fn parse_stanza<'a>(i: &str) -> IResult<&str, Stanza> { | 400 | pub fn parse_stanza(i: &str) -> IResult<&str, Stanza> { |
408 | map( | 401 | map( |
409 | tuple((parse_pattern, parse_block)), | 402 | tuple((parse_pattern, parse_block)), |
410 | |(pattern, statements)| Stanza { | 403 | |(pattern, statements)| Stanza { |