add: parse call statements

1 files changed, 45 insertions, 0 deletions

diff --git a/src/parser/parser.c b/src/parser/parser.c
index 2ae52f2..55203ca 100644
--- a/src/parser/parser.c
+++ b/src/parser/parser.c
@@ -216,6 +216,48 @@ static bool is_compare_op(enum TokenType type)
 		|| type == TOKEN_GREATER_EQUAL;
 }
 
+static bool parse_call(struct Parser* parser, struct Token name, struct Statement* out)
+{
+	struct Expr** args = NULL;
+	size_t count = 0;
+	size_t capacity = 0;
+
+	if (!check(parser, TOKEN_RIGHT_PAREN))
+	{
+		do
+		{
+			struct Expr* arg = parse_expression(parser);
+			if (arg == NULL)
+				goto error;
+
+			if (count == capacity)
+			{
+				capacity = capacity < 4 ? 4 : capacity * 2;
+				args = realloc(args, capacity * sizeof(struct Expr*));
+			}
+			args[count] = arg;
+			count += 1;
+		}
+		while (match_token(parser, TOKEN_COMMA));
+	}
+
+	if (!consume(parser, TOKEN_RIGHT_PAREN, "expected ')' after arguments"))
+		goto error;
+
+	out->kind = STATEMENT_CALL;
+	out->call.name = name;
+	out->call.args = args;
+	out->call.arg_count = count;
+	out->call.arg_capacity = capacity;
+	return true;
+
+error:
+	for (size_t i = 0; i < count; i += 1)
+		free_expr(args[i]);
+	free(args);
+	return false;
+}
+
 static bool parse_statement(struct Parser* parser, struct Statement* out);
 
 static bool parse_if(struct Parser* parser, struct Statement* out)
@@ -284,6 +326,9 @@ static bool parse_statement(struct Parser* parser, struct Statement* out)
 		return false;
 	struct Token name = parser->previous;
 
+	if (!deref && match_token(parser, TOKEN_LEFT_PAREN))
+		return parse_call(parser, name, out);
+
 	if (!deref && match_token(parser, TOKEN_COLON))
 	{
 		out->kind = STATEMENT_LABEL;