feat: Add typed lambda registry to Optimizer

The Optimizer now accepts a typed lambda registry, allowing it to
perform more aggressive inlining and beta-reduction on globally defined
functions. This is a significant step towards optimizing recursive and
globally defined lambdas more effectively.

src/ast/environment.rs

@@ -24,6 +24,7 @@ pub struct Environment {
     pub global_purity: Rc<RefCell<HashMap<u32, bool>>>,
     pub global_values: Rc<RefCell<Vec<Value>>>,
     pub function_registry: Rc<RefCell<HashMap<u32, BoundNode>>>,
+    pub typed_function_registry: Rc<RefCell<HashMap<u32, TypedNode>>>,
     pub monomorph_cache: Rc<RefCell<MonoCache>>,
     pub debug_mode: bool,
     pub optimization: bool,
@@ -89,6 +90,7 @@ impl Environment {
             global_purity: Rc::new(RefCell::new(HashMap::new())),
             global_values: Rc::new(RefCell::new(Vec::new())),
             function_registry: Rc::new(RefCell::new(HashMap::new())),
+            typed_function_registry: Rc::new(RefCell::new(HashMap::new())),
             monomorph_cache: Rc::new(RefCell::new(HashMap::new())),
             debug_mode: false,
             optimization: true,
@@ -180,6 +182,9 @@ impl Environment {
         let checker = TypeChecker::new(self.global_types.clone());
         let typed_ast = checker.check(bound_ast, &[])?;
 
+        // 7. Collect Typed Lambdas
+        LambdaCollector::collect(&typed_ast, &mut self.typed_function_registry.borrow_mut());
+
         Ok(typed_ast)
     }
 
@@ -191,7 +196,8 @@ impl Environment {
         // 2. Optimize (Level 1: Cracking, Level 2: Collapsing)
         let optimizer = Optimizer::new(self.optimization)
             .with_globals(self.global_values.clone())
-            .with_purity(self.global_purity.clone());
+            .with_purity(self.global_purity.clone())
+            .with_registry(self.typed_function_registry.clone());
         let optimized = optimizer.optimize(specialized);
 
         // 3. TCO (Always performed, converts to ExecNode)