Fix module scopes being linked

reid-llvm-lib/src/compile.rs

@@ -265,8 +265,15 @@ impl FunctionHolder {
                 return;
             }
 
-            if self.data.flags.is_imported && !in_main_module {
-                LLVMSetLinkage(own_function.value_ref, LLVMLinkage::LLVMExternalLinkage);
+            if self.data.flags.is_imported {
+                if self.data.flags.is_extern {
+                    LLVMSetLinkage(
+                        own_function.value_ref,
+                        LLVMLinkage::LLVMAvailableExternallyLinkage,
+                    );
+                } else {
+                    LLVMSetLinkage(own_function.value_ref, LLVMLinkage::LLVMExternalLinkage);
+                }
             } else {
                 LLVMSetLinkage(own_function.value_ref, LLVMLinkage::LLVMPrivateLinkage);
             }

reid/examples/cli.rs

@@ -12,6 +12,8 @@ fn main() -> Result<(), std::io::Error> {
         let object_path = parent.with_extension("o");
         let asm_path = parent.with_extension("asm");
 
+        let before = std::time::SystemTime::now();
+
         let text = fs::read_to_string(&path)?;
         match compile(&text, PathBuf::from(&path)) {
             Ok(CompileOutput {
@@ -20,15 +22,20 @@ fn main() -> Result<(), std::io::Error> {
                 obj_buffer,
                 llvm_ir,
             }) => {
-                println!("Compiled with triple: {}", &triple);
+                println!("{}", llvm_ir);
+
+                let after = std::time::SystemTime::now();
+                println!("Compiled with triple: {}\n", &triple);
                 fs::write(&llvm_ir_path, &llvm_ir).expect("Could not write LLVM IR -file!");
                 println!("Output LLVM IR to {:?}", llvm_ir_path);
                 fs::write(&asm_path, &assembly).expect("Could not write Assembly-file!");
                 println!("Output Assembly to {:?}", asm_path);
                 fs::write(&object_path, &obj_buffer).expect("Could not write Object-file!");
-                println!("Output Object-file to {:?}", object_path);
-
-                println!("{}", llvm_ir);
+                println!("Output Object-file to {:?}\n", object_path);
+                println!(
+                    "Compilation took: {:.2}ms\n",
+                    (after.duration_since(before).unwrap().as_micros() as f32) / 1000.
+                );
             }
             Err(e) => panic!("{}", e),
         };

reid/src/ast/process.rs

@@ -58,7 +58,7 @@ impl ast::Module {
                             .cloned()
                             .map(|p| (p.0, p.1.into()))
                             .collect(),
-                        kind: mir::FunctionDefinitionKind::Extern,
+                        kind: mir::FunctionDefinitionKind::Extern(false),
                     };
                     functions.push(def);
                 }

reid/src/codegen.rs

@@ -69,12 +69,13 @@ impl mir::Module {
                         ..FunctionFlags::default()
                     },
                 ),
-                mir::FunctionDefinitionKind::Extern => module.function(
+                mir::FunctionDefinitionKind::Extern(imported) => module.function(
                     &function.name,
                     function.return_type.get_type(),
                     param_types,
                     FunctionFlags {
                         is_extern: true,
+                        is_imported: *imported,
                         ..FunctionFlags::default()
                     },
                 ),
@@ -117,7 +118,7 @@ impl mir::Module {
                         }
                     }
                 }
-                mir::FunctionDefinitionKind::Extern => {}
+                mir::FunctionDefinitionKind::Extern(_) => {}
             }
         }
 

reid/src/lib.rs

@@ -103,13 +103,18 @@ pub fn compile_module(
 }
 
 pub fn perform_all_passes(context: &mut mir::Context) -> Result<(), ReidError> {
+    #[cfg(debug_assertions)]
+    dbg!(&context);
+
     #[cfg(debug_assertions)]
     println!("{}", &context);
 
     let state = context.pass(&mut LinkerPass);
 
     #[cfg(debug_assertions)]
-    println!("{:?}\n{}", &context, &context);
+    println!("{}", &context);
+    #[cfg(debug_assertions)]
+    dbg!(&state);
 
     if !state.errors.is_empty() {
         return Err(ReidError::LinkerErrors(state.errors));
@@ -119,10 +124,10 @@ pub fn perform_all_passes(context: &mut mir::Context) -> Result<(), ReidError> {
 
     let state = context.pass(&mut TypeInference { refs: &refs });
 
-    #[cfg(debug_assertions)]
-    dbg!(&state, &refs);
     #[cfg(debug_assertions)]
     println!("{}", &context);
+    #[cfg(debug_assertions)]
+    dbg!(&state, &refs);
 
     if !state.errors.is_empty() {
         return Err(ReidError::TypeInferenceErrors(state.errors));
@@ -130,10 +135,10 @@ pub fn perform_all_passes(context: &mut mir::Context) -> Result<(), ReidError> {
 
     let state = context.pass(&mut TypeCheck { refs: &refs });
 
-    #[cfg(debug_assertions)]
-    dbg!(&state);
     #[cfg(debug_assertions)]
     println!("{}", &context);
+    #[cfg(debug_assertions)]
+    dbg!(&state);
 
     if !state.errors.is_empty() {
         return Err(ReidError::TypeCheckErrors(state.errors));

reid/src/mir/display.rs

@@ -61,7 +61,8 @@ impl Display for FunctionDefinitionKind {
                 write!(f, "{}", block)?;
                 Ok(())
             }
-            Self::Extern => write!(f, "<External>"),
+            Self::Extern(true) => write!(f, "<Imported Extern>"),
+            Self::Extern(false) => write!(f, "<Linked Extern>"),
         }
     }
 }

reid/src/mir/linker.rs

@@ -193,7 +193,7 @@ impl Pass for LinkerPass {
                     is_imported: false,
                     return_type: func.return_type.clone(),
                     parameters: func.parameters.clone(),
-                    kind: super::FunctionDefinitionKind::Extern,
+                    kind: super::FunctionDefinitionKind::Extern(true),
                 });
             }
         }

