git.delta.rocks / jrsonnet / refs/commits / 9f3d17001fc1

1use crate::{2	context_creator, create_error, future_wrapper, lazy_val, push, with_state, Context,3	ContextCreator, Error, FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue, Result, Val,4};5use closure::closure;6use jsonnet_parser::{7	el, Arg, ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember,8	ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,9	Visibility,10};11use std::{12	collections::{BTreeMap, HashMap},13	rc::Rc,14};1516pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, LazyBinding) {17	let b = b.clone();18	if let Some(params) = &b.params {19		let params = params.clone();20		(21			b.name.clone(),22			LazyBinding::Bindable(Rc::new(move |this, super_obj| {23				Ok(lazy_val!(24					closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(25						context_creator.0(this.clone(), super_obj.clone())?,26						params.clone(),27						b.value.clone(),28					)))29				))30			})),31		)32	} else {33		(34			b.name.clone(),35			LazyBinding::Bindable(Rc::new(move |this, super_obj| {36				Ok(lazy_val!(closure!(clone context_creator, clone b, ||37					push(b.value.clone(), "thunk".to_owned(), ||{38						evaluate(39							context_creator.0(this.clone(), super_obj.clone())?,40							&b.value41						)42					})43				)))44			})),45		)46	}47}4849pub fn evaluate_method(ctx: Context, params: ParamsDesc, body: LocExpr) -> Val {50	Val::Func(FuncDesc { ctx, params, body })51}5253pub fn evaluate_field_name(54	context: Context,55	field_name: &jsonnet_parser::FieldName,56) -> Result<Option<String>> {57	Ok(match field_name {58		jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),59		jsonnet_parser::FieldName::Dyn(expr) => {60			let value = evaluate(context, expr)?.unwrap_if_lazy()?;61			if matches!(value, Val::Null) {62				None63			} else {64				Some(value.try_cast_str("dynamic field name")?)65			}66		}67	})68}6970pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {71	Ok(match (op, b) {72		(o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,73		(UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),74		(UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),75		(UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),76		(op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),77	})78}7980pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {81	Ok(match (a, b) {82		(Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),8384		(Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),85		(Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),86		(Val::Str(v1), Val::Bool(v2)) => Val::Str(format!("{}{}", v1, v2)),87		(Val::Bool(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),88		(Val::Str(v1), Val::Null) => Val::Str(format!("{}null", v1)),89		(Val::Null, Val::Str(v2)) => Val::Str(format!("null{}", v2)),9091		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),92		(Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),93		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),94		_ => panic!("can't add: {:?} and {:?}", a, b),95	})96}9798pub fn evaluate_binary_op_special(99	context: Context,100	a: &LocExpr,101	op: BinaryOpType,102	b: &LocExpr,103) -> Result<Val> {104	Ok(105		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {106			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),107			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),108			(a, op, eb) => {109				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?110			}111		},112	)113}114115pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {116	Ok(match (a, op, b) {117		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,118119		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),120121		// Bool X Bool122		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),123		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),124125		// Str X Str126		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),127		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),128		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),129		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),130131		// Num X Num132		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),133		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {134			if *v2 <= f64::EPSILON {135				create_error(crate::Error::DivisionByZero)?136			}137			Val::Num(v1 / v2)138		}139140		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),141142		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),143		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),144		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),145		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),146147		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {148			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)149		}150		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {151			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)152		}153		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {154			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)155		}156		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {157			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)158		}159		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {160			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)161		}162163		_ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),164	})165}166167future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);168future_wrapper!(ObjValue, FutureObjValue);169170#[inline(always)]171pub fn evaluate_comp<T>(172	context: Context,173	value: &impl Fn(Context) -> Result<T>,174	specs: &[CompSpec],175) -> Result<Option<Vec<T>>> {176	Ok(match specs.get(0) {177		None => Some(vec![value(context)?]),178		Some(CompSpec::IfSpec(IfSpecData(cond))) => {179			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {180				evaluate_comp(context, value, &specs[1..])?181			} else {182				None183			}184		}185		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {186			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {187				Val::Arr(list) => {188					let mut out = Vec::new();189					for item in list {190						let item = item.clone().unwrap_if_lazy()?;191						out.push(evaluate_comp(192							context.with_var(var.clone(), item)?,193							value,194							&specs[1..],195						)?);196					}197					Some(out.into_iter().flatten().flatten().collect())198				}199				_ => panic!("for expression evaluated to non-iterable value"),200			}201		}202	})203}204205// TODO: Asserts206pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {207	Ok(match object {208		ObjBody::MemberList(members) => {209			let new_bindings = FutureNewBindings::new();210			let future_this = FutureObjValue::new();211			let context_creator = context_creator!(212				closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {213					Ok(context.clone().extend_unbound(214						new_bindings.clone().unwrap(),215						context.clone().dollar().clone().or_else(||this.clone()),216						Some(this.unwrap()),217						super_obj218					)?)219				})220			);221			{222				let mut bindings: HashMap<String, LazyBinding> = HashMap::new();223				for (n, b) in members224					.iter()225					.filter_map(|m| match m {226						Member::BindStmt(b) => Some(b.clone()),227						_ => None,228					})229					.map(|b| evaluate_binding(&b, context_creator.clone()))230				{231					bindings.insert(n, b);232				}233				new_bindings.fill(bindings);234			}235236			let mut new_members = BTreeMap::new();237			for member in members.into_iter() {238				match member {239					Member::Field(FieldMember {240						name,241						plus,242						params: None,243						visibility,244						value,245					}) => {246						let name = evaluate_field_name(context.clone(), &name)?;247						if name.is_none() {248							continue;249						}250						let name = name.unwrap();251						new_members.insert(252							name.clone(),253							ObjMember {254								add: plus,255								visibility: visibility.clone(),256								invoke: LazyBinding::Bindable(Rc::new(257									closure!(clone name, clone value, clone context_creator, |this, super_obj| {258										Ok(LazyVal::new_resolved(push(value.clone(), "object ".to_owned()+&name+" field", ||{259											let context = context_creator.0(this, super_obj)?;260											evaluate(261												context,262												&value,263											)?.unwrap_if_lazy()264										})?))265									}),266								)),267							},268						);269					}270					Member::Field(FieldMember {271						name,272						params: Some(params),273						value,274						..275					}) => {276						let name = evaluate_field_name(context.clone(), &name)?;277						if name.is_none() {278							continue;279						}280						let name = name.unwrap();281						new_members.insert(282							name,283							ObjMember {284								add: false,285								visibility: Visibility::Hidden,286								invoke: LazyBinding::Bindable(Rc::new(287									closure!(clone value, clone context_creator, |this, super_obj| {288										// TODO: Assert289										Ok(LazyVal::new_resolved(evaluate_method(290											context_creator.0(this, super_obj)?,291											params.clone(),292											value.clone(),293										)))294									}),295								)),296							},297						);298					}299					Member::BindStmt(_) => {}300					Member::AssertStmt(_) => {}301				}302			}303			future_this.fill(ObjValue::new(None, Rc::new(new_members)))304		}305		ObjBody::ObjComp {306			pre_locals,307			key,308			value,309			post_locals,310			compspecs,311		} => {312			let future_this = FutureObjValue::new();313			let mut new_members = BTreeMap::new();314			for (k, v) in evaluate_comp(315				context.clone(),316				&|ctx| {317					let new_bindings = FutureNewBindings::new();318					let context_creator = context_creator!(319						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {320							Ok(context.clone().extend_unbound(321								new_bindings.clone().unwrap(),322								context.clone().dollar().clone().or_else(||this.clone()),323								None,324								super_obj325							)?)326						})327					);328					let mut bindings: HashMap<String, LazyBinding> = HashMap::new();329					for (n, b) in pre_locals330						.iter()331						.chain(post_locals.iter())332						.map(|b| evaluate_binding(b, context_creator.clone()))333					{334						bindings.insert(n, b);335					}336					let bindings = new_bindings.fill(bindings);337					let ctx = ctx.extend_unbound(bindings, None, None, None)?;338					let key = evaluate(ctx.clone(), &key)?;339					let value = LazyBinding::Bindable(Rc::new(340						closure!(clone ctx, clone value, |this, _super_obj| {341							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))342						}),343					));344345					Ok((key, value))346				},347				&compspecs,348			)?349			.unwrap()350			{351				match k {352					Val::Null => {}353					Val::Str(n) => {354						new_members.insert(355							n,356							ObjMember {357								add: false,358								visibility: Visibility::Normal,359								invoke: v,360							},361						);362					}363					v => create_error(Error::FieldMustBeStringGot(v.value_type()?))?,364				}365			}366367			future_this.fill(ObjValue::new(None, Rc::new(new_members)))368		}369	})370}371372#[inline(always)]373pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {374	use Expr::*;375	let locexpr = expr.clone();376	let LocExpr(expr, loc) = expr;377	Ok(match &**expr {378		Literal(LiteralType::This) => Val::Obj(379			context380				.this()381				.clone()382				.unwrap_or_else(|| panic!("this not found")),383		),384		Literal(LiteralType::Dollar) => Val::Obj(385			context386				.dollar()387				.clone()388				.unwrap_or_else(|| panic!("dollar not found")),389		),390		Literal(LiteralType::True) => Val::Bool(true),391		Literal(LiteralType::False) => Val::Bool(false),392		Literal(LiteralType::Null) => Val::Null,393		Parened(e) => evaluate(context, e)?,394		Str(v) => Val::Str(v.clone()),395		Num(v) => Val::Num(*v),396		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,397		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,398		Var(name) => push(locexpr, "var".to_owned(), || {399			Val::Lazy(context.binding(&name)).unwrap_if_lazy()400		})?,401		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {402			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;403			context404				.super_obj()405				.clone()406				.expect("no super found")407				.get_raw(&name, &context.this().clone().expect("no this found"))?408				.expect("value not found")409		}410		Index(value, index) => {411			match (412				evaluate(context.clone(), value)?.unwrap_if_lazy()?,413				evaluate(context, index)?,414			) {415				(Val::Obj(v), Val::Str(s)) => {416					if let Some(v) = v.get(&s)? {417						v.unwrap_if_lazy()?418					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {419						Val::Intristic(n, s)420					} else {421						create_error(crate::Error::NoSuchField(s))?422					}423				}424				(Val::Arr(v), Val::Num(n)) => {425					if n.fract() > f64::EPSILON {426						create_error(crate::Error::FractionalIndex)?427					}428					v.get(n as usize)429						.unwrap_or_else(|| panic!("out of bounds"))430						.clone()431						.unwrap_if_lazy()?432				}433				(Val::Str(s), Val::Num(n)) => {434					Val::Str(s.chars().skip(n as usize).take(1).collect())435				}436				(v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),437			}438		}439		LocalExpr(bindings, returned) => {440			let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();441			let future_context = Context::new_future();442443			let context_creator = context_creator!(444				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))445			);446447			for (k, v) in bindings448				.iter()449				.map(|b| evaluate_binding(b, context_creator.clone()))450			{451				new_bindings.insert(k, v);452			}453454			let context = context455				.extend_unbound(new_bindings, None, None, None)?456				.into_future(future_context);457			evaluate(context, &returned.clone())?458		}459		Arr(items) => {460			let mut out = Vec::with_capacity(items.len());461			for item in items {462				out.push(Val::Lazy(lazy_val!(463					closure!(clone context, clone item, || {464						evaluate(context.clone(), &item)465					})466				)));467			}468			Val::Arr(out)469		}470		ArrComp(expr, compspecs) => Val::Arr(471			// First compspec should be forspec, so no "None" possible here472			evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap(),473		),474		Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),475		ObjExtend(s, t) => evaluate_add_op(476			&evaluate(context.clone(), s)?,477			&Val::Obj(evaluate_object(context, t.clone())?),478		)?,479		Apply(value, args, tailstrict) => {480			let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;481			match value {482				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {483					// arr/string/function484					("std", "length") => {485						assert_eq!(args.len(), 1);486						let expr = &args.get(0).unwrap().1;487						match evaluate(context, expr)? {488							Val::Str(n) => Val::Num(n.chars().count() as f64),489							Val::Arr(i) => Val::Num(i.len() as f64),490							Val::Obj(o) => Val::Num(491								o.fields_visibility()492									.into_iter()493									.filter(|(_k, v)| *v)494									.count() as f64,495							),496							v => panic!("can't get length of {:?}", v),497						}498					}499					// any500					("std", "type") => {501						assert_eq!(args.len(), 1);502						let expr = &args.get(0).unwrap().1;503						Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())504					}505					// length, idx=>any506					("std", "makeArray") => {507						assert_eq!(args.len(), 2);508						if let (Val::Num(v), Val::Func(d)) = (509							evaluate(context.clone(), &args[0].1)?,510							evaluate(context, &args[1].1)?,511						) {512							assert!(v >= 0.0);513							let mut out = Vec::with_capacity(v as usize);514							for i in 0..v as usize {515								let call_ctx =516									Context::new().with_var("v".to_owned(), Val::Num(i as f64))?;517								out.push(d.evaluate(518									call_ctx,519									&ArgsDesc(vec![Arg(None, el!(Expr::Var("v".to_owned())))]),520									true,521								)?)522							}523							Val::Arr(out)524						} else {525							panic!("bad makeArray call");526						}527					}528					// string529					("std", "codepoint") => {530						assert_eq!(args.len(), 1);531						if let Val::Str(s) = evaluate(context, &args[0].1)? {532							assert!(533								s.chars().count() == 1,534								"std.codepoint should receive single char string"535							);536							Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)537						} else {538							panic!("bad codepoint call");539						}540					}541					// object, includeHidden542					("std", "objectFieldsEx") => {543						assert_eq!(args.len(), 2);544						if let (Val::Obj(body), Val::Bool(include_hidden)) = (545							evaluate(context.clone(), &args[0].1)?,546							evaluate(context, &args[1].1)?,547						) {548							Val::Arr(549								body.fields_visibility()550									.into_iter()551									.filter(|(_k, v)| *v || include_hidden)552									.map(|(k, _v)| Val::Str(k))553									.collect(),554							)555						} else {556							panic!("bad objectFieldsEx call");557						}558					}559					("std", "primitiveEquals") => {560						assert_eq!(args.len(), 2);561						let (a, b) = (562							evaluate(context.clone(), &args[0].1)?,563							evaluate(context, &args[1].1)?,564						);565						Val::Bool(a == b)566					}567					("std", "modulo") => {568						assert_eq!(args.len(), 2);569						if let (Val::Num(a), Val::Num(b)) = (570							evaluate(context.clone(), &args[0].1)?,571							evaluate(context, &args[1].1)?,572						) {573							Val::Num(a % b)574						} else {575							panic!("bad modulo call");576						}577					}578					("std", "floor") => {579						assert_eq!(args.len(), 1);580						if let Val::Num(a) = evaluate(context, &args[0].1)? {581							Val::Num(a.floor())582						} else {583							panic!("bad floor call");584						}585					}586					("std", "trace") => {587						assert_eq!(args.len(), 2);588						if let (Val::Str(a), b) = (589							evaluate(context.clone(), &args[0].1)?,590							evaluate(context, &args[1].1)?,591						) {592							// TODO: Line numbers as in original jsonnet593							println!("TRACE: {}", a);594							b595						} else {596							panic!("bad trace call");597						}598					}599					("std", "pow") => {600						assert_eq!(args.len(), 2);601						if let (Val::Num(a), Val::Num(b)) = (602							evaluate(context.clone(), &args[0].1)?,603							evaluate(context, &args[1].1)?,604						) {605							Val::Num(a.powf(b))606						} else {607							panic!("bad pow call");608						}609					}610					("std", "extVar") => {611						assert_eq!(args.len(), 1);612						if let Val::Str(a) = evaluate(context, &args[0].1)? {613							with_state(|s| s.0.ext_vars.borrow().get(&a).cloned()).ok_or_else(614								|| {615									create_error::<()>(crate::Error::UndefinedExternalVariable(a))616										.err()617										.unwrap()618								},619							)?620						} else {621							panic!("bad extVar call");622						}623					}624					(ns, name) => panic!("Intristic not found: {}.{}", ns, name),625				},626				Val::Func(f) => {627					let body = #[inline(always)]628					|| f.evaluate(context, args, *tailstrict);629					if *tailstrict {630						body()?631					} else {632						push(locexpr, "function call".to_owned(), body)?633					}634				}635				_ => panic!("{:?} is not a function", value),636			}637		}638		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),639		AssertExpr(AssertStmt(value, msg), returned) => {640			let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {641				evaluate(context.clone(), &value)?642					.try_cast_bool("assertion condition should be boolean")643			})?;644			if assertion_result {645				push(646					returned.clone(),647					"assert 'return' branch".to_owned(),648					|| evaluate(context, returned),649				)?650			} else if let Some(msg) = msg {651				panic!(652					"assertion failed ({:?}): {}",653					value,654					evaluate(context, msg)?.try_cast_str("assertion message should be string")?655				);656			} else {657				panic!("assertion failed ({:?}): no message", value);658			}659		}660		Error(e) => create_error(crate::Error::RuntimeError(661			evaluate(context, e)?.try_cast_str("error text should be string")?,662		))?,663		IfElse {664			cond,665			cond_then,666			cond_else,667		} => {668			let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {669				evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")670			})?;671			if condition_result {672				push(673					cond_then.clone(),674					"if condition 'then' branch".to_owned(),675					|| evaluate(context, cond_then),676				)?677			} else {678				match cond_else {679					Some(v) => evaluate(context, v)?,680					None => Val::Null,681				}682			}683		}684		Import(path) => {685			let mut lib_path = loc686				.clone()687				.expect("imports can't be used without loc_data")688				.0689				.clone();690			lib_path.pop();691			lib_path.push(path);692			with_state(|s| s.import_file(&lib_path))?693		}694		_ => panic!(695			"evaluation not implemented: {:?}",696			LocExpr(expr.clone(), loc.clone())697		),698	})699}