Move linking to its own pass

src/semantics/link_impls.rs

@@ -0,0 +1,37 @@
+use derive_visitor::VisitorMut;
+use log::debug;
+
+use crate::hir::{Call, Generic, Type};
+
+use super::Context;
+
+#[derive(VisitorMut)]
+#[visitor(Call(exit))]
+pub struct TraitFunctionsLinker<'ctx, C: Context> {
+    context: &'ctx mut C,
+}
+
+impl<'ctx, C: Context> TraitFunctionsLinker<'ctx, C> {
+    pub fn new(context: &'ctx mut C) -> Self {
+        Self { context }
+    }
+
+    fn exit_call(&mut self, call: &mut Call) {
+        let f = call.function.read().unwrap();
+        // FIXME: definition may be overrided
+        if f.is_generic() || !f.is_from_trait() || f.is_definition() {
+            return;
+        }
+
+        debug!(target: "linking-trait-fn-from", "{f}");
+        // Unknown type here is ok, because we don't have selfs any more
+        let real_impl = self
+            .context
+            .find_implementation(&f, &Type::Unknown)
+            .unwrap();
+        drop(f);
+
+        call.function = real_impl;
+        debug!(target: "linking-trait-fn-to", "{}", call.function);
+    }
+}

src/semantics/mod.rs

@@ -40,3 +40,6 @@ pub use unnamed::*;
 
 mod replace_self;
 pub use replace_self::*;
+
+mod link_impls;
+pub use link_impls::*;

src/semantics/to_hir.rs

@@ -15,7 +15,9 @@ use crate::hir::{
 use crate::mutability::{Mutability, Mutable};
 use crate::named::Named;
 use crate::semantics::clone::Clonner;
-use crate::semantics::{InsertDestructors, ParameterNamer, TemporariesInserter};
+use crate::semantics::{
+    InsertDestructors, ParameterNamer, TemporariesInserter, TraitFunctionsLinker,
+};
 use crate::syntax::{Identifier, Keyword, Ranged};
 use crate::{AddSourceLocation, ErrVec, SourceLocation, WithSourceLocation};
 
@@ -975,6 +977,7 @@ impl ToHIR for ast::Module {
         debug!(target: &format!("{name}-hir"), "\n{:#}", module);
         trace!(target: "steps", "Running passes on `{}`", module.source_file.path().display());
         module.drive_mut(&mut ParameterNamer::new());
+        module.drive_mut(&mut TraitFunctionsLinker::new(context));
         module.drive_mut(&mut Clonner::new(context));
         module.drive_mut(&mut TemporariesInserter::new());
         module.insert_destructors(context);