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
|
extern crate peg;
use arithmetic::expression;
peg::parser!( grammar arithmetic() for str {
pub rule expression() -> i64
= sum()
rule sum() -> i64
= l:product() "+" r:product() { l+r }
/ product()
rule product() -> i64
= l:atom() "*" r:atom() { l*r }
/ atom()
rule atom() -> i64
= number()
/ "(" v:sum() ")" { v }
rule number() -> i64
= n:$(['0'..='9']+) { n.parse().unwrap() }
});
fn main() {
assert_eq!(expression("1+1"), Ok(2));
assert_eq!(expression("5*5"), Ok(25));
assert_eq!(expression("222+3333"), Ok(3555));
assert_eq!(expression("2+3*4"), Ok(14));
assert_eq!(expression("(2+2)*3"), Ok(12));
assert!(expression("(22+)+1").is_err());
assert!(expression("1++1").is_err());
assert!(expression("3)+1").is_err());
}
|