Fix struct literals in if expressions

With special case-ing, sigh :( Also will need to do this for match
expressions but I'll cross that bridge when I come to it
This commit is contained in:
greg 2018-02-11 22:10:21 -08:00
parent a452bccd1c
commit 1f2a4c706f

View File

@ -284,12 +284,10 @@ type_names := ε | type_name (, type_name)*
type_params := '<' type_name (, type_name)* '>' type_params := '<' type_name (, type_name)* '>'
expression := precedence_expr type_anno+ expression := precedence_expr type_anno+
precedence_expr := prefix_expr precedence_expr := prefix_expr
prefix_expr := prefix_op primary prefix_expr := prefix_op primary
prefix_op := '+' | '-' | '!' | '~' prefix_op := '+' | '-' | '!' | '~'
primary := literal | paren_expr | if_expr | match_expr | for_expr | identifier_expr primary := literal | paren_expr | if_expr | match_expr | for_expr | identifier_expr
paren_expr := LParen paren_inner RParen paren_expr := LParen paren_inner RParen
paren_inner := (expression ',')* paren_inner := (expression ',')*
identifier_expr := named_struct | call_expr | index_expr | IDENTIFIER identifier_expr := named_struct | call_expr | index_expr | IDENTIFIER
@ -348,12 +346,22 @@ pub struct ParseRecord {
struct Parser { struct Parser {
tokens: TokenIter, tokens: TokenIter,
parse_record: Vec<ParseRecord>, parse_record: Vec<ParseRecord>,
parse_level: u32 parse_level: u32,
restrictions: ParserRestrictions,
}
struct ParserRestrictions {
no_struct_literal: bool
} }
impl Parser { impl Parser {
fn new(input: Vec<Token>) -> Parser { fn new(input: Vec<Token>) -> Parser {
Parser { tokens: input.into_iter().peekable(), parse_record: vec![], parse_level: 0 } Parser {
tokens: input.into_iter().peekable(),
parse_record: vec![],
parse_level: 0,
restrictions: ParserRestrictions { no_struct_literal: false }
}
} }
fn peek(&mut self) -> TokenType { fn peek(&mut self) -> TokenType {
@ -812,19 +820,25 @@ impl Parser {
parse_method!(paren_expr(&mut self) -> ParseResult<Expression> { parse_method!(paren_expr(&mut self) -> ParseResult<Expression> {
use self::ExpressionType::*; use self::ExpressionType::*;
let mut inner = delimited!(self, LParen, '(', expression, Comma, RParen, ')'); let old_struct_value = self.restrictions.no_struct_literal;
match inner.len() { self.restrictions.no_struct_literal = false;
0 => Ok(Expression(TupleLiteral(vec![]), None)), let output = (|| {
1 => Ok(inner.pop().unwrap()), let mut inner = delimited!(self, LParen, '(', expression, Comma, RParen, ')');
_ => Ok(Expression(TupleLiteral(inner), None)), match inner.len() {
} 0 => Ok(Expression(TupleLiteral(vec![]), None)),
1 => Ok(inner.pop().unwrap()),
_ => Ok(Expression(TupleLiteral(inner), None)),
}
})();
self.restrictions.no_struct_literal = old_struct_value;
output
}); });
parse_method!(identifier_expr(&mut self) -> ParseResult<Expression> { parse_method!(identifier_expr(&mut self) -> ParseResult<Expression> {
use self::ExpressionType::*; use self::ExpressionType::*;
let identifier = self.identifier()?; let identifier = self.identifier()?;
Ok(match self.peek() { Ok(match self.peek() {
LCurlyBrace => { LCurlyBrace if !self.restrictions.no_struct_literal => {
let fields = self.record_block()?; let fields = self.record_block()?;
Expression(Value(identifier, fields), None) Expression(Value(identifier, fields), None)
}, },
@ -866,7 +880,12 @@ impl Parser {
parse_method!(if_expr(&mut self) -> ParseResult<Expression> { parse_method!(if_expr(&mut self) -> ParseResult<Expression> {
expect!(self, Keyword(Kw::If), "'if'"); expect!(self, Keyword(Kw::If), "'if'");
let condition = self.expression()?; let condition = (|| {
self.restrictions.no_struct_literal = true;
let x = self.expression();
self.restrictions.no_struct_literal = false;
x
})()?;
let then_clause = self.block()?; let then_clause = self.block()?;
let else_clause = self.else_clause()?; let else_clause = self.else_clause()?;
Ok(Expression(ExpressionType::IfExpression(Box::new(condition), then_clause, else_clause), None)) Ok(Expression(ExpressionType::IfExpression(Box::new(condition), then_clause, else_clause), None))
@ -1243,6 +1262,18 @@ mod parse_tests {
exprstatement!(Value(rc!(b), vec![]))], exprstatement!(Value(rc!(b), vec![]))],
Some(vec![exprstatement!(Value(rc!(c), vec![]))])))]) Some(vec![exprstatement!(Value(rc!(c), vec![]))])))])
); );
parse_test!("if a { b } else { c }", AST(vec![exprstatement!(
IfExpression(Box::new(ex!(val!("a"))),
vec![exprstatement!(val!("b"))],
Some(vec![exprstatement!(val!("c"))])))]));
parse_test!("if (A {a: 1}) { b } else { c }", AST(vec![exprstatement!(
IfExpression(Box::new(ex!(Value(rc!(A), vec![(rc!(a), ex!(IntLiteral(1)))]))),
vec![exprstatement!(val!("b"))],
Some(vec![exprstatement!(val!("c"))])))]));
parse_error!("if A {a: 1} { b } else { c }");
} }
#[test] #[test]