git.delta.rocks / jrsonnet / refs/commits / f8b60736ba96

difftreelog

fix(evaluator) str+num infinite recursion

Лач2020-06-25parent: #0b9856e.patch.diff
in: master

1 file changed

modifiedcrates/jsonnet-evaluator/src/evaluate.rsdiffbeforeafterboth
before · crates/jsonnet-evaluator/src/evaluate.rs
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	ValType,5};6use closure::closure;7use jsonnet_parser::{8	el, Arg, ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember,9	ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,10	Visibility,11};12use std::{13	collections::{BTreeMap, HashMap},14	rc::Rc,15};1617pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, LazyBinding) {18	let b = b.clone();19	if let Some(params) = &b.params {20		let params = params.clone();21		(22			b.name.clone(),23			LazyBinding::Bindable(Rc::new(move |this, super_obj| {24				Ok(lazy_val!(25					closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(26						context_creator.0(this.clone(), super_obj.clone())?,27						params.clone(),28						b.value.clone(),29					)))30				))31			})),32		)33	} else {34		(35			b.name.clone(),36			LazyBinding::Bindable(Rc::new(move |this, super_obj| {37				Ok(lazy_val!(closure!(clone context_creator, clone b, ||38					push(b.value.clone(), "thunk".to_owned(), ||{39						evaluate(40							context_creator.0(this.clone(), super_obj.clone())?,41							&b.value42						)43					})44				)))45			})),46		)47	}48}4950pub fn evaluate_method(ctx: Context, params: ParamsDesc, body: LocExpr) -> Val {51	Val::Func(FuncDesc { ctx, params, body })52}5354pub fn evaluate_field_name(55	context: Context,56	field_name: &jsonnet_parser::FieldName,57) -> Result<Option<String>> {58	Ok(match field_name {59		jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),60		jsonnet_parser::FieldName::Dyn(expr) => {61			let value = evaluate(context, expr)?.unwrap_if_lazy()?;62			if matches!(value, Val::Null) {63				None64			} else {65				Some(value.try_cast_str("dynamic field name")?)66			}67		}68	})69}7071pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {72	Ok(match (op, b) {73		(o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,74		(UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),75		(UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),76		(UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),77		(op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),78	})79}8081pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {82	Ok(match (a, b) {83		(Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),8485		(Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?)),86		(o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s)),8788		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),89		(Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),90		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),91		_ => panic!("can't add: {:?} and {:?}", a, b),92	})93}9495pub fn evaluate_binary_op_special(96	context: Context,97	a: &LocExpr,98	op: BinaryOpType,99	b: &LocExpr,100) -> Result<Val> {101	Ok(102		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {103			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),104			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),105			(a, op, eb) => {106				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?107			}108		},109	)110}111112pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {113	Ok(match (a, op, b) {114		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,115116		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),117118		// Bool X Bool119		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),120		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),121122		// Str X Str123		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),124		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),125		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),126		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),127128		// Num X Num129		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),130		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {131			if *v2 <= f64::EPSILON {132				create_error(crate::Error::DivisionByZero)?133			}134			Val::Num(v1 / v2)135		}136137		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),138139		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),140		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),141		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),142		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),143144		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {145			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)146		}147		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {148			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)149		}150		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {151			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)152		}153		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {154			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)155		}156		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {157			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)158		}159160		_ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),161	})162}163164future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);165future_wrapper!(ObjValue, FutureObjValue);166167#[inline(always)]168pub fn evaluate_comp<T>(169	context: Context,170	value: &impl Fn(Context) -> Result<T>,171	specs: &[CompSpec],172) -> Result<Option<Vec<T>>> {173	Ok(match specs.get(0) {174		None => Some(vec![value(context)?]),175		Some(CompSpec::IfSpec(IfSpecData(cond))) => {176			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {177				evaluate_comp(context, value, &specs[1..])?178			} else {179				None180			}181		}182		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {183			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {184				Val::Arr(list) => {185					let mut out = Vec::new();186					for item in list {187						let item = item.clone().unwrap_if_lazy()?;188						out.push(evaluate_comp(189							context.with_var(var.clone(), item)?,190							value,191							&specs[1..],192						)?);193					}194					Some(out.into_iter().flatten().flatten().collect())195				}196				_ => panic!("for expression evaluated to non-iterable value"),197			}198		}199	})200}201202// TODO: Asserts203pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {204	Ok(match object {205		ObjBody::MemberList(members) => {206			let new_bindings = FutureNewBindings::new();207			let future_this = FutureObjValue::new();208			let context_creator = context_creator!(209				closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {210					Ok(context.clone().extend_unbound(211						new_bindings.clone().unwrap(),212						context.clone().dollar().clone().or_else(||this.clone()),213						Some(this.unwrap()),214						super_obj215					)?)216				})217			);218			{219				let mut bindings: HashMap<String, LazyBinding> = HashMap::new();220				for (n, b) in members221					.iter()222					.filter_map(|m| match m {223						Member::BindStmt(b) => Some(b.clone()),224						_ => None,225					})226					.map(|b| evaluate_binding(&b, context_creator.clone()))227				{228					bindings.insert(n, b);229				}230				new_bindings.fill(bindings);231			}232233			let mut new_members = BTreeMap::new();234			for member in members.into_iter() {235				match member {236					Member::Field(FieldMember {237						name,238						plus,239						params: None,240						visibility,241						value,242					}) => {243						let name = evaluate_field_name(context.clone(), &name)?;244						if name.is_none() {245							continue;246						}247						let name = name.unwrap();248						new_members.insert(249							name.clone(),250							ObjMember {251								add: plus,252								visibility: visibility.clone(),253								invoke: LazyBinding::Bindable(Rc::new(254									closure!(clone name, clone value, clone context_creator, |this, super_obj| {255										Ok(LazyVal::new_resolved(push(value.clone(), "object ".to_owned()+&name+" field", ||{256											let context = context_creator.0(this, super_obj)?;257											evaluate(258												context,259												&value,260											)?.unwrap_if_lazy()261										})?))262									}),263								)),264							},265						);266					}267					Member::Field(FieldMember {268						name,269						params: Some(params),270						value,271						..272					}) => {273						let name = evaluate_field_name(context.clone(), &name)?;274						if name.is_none() {275							continue;276						}277						let name = name.unwrap();278						new_members.insert(279							name,280							ObjMember {281								add: false,282								visibility: Visibility::Hidden,283								invoke: LazyBinding::Bindable(Rc::new(284									closure!(clone value, clone context_creator, |this, super_obj| {285										// TODO: Assert286										Ok(LazyVal::new_resolved(evaluate_method(287											context_creator.0(this, super_obj)?,288											params.clone(),289											value.clone(),290										)))291									}),292								)),293							},294						);295					}296					Member::BindStmt(_) => {}297					Member::AssertStmt(_) => {}298				}299			}300			future_this.fill(ObjValue::new(None, Rc::new(new_members)))301		}302		ObjBody::ObjComp {303			pre_locals,304			key,305			value,306			post_locals,307			compspecs,308		} => {309			let future_this = FutureObjValue::new();310			let mut new_members = BTreeMap::new();311			for (k, v) in evaluate_comp(312				context.clone(),313				&|ctx| {314					let new_bindings = FutureNewBindings::new();315					let context_creator = context_creator!(316						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {317							Ok(context.clone().extend_unbound(318								new_bindings.clone().unwrap(),319								context.clone().dollar().clone().or_else(||this.clone()),320								None,321								super_obj322							)?)323						})324					);325					let mut bindings: HashMap<String, LazyBinding> = HashMap::new();326					for (n, b) in pre_locals327						.iter()328						.chain(post_locals.iter())329						.map(|b| evaluate_binding(b, context_creator.clone()))330					{331						bindings.insert(n, b);332					}333					let bindings = new_bindings.fill(bindings);334					let ctx = ctx.extend_unbound(bindings, None, None, None)?;335					let key = evaluate(ctx.clone(), &key)?;336					let value = LazyBinding::Bindable(Rc::new(337						closure!(clone ctx, clone value, |this, _super_obj| {338							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))339						}),340					));341342					Ok((key, value))343				},344				&compspecs,345			)?346			.unwrap()347			{348				match k {349					Val::Null => {}350					Val::Str(n) => {351						new_members.insert(352							n,353							ObjMember {354								add: false,355								visibility: Visibility::Normal,356								invoke: v,357							},358						);359					}360					v => create_error(Error::FieldMustBeStringGot(v.value_type()?))?,361				}362			}363364			future_this.fill(ObjValue::new(None, Rc::new(new_members)))365		}366	})367}368369#[inline(always)]370pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {371	use Expr::*;372	let locexpr = expr.clone();373	let LocExpr(expr, loc) = expr;374	Ok(match &**expr {375		Literal(LiteralType::This) => Val::Obj(376			context377				.this()378				.clone()379				.unwrap_or_else(|| panic!("this not found")),380		),381		Literal(LiteralType::Dollar) => Val::Obj(382			context383				.dollar()384				.clone()385				.unwrap_or_else(|| panic!("dollar not found")),386		),387		Literal(LiteralType::True) => Val::Bool(true),388		Literal(LiteralType::False) => Val::Bool(false),389		Literal(LiteralType::Null) => Val::Null,390		Parened(e) => evaluate(context, e)?,391		Str(v) => Val::Str(v.clone()),392		Num(v) => Val::Num(*v),393		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,394		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,395		Var(name) => push(locexpr, "var".to_owned(), || {396			Val::Lazy(context.binding(&name)).unwrap_if_lazy()397		})?,398		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {399			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;400			context401				.super_obj()402				.clone()403				.expect("no super found")404				.get_raw(&name, &context.this().clone().expect("no this found"))?405				.expect("value not found")406		}407		Index(value, index) => {408			match (409				evaluate(context.clone(), value)?.unwrap_if_lazy()?,410				evaluate(context, index)?,411			) {412				(Val::Obj(v), Val::Str(s)) => {413					if let Some(v) = v.get(&s)? {414						v.unwrap_if_lazy()?415					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {416						Val::Intristic(n, s)417					} else {418						create_error(crate::Error::NoSuchField(s))?419					}420				}421				(Val::Obj(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(422					ValType::Obj,423					ValType::Str,424					n.value_type()?,425				))?,426427				(Val::Arr(v), Val::Num(n)) => {428					if n.fract() > f64::EPSILON {429						create_error(crate::Error::FractionalIndex)?430					}431					v.get(n as usize)432						.unwrap_or_else(|| panic!("out of bounds"))433						.clone()434						.unwrap_if_lazy()?435				}436				(Val::Arr(_), Val::Str(n)) => {437					create_error(crate::Error::AttemptedIndexAnArrayWithString(n))?438				}439				(Val::Arr(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(440					ValType::Arr,441					ValType::Num,442					n.value_type()?,443				))?,444445				(Val::Str(s), Val::Num(n)) => {446					Val::Str(s.chars().skip(n as usize).take(1).collect())447				}448				(Val::Str(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(449					ValType::Str,450					ValType::Num,451					n.value_type()?,452				))?,453454				(v, _) => create_error(crate::Error::CantIndexInto(v.value_type()?))?,455			}456		}457		LocalExpr(bindings, returned) => {458			let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();459			let future_context = Context::new_future();460461			let context_creator = context_creator!(462				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))463			);464465			for (k, v) in bindings466				.iter()467				.map(|b| evaluate_binding(b, context_creator.clone()))468			{469				new_bindings.insert(k, v);470			}471472			let context = context473				.extend_unbound(new_bindings, None, None, None)?474				.into_future(future_context);475			evaluate(context, &returned.clone())?476		}477		Arr(items) => {478			let mut out = Vec::with_capacity(items.len());479			for item in items {480				out.push(Val::Lazy(lazy_val!(481					closure!(clone context, clone item, || {482						evaluate(context.clone(), &item)483					})484				)));485			}486			Val::Arr(out)487		}488		ArrComp(expr, compspecs) => Val::Arr(489			// First compspec should be forspec, so no "None" possible here490			evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap(),491		),492		Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),493		ObjExtend(s, t) => evaluate_add_op(494			&evaluate(context.clone(), s)?,495			&Val::Obj(evaluate_object(context, t.clone())?),496		)?,497		Apply(value, args, tailstrict) => {498			let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;499			match value {500				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {501					// arr/string/function502					("std", "length") => {503						assert_eq!(args.len(), 1);504						let expr = &args.get(0).unwrap().1;505						match evaluate(context, expr)? {506							Val::Str(n) => Val::Num(n.chars().count() as f64),507							Val::Arr(i) => Val::Num(i.len() as f64),508							Val::Obj(o) => Val::Num(509								o.fields_visibility()510									.into_iter()511									.filter(|(_k, v)| *v)512									.count() as f64,513							),514							v => panic!("can't get length of {:?}", v),515						}516					}517					// any518					("std", "type") => {519						assert_eq!(args.len(), 1);520						let expr = &args.get(0).unwrap().1;521						Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())522					}523					// length, idx=>any524					("std", "makeArray") => {525						assert_eq!(args.len(), 2);526						if let (Val::Num(v), Val::Func(d)) = (527							evaluate(context.clone(), &args[0].1)?,528							evaluate(context, &args[1].1)?,529						) {530							assert!(v >= 0.0);531							let mut out = Vec::with_capacity(v as usize);532							for i in 0..v as usize {533								let call_ctx =534									Context::new().with_var("v".to_owned(), Val::Num(i as f64))?;535								out.push(d.evaluate(536									call_ctx,537									&ArgsDesc(vec![Arg(None, el!(Expr::Var("v".to_owned())))]),538									true,539								)?)540							}541							Val::Arr(out)542						} else {543							panic!("bad makeArray call");544						}545					}546					// string547					("std", "codepoint") => {548						assert_eq!(args.len(), 1);549						if let Val::Str(s) = evaluate(context, &args[0].1)? {550							assert!(551								s.chars().count() == 1,552								"std.codepoint should receive single char string"553							);554							Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)555						} else {556							panic!("bad codepoint call");557						}558					}559					// object, includeHidden560					("std", "objectFieldsEx") => {561						assert_eq!(args.len(), 2);562						if let (Val::Obj(body), Val::Bool(include_hidden)) = (563							evaluate(context.clone(), &args[0].1)?,564							evaluate(context, &args[1].1)?,565						) {566							Val::Arr(567								body.fields_visibility()568									.into_iter()569									.filter(|(_k, v)| *v || include_hidden)570									.map(|(k, _v)| Val::Str(k))571									.collect(),572							)573						} else {574							panic!("bad objectFieldsEx call");575						}576					}577					// object, field, includeHidden578					("std", "objectHasEx") => {579						assert_eq!(args.len(), 3);580						if let (Val::Obj(body), Val::Str(name), Val::Bool(include_hidden)) = (581							evaluate(context.clone(), &args[0].1)?,582							evaluate(context.clone(), &args[1].1)?,583							evaluate(context, &args[2].1)?,584						) {585							Val::Bool(586								body.fields_visibility()587									.into_iter()588									.filter(|(_k, v)| *v || include_hidden)589									.any(|(k, _v)| k == name),590							)591						} else {592							panic!("bad objectHasEx call");593						}594					}595					("std", "primitiveEquals") => {596						assert_eq!(args.len(), 2);597						let (a, b) = (598							evaluate(context.clone(), &args[0].1)?,599							evaluate(context, &args[1].1)?,600						);601						Val::Bool(a == b)602					}603					("std", "modulo") => {604						assert_eq!(args.len(), 2);605						if let (Val::Num(a), Val::Num(b)) = (606							evaluate(context.clone(), &args[0].1)?,607							evaluate(context, &args[1].1)?,608						) {609							Val::Num(a % b)610						} else {611							panic!("bad modulo call");612						}613					}614					("std", "floor") => {615						assert_eq!(args.len(), 1);616						if let Val::Num(a) = evaluate(context, &args[0].1)? {617							Val::Num(a.floor())618						} else {619							panic!("bad floor call");620						}621					}622					("std", "trace") => {623						assert_eq!(args.len(), 2);624						if let (Val::Str(a), b) = (625							evaluate(context.clone(), &args[0].1)?,626							evaluate(context, &args[1].1)?,627						) {628							// TODO: Line numbers as in original jsonnet629							println!("TRACE: {}", a);630							b631						} else {632							panic!("bad trace call");633						}634					}635					("std", "pow") => {636						assert_eq!(args.len(), 2);637						if let (Val::Num(a), Val::Num(b)) = (638							evaluate(context.clone(), &args[0].1)?,639							evaluate(context, &args[1].1)?,640						) {641							Val::Num(a.powf(b))642						} else {643							panic!("bad pow call");644						}645					}646					("std", "extVar") => {647						assert_eq!(args.len(), 1);648						if let Val::Str(a) = evaluate(context, &args[0].1)? {649							with_state(|s| s.0.ext_vars.borrow().get(&a).cloned()).ok_or_else(650								|| {651									create_error::<()>(crate::Error::UndefinedExternalVariable(a))652										.err()653										.unwrap()654								},655							)?656						} else {657							panic!("bad extVar call");658						}659					}660					(ns, name) => panic!("Intristic not found: {}.{}", ns, name),661				},662				Val::Func(f) => {663					let body = #[inline(always)]664					|| f.evaluate(context, args, *tailstrict);665					if *tailstrict {666						body()?667					} else {668						push(locexpr, "function call".to_owned(), body)?669					}670				}671				_ => panic!("{:?} is not a function", value),672			}673		}674		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),675		AssertExpr(AssertStmt(value, msg), returned) => {676			let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {677				evaluate(context.clone(), &value)?678					.try_cast_bool("assertion condition should be boolean")679			})?;680			if assertion_result {681				push(682					returned.clone(),683					"assert 'return' branch".to_owned(),684					|| evaluate(context, returned),685				)?686			} else if let Some(msg) = msg {687				panic!(688					"assertion failed ({:?}): {}",689					value,690					evaluate(context, msg)?.try_cast_str("assertion message should be string")?691				);692			} else {693				panic!("assertion failed ({:?}): no message", value);694			}695		}696		Error(e) => create_error(crate::Error::RuntimeError(697			evaluate(context, e)?.try_cast_str("error text should be string")?,698		))?,699		IfElse {700			cond,701			cond_then,702			cond_else,703		} => {704			let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {705				evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")706			})?;707			if condition_result {708				push(709					cond_then.clone(),710					"if condition 'then' branch".to_owned(),711					|| evaluate(context, cond_then),712				)?713			} else {714				match cond_else {715					Some(v) => evaluate(context, v)?,716					None => Val::Null,717				}718			}719		}720		Import(path) => {721			let mut lib_path = loc722				.clone()723				.expect("imports can't be used without loc_data")724				.0725				.clone();726			lib_path.pop();727			lib_path.push(path);728			with_state(|s| s.import_file(&lib_path))?729		}730		ImportStr(path) => {731			let mut file_path = loc732				.clone()733				.expect("imports can't be used without loc_data")734				.0735				.clone();736			file_path.pop();737			file_path.push(path);738			Val::Str(with_state(|s| s.import_file_str(&file_path))?)739		}740	})741}
after · crates/jsonnet-evaluator/src/evaluate.rs
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	ValType,5};6use closure::closure;7use jsonnet_parser::{8	el, Arg, ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember,9	ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,10	Visibility,11};12use std::{13	collections::{BTreeMap, HashMap},14	rc::Rc,15};1617pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, LazyBinding) {18	let b = b.clone();19	if let Some(params) = &b.params {20		let params = params.clone();21		(22			b.name.clone(),23			LazyBinding::Bindable(Rc::new(move |this, super_obj| {24				Ok(lazy_val!(25					closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(26						context_creator.0(this.clone(), super_obj.clone())?,27						params.clone(),28						b.value.clone(),29					)))30				))31			})),32		)33	} else {34		(35			b.name.clone(),36			LazyBinding::Bindable(Rc::new(move |this, super_obj| {37				Ok(lazy_val!(closure!(clone context_creator, clone b, ||38					push(b.value.clone(), "thunk".to_owned(), ||{39						evaluate(40							context_creator.0(this.clone(), super_obj.clone())?,41							&b.value42						)43					})44				)))45			})),46		)47	}48}4950pub fn evaluate_method(ctx: Context, params: ParamsDesc, body: LocExpr) -> Val {51	Val::Func(FuncDesc { ctx, params, body })52}5354pub fn evaluate_field_name(55	context: Context,56	field_name: &jsonnet_parser::FieldName,57) -> Result<Option<String>> {58	Ok(match field_name {59		jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),60		jsonnet_parser::FieldName::Dyn(expr) => {61			let value = evaluate(context, expr)?.unwrap_if_lazy()?;62			if matches!(value, Val::Null) {63				None64			} else {65				Some(value.try_cast_str("dynamic field name")?)66			}67		}68	})69}7071pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {72	Ok(match (op, b) {73		(o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,74		(UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),75		(UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),76		(UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),77		(op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),78	})79}8081pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {82	Ok(match (a, b) {83		(Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),8485		// Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)86		(Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o)),87		(Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n)),8889		(Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?)),90		(o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s)),9192		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),93		(Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),94		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),95		_ => panic!("can't add: {:?} and {:?}", a, b),96	})97}9899pub fn evaluate_binary_op_special(100	context: Context,101	a: &LocExpr,102	op: BinaryOpType,103	b: &LocExpr,104) -> Result<Val> {105	Ok(106		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {107			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),108			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),109			(a, op, eb) => {110				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?111			}112		},113	)114}115116pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {117	Ok(match (a, op, b) {118		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,119120		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),121122		// Bool X Bool123		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),124		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),125126		// Str X Str127		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),128		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),129		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),130		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),131132		// Num X Num133		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),134		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {135			if *v2 <= f64::EPSILON {136				create_error(crate::Error::DivisionByZero)?137			}138			Val::Num(v1 / v2)139		}140141		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),142143		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),144		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),145		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),146		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),147148		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {149			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)150		}151		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {152			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)153		}154		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {155			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)156		}157		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {158			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)159		}160		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {161			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)162		}163164		_ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),165	})166}167168future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);169future_wrapper!(ObjValue, FutureObjValue);170171#[inline(always)]172pub fn evaluate_comp<T>(173	context: Context,174	value: &impl Fn(Context) -> Result<T>,175	specs: &[CompSpec],176) -> Result<Option<Vec<T>>> {177	Ok(match specs.get(0) {178		None => Some(vec![value(context)?]),179		Some(CompSpec::IfSpec(IfSpecData(cond))) => {180			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {181				evaluate_comp(context, value, &specs[1..])?182			} else {183				None184			}185		}186		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {187			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {188				Val::Arr(list) => {189					let mut out = Vec::new();190					for item in list {191						let item = item.clone().unwrap_if_lazy()?;192						out.push(evaluate_comp(193							context.with_var(var.clone(), item)?,194							value,195							&specs[1..],196						)?);197					}198					Some(out.into_iter().flatten().flatten().collect())199				}200				_ => panic!("for expression evaluated to non-iterable value"),201			}202		}203	})204}205206// TODO: Asserts207pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {208	Ok(match object {209		ObjBody::MemberList(members) => {210			let new_bindings = FutureNewBindings::new();211			let future_this = FutureObjValue::new();212			let context_creator = context_creator!(213				closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {214					Ok(context.clone().extend_unbound(215						new_bindings.clone().unwrap(),216						context.clone().dollar().clone().or_else(||this.clone()),217						Some(this.unwrap()),218						super_obj219					)?)220				})221			);222			{223				let mut bindings: HashMap<String, LazyBinding> = HashMap::new();224				for (n, b) in members225					.iter()226					.filter_map(|m| match m {227						Member::BindStmt(b) => Some(b.clone()),228						_ => None,229					})230					.map(|b| evaluate_binding(&b, context_creator.clone()))231				{232					bindings.insert(n, b);233				}234				new_bindings.fill(bindings);235			}236237			let mut new_members = BTreeMap::new();238			for member in members.into_iter() {239				match member {240					Member::Field(FieldMember {241						name,242						plus,243						params: None,244						visibility,245						value,246					}) => {247						let name = evaluate_field_name(context.clone(), &name)?;248						if name.is_none() {249							continue;250						}251						let name = name.unwrap();252						new_members.insert(253							name.clone(),254							ObjMember {255								add: plus,256								visibility: visibility.clone(),257								invoke: LazyBinding::Bindable(Rc::new(258									closure!(clone name, clone value, clone context_creator, |this, super_obj| {259										Ok(LazyVal::new_resolved(push(value.clone(), "object ".to_owned()+&name+" field", ||{260											let context = context_creator.0(this, super_obj)?;261											evaluate(262												context,263												&value,264											)?.unwrap_if_lazy()265										})?))266									}),267								)),268							},269						);270					}271					Member::Field(FieldMember {272						name,273						params: Some(params),274						value,275						..276					}) => {277						let name = evaluate_field_name(context.clone(), &name)?;278						if name.is_none() {279							continue;280						}281						let name = name.unwrap();282						new_members.insert(283							name,284							ObjMember {285								add: false,286								visibility: Visibility::Hidden,287								invoke: LazyBinding::Bindable(Rc::new(288									closure!(clone value, clone context_creator, |this, super_obj| {289										// TODO: Assert290										Ok(LazyVal::new_resolved(evaluate_method(291											context_creator.0(this, super_obj)?,292											params.clone(),293											value.clone(),294										)))295									}),296								)),297							},298						);299					}300					Member::BindStmt(_) => {}301					Member::AssertStmt(_) => {}302				}303			}304			future_this.fill(ObjValue::new(None, Rc::new(new_members)))305		}306		ObjBody::ObjComp {307			pre_locals,308			key,309			value,310			post_locals,311			compspecs,312		} => {313			let future_this = FutureObjValue::new();314			let mut new_members = BTreeMap::new();315			for (k, v) in evaluate_comp(316				context.clone(),317				&|ctx| {318					let new_bindings = FutureNewBindings::new();319					let context_creator = context_creator!(320						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {321							Ok(context.clone().extend_unbound(322								new_bindings.clone().unwrap(),323								context.clone().dollar().clone().or_else(||this.clone()),324								None,325								super_obj326							)?)327						})328					);329					let mut bindings: HashMap<String, LazyBinding> = HashMap::new();330					for (n, b) in pre_locals331						.iter()332						.chain(post_locals.iter())333						.map(|b| evaluate_binding(b, context_creator.clone()))334					{335						bindings.insert(n, b);336					}337					let bindings = new_bindings.fill(bindings);338					let ctx = ctx.extend_unbound(bindings, None, None, None)?;339					let key = evaluate(ctx.clone(), &key)?;340					let value = LazyBinding::Bindable(Rc::new(341						closure!(clone ctx, clone value, |this, _super_obj| {342							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))343						}),344					));345346					Ok((key, value))347				},348				&compspecs,349			)?350			.unwrap()351			{352				match k {353					Val::Null => {}354					Val::Str(n) => {355						new_members.insert(356							n,357							ObjMember {358								add: false,359								visibility: Visibility::Normal,360								invoke: v,361							},362						);363					}364					v => create_error(Error::FieldMustBeStringGot(v.value_type()?))?,365				}366			}367368			future_this.fill(ObjValue::new(None, Rc::new(new_members)))369		}370	})371}372373#[inline(always)]374pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {375	use Expr::*;376	let locexpr = expr.clone();377	let LocExpr(expr, loc) = expr;378	Ok(match &**expr {379		Literal(LiteralType::This) => Val::Obj(380			context381				.this()382				.clone()383				.unwrap_or_else(|| panic!("this not found")),384		),385		Literal(LiteralType::Dollar) => Val::Obj(386			context387				.dollar()388				.clone()389				.unwrap_or_else(|| panic!("dollar not found")),390		),391		Literal(LiteralType::True) => Val::Bool(true),392		Literal(LiteralType::False) => Val::Bool(false),393		Literal(LiteralType::Null) => Val::Null,394		Parened(e) => evaluate(context, e)?,395		Str(v) => Val::Str(v.clone()),396		Num(v) => Val::Num(*v),397		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,398		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,399		Var(name) => push(locexpr, "var".to_owned(), || {400			Val::Lazy(context.binding(&name)).unwrap_if_lazy()401		})?,402		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {403			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;404			context405				.super_obj()406				.clone()407				.expect("no super found")408				.get_raw(&name, &context.this().clone().expect("no this found"))?409				.expect("value not found")410		}411		Index(value, index) => {412			match (413				evaluate(context.clone(), value)?.unwrap_if_lazy()?,414				evaluate(context, index)?,415			) {416				(Val::Obj(v), Val::Str(s)) => {417					if let Some(v) = v.get(&s)? {418						v.unwrap_if_lazy()?419					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {420						Val::Intristic(n, s)421					} else {422						create_error(crate::Error::NoSuchField(s))?423					}424				}425				(Val::Obj(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(426					ValType::Obj,427					ValType::Str,428					n.value_type()?,429				))?,430431				(Val::Arr(v), Val::Num(n)) => {432					if n.fract() > f64::EPSILON {433						create_error(crate::Error::FractionalIndex)?434					}435					v.get(n as usize)436						.unwrap_or_else(|| panic!("out of bounds"))437						.clone()438						.unwrap_if_lazy()?439				}440				(Val::Arr(_), Val::Str(n)) => {441					create_error(crate::Error::AttemptedIndexAnArrayWithString(n))?442				}443				(Val::Arr(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(444					ValType::Arr,445					ValType::Num,446					n.value_type()?,447				))?,448449				(Val::Str(s), Val::Num(n)) => {450					Val::Str(s.chars().skip(n as usize).take(1).collect())451				}452				(Val::Str(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(453					ValType::Str,454					ValType::Num,455					n.value_type()?,456				))?,457458				(v, _) => create_error(crate::Error::CantIndexInto(v.value_type()?))?,459			}460		}461		LocalExpr(bindings, returned) => {462			let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();463			let future_context = Context::new_future();464465			let context_creator = context_creator!(466				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))467			);468469			for (k, v) in bindings470				.iter()471				.map(|b| evaluate_binding(b, context_creator.clone()))472			{473				new_bindings.insert(k, v);474			}475476			let context = context477				.extend_unbound(new_bindings, None, None, None)?478				.into_future(future_context);479			evaluate(context, &returned.clone())?480		}481		Arr(items) => {482			let mut out = Vec::with_capacity(items.len());483			for item in items {484				out.push(Val::Lazy(lazy_val!(485					closure!(clone context, clone item, || {486						evaluate(context.clone(), &item)487					})488				)));489			}490			Val::Arr(out)491		}492		ArrComp(expr, compspecs) => Val::Arr(493			// First compspec should be forspec, so no "None" possible here494			evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap(),495		),496		Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),497		ObjExtend(s, t) => evaluate_add_op(498			&evaluate(context.clone(), s)?,499			&Val::Obj(evaluate_object(context, t.clone())?),500		)?,501		Apply(value, args, tailstrict) => {502			let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;503			match value {504				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {505					// arr/string/function506					("std", "length") => {507						assert_eq!(args.len(), 1);508						let expr = &args.get(0).unwrap().1;509						match evaluate(context, expr)? {510							Val::Str(n) => Val::Num(n.chars().count() as f64),511							Val::Arr(i) => Val::Num(i.len() as f64),512							Val::Obj(o) => Val::Num(513								o.fields_visibility()514									.into_iter()515									.filter(|(_k, v)| *v)516									.count() as f64,517							),518							v => panic!("can't get length of {:?}", v),519						}520					}521					// any522					("std", "type") => {523						assert_eq!(args.len(), 1);524						let expr = &args.get(0).unwrap().1;525						Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())526					}527					// length, idx=>any528					("std", "makeArray") => {529						assert_eq!(args.len(), 2);530						if let (Val::Num(v), Val::Func(d)) = (531							evaluate(context.clone(), &args[0].1)?,532							evaluate(context, &args[1].1)?,533						) {534							assert!(v >= 0.0);535							let mut out = Vec::with_capacity(v as usize);536							for i in 0..v as usize {537								let call_ctx =538									Context::new().with_var("v".to_owned(), Val::Num(i as f64))?;539								out.push(d.evaluate(540									call_ctx,541									&ArgsDesc(vec![Arg(None, el!(Expr::Var("v".to_owned())))]),542									true,543								)?)544							}545							Val::Arr(out)546						} else {547							panic!("bad makeArray call");548						}549					}550					// string551					("std", "codepoint") => {552						assert_eq!(args.len(), 1);553						if let Val::Str(s) = evaluate(context, &args[0].1)? {554							assert!(555								s.chars().count() == 1,556								"std.codepoint should receive single char string"557							);558							Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)559						} else {560							panic!("bad codepoint call");561						}562					}563					// object, includeHidden564					("std", "objectFieldsEx") => {565						assert_eq!(args.len(), 2);566						if let (Val::Obj(body), Val::Bool(include_hidden)) = (567							evaluate(context.clone(), &args[0].1)?,568							evaluate(context, &args[1].1)?,569						) {570							Val::Arr(571								body.fields_visibility()572									.into_iter()573									.filter(|(_k, v)| *v || include_hidden)574									.map(|(k, _v)| Val::Str(k))575									.collect(),576							)577						} else {578							panic!("bad objectFieldsEx call");579						}580					}581					// object, field, includeHidden582					("std", "objectHasEx") => {583						assert_eq!(args.len(), 3);584						if let (Val::Obj(body), Val::Str(name), Val::Bool(include_hidden)) = (585							evaluate(context.clone(), &args[0].1)?,586							evaluate(context.clone(), &args[1].1)?,587							evaluate(context, &args[2].1)?,588						) {589							Val::Bool(590								body.fields_visibility()591									.into_iter()592									.filter(|(_k, v)| *v || include_hidden)593									.any(|(k, _v)| k == name),594							)595						} else {596							panic!("bad objectHasEx call");597						}598					}599					("std", "primitiveEquals") => {600						assert_eq!(args.len(), 2);601						let (a, b) = (602							evaluate(context.clone(), &args[0].1)?,603							evaluate(context, &args[1].1)?,604						);605						Val::Bool(a == b)606					}607					("std", "modulo") => {608						assert_eq!(args.len(), 2);609						if let (Val::Num(a), Val::Num(b)) = (610							evaluate(context.clone(), &args[0].1)?,611							evaluate(context, &args[1].1)?,612						) {613							Val::Num(a % b)614						} else {615							panic!("bad modulo call");616						}617					}618					("std", "floor") => {619						assert_eq!(args.len(), 1);620						if let Val::Num(a) = evaluate(context, &args[0].1)? {621							Val::Num(a.floor())622						} else {623							panic!("bad floor call");624						}625					}626					("std", "trace") => {627						assert_eq!(args.len(), 2);628						if let (Val::Str(a), b) = (629							evaluate(context.clone(), &args[0].1)?,630							evaluate(context, &args[1].1)?,631						) {632							// TODO: Line numbers as in original jsonnet633							println!("TRACE: {}", a);634							b635						} else {636							panic!("bad trace call");637						}638					}639					("std", "pow") => {640						assert_eq!(args.len(), 2);641						if let (Val::Num(a), Val::Num(b)) = (642							evaluate(context.clone(), &args[0].1)?,643							evaluate(context, &args[1].1)?,644						) {645							Val::Num(a.powf(b))646						} else {647							panic!("bad pow call");648						}649					}650					("std", "extVar") => {651						assert_eq!(args.len(), 1);652						if let Val::Str(a) = evaluate(context, &args[0].1)? {653							with_state(|s| s.0.ext_vars.borrow().get(&a).cloned()).ok_or_else(654								|| {655									create_error::<()>(crate::Error::UndefinedExternalVariable(a))656										.err()657										.unwrap()658								},659							)?660						} else {661							panic!("bad extVar call");662						}663					}664					(ns, name) => panic!("Intristic not found: {}.{}", ns, name),665				},666				Val::Func(f) => {667					let body = #[inline(always)]668					|| f.evaluate(context, args, *tailstrict);669					if *tailstrict {670						body()?671					} else {672						push(locexpr, "function call".to_owned(), body)?673					}674				}675				_ => panic!("{:?} is not a function", value),676			}677		}678		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),679		AssertExpr(AssertStmt(value, msg), returned) => {680			let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {681				evaluate(context.clone(), &value)?682					.try_cast_bool("assertion condition should be boolean")683			})?;684			if assertion_result {685				push(686					returned.clone(),687					"assert 'return' branch".to_owned(),688					|| evaluate(context, returned),689				)?690			} else if let Some(msg) = msg {691				panic!(692					"assertion failed ({:?}): {}",693					value,694					evaluate(context, msg)?.try_cast_str("assertion message should be string")?695				);696			} else {697				panic!("assertion failed ({:?}): no message", value);698			}699		}700		Error(e) => create_error(crate::Error::RuntimeError(701			evaluate(context, e)?.try_cast_str("error text should be string")?,702		))?,703		IfElse {704			cond,705			cond_then,706			cond_else,707		} => {708			let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {709				evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")710			})?;711			if condition_result {712				push(713					cond_then.clone(),714					"if condition 'then' branch".to_owned(),715					|| evaluate(context, cond_then),716				)?717			} else {718				match cond_else {719					Some(v) => evaluate(context, v)?,720					None => Val::Null,721				}722			}723		}724		Import(path) => {725			let mut lib_path = loc726				.clone()727				.expect("imports can't be used without loc_data")728				.0729				.clone();730			lib_path.pop();731			lib_path.push(path);732			with_state(|s| s.import_file(&lib_path))?733		}734		ImportStr(path) => {735			let mut file_path = loc736				.clone()737				.expect("imports can't be used without loc_data")738				.0739				.clone();740			file_path.pop();741			file_path.push(path);742			Val::Str(with_state(|s| s.import_file_str(&file_path))?)743		}744	})745}