Implement optional break expressions

1 files changed, 15 insertions, 8 deletions

diff --git a/src/parse.c b/src/parse.c
index 3fa1498..ee115d8 100644
--- a/src/parse.c
+++ b/src/parse.c
@@ -1107,14 +1107,14 @@ parse_return(struct lexer *lexer, struct ast_return *out)
 }
 
 static bool
-parse_break(struct lexer *lexer, const char **out)
+parse_break(struct lexer *lexer, struct ast_break *out)
 {
 	if (!match(lexer, T_BREAK)) {
 		return false;
 	}
 
 	if (!match(lexer, T_COLON)) {
-		*out = NULL;
+		out->label = NULL;
 		return true;
 	}
 
@@ -1122,20 +1122,26 @@ parse_break(struct lexer *lexer, const char **out)
 	if (lex(lexer, &label) != T_NAME) {
 		error(lex_loc(lexer), "syntax error: expected name");
 	}
-	*out = label.info.str;
+	out->label = label.info.str;
+
+	out->value = must_malloc(sizeof(struct ast_expr));
+	if (!parse_expr(lexer, out->value)) {
+		free(out->value);
+		out->value = NULL;
+	}
 
 	return true;
 }
 
 static bool
-parse_continue(struct lexer *lexer, const char **out)
+parse_continue(struct lexer *lexer, struct ast_break *out)
 {
 	if (!match(lexer, T_CONTINUE)) {
 		return false;
 	}
 
 	if (!match(lexer, T_COLON)) {
-		*out = NULL;
+		out->label = NULL;
 		return true;
 	}
 
@@ -1143,8 +1149,9 @@ parse_continue(struct lexer *lexer, const char **out)
 	if (lex(lexer, &label) != T_NAME) {
 		error(lex_loc(lexer), "syntax error: expected name");
 	}
-	*out = label.info.str;
+	out->label = label.info.str;
 
+	out->value = NULL;
 	return true;
 }
 
@@ -1307,14 +1314,14 @@ parse_statement(struct lexer *lexer, struct ast_statement *out)
 		}
 		return true;
 	}
-	if (parse_break(lexer, &out->stmt.brklabel)) {
+	if (parse_break(lexer, &out->stmt.brk)) {
 		out->action = STM_BREAK;
 		if (!match(lexer, T_SEMICOLON)) {
 			error(lex_loc(lexer), "syntax error: expected semicolon");
 		}
 		return true;
 	}
-	if (parse_continue(lexer, &out->stmt.brklabel)) {
+	if (parse_continue(lexer, &out->stmt.brk)) {
 		out->action = STM_CONTINUE;
 		if (!match(lexer, T_SEMICOLON)) {
 			error(lex_loc(lexer), "syntax error: expected semicolon");