Decleration identifier

semantic/src/hir.rs

@@ -97,6 +97,9 @@ impl TopLevelHirScope {
         tlss: &TopLevelSymbolScope,
         symbol: &StructSymbol,
     ) -> Arc<TypeRef> {
+        if let Some(rf) = self.variables.get(&symbol.name) {
+            rf.var_type.clone()
+        } else {
             let name = self.get_name(symbol.export);
 
             let rf = Arc::new(TypeRef::Struct(StructRef {
@@ -119,11 +122,12 @@ impl TopLevelHirScope {
 
             rf
         }
+    }
 
     pub fn var_ref(&mut self, tlss: &TopLevelSymbolScope, symbol: &VarSymbol) -> Arc<VarRef> {
         Arc::new(VarRef {
             var_type: self.type_ref(tlss, &symbol.var_type),
-            name: symbol.name.clone(),
+            name: self.get_name(true),
         })
     }
 
@@ -132,15 +136,7 @@ impl TopLevelHirScope {
             var_ref.var_type.clone()
         } else {
             if let Some(tlss_rf) = tlss.structs.get(&symbol.0) {
-                let struct_ref = self.struct_ref(tlss, tlss_rf);
+                self.struct_ref(tlss, tlss_rf)
-
-                let var_ref = Arc::new(VarRef {
-                    name: symbol.0.clone(),
-                    var_type: struct_ref,
-                });
-                self.variables.insert(symbol.0.clone(), var_ref.clone());
-
-                var_ref.var_type.clone()
             } else {
                 match symbol.0.as_str() {
                     "int" => {

semantic/src/lib.rs

@@ -11,7 +11,11 @@ pub fn walk_ast(top_scope: Arc<Scope>, tl: &mut Vec<parser::ast::TopLevel>) {
         match tl {
             parser::ast::TopLevel::Import(_) => unimplemented!(),
 
-            parser::ast::TopLevel::FunctionDecl { params, body, .. } => {
+            parser::ast::TopLevel::FunctionDecl {
+                params, body, name, ..
+            } => {
+                *name = top_scope.get_reference(name).unwrap().name.clone();
+
                 let scope = LocalScope::new(top_scope.clone());
 
                 scope.with_params(params);
@@ -19,7 +23,9 @@ pub fn walk_ast(top_scope: Arc<Scope>, tl: &mut Vec<parser::ast::TopLevel>) {
                 scope.with_block(body);
             }
 
-            parser::ast::TopLevel::StructDecl { .. } => {}
+            parser::ast::TopLevel::StructDecl { name, .. } => {
+                *name = top_scope.get_reference(name).unwrap().name.clone();
+            }
         }
     }
 }

test/src/main.ms

@@ -1,8 +1,8 @@
-struct MyType {
+export struct MyType {
     x: int,
 }
 
-function test(): MyType { }
+export function test(): MyType { }
 
 function add() {
     let output = test().x;