Can parse variables

v0/parser.c

@@ -264,6 +264,57 @@ static bool parse_alias_declaration(Parser* p, Module* module, bool is_public) {
     return true;
 }
 
+/**
+ * Parses a variable declaration.
+ * 
+ * @param p The parser state.
+ * @param module The module to add the variable to.
+ * @param is_public Whether the variable is public.
+ * @param is_static Whether the variable is static.
+ * @param is_const Whether the variable is constant.
+ * @return true if successful, false otherwise.
+ */
+static bool parse_variable_declaration(Parser* p, Module* module, bool is_public, bool is_static, bool is_const) {
+    module->variable_count++;
+    module->variables = realloc(module->variables, sizeof(VariableDeclaration) * module->variable_count);
+
+    VariableDeclaration* var = &module->variables[module->variable_count - 1];
+    memset(var, 0, sizeof(VariableDeclaration));
+    var->is_public = is_public;
+    var->is_static = is_static;
+    var->is_const = is_const;
+
+    if (!parse_type_expression(p, &var->type)) {
+        return false;
+    }
+
+    if (!parser_require(p, TOKEN_IDENTIFIER, "expected variable identifier")) {
+        return false;
+    }
+    var->name = parser_to_text(p);
+
+    if (!parser_expect(p, TOKEN_SEMICOLON, "expected ';' after variable declaration")) {
+        return false;
+    }
+
+    return true;
+}
+
+/**
+ * Checks if the current token is a primitive type.
+ * 
+ * The token will not be consumed by this function, so the caller can decide how to handle it if it is a primitive type.
+ * 
+ * @param p The parser state.
+ * @return true if the current token is a primitive type, false otherwise.
+ */
+static bool parser_accept_primitive(Parser* p) {
+    return parser_peek(p, TOKEN_I8) || parser_peek(p, TOKEN_I16) || 
+           parser_peek(p, TOKEN_I32) || parser_peek(p, TOKEN_I64) ||
+           parser_peek(p, TOKEN_U8) || parser_peek(p, TOKEN_U16) || 
+           parser_peek(p, TOKEN_U32) || parser_peek(p, TOKEN_U64);
+}
+
 Module* parser_parse(TokenStream* ts) {
     Parser* p = malloc(sizeof(Parser));
     p->ts = ts;
@@ -277,25 +328,53 @@ Module* parser_parse(TokenStream* ts) {
 
     while (!parser_peek(p, TOKEN_EOF)) {
         bool is_public = false;
+        bool is_static = false;
+        bool is_const = false;
         bool terminal = false;
-        do {
+        
+        while (!terminal) {
             if (parser_accept(p, TOKEN_IMPORT)) {
+                if (is_static) {
+                    log_on_line(&p->token.location, "import declarations cannot be static or const");
+                    goto fail;
+                }
+                if (is_const) {
+                    log_on_line(&p->token.location, "import declarations cannot be static or const");
+                    goto fail;
+                }
                 if (!parse_import_declaration(p, module, is_public)) {
                     goto fail;
                 }
                 terminal = true;
             } else if (parser_accept(p, TOKEN_ALIAS)) {
+                if (is_static) {
+                    log_on_line(&p->token.location, "alias declarations cannot be static or const");
+                    goto fail;
+                }
+                if (is_const) {
+                    log_on_line(&p->token.location, "alias declarations cannot be static or const");
+                    goto fail;
+                }
                 if (!parse_alias_declaration(p, module, is_public)) {
                     goto fail;
                 }
                 terminal = true;
             } else if (parser_accept(p, TOKEN_PUBLIC)) {
                 is_public = true;
+            } else if (parser_accept(p, TOKEN_STATIC)) {
+                is_static = true;
+            } else if (parser_accept(p, TOKEN_CONST)) {
+                is_const = true;
+            } else if (parser_accept_primitive(p)) {
+                if (!parse_variable_declaration(p, module, is_public, is_static, is_const)) {
+                    goto fail;
+                }
+                terminal = true;
             } else {
                 log_on_line(&p->token.location, "unexpected token");
                 goto fail;
             }
-        } while (!terminal);
+        }
     }
 
     free(p);
@@ -333,6 +412,14 @@ void parser_free(Module* module) {
         free(module->aliases);
     }
 
+    if (module->variables != NULL) {
+        for(size_t i = 0; i < module->variable_count; i++) {
+            free(module->variables[i].name);
+            free_type_expression(&module->variables[i].type);
+        }
+        free(module->variables);
+    }
+
     free(module->name);
-	free(module);
+    free(module);
 }