diff options
Diffstat (limited to 'src/parse/ast')
| -rw-r--r-- | src/parse/ast/statement.rs | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/parse/ast/statement.rs b/src/parse/ast/statement.rs index 7ed91dc..899190b 100644 --- a/src/parse/ast/statement.rs +++ b/src/parse/ast/statement.rs @@ -16,7 +16,7 @@ pub enum StatementKind { Print(Expression), Break(Option<Expression>), Continue, - Return(Expression), + Return(Option<Expression>), } impl Display for Statement { @@ -39,8 +39,12 @@ impl Statement { expression.nested_fmt(f, depth + 1)?; } StatementKind::Return(expression) => { - writeln!(f, "{}Return:", pad)?; - expression.nested_fmt(f, depth + 1)?; + if let Some(returned) = expression { + writeln!(f, "{}Return:", pad)?; + returned.nested_fmt(f, depth + 1)?; + } else { + writeln!(f, "{}Return", pad)?; + } } StatementKind::Break(expression) => { if let Some(returned_on_break) = expression { |