VM: add support for type deduction

mrunix00 · Jul 9, 2024 · cc1247d · cc1247d
1 parent 260498e
commit cc1247d
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/src/ast.cpp b/src/ast.cpp
@@ -116,8 +116,17 @@ bool Declaration::operator==(const AbstractSyntaxTree &other) const {
 }
 
 void Declaration::compile(Program &program, Segment &segment) const {
-    if (!type.has_value())
-        throw std::runtime_error("[Declaration::compile] Type deduction is not implemented!");
+    if (!type.has_value()) {
+        if (value.has_value()) {
+            auto varType = deduceType(program, segment, value.value());
+            value.value()->compile(program, segment);
+            segment.declare_variable(identifier.token.value, varType);
+            emitStore(program, segment, identifier.token.value);
+        } else {
+            throw std::runtime_error("[Declaration::compile] Cannot deduce the variable type!");
+        }
+        return;
+    }
     switch (type.value()->nodeType) {
         case AbstractSyntaxTree::Type::Node: {
             Node *initNode = (Node *) value.value();

diff --git a/tests/vm_tests.cpp b/tests/vm_tests.cpp
@@ -58,6 +58,15 @@ TEST(VM, SimpleVariableDeclaration) {
     ASSERT_EQ(*static_cast<int32_t *>(vm.topStack(sizeof(int32_t))), 42);
 }
 
+TEST(VM, TypeDeduction) {
+    const char *input = "define a = 42;"
+                        "a;";
+    VM vm;
+    auto program = compile(input);
+    vm.run(program);
+    ASSERT_EQ(*static_cast<int32_t *>(vm.topStack(sizeof(int32_t))), 42);
+}
+
 TEST(VM, SimpleI64VariableDeclaration) {
     const char *input = "define a : i64 = 42;"
                         "a;";