Fix type updating for associated function calls

examples/associated_functions_shorthand.reid

@@ -1,6 +1,5 @@
 import std::print;
 import std::from_str;
-import std::String;
 
 struct Otus {
     field: u32,

examples/hello_world_harder.reid

@@ -1,10 +1,12 @@
 import std::print;
+import std::from_str;
 import std::add_char;
 import std::set_char;
 import std::free_string;
-import std::new_string;
 import std::add_num_to_str;
 import std::concat_strings;
+import std::String;
+
 
 fn main() {
     let mut test = String::new();

reid/src/mir/linker.rs

@@ -401,9 +401,14 @@ impl<'map> Pass for LinkerPass<'map> {
                 }
                 super::ExprKind::BinOp(.., type_kind) => *type_kind = type_kind.update_imported(extern_types, mod_id),
 
-                super::ExprKind::Borrow(expr, _) => {}
-                super::ExprKind::Deref(expr) => {}
+                super::ExprKind::Borrow(..) => {}
+                super::ExprKind::Deref(..) => {}
                 super::ExprKind::CastTo(_, type_kind) => *type_kind = type_kind.update_imported(extern_types, mod_id),
+                super::ExprKind::AssociatedFunctionCall(type_kind, _) => {
+                    dbg!(&type_kind);
+                    dbg!(extern_types);
+                    *type_kind = type_kind.update_imported(extern_types, mod_id)
+                }
                 _ => {}
             }
         }

reid/src/mir/pass.rs

@@ -519,7 +519,8 @@ impl Statement {
             StmtKind::Let(_, _, expression) => {
                 expression.pass(pass, state, scope, mod_id)?;
             }
-            StmtKind::Set(_, expression) => {
+            StmtKind::Set(set_expr, expression) => {
+                set_expr.pass(pass, state, scope, mod_id)?;
                 expression.pass(pass, state, scope, mod_id)?;
             }
             StmtKind::Import(_) => {}
@@ -565,6 +566,52 @@ impl Expression {
         mod_id: SourceModuleId,
     ) -> PassResult {
         pass.expr(self, PassState::from(state, scope, Some(mod_id)))?;
+        match &mut self.0 {
+            ExprKind::Variable(_) => {}
+            ExprKind::Indexed(value_expr, _, index_expr) => {
+                pass.expr(value_expr.as_mut(), PassState::from(state, scope, Some(mod_id)))?;
+                pass.expr(index_expr.as_mut(), PassState::from(state, scope, Some(mod_id)))?;
+            }
+            ExprKind::Accessed(value_expr, ..) => {
+                pass.expr(value_expr.as_mut(), PassState::from(state, scope, Some(mod_id)))?;
+            }
+            ExprKind::Array(expressions) => {
+                for expr in expressions {
+                    pass.expr(expr, PassState::from(state, scope, Some(mod_id)))?;
+                }
+            }
+            ExprKind::Struct(_, items) => {
+                for (_, expr) in items {
+                    pass.expr(expr, PassState::from(state, scope, Some(mod_id)))?;
+                }
+            }
+            ExprKind::Literal(_) => {}
+            ExprKind::BinOp(_, lhs, rhs, _) => {
+                pass.expr(lhs.as_mut(), PassState::from(state, scope, Some(mod_id)))?;
+                pass.expr(rhs.as_mut(), PassState::from(state, scope, Some(mod_id)))?;
+            }
+            ExprKind::FunctionCall(FunctionCall { parameters, .. }) => {
+                for expr in parameters {
+                    pass.expr(expr, PassState::from(state, scope, Some(mod_id)))?;
+                }
+            }
+            ExprKind::AssociatedFunctionCall(_, FunctionCall { parameters, .. }) => {
+                for expr in parameters {
+                    pass.expr(expr, PassState::from(state, scope, Some(mod_id)))?;
+                }
+            }
+            ExprKind::If(IfExpression(cond, lhs, rhs)) => {
+                pass.expr(cond.as_mut(), PassState::from(state, scope, Some(mod_id)))?;
+                pass.expr(lhs.as_mut(), PassState::from(state, scope, Some(mod_id)))?;
+                if let Some(rhs) = rhs.as_mut() {
+                    pass.expr(rhs, PassState::from(state, scope, Some(mod_id)))?;
+                }
+            }
+            ExprKind::Block(block) => pass.block(block, PassState::from(state, scope, Some(mod_id)))?,
+            ExprKind::Borrow(expression, _) => pass.expr(expression, PassState::from(state, scope, Some(mod_id)))?,
+            ExprKind::Deref(expression) => pass.expr(expression, PassState::from(state, scope, Some(mod_id)))?,
+            ExprKind::CastTo(expression, _) => pass.expr(expression, PassState::from(state, scope, Some(mod_id)))?,
+        }
         Ok(())
     }
 }

reid/src/mir/typecheck/typeinference.rs

@@ -594,7 +594,10 @@ impl Expression {
                 Ok(type_refs.from_type(type_kind).unwrap())
             }
             ExprKind::AssociatedFunctionCall(type_kind, function_call) => {
-                if type_kind.is_known(state).is_err() {
+                match type_kind.is_known(state) {
+                    Ok(_) => {}
+                    Err(ErrorKind::NoSuchType(name, mod_id)) => return Err(ErrorKind::NoSuchType(name, mod_id)),
+                    Err(_) => {
                         let first_param = function_call
                             .parameters
                             .get_mut(0)
@@ -613,6 +616,7 @@ impl Expression {
                             first_param.remove_borrow_mutability();
                         }
                     }
+                }
 
                 // Get function definition and types
                 let fn_call = state