fix(type_checker,compiler): add support for function assignment + add…

…itional checks
YakshaLang · Jul 1, 2023 · 1a5fd71 · 1a5fd71
1 parent 98c94d0
commit 1a5fd71
Show file tree

Hide file tree

Showing 5 changed files with 510 additions and 407 deletions.
diff --git a/compiler/src/compiler/compiler.cpp b/compiler/src/compiler/compiler.cpp
@@ -679,8 +679,15 @@ void compiler::visit_let_stmt(let_stmt *obj) {
   if (obj->data_type_ == nullptr) {
     visited_expr = true;
     resulting_pair = compile_expression(obj->expression_);
-    obj->data_type_ =
-        resulting_pair.second.datatype_; /* set our data type here */
+    if (resulting_pair.second.is_a_function()) {
+      obj->data_type_ = function_to_datatype(resulting_pair.second);
+    } else {
+      obj->data_type_ =
+          resulting_pair.second.datatype_; /* set our data type here */
+    }
+  }
+  if (obj->data_type_->is_none()) {
+    error(obj->name_, "Failed to compile let statement. (Use of non compilable data type)");
   }
   if (obj->data_type_->is_str()) {
     object = ykobject(std::string("str"), dt_pool_);

diff --git a/compiler/src/compiler/type_checker.cpp b/compiler/src/compiler/type_checker.cpp
@@ -19,6 +19,9 @@ void type_checker::visit_assign_expr(assign_expr *obj) {
   if (scope_.is_defined(name)) {
     object = scope_.get(name);
   } else {
+    if (rhs.is_a_function()) {
+      rhs.datatype_ = function_to_datatype(rhs);
+    }
     object = ykobject(rhs.datatype_);
     obj->promoted_ = true;
   }

diff --git a/compiler/test_data/compiler_tests/on_stack_test.yaka b/compiler/test_data/compiler_tests/on_stack_test.yaka
@@ -12,7 +12,13 @@ struct Friend:
 class AnotherFriend:
     x: int
 
+def a() -> int:
+    return 0
+
 def main() -> int:
+    b = a
+    d = a
+    e = "Hello world"
     enemies = arrnew("Enemy", 2)
     enemies[0].x = 1
     enemies[1].x = 2