fix: user type renaming bug

erg-lang · Sep 6, 2023 · 6ca5e07 · 6ca5e07
1 parent 2e0ee93
commit 6ca5e07
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/crates/erg_compiler/context/inquire.rs b/crates/erg_compiler/context/inquire.rs
@@ -1329,7 +1329,7 @@ impl Context {
         ))
     }
 
-    fn validate_visibility(
+    pub(crate) fn validate_visibility(
         &self,
         ident: &Identifier,
         vi: &VarInfo,

diff --git a/crates/erg_compiler/context/instantiate_spec.rs b/crates/erg_compiler/context/instantiate_spec.rs
@@ -551,12 +551,18 @@ impl Context {
                     }
                 }
                 if let Some((typ, _)) = self.get_type_and_ctx(ident.inspect()) {
+                    if let Some((_, vi)) = self.get_var_info(ident.inspect()) {
+                        self.inc_ref(ident.inspect(), vi, ident, self);
+                    }
                     Ok(typ.clone())
                 } else if let Some(typ) = self
                     .consts
                     .get(ident.inspect())
                     .and_then(|v| self.convert_value_into_type(v.clone()).ok())
                 {
+                    if let Some((_, vi)) = self.get_var_info(ident.inspect()) {
+                        self.inc_ref(ident.inspect(), vi, ident, self);
+                    }
                     Ok(typ)
                 } else if not_found_is_qvar {
                     let tyvar = named_free_var(Str::rc(other), self.level, Constraint::Uninited);