git.delta.rocks / jrsonnet / refs/commits / 8dc1868a104f

difftreelog

refactor(evaluator) extract Apply evaluation

Лач2020-06-28parent: #be6bedc.patch.diff
in: master

2 files changed

modifiedcrates/jrsonnet-evaluator/src/evaluate.rsdiffbeforeafterboth
before · crates/jrsonnet-evaluator/src/evaluate.rs
1use crate::{2	context_creator, create_error, create_error_result, escape_string_json, future_wrapper,3	lazy_val, manifest_json_ex, parse_args, push, with_state, Context, ContextCreator, Error,4	FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue, Result, Val, ValType,5};6use closure::closure;7use jrsonnet_parser::{8	AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember, ForSpecData, IfSpecData,9	LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType, Visibility,10};11use std::{12	collections::{BTreeMap, HashMap},13	rc::Rc,14};1516pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (Rc<str>, 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.1, "thunk", ||{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: &jrsonnet_parser::FieldName,56) -> Result<Option<Rc<str>>> {57	Ok(match field_name {58		jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),59		jrsonnet_parser::FieldName::Dyn(expr) => {60			let lazy = evaluate(context, expr)?;61			let value = lazy.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) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(78			op,79			o.value_type()?,80		))?,81	})82}8384pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {85	Ok(match (a, b) {86		(Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),8788		// Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)89		(Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),90		(Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),9192		(Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),93		(o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9495		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),96		(Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),97		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),98		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(99			BinaryOpType::Add,100			a.value_type()?,101			b.value_type()?,102		))?,103	})104}105106pub fn evaluate_binary_op_special(107	context: Context,108	a: &LocExpr,109	op: BinaryOpType,110	b: &LocExpr,111) -> Result<Val> {112	Ok(113		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {114			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),115			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),116			(a, op, eb) => {117				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?118			}119		},120	)121}122123pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {124	Ok(match (a, op, b) {125		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,126127		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),128129		// Bool X Bool130		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),131		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),132133		// Str X Str134		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),135		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),136		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),137		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),138139		// Num X Num140		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),141		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {142			if *v2 <= f64::EPSILON {143				create_error_result(crate::Error::DivisionByZero)?144			}145			Val::Num(v1 / v2)146		}147148		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),149150		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),151		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),152		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),153		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),154155		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {156			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)157		}158		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {159			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)160		}161		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {162			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)163		}164		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {165			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)166		}167		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {168			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)169		}170171		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(172			op,173			a.value_type()?,174			b.value_type()?,175		))?,176	})177}178179future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);180future_wrapper!(ObjValue, FutureObjValue);181182pub fn evaluate_comp<T>(183	context: Context,184	value: &impl Fn(Context) -> Result<T>,185	specs: &[CompSpec],186) -> Result<Option<Vec<T>>> {187	Ok(match specs.get(0) {188		None => Some(vec![value(context)?]),189		Some(CompSpec::IfSpec(IfSpecData(cond))) => {190			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {191				evaluate_comp(context, value, &specs[1..])?192			} else {193				None194			}195		}196		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {197			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {198				Val::Arr(list) => {199					let mut out = Vec::new();200					for item in list.iter() {201						let item = item.unwrap_if_lazy()?;202						out.push(evaluate_comp(203							context.with_var(var.clone(), item.clone())?,204							value,205							&specs[1..],206						)?);207					}208					Some(out.into_iter().flatten().flatten().collect())209				}210				_ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,211			}212		}213	})214}215216pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {217	let new_bindings = FutureNewBindings::new();218	let future_this = FutureObjValue::new();219	let context_creator = context_creator!(220		closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {221			Ok(context.extend_unbound(222				new_bindings.clone().unwrap(),223				context.dollar().clone().or_else(||this.clone()),224				Some(this.unwrap()),225				super_obj226			)?)227		})228	);229	{230		let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();231		for (n, b) in members232			.iter()233			.filter_map(|m| match m {234				Member::BindStmt(b) => Some(b.clone()),235				_ => None,236			})237			.map(|b| evaluate_binding(&b, context_creator.clone()))238		{239			bindings.insert(n, b);240		}241		new_bindings.fill(bindings);242	}243244	let mut new_members = BTreeMap::new();245	for member in members.iter() {246		match member {247			Member::Field(FieldMember {248				name,249				plus,250				params: None,251				visibility,252				value,253			}) => {254				let name = evaluate_field_name(context.clone(), &name)?;255				if name.is_none() {256					continue;257				}258				let name = name.unwrap();259				new_members.insert(260					name.clone(),261					ObjMember {262						add: *plus,263						visibility: *visibility,264						invoke: LazyBinding::Bindable(Rc::new(265							closure!(clone name, clone value, clone context_creator, |this, super_obj| {266								Ok(LazyVal::new_resolved(push(&value.1, "object field", ||{267									let context = context_creator.0(this, super_obj)?;268									evaluate(269										context,270										&value,271									)272								})?))273							}),274						)),275					},276				);277			}278			Member::Field(FieldMember {279				name,280				params: Some(params),281				value,282				..283			}) => {284				let name = evaluate_field_name(context.clone(), &name)?;285				if name.is_none() {286					continue;287				}288				let name = name.unwrap();289				new_members.insert(290					name,291					ObjMember {292						add: false,293						visibility: Visibility::Hidden,294						invoke: LazyBinding::Bindable(Rc::new(295							closure!(clone value, clone context_creator, clone params, |this, super_obj| {296								// TODO: Assert297								Ok(LazyVal::new_resolved(evaluate_method(298									context_creator.0(this, super_obj)?,299									params.clone(),300									value.clone(),301								)))302							}),303						)),304					},305				);306			}307			Member::BindStmt(_) => {}308			Member::AssertStmt(_) => {}309		}310	}311	Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))312}313314pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {315	Ok(match object {316		ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,317		ObjBody::ObjComp(obj) => {318			let future_this = FutureObjValue::new();319			let mut new_members = BTreeMap::new();320			for (k, v) in evaluate_comp(321				context.clone(),322				&|ctx| {323					let new_bindings = FutureNewBindings::new();324					let context_creator = context_creator!(325						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {326							Ok(context.extend_unbound(327								new_bindings.clone().unwrap(),328								context.dollar().clone().or_else(||this.clone()),329								None,330								super_obj331							)?)332						})333					);334					let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();335					for (n, b) in obj336						.pre_locals337						.iter()338						.chain(obj.post_locals.iter())339						.map(|b| evaluate_binding(b, context_creator.clone()))340					{341						bindings.insert(n, b);342					}343					let bindings = new_bindings.fill(bindings);344					let ctx = ctx.extend_unbound(bindings, None, None, None)?;345					let key = evaluate(ctx.clone(), &obj.key)?;346					let value = LazyBinding::Bindable(Rc::new(347						closure!(clone ctx, clone obj.value, |this, _super_obj| {348							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))349						}),350					));351352					Ok((key, value))353				},354				&obj.compspecs,355			)?356			.unwrap()357			{358				match k {359					Val::Null => {}360					Val::Str(n) => {361						new_members.insert(362							n,363							ObjMember {364								add: false,365								visibility: Visibility::Normal,366								invoke: v,367							},368						);369					}370					v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,371				}372			}373374			future_this.fill(ObjValue::new(None, Rc::new(new_members)))375		}376	})377}378379pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {380	use Expr::*;381	let LocExpr(expr, loc) = expr;382	Ok(match &**expr {383		Literal(LiteralType::This) => Val::Obj(384			context385				.this()386				.clone()387				.ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,388		),389		Literal(LiteralType::Dollar) => Val::Obj(390			context391				.dollar()392				.clone()393				.ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,394		),395		Literal(LiteralType::True) => Val::Bool(true),396		Literal(LiteralType::False) => Val::Bool(false),397		Literal(LiteralType::Null) => Val::Null,398		Parened(e) => evaluate(context, e)?,399		Str(v) => Val::Str(v.clone()),400		Num(v) => Val::Num(*v),401		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,402		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,403		Var(name) => push(loc, "var", || {404			Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?)405		})?,406		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {407			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;408			context409				.super_obj()410				.clone()411				.expect("no super found")412				.get_raw(&name, &context.this().clone().expect("no this found"))?413				.expect("value not found")414		}415		Index(value, index) => {416			match (417				evaluate(context.clone(), value)?.unwrap_if_lazy()?,418				evaluate(context, index)?,419			) {420				(Val::Obj(v), Val::Str(s)) => {421					if let Some(v) = v.get(s.clone())? {422						v.unwrap_if_lazy()?423					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {424						Val::Intristic(n, s)425					} else {426						create_error_result(crate::Error::NoSuchField(s))?427					}428				}429				(Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(430					ValType::Obj,431					ValType::Str,432					n.value_type()?,433				))?,434435				(Val::Arr(v), Val::Num(n)) => {436					if n.fract() > f64::EPSILON {437						create_error_result(crate::Error::FractionalIndex)?438					}439					v.get(n as usize)440						.ok_or_else(|| {441							create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))442						})?443						.clone()444						.unwrap_if_lazy()?445				}446				(Val::Arr(_), Val::Str(n)) => {447					create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?448				}449				(Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(450					ValType::Arr,451					ValType::Num,452					n.value_type()?,453				))?,454455				(Val::Str(s), Val::Num(n)) => Val::Str(456					s.chars()457						.skip(n as usize)458						.take(1)459						.collect::<String>()460						.into(),461				),462				(Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(463					ValType::Str,464					ValType::Num,465					n.value_type()?,466				))?,467468				(v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,469			}470		}471		LocalExpr(bindings, returned) => {472			let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();473			let future_context = Context::new_future();474475			let context_creator = context_creator!(476				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))477			);478479			for (k, v) in bindings480				.iter()481				.map(|b| evaluate_binding(b, context_creator.clone()))482			{483				new_bindings.insert(k, v);484			}485486			let context = context487				.extend_unbound(new_bindings, None, None, None)?488				.into_future(future_context);489			evaluate(context, &returned.clone())?490		}491		Arr(items) => {492			let mut out = Vec::with_capacity(items.len());493			for item in items {494				out.push(Val::Lazy(lazy_val!(495					closure!(clone context, clone item, || {496						evaluate(context.clone(), &item)497					})498				)));499			}500			Val::Arr(Rc::new(out))501		}502		ArrComp(expr, compspecs) => Val::Arr(503			// First compspec should be forspec, so no "None" possible here504			Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),505		),506		Obj(body) => Val::Obj(evaluate_object(context, body)?),507		ObjExtend(s, t) => evaluate_add_op(508			&evaluate(context.clone(), s)?,509			&Val::Obj(evaluate_object(context, t)?),510		)?,511		Apply(value, args, tailstrict) => {512			let lazy = evaluate(context.clone(), value)?;513			let value = lazy.unwrap_if_lazy()?;514			match value {515				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {516					// arr/string/function517					("std", "length") => parse_args!(context, "std.length", args, 1, [518						0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];519					], {520						match x {521							Val::Str(n) => Val::Num(n.chars().count() as f64),522							Val::Arr(i) => Val::Num(i.len() as f64),523							Val::Obj(o) => Val::Num(524								o.fields_visibility()525									.into_iter()526									.filter(|(_k, v)| *v)527									.count() as f64,528							),529							_ => unreachable!(),530						}531					}),532					// any533					("std", "type") => parse_args!(context, "std.type", args, 1, [534						0, x, vec![];535					], {536						Val::Str(x.value_type()?.name().into())537					}),538					// length, idx=>any539					("std", "makeArray") => parse_args!(context, "std.makeArray", args, 2, [540						0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];541						1, func: [Val::Func]!!Val::Func, vec![ValType::Func];542					], {543						assert!(sz >= 0.0);544						let mut out = Vec::with_capacity(sz as usize);545						for i in 0..sz as usize {546							out.push(func.evaluate_values(547								Context::new(),548								&[Val::Num(i as f64)]549							)?)550						}551						Val::Arr(Rc::new(out))552					}),553					// string554					("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [555						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];556					], {557						assert!(558							str.chars().count() == 1,559							"std.codepoint should receive single char string"560						);561						Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)562					}),563					// object, includeHidden564					("std", "objectFieldsEx") => {565						parse_args!(context, "std.objectFieldsEx",args, 2, [566							0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];567							1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];568						], {569							Val::Arr(Rc::new(570								obj.fields_visibility()571									.into_iter()572									.filter(|(_k, v)| *v || inc_hidden)573									.map(|(k, _v)| Val::Str(k))574									.collect(),575							))576						})577					}578					// object, field, includeHidden579					("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [580						0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];581						1, f: [Val::Str]!!Val::Str, vec![ValType::Str];582						2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];583					], {584						Val::Bool(585							obj.fields_visibility()586								.into_iter()587								.filter(|(_k, v)| *v || inc_hidden)588								.any(|(k, _v)| *k == *f),589						)590					}),591					("std", "primitiveEquals") => {592						parse_args!(context, "std.primitiveEquals", args, 2, [593							0, a, vec![];594							1, b, vec![];595						], {596							Val::Bool(a == b)597						})598					}599					("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [600						0, a: [Val::Num]!!Val::Num, vec![ValType::Num];601						1, b: [Val::Num]!!Val::Num, vec![ValType::Num];602					], {603						Val::Num(a % b)604					}),605					("std", "floor") => parse_args!(context, "std.floor", args, 1, [606						0, x: [Val::Num]!!Val::Num, vec![ValType::Num];607					], {608						Val::Num(x.floor())609					}),610					("std", "trace") => parse_args!(context, "std.trace", args, 2, [611						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];612						1, rest, vec![];613					], {614						// TODO: Line numbers as in original jsonnet615						println!("TRACE: {}", str);616						rest617					}),618					("std", "pow") => parse_args!(context, "std.modulo", args, 2, [619						0, x: [Val::Num]!!Val::Num, vec![ValType::Num];620						1, n: [Val::Num]!!Val::Num, vec![ValType::Num];621					], {622						Val::Num(x.powf(n))623					}),624					("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [625						0, x: [Val::Str]!!Val::Str, vec![ValType::Str];626					], {627						with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(628							|| create_error(crate::Error::UndefinedExternalVariable(x)),629						)?630					}),631					("std", "filter") => parse_args!(context, "std.filter", args, 2, [632						0, func: [Val::Func]!!Val::Func, vec![ValType::Func];633						1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];634					], {635						Val::Arr(Rc::new(636							arr.iter()637								.cloned()638								.filter(|e| {639									func640										.evaluate_values(context.clone(), &[e.clone()])641										.unwrap()642										.try_cast_bool("filter predicate")643										.unwrap()644								})645								.collect(),646						))647					}),648					("std", "char") => parse_args!(context, "std.char", args, 1, [649						0, n: [Val::Num]!!Val::Num, vec![ValType::Num];650					], {651						let mut out = String::new();652						out.push(std::char::from_u32(n as u32).unwrap());653						Val::Str(out.into())654					}),655					("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [656						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];657					], {658						Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))659					}),660					("std", "md5") => parse_args!(context, "std.md5", args, 1, [661						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];662					], {663						Val::Str(format!("{:x}", md5::compute(str.as_bytes())).into())664					}),665					// faster666					("std", "join") => parse_args!(context, "std.join", args, 2, [667						0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];668						1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];669					], {670						match sep {671							Val::Arr(joiner_items) => {672								let mut out = Vec::new();673674								let mut first = true;675								for item in arr.iter().cloned() {676									if let Val::Arr(items) = item.unwrap_if_lazy()? {677										if !first {678											out.reserve(joiner_items.len());679											out.extend(joiner_items.iter().cloned());680										}681										first = false;682										out.reserve(items.len());683										out.extend(items.iter().cloned());684									} else {685										create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;686									}687								}688689								Val::Arr(Rc::new(out))690							},691							Val::Str(sep) => {692								let mut out = String::new();693694								let mut first = true;695								for item in arr.iter().cloned() {696									if let Val::Str(item) = item.unwrap_if_lazy()? {697										if !first {698											out += &sep;699										}700										first = false;701										out += &item;702									} else {703										create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;704									}705								}706707								Val::Str(out.into())708							},709							_ => unreachable!()710						}711					}),712					// Faster713					("std", "escapeStringJson") => {714						parse_args!(context, "std.escapeStringJson", args, 1, [715							0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];716						], {717							Val::Str(escape_string_json(&str_).into())718						})719					}720					// Faster721					("std", "manifestJsonEx") => {722						parse_args!(context, "std.manifestJsonEx", args, 2, [723							0, value, vec![];724							1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];725						], {726							Val::Str(manifest_json_ex(&value, &indent)?.into())727						})728					}729					(ns, name) => create_error_result(crate::Error::IntristicNotFound(730						ns.into(),731						name.into(),732					))?,733				},734				Val::Func(f) => {735					let body = || f.evaluate(context, args, *tailstrict);736					if *tailstrict {737						body()?738					} else {739						push(loc, "function call", body)?740					}741				}742				v => {743					create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?744				}745			}746		}747		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),748		AssertExpr(AssertStmt(value, msg), returned) => {749			let assertion_result = push(&value.1, "assertion condition", || {750				evaluate(context.clone(), &value)?751					.try_cast_bool("assertion condition should be boolean")752			})?;753			if assertion_result {754				evaluate(context, returned)?755			} else if let Some(msg) = msg {756				create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?757			} else {758				create_error_result(crate::Error::AssertionFailed(Val::Null))?759			}760		}761		Error(e) => create_error_result(crate::Error::RuntimeError(762			evaluate(context, e)?.try_cast_str("error text should be string")?,763		))?,764		IfElse {765			cond,766			cond_then,767			cond_else,768		} => {769			if evaluate(context.clone(), &cond.0)?770				.try_cast_bool("if condition should be boolean")?771			{772				evaluate(context, cond_then)?773			} else {774				match cond_else {775					Some(v) => evaluate(context, v)?,776					None => Val::Null,777				}778			}779		}780		Import(path) => {781			let mut tmp = loc782				.clone()783				.expect("imports can't be used without loc_data")784				.0;785			let import_location = Rc::make_mut(&mut tmp);786			import_location.pop();787			with_state(|s| s.import_file(&import_location, path))?788		}789		ImportStr(path) => {790			let mut tmp = loc791				.clone()792				.expect("imports can't be used without loc_data")793				.0;794			let import_location = Rc::make_mut(&mut tmp);795			import_location.pop();796			Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)797		}798		Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),799	})800}
after · crates/jrsonnet-evaluator/src/evaluate.rs
1use crate::{2	context_creator, create_error, create_error_result, escape_string_json, future_wrapper,3	lazy_val, manifest_json_ex, parse_args, push, with_state, Context, ContextCreator, Error,4	FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue, Result, Val, ValType,5};6use closure::closure;7use jrsonnet_parser::{8	ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, ExprLocation, 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) -> (Rc<str>, 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.1, "thunk", ||{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: &jrsonnet_parser::FieldName,57) -> Result<Option<Rc<str>>> {58	Ok(match field_name {59		jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),60		jrsonnet_parser::FieldName::Dyn(expr) => {61			let lazy = evaluate(context, expr)?;62			let value = lazy.unwrap_if_lazy()?;63			if matches!(value, Val::Null) {64				None65			} else {66				Some(value.try_cast_str("dynamic field name")?)67			}68		}69	})70}7172pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {73	Ok(match (op, b) {74		(o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,75		(UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),76		(UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),77		(UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),78		(op, o) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(79			op,80			o.value_type()?,81		))?,82	})83}8485pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {86	Ok(match (a, b) {87		(Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),8889		// Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)90		(Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),91		(Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),9293		(Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),94		(o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9596		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),97		(Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),98		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),99		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(100			BinaryOpType::Add,101			a.value_type()?,102			b.value_type()?,103		))?,104	})105}106107pub fn evaluate_binary_op_special(108	context: Context,109	a: &LocExpr,110	op: BinaryOpType,111	b: &LocExpr,112) -> Result<Val> {113	Ok(114		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {115			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),116			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),117			(a, op, eb) => {118				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?119			}120		},121	)122}123124pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {125	Ok(match (a, op, b) {126		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,127128		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),129130		// Bool X Bool131		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),132		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),133134		// Str X Str135		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),136		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),137		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),138		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),139140		// Num X Num141		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),142		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {143			if *v2 <= f64::EPSILON {144				create_error_result(crate::Error::DivisionByZero)?145			}146			Val::Num(v1 / v2)147		}148149		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),150151		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),152		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),153		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),154		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),155156		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {157			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)158		}159		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {160			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)161		}162		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {163			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)164		}165		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {166			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)167		}168		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {169			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)170		}171172		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(173			op,174			a.value_type()?,175			b.value_type()?,176		))?,177	})178}179180future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);181future_wrapper!(ObjValue, FutureObjValue);182183pub fn evaluate_comp<T>(184	context: Context,185	value: &impl Fn(Context) -> Result<T>,186	specs: &[CompSpec],187) -> Result<Option<Vec<T>>> {188	Ok(match specs.get(0) {189		None => Some(vec![value(context)?]),190		Some(CompSpec::IfSpec(IfSpecData(cond))) => {191			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {192				evaluate_comp(context, value, &specs[1..])?193			} else {194				None195			}196		}197		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {198			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {199				Val::Arr(list) => {200					let mut out = Vec::new();201					for item in list.iter() {202						let item = item.unwrap_if_lazy()?;203						out.push(evaluate_comp(204							context.with_var(var.clone(), item.clone())?,205							value,206							&specs[1..],207						)?);208					}209					Some(out.into_iter().flatten().flatten().collect())210				}211				_ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,212			}213		}214	})215}216217pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {218	let new_bindings = FutureNewBindings::new();219	let future_this = FutureObjValue::new();220	let context_creator = context_creator!(221		closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {222			Ok(context.extend_unbound(223				new_bindings.clone().unwrap(),224				context.dollar().clone().or_else(||this.clone()),225				Some(this.unwrap()),226				super_obj227			)?)228		})229	);230	{231		let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();232		for (n, b) in members233			.iter()234			.filter_map(|m| match m {235				Member::BindStmt(b) => Some(b.clone()),236				_ => None,237			})238			.map(|b| evaluate_binding(&b, context_creator.clone()))239		{240			bindings.insert(n, b);241		}242		new_bindings.fill(bindings);243	}244245	let mut new_members = BTreeMap::new();246	for member in members.iter() {247		match member {248			Member::Field(FieldMember {249				name,250				plus,251				params: None,252				visibility,253				value,254			}) => {255				let name = evaluate_field_name(context.clone(), &name)?;256				if name.is_none() {257					continue;258				}259				let name = name.unwrap();260				new_members.insert(261					name.clone(),262					ObjMember {263						add: *plus,264						visibility: *visibility,265						invoke: LazyBinding::Bindable(Rc::new(266							closure!(clone name, clone value, clone context_creator, |this, super_obj| {267								Ok(LazyVal::new_resolved(push(&value.1, "object field", ||{268									let context = context_creator.0(this, super_obj)?;269									evaluate(270										context,271										&value,272									)273								})?))274							}),275						)),276					},277				);278			}279			Member::Field(FieldMember {280				name,281				params: Some(params),282				value,283				..284			}) => {285				let name = evaluate_field_name(context.clone(), &name)?;286				if name.is_none() {287					continue;288				}289				let name = name.unwrap();290				new_members.insert(291					name,292					ObjMember {293						add: false,294						visibility: Visibility::Hidden,295						invoke: LazyBinding::Bindable(Rc::new(296							closure!(clone value, clone context_creator, clone params, |this, super_obj| {297								// TODO: Assert298								Ok(LazyVal::new_resolved(evaluate_method(299									context_creator.0(this, super_obj)?,300									params.clone(),301									value.clone(),302								)))303							}),304						)),305					},306				);307			}308			Member::BindStmt(_) => {}309			Member::AssertStmt(_) => {}310		}311	}312	Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))313}314315pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {316	Ok(match object {317		ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,318		ObjBody::ObjComp(obj) => {319			let future_this = FutureObjValue::new();320			let mut new_members = BTreeMap::new();321			for (k, v) in evaluate_comp(322				context.clone(),323				&|ctx| {324					let new_bindings = FutureNewBindings::new();325					let context_creator = context_creator!(326						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {327							Ok(context.extend_unbound(328								new_bindings.clone().unwrap(),329								context.dollar().clone().or_else(||this.clone()),330								None,331								super_obj332							)?)333						})334					);335					let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();336					for (n, b) in obj337						.pre_locals338						.iter()339						.chain(obj.post_locals.iter())340						.map(|b| evaluate_binding(b, context_creator.clone()))341					{342						bindings.insert(n, b);343					}344					let bindings = new_bindings.fill(bindings);345					let ctx = ctx.extend_unbound(bindings, None, None, None)?;346					let key = evaluate(ctx.clone(), &obj.key)?;347					let value = LazyBinding::Bindable(Rc::new(348						closure!(clone ctx, clone obj.value, |this, _super_obj| {349							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))350						}),351					));352353					Ok((key, value))354				},355				&obj.compspecs,356			)?357			.unwrap()358			{359				match k {360					Val::Null => {}361					Val::Str(n) => {362						new_members.insert(363							n,364							ObjMember {365								add: false,366								visibility: Visibility::Normal,367								invoke: v,368							},369						);370					}371					v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,372				}373			}374375			future_this.fill(ObjValue::new(None, Rc::new(new_members)))376		}377	})378}379380/// Extracts code block and disables inlining for them381/// Fixes WASM to java bytecode compilation failing because of very large method382macro_rules! noinline {383	($e:expr) => {384		(#[inline(never)]385		move || $e)()386	};387}388389pub fn evaluate_apply(390	context: Context,391	value: &LocExpr,392	args: &ArgsDesc,393	loc: &Option<ExprLocation>,394	tailstrict: bool,395) -> Result<Val> {396	let lazy = evaluate(context.clone(), value)?;397	let value = lazy.unwrap_if_lazy()?;398	Ok(match value {399		Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {400			// arr/string/function401			("std", "length") => noinline!(parse_args!(context, "std.length", args, 1, [402				0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];403			], {404				Ok(match x {405					Val::Str(n) => Val::Num(n.chars().count() as f64),406					Val::Arr(i) => Val::Num(i.len() as f64),407					Val::Obj(o) => Val::Num(408						o.fields_visibility()409							.into_iter()410							.filter(|(_k, v)| *v)411							.count() as f64,412					),413					_ => unreachable!(),414				})415			}))?,416			// any417			("std", "type") => parse_args!(context, "std.type", args, 1, [418				0, x, vec![];419			], {420				Val::Str(x.value_type()?.name().into())421			}),422			// length, idx=>any423			("std", "makeArray") => noinline!(parse_args!(context, "std.makeArray", args, 2, [424				0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];425				1, func: [Val::Func]!!Val::Func, vec![ValType::Func];426			], {427				assert!(sz >= 0.0);428				let mut out = Vec::with_capacity(sz as usize);429				for i in 0..sz as usize {430					out.push(func.evaluate_values(431						Context::new(),432						&[Val::Num(i as f64)]433					)?)434				}435				Ok(Val::Arr(Rc::new(out)))436			}))?,437			// string438			("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [439				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];440			], {441				assert!(442					str.chars().count() == 1,443					"std.codepoint should receive single char string"444				);445				Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)446			}),447			// object, includeHidden448			("std", "objectFieldsEx") => {449				noinline!(parse_args!(context, "std.objectFieldsEx",args, 2, [450					0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];451					1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];452				], {453					Ok(Val::Arr(Rc::new(454						obj.fields_visibility()455							.into_iter()456							.filter(|(_k, v)| *v || inc_hidden)457							.map(|(k, _v)| Val::Str(k))458							.collect(),459					)))460				}))?461			}462			// object, field, includeHidden463			("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [464				0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];465				1, f: [Val::Str]!!Val::Str, vec![ValType::Str];466				2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];467			], {468				Val::Bool(469					obj.fields_visibility()470						.into_iter()471						.filter(|(_k, v)| *v || inc_hidden)472						.any(|(k, _v)| *k == *f),473				)474			}),475			("std", "primitiveEquals") => parse_args!(context, "std.primitiveEquals", args, 2, [476				0, a, vec![];477				1, b, vec![];478			], {479				Val::Bool(a == b)480			}),481			("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [482				0, a: [Val::Num]!!Val::Num, vec![ValType::Num];483				1, b: [Val::Num]!!Val::Num, vec![ValType::Num];484			], {485				Val::Num(a % b)486			}),487			("std", "floor") => parse_args!(context, "std.floor", args, 1, [488				0, x: [Val::Num]!!Val::Num, vec![ValType::Num];489			], {490				Val::Num(x.floor())491			}),492			("std", "trace") => parse_args!(context, "std.trace", args, 2, [493				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];494				1, rest, vec![];495			], {496				// TODO: Line numbers as in original jsonnet497				println!("TRACE: {}", str);498				rest499			}),500			("std", "pow") => parse_args!(context, "std.modulo", args, 2, [501				0, x: [Val::Num]!!Val::Num, vec![ValType::Num];502				1, n: [Val::Num]!!Val::Num, vec![ValType::Num];503			], {504				Val::Num(x.powf(n))505			}),506			("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [507				0, x: [Val::Str]!!Val::Str, vec![ValType::Str];508			], {509				with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(510					|| create_error(crate::Error::UndefinedExternalVariable(x)),511				)?512			}),513			("std", "filter") => noinline!(parse_args!(context, "std.filter", args, 2, [514				0, func: [Val::Func]!!Val::Func, vec![ValType::Func];515				1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];516			], {517				Ok(Val::Arr(Rc::new(518					arr.iter()519						.cloned()520						.filter(|e| {521							func522								.evaluate_values(context.clone(), &[e.clone()])523								.unwrap()524								.try_cast_bool("filter predicate")525								.unwrap()526						})527						.collect(),528				)))529			}))?,530			("std", "char") => parse_args!(context, "std.char", args, 1, [531				0, n: [Val::Num]!!Val::Num, vec![ValType::Num];532			], {533				let mut out = String::new();534				out.push(std::char::from_u32(n as u32).unwrap());535				Val::Str(out.into())536			}),537			("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [538				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];539			], {540				Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))541			}),542			("std", "md5") => noinline!(parse_args!(context, "std.md5", args, 1, [543				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];544			], {545				Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))546			}))?,547			// faster548			("std", "join") => noinline!(parse_args!(context, "std.join", args, 2, [549				0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];550				1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];551			], {552				Ok(match sep {553					Val::Arr(joiner_items) => {554						let mut out = Vec::new();555556						let mut first = true;557						for item in arr.iter().cloned() {558							if let Val::Arr(items) = item.unwrap_if_lazy()? {559								if !first {560									out.reserve(joiner_items.len());561									out.extend(joiner_items.iter().cloned());562								}563								first = false;564								out.reserve(items.len());565								out.extend(items.iter().cloned());566							} else {567								create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;568							}569						}570571						Val::Arr(Rc::new(out))572					},573					Val::Str(sep) => {574						let mut out = String::new();575576						let mut first = true;577						for item in arr.iter().cloned() {578							if let Val::Str(item) = item.unwrap_if_lazy()? {579								if !first {580									out += &sep;581								}582								first = false;583								out += &item;584							} else {585								create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;586							}587						}588589						Val::Str(out.into())590					},591					_ => unreachable!()592				})593			}))?,594			// Faster595			("std", "escapeStringJson") => parse_args!(context, "std.escapeStringJson", args, 1, [596				0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];597			], {598				Val::Str(escape_string_json(&str_).into())599			}),600			// Faster601			("std", "manifestJsonEx") => parse_args!(context, "std.manifestJsonEx", args, 2, [602				0, value, vec![];603				1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];604			], {605				Val::Str(manifest_json_ex(&value, &indent)?.into())606			}),607			(ns, name) => {608				create_error_result(crate::Error::IntristicNotFound(ns.into(), name.into()))?609			}610		},611		Val::Func(f) => {612			let body = || f.evaluate(context, args, tailstrict);613			if tailstrict {614				body()?615			} else {616				push(loc, "function call", body)?617			}618		}619		v => create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?,620	})621}622623pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {624	use Expr::*;625	let LocExpr(expr, loc) = expr;626	Ok(match &**expr {627		Literal(LiteralType::This) => Val::Obj(628			context629				.this()630				.clone()631				.ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,632		),633		Literal(LiteralType::Dollar) => Val::Obj(634			context635				.dollar()636				.clone()637				.ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,638		),639		Literal(LiteralType::True) => Val::Bool(true),640		Literal(LiteralType::False) => Val::Bool(false),641		Literal(LiteralType::Null) => Val::Null,642		Parened(e) => evaluate(context, e)?,643		Str(v) => Val::Str(v.clone()),644		Num(v) => Val::Num(*v),645		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,646		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,647		Var(name) => push(loc, "var", || {648			Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?)649		})?,650		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {651			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;652			context653				.super_obj()654				.clone()655				.expect("no super found")656				.get_raw(&name, &context.this().clone().expect("no this found"))?657				.expect("value not found")658		}659		Index(value, index) => {660			match (661				evaluate(context.clone(), value)?.unwrap_if_lazy()?,662				evaluate(context, index)?,663			) {664				(Val::Obj(v), Val::Str(s)) => {665					if let Some(v) = v.get(s.clone())? {666						v.unwrap_if_lazy()?667					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {668						Val::Intristic(n, s)669					} else {670						create_error_result(crate::Error::NoSuchField(s))?671					}672				}673				(Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(674					ValType::Obj,675					ValType::Str,676					n.value_type()?,677				))?,678679				(Val::Arr(v), Val::Num(n)) => {680					if n.fract() > f64::EPSILON {681						create_error_result(crate::Error::FractionalIndex)?682					}683					v.get(n as usize)684						.ok_or_else(|| {685							create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))686						})?687						.clone()688						.unwrap_if_lazy()?689				}690				(Val::Arr(_), Val::Str(n)) => {691					create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?692				}693				(Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(694					ValType::Arr,695					ValType::Num,696					n.value_type()?,697				))?,698699				(Val::Str(s), Val::Num(n)) => Val::Str(700					s.chars()701						.skip(n as usize)702						.take(1)703						.collect::<String>()704						.into(),705				),706				(Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(707					ValType::Str,708					ValType::Num,709					n.value_type()?,710				))?,711712				(v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,713			}714		}715		LocalExpr(bindings, returned) => {716			let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();717			let future_context = Context::new_future();718719			let context_creator = context_creator!(720				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))721			);722723			for (k, v) in bindings724				.iter()725				.map(|b| evaluate_binding(b, context_creator.clone()))726			{727				new_bindings.insert(k, v);728			}729730			let context = context731				.extend_unbound(new_bindings, None, None, None)?732				.into_future(future_context);733			evaluate(context, &returned.clone())?734		}735		Arr(items) => {736			let mut out = Vec::with_capacity(items.len());737			for item in items {738				out.push(Val::Lazy(lazy_val!(739					closure!(clone context, clone item, || {740						evaluate(context.clone(), &item)741					})742				)));743			}744			Val::Arr(Rc::new(out))745		}746		ArrComp(expr, compspecs) => Val::Arr(747			// First compspec should be forspec, so no "None" possible here748			Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),749		),750		Obj(body) => Val::Obj(evaluate_object(context, body)?),751		ObjExtend(s, t) => evaluate_add_op(752			&evaluate(context.clone(), s)?,753			&Val::Obj(evaluate_object(context, t)?),754		)?,755		Apply(value, args, tailstrict) => evaluate_apply(context, value, args, loc, *tailstrict)?,756		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),757		AssertExpr(AssertStmt(value, msg), returned) => {758			let assertion_result = push(&value.1, "assertion condition", || {759				evaluate(context.clone(), &value)?760					.try_cast_bool("assertion condition should be boolean")761			})?;762			if assertion_result {763				evaluate(context, returned)?764			} else if let Some(msg) = msg {765				create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?766			} else {767				create_error_result(crate::Error::AssertionFailed(Val::Null))?768			}769		}770		Error(e) => create_error_result(crate::Error::RuntimeError(771			evaluate(context, e)?.try_cast_str("error text should be string")?,772		))?,773		IfElse {774			cond,775			cond_then,776			cond_else,777		} => {778			if evaluate(context.clone(), &cond.0)?779				.try_cast_bool("if condition should be boolean")?780			{781				evaluate(context, cond_then)?782			} else {783				match cond_else {784					Some(v) => evaluate(context, v)?,785					None => Val::Null,786				}787			}788		}789		Import(path) => {790			let mut tmp = loc791				.clone()792				.expect("imports can't be used without loc_data")793				.0;794			let import_location = Rc::make_mut(&mut tmp);795			import_location.pop();796			with_state(|s| s.import_file(&import_location, path))?797		}798		ImportStr(path) => {799			let mut tmp = loc800				.clone()801				.expect("imports can't be used without loc_data")802				.0;803			let import_location = Rc::make_mut(&mut tmp);804			import_location.pop();805			Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)806		}807		Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),808	})809}
modifiedcrates/jrsonnet-evaluator/src/lib.rsdiffbeforeafterboth
--- a/crates/jrsonnet-evaluator/src/lib.rs
+++ b/crates/jrsonnet-evaluator/src/lib.rs
@@ -2,6 +2,7 @@
 #![feature(type_alias_impl_trait)]
 #![feature(debug_non_exhaustive)]
 #![feature(test)]
+#![feature(stmt_expr_attributes)]
 #![allow(macro_expanded_macro_exports_accessed_by_absolute_paths)]
 
 extern crate test;