1use crate::{2 future_wrapper, map::LayeredHashMap, rc_fn_helper, resolved_lazy_val, LazyBinding, LazyVal,3 ObjValue, Result, Val,4};5use std::{cell::RefCell, collections::HashMap, fmt::Debug, rc::Rc};67rc_fn_helper!(8 ContextCreator,9 context_creator,10 dyn Fn(Option<ObjValue>, Option<ObjValue>) -> Result<Context>11);1213future_wrapper!(Context, FutureContext);1415struct ContextInternals {16 dollar: Option<ObjValue>,17 this: Option<ObjValue>,18 super_obj: Option<ObjValue>,19 bindings: LayeredHashMap<String, LazyVal>,20}21pub struct Context(Rc<ContextInternals>);22impl Debug for Context {23 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {24 f.debug_struct("Context")25 .field("this", &self.0.this.clone().map(|e| Rc::as_ptr(&e.0)))26 .finish()27 }28}29impl Context {30 pub fn new_future() -> FutureContext {31 FutureContext(Rc::new(RefCell::new(None)))32 }3334 pub fn dollar(&self) -> &Option<ObjValue> {35 &self.0.dollar36 }3738 pub fn this(&self) -> &Option<ObjValue> {39 &self.0.this40 }4142 pub fn super_obj(&self) -> &Option<ObjValue> {43 &self.0.super_obj44 }4546 pub fn new() -> Context {47 Context(Rc::new(ContextInternals {48 dollar: None,49 this: None,50 super_obj: None,51 bindings: LayeredHashMap::default(),52 }))53 }5455 pub fn binding(&self, name: &str) -> Result<LazyVal> {56 self.0.bindings.get(name).cloned().ok_or_else(|| {57 create_error::<()>(Error::UnknownVariable(name.to_owned()))58 .err()59 .unwrap()60 })61 }62 pub fn into_future(self, ctx: FutureContext) -> Context {63 {64 ctx.0.borrow_mut().replace(self);65 }66 ctx.unwrap()67 }6869 pub fn with_var(&self, name: String, value: Val) -> Result<Context> {70 let mut new_bindings = HashMap::with_capacity(1);71 new_bindings.insert(name, resolved_lazy_val!(value));72 self.extend(new_bindings, None, None, None)73 }7475 pub fn extend(76 &self,77 new_bindings: HashMap<String, LazyVal>,78 new_dollar: Option<ObjValue>,79 new_this: Option<ObjValue>,80 new_super_obj: Option<ObjValue>,81 ) -> Result<Context> {82 let dollar = new_dollar.or_else(|| self.0.dollar.clone());83 let this = new_this.or_else(|| self.0.this.clone());84 let super_obj = new_super_obj.or_else(|| self.0.super_obj.clone());85 let bindings = if new_bindings.is_empty() {86 self.0.bindings.clone()87 } else {88 self.0.bindings.extend(new_bindings)89 };90 Ok(Context(Rc::new(ContextInternals {91 dollar,92 this,93 super_obj,94 bindings,95 })))96 }97 pub fn extend_unbound(98 &self,99 new_bindings: HashMap<String, LazyBinding>,100 new_dollar: Option<ObjValue>,101 new_this: Option<ObjValue>,102 new_super_obj: Option<ObjValue>,103 ) -> Result<Context> {104 let this = new_this.or_else(|| self.0.this.clone());105 let super_obj = new_super_obj.or_else(|| self.0.super_obj.clone());106 let mut new = HashMap::with_capacity(new_bindings.len());107 for (k, v) in new_bindings.into_iter() {108 new.insert(k, v.evaluate(this.clone(), super_obj.clone())?);109 }110 self.extend(new, new_dollar, this, super_obj)111 }112}113114impl Default for Context {115 fn default() -> Self {116 Self::new()117 }118}119120impl PartialEq for Context {121 fn eq(&self, other: &Self) -> bool {122 Rc::ptr_eq(&self.0, &other.0)123 }124}125126impl Clone for Context {127 fn clone(&self) -> Self {128 Context(self.0.clone())129 }130}