From a07b22097f3758558f4b35aa5aa5d8c1afc3402b Mon Sep 17 00:00:00 2001 From: hachem Date: Sun, 30 Aug 2026 16:09:03 +0200 Subject: add: parse call statements --- src/main.c | 10 ++++++++++ src/parser/ast.c | 5 +++++ src/parser/ast.h | 10 ++++++++++ src/parser/parser.c | 45 +++++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 70 insertions(+) (limited to 'src') diff --git a/src/main.c b/src/main.c index 63f87fd..e33734d 100644 --- a/src/main.c +++ b/src/main.c @@ -57,6 +57,16 @@ static void print_statement(struct Statement* statement, const char* indent) printf("\n"); print_statement(statement->branch.body, " "); break; + case STATEMENT_CALL: + printf("%s%.*s(", indent, (int)statement->call.name.length, statement->call.name.start); + for (size_t i = 0; i < statement->call.arg_count; i += 1) + { + if (i > 0) + printf(", "); + print_expr(statement->call.args[i]); + } + printf(")\n"); + break; } } diff --git a/src/parser/ast.c b/src/parser/ast.c index 9bdf4d1..1aa1d38 100644 --- a/src/parser/ast.c +++ b/src/parser/ast.c @@ -36,6 +36,11 @@ static void free_statement(struct Statement* statement) free_statement(statement->branch.body); free(statement->branch.body); break; + case STATEMENT_CALL: + for (size_t i = 0; i < statement->call.arg_count; i += 1) + free_expr(statement->call.args[i]); + free(statement->call.args); + break; default: break; } diff --git a/src/parser/ast.h b/src/parser/ast.h index e85e72f..3fa80da 100644 --- a/src/parser/ast.h +++ b/src/parser/ast.h @@ -66,6 +66,7 @@ enum StatementKind STATEMENT_GOTO, STATEMENT_SYSCALL, STATEMENT_IF, + STATEMENT_CALL, }; struct AssignStatement @@ -94,6 +95,14 @@ struct IfStatement struct Statement* body; }; +struct CallStatement +{ + struct Token name; + struct Expr** args; + size_t arg_count; + size_t arg_capacity; +}; + struct Statement { enum StatementKind kind; @@ -103,6 +112,7 @@ struct Statement struct LabelStatement label; struct GotoStatement jump; struct IfStatement branch; + struct CallStatement call; }; }; 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; -- cgit v1.3