ProvableHQ/leo
1
1
Fork 0
mirror of https://github.com/ProvableHQ/leo.git synced 2024-11-30 03:34:51 +03:00
Code Issues Packages Projects Releases Wiki Activity

Parse empty return statement as a unit expression

Browse Source
This commit is contained in:
d0cd 2022-10-13 13:39:30 -05:00
parent 6789bc8aae
commit 34fa2cb2c6
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
compiler/parser/src/parser/statement.rs
View File

@ -116,8 +116,13 @@ impl ParserContext<'_> {
/// Returns a [`ReturnStatement`] AST node if the next tokens represent a return statement.
fn parse_return_statement(&mut self) -> Result<ReturnStatement> {
let start = self.expect(&Token::Return)?;
let expression = self.parse_expression()?;
self.expect(&Token::Semicolon)?;
let expression = match self.token.token {
Token::Semicolon => {
let span = self.expect(&Token::Semicolon)?;
Expression::Unit(UnitExpression { span })
}
_ => self.parse_expression()?,
};
let span = start + expression.span();
Ok(ReturnStatement { span, expression })
}