feat: use hashbrown

src/vm/env.rs

@@ -1,13 +1,12 @@
 use std::rc::Rc;
-
-use rpds::HashTrieMap;
+use hashbrown::HashMap;
 
 use crate::ty::internal::{AttrSet, Value};
 
 #[derive(Debug, Default, Clone)]
 pub struct Env<'vm> {
     last: Option<Rc<Env<'vm>>>,
-    map: HashTrieMap<usize, Value<'vm>>,
+    map: Rc<HashMap<usize, Value<'vm>>>,
 }
 
 impl<'vm> Env<'vm> {
@@ -16,18 +15,22 @@ impl<'vm> Env<'vm> {
     }
 
     pub fn lookup(&self, symbol: usize) -> Option<Value<'vm>> {
-        self.map.get(&symbol).cloned()
+        if let Some(val) = self.map.get(&symbol).cloned() {
+            return Some(val)
+        }
+        self.last.as_ref().map(|env| env.lookup(symbol)).flatten()
     }
 
     pub fn insert(&mut self, symbol: usize, value: Value<'vm>) {
-        self.map.insert_mut(symbol, value);
+        Rc::make_mut(&mut self.map).insert(symbol, value);
     }
 
     pub fn enter(self, new: impl Iterator<Item = (usize, Value<'vm>)>) -> Self {
-        let mut map = self.map.clone();
+        /* let mut map = self.map.clone();
         for (k, v) in new {
             map.insert_mut(k, v);
-        }
+        } */
+        let map = Rc::new(new.collect());
         let last = Some(
             Env {
                 last: self.last,
@@ -39,7 +42,7 @@ impl<'vm> Env<'vm> {
     }
 
     pub fn enter_with(self, new: Rc<AttrSet<'vm>>) -> Self {
-        let mut map = self.map.clone();
+        /* let mut map = self.map.clone();
         for (k, v) in new.as_inner().iter() {
             let v = if let Value::Builtins = v {
                 Value::AttrSet(new.clone())
@@ -47,7 +50,14 @@ impl<'vm> Env<'vm> {
                 v.clone()
             };
             map.insert_mut(k.clone(), v);
-        }
+        } */
+        let map = Rc::new(new.as_inner().iter().map(|(&k, v)| {
+            (k, if let Value::Builtins = v {
+                Value::AttrSet(new.clone())
+            } else {
+                v.clone()
+            })
+        }).collect());
         let last = Some(
             Env {
                 last: self.last.clone(),