added parsing logic for function arguments on function definitions

diff --git a/example.src b/example.src
index 2a2416d933511ac0229e4ad5468ff3d0595291dc..c6f51e66133817c62ffc61252db836b75ad5cdda 100644 (file)
--- a/example.src
+++ b/example.src
@@ -26,7 +26,7 @@ type type_union = union {
     bar = float;
 };
 
-fun main() int {
+fun main(args string[]) int {
     let foo int = 123;
     var bar int = 123;
     {}

diff --git a/source/parser.c b/source/parser.c
index 524df58bf98e89c188ea417c224100720eaa1191..374116bffb75373e92aab2bf5cf85bf855bd7c93 100644 (file)
--- a/source/parser.c
+++ b/source/parser.c
@@ -140,8 +140,20 @@ static AST* type_definition(Parser* p) {
 static AST* func_definition(Parser* p) {
     expect(p, T_FUN);
     expect(p, T_ID);
+
     expect(p, '(');
+    if (!matches(p, ')')) {
+        while (true) {
+            expect(p, T_ID);
+            type_expression(p);
+            if (!matches(p, ')'))
+                expect(p, ',');
+            else
+                break;
+        }
+    }
     expect(p, ')');
+
     type_expression(p);
     expression_block(p);
     return NULL;