reid/src/mir/mod.rs

@@ -254,21 +254,22 @@ pub struct FunctionDefinition {
 pub enum FunctionDefinitionKind {
     /// Actual definition block and surrounding signature range
     Local(Block, Metadata),
-    Extern,
+    /// True = imported from other module, False = Is user defined extern
+    Extern(bool),
 }
 
 impl FunctionDefinition {
     fn block_meta(&self) -> Metadata {
         match &self.kind {
             FunctionDefinitionKind::Local(block, _) => block.meta,
-            FunctionDefinitionKind::Extern => Metadata::default(),
+            FunctionDefinitionKind::Extern(_) => Metadata::default(),
         }
     }
 
     fn signature(&self) -> Metadata {
         match &self.kind {
             FunctionDefinitionKind::Local(_, metadata) => *metadata,
-            FunctionDefinitionKind::Extern => Metadata::default(),
+            FunctionDefinitionKind::Extern(_) => Metadata::default(),
         }
     }
 }

reid/src/mir/pass.rs

@@ -210,7 +210,7 @@ impl Context {
         let mut scope = Scope::default();
         pass.context(self, PassState::from(&mut state, &mut scope));
         for module in &mut self.modules {
-            module.pass(pass, &mut state, &mut scope);
+            module.pass(pass, &mut state, &mut scope.inner());
         }
         state
     }
@@ -261,7 +261,7 @@ impl FunctionDefinition {
                 scope.return_type_hint = Some(self.return_type.clone());
                 block.pass(pass, state, scope);
             }
-            FunctionDefinitionKind::Extern => {}
+            FunctionDefinitionKind::Extern(_) => {}
         };
     }
 }

reid/src/mir/typecheck.rs

@@ -94,7 +94,7 @@ impl FunctionDefinition {
                 state.scope.return_type_hint = Some(self.return_type.clone());
                 block.typecheck(&mut state.inner(), &hints, Some(&return_type))
             }
-            FunctionDefinitionKind::Extern => {
+            FunctionDefinitionKind::Extern(_) => {
                 Ok((ReturnKind::Soft, TypeKind::Vague(Vague::Unknown)))
             }
         };

reid/src/mir/typeinference.rs

@@ -64,7 +64,7 @@ impl FunctionDefinition {
                     ret_ty.narrow(&scope_hints.from_type(&self.return_type).unwrap());
                 }
             }
-            FunctionDefinitionKind::Extern => {}
+            FunctionDefinitionKind::Extern(_) => {}
         };
 
         Ok(())

reid/src/mir/types.rs

@@ -181,14 +181,16 @@ impl IndexedVariableReference {
     }
 }
 
-#[derive(Debug, Clone, Copy, thiserror::Error)]
+#[derive(Debug, Clone, thiserror::Error)]
 pub enum EqualsIssue {
     #[error("Function is already defined locally at {:?}", (.0).range)]
     ExistsLocally(Metadata),
-    #[error("asd")]
+    #[error("Equals")]
     Equals,
-    #[error("asd")]
-    ConflictingImports,
+    #[error("Function {0} is already declared locally at {:?}", (.1).range)]
+    AlreadyExtern(String, Metadata),
+    #[error("Function {0} is already imported from another module")]
+    ConflictWithImport(String),
 }
 
 impl FunctionDefinition {
@@ -197,15 +199,22 @@ impl FunctionDefinition {
             FunctionDefinitionKind::Local(_, metadata) => {
                 Err(EqualsIssue::ExistsLocally(*metadata))
             }
-            FunctionDefinitionKind::Extern => {
-                if self.is_pub == other.is_pub
-                    && self.name == other.name
-                    && self.parameters == other.parameters
-                    && self.return_type == other.return_type
-                {
-                    Ok(())
+            FunctionDefinitionKind::Extern(imported) => {
+                if *imported {
+                    Err(EqualsIssue::ConflictWithImport(self.name.clone()))
                 } else {
-                    Err(EqualsIssue::ConflictingImports)
+                    if self.is_pub == other.is_pub
+                        && self.name == other.name
+                        && self.parameters == other.parameters
+                        && self.return_type == other.return_type
+                    {
+                        Ok(())
+                    } else {
+                        Err(EqualsIssue::AlreadyExtern(
+                            self.name.clone(),
+                            self.signature(),
+                        ))
+                    }
                 }
             }
         }