Merge pull request #5059 from stacks-network/clarity-wasm-develop-typemap-patch

add methods to TypeMap

Author: krl

clarity/src/vm/analysis/type_checker/contexts.rs

@@ -86,6 +86,20 @@ impl TypeMap {
         }
     }
 
+    /// Like set_type but forcing a change if already set
+    pub fn overwrite_type(&mut self, expr: &SymbolicExpression, type_sig: TypeSignature) {
+        if let TypeMapDataType::Map(ref mut map) = self.map {
+            map.insert(expr.id, type_sig);
+        }
+    }
+
+    pub fn get_type(&self, expr: &SymbolicExpression) -> Option<&TypeSignature> {
+        match self.map {
+            TypeMapDataType::Map(ref map) => map.get(&expr.id),
+            _ => None,
+        }
+    }
+
     pub fn get_type_expected(&self, expr: &SymbolicExpression) -> Option<&TypeSignature> {
         match self.map {
             TypeMapDataType::Map(ref map) => map.get(&expr.id),