git.delta.rocks / jrsonnet / refs/commits / 0ba2642a3a69

difftreelog

style(evaluator) fix clippy warnings

Лач2020-06-27parent: #0afc5f7.patch.diff
in: master

3 files changed

modifiedcrates/jsonnet-evaluator/src/evaluate.rsdiffbeforeafterboth
before · crates/jsonnet-evaluator/src/evaluate.rs
1use crate::{2	context_creator, create_error, escape_string_json, future_wrapper, lazy_val, manifest_json_ex,3	parse_args, push, with_state, Context, ContextCreator, Error, FuncDesc, LazyBinding, LazyVal,4	ObjMember, ObjValue, Result, Val, 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) -> (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: &jsonnet_parser::FieldName,57) -> Result<Option<Rc<str>>> {58	Ok(match field_name {59		jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),60		jsonnet_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) => panic!("unary op not implemented: {:?} {:?}", op, o),79	})80}8182pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {83	Ok(match (a, b) {84		(Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),8586		// Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)87		(Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),88		(Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),8990		(Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),91		(o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9293		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),94		(Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),95		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),96		_ => panic!("can't add: {:?} and {:?}", a, b),97	})98}99100pub fn evaluate_binary_op_special(101	context: Context,102	a: &LocExpr,103	op: BinaryOpType,104	b: &LocExpr,105) -> Result<Val> {106	Ok(107		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {108			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),109			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),110			(a, op, eb) => {111				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?112			}113		},114	)115}116117pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {118	Ok(match (a, op, b) {119		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,120121		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),122123		// Bool X Bool124		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),125		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),126127		// Str X Str128		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),129		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),130		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),131		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),132133		// Num X Num134		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),135		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {136			if *v2 <= f64::EPSILON {137				create_error(crate::Error::DivisionByZero)?138			}139			Val::Num(v1 / v2)140		}141142		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),143144		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),145		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),146		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),147		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),148149		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {150			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)151		}152		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {153			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)154		}155		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {156			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)157		}158		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {159			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)160		}161		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {162			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)163		}164165		_ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),166	})167}168169future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);170future_wrapper!(ObjValue, FutureObjValue);171172pub 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.iter() {191						let item = item.unwrap_if_lazy()?;192						out.push(evaluate_comp(193							context.with_var(var.clone(), item.clone())?,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}205206pub fn evaluate_member_list_object(context: Context, members: &Vec<Member>) -> Result<ObjValue> {207	let new_bindings = FutureNewBindings::new();208	let future_this = FutureObjValue::new();209	let context_creator = context_creator!(210		closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {211			Ok(context.extend_unbound(212				new_bindings.clone().unwrap(),213				context.dollar().clone().or_else(||this.clone()),214				Some(this.unwrap()),215				super_obj216			)?)217		})218	);219	{220		let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();221		for (n, b) in members222			.iter()223			.filter_map(|m| match m {224				Member::BindStmt(b) => Some(b.clone()),225				_ => None,226			})227			.map(|b| evaluate_binding(&b, context_creator.clone()))228		{229			bindings.insert(n, b);230		}231		new_bindings.fill(bindings);232	}233234	let mut new_members = BTreeMap::new();235	for member in members.iter() {236		match member {237			Member::Field(FieldMember {238				name,239				plus,240				params: None,241				visibility,242				value,243			}) => {244				let name = evaluate_field_name(context.clone(), &name)?;245				if name.is_none() {246					continue;247				}248				let name = name.unwrap();249				new_members.insert(250					name.clone(),251					ObjMember {252						add: *plus,253						visibility: *visibility,254						invoke: LazyBinding::Bindable(Rc::new(255							closure!(clone name, clone value, clone context_creator, |this, super_obj| {256								Ok(LazyVal::new_resolved(push(&value.1, "object field", ||{257									let context = context_creator.0(this, super_obj)?;258									evaluate(259										context,260										&value,261									)262								})?))263							}),264						)),265					},266				);267			}268			Member::Field(FieldMember {269				name,270				params: Some(params),271				value,272				..273			}) => {274				let name = evaluate_field_name(context.clone(), &name)?;275				if name.is_none() {276					continue;277				}278				let name = name.unwrap();279				new_members.insert(280					name,281					ObjMember {282						add: false,283						visibility: Visibility::Hidden,284						invoke: LazyBinding::Bindable(Rc::new(285							closure!(clone value, clone context_creator, clone params, |this, super_obj| {286								// TODO: Assert287								Ok(LazyVal::new_resolved(evaluate_method(288									context_creator.0(this, super_obj)?,289									params.clone(),290									value.clone(),291								)))292							}),293						)),294					},295				);296			}297			Member::BindStmt(_) => {}298			Member::AssertStmt(_) => {}299		}300	}301	Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))302}303304pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {305	Ok(match object {306		ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,307		ObjBody::ObjComp(obj) => {308			let future_this = FutureObjValue::new();309			let mut new_members = BTreeMap::new();310			for (k, v) in evaluate_comp(311				context.clone(),312				&|ctx| {313					let new_bindings = FutureNewBindings::new();314					let context_creator = context_creator!(315						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {316							Ok(context.extend_unbound(317								new_bindings.clone().unwrap(),318								context.dollar().clone().or_else(||this.clone()),319								None,320								super_obj321							)?)322						})323					);324					let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();325					for (n, b) in obj326						.pre_locals327						.iter()328						.chain(obj.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(), &obj.key)?;336					let value = LazyBinding::Bindable(Rc::new(337						closure!(clone ctx, clone obj.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				&obj.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}368369pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {370	use Expr::*;371	let LocExpr(expr, loc) = expr;372	Ok(match &**expr {373		Literal(LiteralType::This) => Val::Obj(374			context375				.this()376				.clone()377				.unwrap_or_else(|| panic!("this not found")),378		),379		Literal(LiteralType::Dollar) => Val::Obj(380			context381				.dollar()382				.clone()383				.unwrap_or_else(|| panic!("dollar not found")),384		),385		Literal(LiteralType::True) => Val::Bool(true),386		Literal(LiteralType::False) => Val::Bool(false),387		Literal(LiteralType::Null) => Val::Null,388		Parened(e) => evaluate(context, e)?,389		Str(v) => Val::Str(v.clone()),390		Num(v) => Val::Num(*v),391		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,392		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,393		Var(name) => push(loc, "var", || {394			Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?)395		})?,396		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {397			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;398			context399				.super_obj()400				.clone()401				.expect("no super found")402				.get_raw(&name, &context.this().clone().expect("no this found"))?403				.expect("value not found")404		}405		Index(value, index) => {406			match (407				evaluate(context.clone(), value)?.unwrap_if_lazy()?,408				evaluate(context, index)?,409			) {410				(Val::Obj(v), Val::Str(s)) => {411					if let Some(v) = v.get(s.clone())? {412						v.unwrap_if_lazy()?413					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {414						Val::Intristic(n, s)415					} else {416						create_error(crate::Error::NoSuchField(s))?417					}418				}419				(Val::Obj(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(420					ValType::Obj,421					ValType::Str,422					n.value_type()?,423				))?,424425				(Val::Arr(v), Val::Num(n)) => {426					if n.fract() > f64::EPSILON {427						create_error(crate::Error::FractionalIndex)?428					}429					v.get(n as usize)430						.unwrap_or_else(|| panic!("out of bounds"))431						.clone()432						.unwrap_if_lazy()?433				}434				(Val::Arr(_), Val::Str(n)) => {435					create_error(crate::Error::AttemptedIndexAnArrayWithString(n))?436				}437				(Val::Arr(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(438					ValType::Arr,439					ValType::Num,440					n.value_type()?,441				))?,442443				(Val::Str(s), Val::Num(n)) => Val::Str(444					s.chars()445						.skip(n as usize)446						.take(1)447						.collect::<String>()448						.into(),449				),450				(Val::Str(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(451					ValType::Str,452					ValType::Num,453					n.value_type()?,454				))?,455456				(v, _) => create_error(crate::Error::CantIndexInto(v.value_type()?))?,457			}458		}459		LocalExpr(bindings, returned) => {460			let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();461			let future_context = Context::new_future();462463			let context_creator = context_creator!(464				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))465			);466467			for (k, v) in bindings468				.iter()469				.map(|b| evaluate_binding(b, context_creator.clone()))470			{471				new_bindings.insert(k, v);472			}473474			let context = context475				.extend_unbound(new_bindings, None, None, None)?476				.into_future(future_context);477			evaluate(context, &returned.clone())?478		}479		Arr(items) => {480			let mut out = Vec::with_capacity(items.len());481			for item in items {482				out.push(Val::Lazy(lazy_val!(483					closure!(clone context, clone item, || {484						evaluate(context.clone(), &item)485					})486				)));487			}488			Val::Arr(Rc::new(out))489		}490		ArrComp(expr, compspecs) => Val::Arr(491			// First compspec should be forspec, so no "None" possible here492			Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),493		),494		Obj(body) => Val::Obj(evaluate_object(context, body)?),495		ObjExtend(s, t) => evaluate_add_op(496			&evaluate(context.clone(), s)?,497			&Val::Obj(evaluate_object(context, t)?),498		)?,499		Apply(value, args, tailstrict) => {500			let lazy = evaluate(context.clone(), value)?;501			let value = lazy.unwrap_if_lazy()?;502			match value {503				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {504					// arr/string/function505					("std", "length") => parse_args!(context, "std.length", args, 1, [506						0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];507					], {508						match x {509							Val::Str(n) => Val::Num(n.chars().count() as f64),510							Val::Arr(i) => Val::Num(i.len() as f64),511							Val::Obj(o) => Val::Num(512								o.fields_visibility()513									.into_iter()514									.filter(|(_k, v)| *v)515									.count() as f64,516							),517							_ => unreachable!(),518						}519					}),520					// any521					("std", "type") => parse_args!(context, "std.type", args, 1, [522						0, x, vec![];523					], {524						Val::Str(x.value_type()?.name().into())525					}),526					// length, idx=>any527					("std", "makeArray") => parse_args!(context, "std.makeArray", args, 2, [528						0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];529						1, func: [Val::Func]!!Val::Func, vec![ValType::Func];530					], {531						assert!(sz >= 0.0);532						let mut out = Vec::with_capacity(sz as usize);533						for i in 0..sz as usize {534							out.push(func.evaluate_values(535								Context::new(),536								&[Val::Num(i as f64)]537							)?)538						}539						Val::Arr(Rc::new(out))540					}),541					// string542					("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [543						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];544					], {545						assert!(546							str.chars().count() == 1,547							"std.codepoint should receive single char string"548						);549						Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)550					}),551					// object, includeHidden552					("std", "objectFieldsEx") => {553						parse_args!(context, "std.objectFieldsEx",args, 2, [554							0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];555							1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];556						], {557							Val::Arr(Rc::new(558								obj.fields_visibility()559									.into_iter()560									.filter(|(_k, v)| *v || inc_hidden)561									.map(|(k, _v)| Val::Str(k))562									.collect(),563							))564						})565					}566					// object, field, includeHidden567					("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [568						0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];569						1, f: [Val::Str]!!Val::Str, vec![ValType::Str];570						2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];571					], {572						Val::Bool(573							obj.fields_visibility()574								.into_iter()575								.filter(|(_k, v)| *v || inc_hidden)576								.any(|(k, _v)| *k == *f),577						)578					}),579					("std", "primitiveEquals") => {580						parse_args!(context, "std.primitiveEquals", args, 2, [581							0, a, vec![];582							1, b, vec![];583						], {584							Val::Bool(a == b)585						})586					}587					("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [588						0, a: [Val::Num]!!Val::Num, vec![ValType::Num];589						1, b: [Val::Num]!!Val::Num, vec![ValType::Num];590					], {591						Val::Num(a % b)592					}),593					("std", "floor") => parse_args!(context, "std.floor", args, 1, [594						0, x: [Val::Num]!!Val::Num, vec![ValType::Num];595					], {596						Val::Num(x.floor())597					}),598					("std", "trace") => parse_args!(context, "std.trace", args, 2, [599						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];600						1, rest, vec![];601					], {602						// TODO: Line numbers as in original jsonnet603						println!("TRACE: {}", str);604						rest605					}),606					("std", "pow") => parse_args!(context, "std.modulo", args, 2, [607						0, x: [Val::Num]!!Val::Num, vec![ValType::Num];608						1, n: [Val::Num]!!Val::Num, vec![ValType::Num];609					], {610						Val::Num(x.powf(n))611					}),612					("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [613						0, x: [Val::Str]!!Val::Str, vec![ValType::Str];614					], {615						with_state(|s| s.0.ext_vars.borrow().get(&x).cloned()).ok_or_else(616							|| {617								create_error::<()>(crate::Error::UndefinedExternalVariable(x))618									.err()619									.unwrap()620							},621						)?622					}),623					("std", "filter") => parse_args!(context, "std.filter", args, 2, [624						0, func: [Val::Func]!!Val::Func, vec![ValType::Func];625						1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];626					], {627						Val::Arr(Rc::new(628							arr.iter()629								.cloned()630								.filter(|e| {631									func632										.evaluate_values(context.clone(), &[e.clone()])633										.unwrap()634										.try_cast_bool("filter predicate")635										.unwrap()636								})637								.collect(),638						))639					}),640					("std", "char") => parse_args!(context, "std.char", args, 1, [641						0, n: [Val::Num]!!Val::Num, vec![ValType::Num];642					], {643						let mut out = String::new();644						out.push(std::char::from_u32(n as u32).unwrap());645						Val::Str(out.into())646					}),647					("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [648						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];649					], {650						Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))651					}),652					("std", "md5") => parse_args!(context, "std.md5", args, 1, [653						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];654					], {655						Val::Str(format!("{:x}", md5::compute(str.as_bytes())).into())656					}),657					// faster658					("std", "join") => parse_args!(context, "std.join", args, 2, [659						0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];660						1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];661					], {662						match sep {663							Val::Arr(joiner_items) => {664								let mut out = Vec::new();665666								let mut first = true;667								for item in arr.iter().cloned() {668									if let Val::Arr(items) = item.unwrap_if_lazy()? {669										if !first {670											out.reserve(joiner_items.len());671											out.extend(joiner_items.iter().cloned());672										}673										first = false;674										out.reserve(items.len());675										out.extend(items.iter().cloned());676									} else {677										panic!("all array items should be arrays")678									}679								}680681								Val::Arr(Rc::new(out))682							},683							Val::Str(sep) => {684								let mut out = String::new();685686								let mut first = true;687								for item in arr.iter().cloned() {688									if let Val::Str(item) = item.unwrap_if_lazy()? {689										if !first {690											out += &sep;691										}692										first = false;693										out += &item;694									} else {695										panic!("all array items should be strings")696									}697								}698699								Val::Str(out.into())700							},701							_ => unreachable!()702						}703					}),704					// Faster705					("std", "escapeStringJson") => {706						parse_args!(context, "std.escapeStringJson", args, 1, [707							0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];708						], {709							Val::Str(escape_string_json(&str_).into())710						})711					}712					// Faster713					("std", "manifestJsonEx") => {714						parse_args!(context, "std.manifestJsonEx", args, 2, [715							0, value, vec![];716							1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];717						], {718							Val::Str(manifest_json_ex(&value, &indent)?.into())719						})720					}721					(ns, name) => create_error(crate::error::Error::IntristicNotFound(722						ns.into(),723						name.into(),724					))?,725				},726				Val::Func(f) => {727					let body = || f.evaluate(context, args, *tailstrict);728					if *tailstrict {729						body()?730					} else {731						push(loc, "function call", body)?732					}733				}734				v => create_error(crate::error::Error::OnlyFunctionsCanBeCalledGot(735					v.value_type()?,736				))?,737			}738		}739		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),740		AssertExpr(AssertStmt(value, msg), returned) => {741			let assertion_result = push(&value.1, "assertion condition", || {742				evaluate(context.clone(), &value)?743					.try_cast_bool("assertion condition should be boolean")744			})?;745			if assertion_result {746				evaluate(context, returned)?747			} else if let Some(msg) = msg {748				panic!(749					"assertion failed ({:?}): {}",750					value,751					evaluate(context, msg)?.try_cast_str("assertion message should be string")?752				);753			} else {754				panic!("assertion failed ({:?}): no message", value);755			}756		}757		Error(e) => create_error(crate::Error::RuntimeError(758			evaluate(context, e)?.try_cast_str("error text should be string")?,759		))?,760		IfElse {761			cond,762			cond_then,763			cond_else,764		} => {765			if evaluate(context.clone(), &cond.0)?766				.try_cast_bool("if condition should be boolean")?767			{768				evaluate(context, cond_then)?769			} else {770				match cond_else {771					Some(v) => evaluate(context, v)?,772					None => Val::Null,773				}774			}775		}776		Import(path) => {777			let mut tmp = loc778				.clone()779				.expect("imports can't be used without loc_data")780				.0;781			let import_location = Rc::make_mut(&mut tmp);782			import_location.pop();783			with_state(|s| s.import_file(&import_location, path))?784		}785		ImportStr(path) => {786			let mut tmp = loc787				.clone()788				.expect("imports can't be used without loc_data")789				.0;790			let import_location = Rc::make_mut(&mut tmp);791			import_location.pop();792			Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)793		}794		Literal(LiteralType::Super) => return create_error(crate::error::Error::StandaloneSuper),795	})796}
after · crates/jsonnet-evaluator/src/evaluate.rs
1use crate::{2	context_creator, create_error, escape_string_json, future_wrapper, lazy_val, manifest_json_ex,3	parse_args, push, with_state, Context, ContextCreator, Error, FuncDesc, LazyBinding, LazyVal,4	ObjMember, ObjValue, Result, Val, ValType,5};6use closure::closure;7use jsonnet_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: &jsonnet_parser::FieldName,56) -> Result<Option<Rc<str>>> {57	Ok(match field_name {58		jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),59		jsonnet_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) => 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).into()),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).into()),87		(Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),8889		(Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),90		(o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9192		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),93		(Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&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).into()),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<Rc<str>, LazyBinding>, FutureNewBindings);169future_wrapper!(ObjValue, FutureObjValue);170171pub fn evaluate_comp<T>(172	context: Context,173	value: &impl Fn(Context) -> Result<T>,174	specs: &[CompSpec],175) -> Result<Option<Vec<T>>> {176	Ok(match specs.get(0) {177		None => Some(vec![value(context)?]),178		Some(CompSpec::IfSpec(IfSpecData(cond))) => {179			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {180				evaluate_comp(context, value, &specs[1..])?181			} else {182				None183			}184		}185		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {186			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {187				Val::Arr(list) => {188					let mut out = Vec::new();189					for item in list.iter() {190						let item = item.unwrap_if_lazy()?;191						out.push(evaluate_comp(192							context.with_var(var.clone(), item.clone())?,193							value,194							&specs[1..],195						)?);196					}197					Some(out.into_iter().flatten().flatten().collect())198				}199				_ => panic!("for expression evaluated to non-iterable value"),200			}201		}202	})203}204205pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {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.extend_unbound(211				new_bindings.clone().unwrap(),212				context.dollar().clone().or_else(||this.clone()),213				Some(this.unwrap()),214				super_obj215			)?)216		})217	);218	{219		let mut bindings: HashMap<Rc<str>, 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.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,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.1, "object field", ||{256									let context = context_creator.0(this, super_obj)?;257									evaluate(258										context,259										&value,260									)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, clone params, |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	Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))301}302303pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {304	Ok(match object {305		ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,306		ObjBody::ObjComp(obj) => {307			let future_this = FutureObjValue::new();308			let mut new_members = BTreeMap::new();309			for (k, v) in evaluate_comp(310				context.clone(),311				&|ctx| {312					let new_bindings = FutureNewBindings::new();313					let context_creator = context_creator!(314						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {315							Ok(context.extend_unbound(316								new_bindings.clone().unwrap(),317								context.dollar().clone().or_else(||this.clone()),318								None,319								super_obj320							)?)321						})322					);323					let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();324					for (n, b) in obj325						.pre_locals326						.iter()327						.chain(obj.post_locals.iter())328						.map(|b| evaluate_binding(b, context_creator.clone()))329					{330						bindings.insert(n, b);331					}332					let bindings = new_bindings.fill(bindings);333					let ctx = ctx.extend_unbound(bindings, None, None, None)?;334					let key = evaluate(ctx.clone(), &obj.key)?;335					let value = LazyBinding::Bindable(Rc::new(336						closure!(clone ctx, clone obj.value, |this, _super_obj| {337							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))338						}),339					));340341					Ok((key, value))342				},343				&obj.compspecs,344			)?345			.unwrap()346			{347				match k {348					Val::Null => {}349					Val::Str(n) => {350						new_members.insert(351							n,352							ObjMember {353								add: false,354								visibility: Visibility::Normal,355								invoke: v,356							},357						);358					}359					v => create_error(Error::FieldMustBeStringGot(v.value_type()?))?,360				}361			}362363			future_this.fill(ObjValue::new(None, Rc::new(new_members)))364		}365	})366}367368pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {369	use Expr::*;370	let LocExpr(expr, loc) = expr;371	Ok(match &**expr {372		Literal(LiteralType::This) => Val::Obj(373			context374				.this()375				.clone()376				.unwrap_or_else(|| panic!("this not found")),377		),378		Literal(LiteralType::Dollar) => Val::Obj(379			context380				.dollar()381				.clone()382				.unwrap_or_else(|| panic!("dollar not found")),383		),384		Literal(LiteralType::True) => Val::Bool(true),385		Literal(LiteralType::False) => Val::Bool(false),386		Literal(LiteralType::Null) => Val::Null,387		Parened(e) => evaluate(context, e)?,388		Str(v) => Val::Str(v.clone()),389		Num(v) => Val::Num(*v),390		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,391		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,392		Var(name) => push(loc, "var", || {393			Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?)394		})?,395		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {396			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;397			context398				.super_obj()399				.clone()400				.expect("no super found")401				.get_raw(&name, &context.this().clone().expect("no this found"))?402				.expect("value not found")403		}404		Index(value, index) => {405			match (406				evaluate(context.clone(), value)?.unwrap_if_lazy()?,407				evaluate(context, index)?,408			) {409				(Val::Obj(v), Val::Str(s)) => {410					if let Some(v) = v.get(s.clone())? {411						v.unwrap_if_lazy()?412					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {413						Val::Intristic(n, s)414					} else {415						create_error(crate::Error::NoSuchField(s))?416					}417				}418				(Val::Obj(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(419					ValType::Obj,420					ValType::Str,421					n.value_type()?,422				))?,423424				(Val::Arr(v), Val::Num(n)) => {425					if n.fract() > f64::EPSILON {426						create_error(crate::Error::FractionalIndex)?427					}428					v.get(n as usize)429						.unwrap_or_else(|| panic!("out of bounds"))430						.clone()431						.unwrap_if_lazy()?432				}433				(Val::Arr(_), Val::Str(n)) => {434					create_error(crate::Error::AttemptedIndexAnArrayWithString(n))?435				}436				(Val::Arr(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(437					ValType::Arr,438					ValType::Num,439					n.value_type()?,440				))?,441442				(Val::Str(s), Val::Num(n)) => Val::Str(443					s.chars()444						.skip(n as usize)445						.take(1)446						.collect::<String>()447						.into(),448				),449				(Val::Str(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(450					ValType::Str,451					ValType::Num,452					n.value_type()?,453				))?,454455				(v, _) => create_error(crate::Error::CantIndexInto(v.value_type()?))?,456			}457		}458		LocalExpr(bindings, returned) => {459			let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();460			let future_context = Context::new_future();461462			let context_creator = context_creator!(463				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))464			);465466			for (k, v) in bindings467				.iter()468				.map(|b| evaluate_binding(b, context_creator.clone()))469			{470				new_bindings.insert(k, v);471			}472473			let context = context474				.extend_unbound(new_bindings, None, None, None)?475				.into_future(future_context);476			evaluate(context, &returned.clone())?477		}478		Arr(items) => {479			let mut out = Vec::with_capacity(items.len());480			for item in items {481				out.push(Val::Lazy(lazy_val!(482					closure!(clone context, clone item, || {483						evaluate(context.clone(), &item)484					})485				)));486			}487			Val::Arr(Rc::new(out))488		}489		ArrComp(expr, compspecs) => Val::Arr(490			// First compspec should be forspec, so no "None" possible here491			Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),492		),493		Obj(body) => Val::Obj(evaluate_object(context, body)?),494		ObjExtend(s, t) => evaluate_add_op(495			&evaluate(context.clone(), s)?,496			&Val::Obj(evaluate_object(context, t)?),497		)?,498		Apply(value, args, tailstrict) => {499			let lazy = evaluate(context.clone(), value)?;500			let value = lazy.unwrap_if_lazy()?;501			match value {502				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {503					// arr/string/function504					("std", "length") => parse_args!(context, "std.length", args, 1, [505						0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];506					], {507						match x {508							Val::Str(n) => Val::Num(n.chars().count() as f64),509							Val::Arr(i) => Val::Num(i.len() as f64),510							Val::Obj(o) => Val::Num(511								o.fields_visibility()512									.into_iter()513									.filter(|(_k, v)| *v)514									.count() as f64,515							),516							_ => unreachable!(),517						}518					}),519					// any520					("std", "type") => parse_args!(context, "std.type", args, 1, [521						0, x, vec![];522					], {523						Val::Str(x.value_type()?.name().into())524					}),525					// length, idx=>any526					("std", "makeArray") => parse_args!(context, "std.makeArray", args, 2, [527						0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];528						1, func: [Val::Func]!!Val::Func, vec![ValType::Func];529					], {530						assert!(sz >= 0.0);531						let mut out = Vec::with_capacity(sz as usize);532						for i in 0..sz as usize {533							out.push(func.evaluate_values(534								Context::new(),535								&[Val::Num(i as f64)]536							)?)537						}538						Val::Arr(Rc::new(out))539					}),540					// string541					("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [542						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];543					], {544						assert!(545							str.chars().count() == 1,546							"std.codepoint should receive single char string"547						);548						Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)549					}),550					// object, includeHidden551					("std", "objectFieldsEx") => {552						parse_args!(context, "std.objectFieldsEx",args, 2, [553							0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];554							1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];555						], {556							Val::Arr(Rc::new(557								obj.fields_visibility()558									.into_iter()559									.filter(|(_k, v)| *v || inc_hidden)560									.map(|(k, _v)| Val::Str(k))561									.collect(),562							))563						})564					}565					// object, field, includeHidden566					("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [567						0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];568						1, f: [Val::Str]!!Val::Str, vec![ValType::Str];569						2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];570					], {571						Val::Bool(572							obj.fields_visibility()573								.into_iter()574								.filter(|(_k, v)| *v || inc_hidden)575								.any(|(k, _v)| *k == *f),576						)577					}),578					("std", "primitiveEquals") => {579						parse_args!(context, "std.primitiveEquals", args, 2, [580							0, a, vec![];581							1, b, vec![];582						], {583							Val::Bool(a == b)584						})585					}586					("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [587						0, a: [Val::Num]!!Val::Num, vec![ValType::Num];588						1, b: [Val::Num]!!Val::Num, vec![ValType::Num];589					], {590						Val::Num(a % b)591					}),592					("std", "floor") => parse_args!(context, "std.floor", args, 1, [593						0, x: [Val::Num]!!Val::Num, vec![ValType::Num];594					], {595						Val::Num(x.floor())596					}),597					("std", "trace") => parse_args!(context, "std.trace", args, 2, [598						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];599						1, rest, vec![];600					], {601						// TODO: Line numbers as in original jsonnet602						println!("TRACE: {}", str);603						rest604					}),605					("std", "pow") => parse_args!(context, "std.modulo", args, 2, [606						0, x: [Val::Num]!!Val::Num, vec![ValType::Num];607						1, n: [Val::Num]!!Val::Num, vec![ValType::Num];608					], {609						Val::Num(x.powf(n))610					}),611					("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [612						0, x: [Val::Str]!!Val::Str, vec![ValType::Str];613					], {614						with_state(|s| s.0.ext_vars.borrow().get(&x).cloned()).ok_or_else(615							|| {616								create_error::<()>(crate::Error::UndefinedExternalVariable(x))617									.err()618									.unwrap()619							},620						)?621					}),622					("std", "filter") => parse_args!(context, "std.filter", args, 2, [623						0, func: [Val::Func]!!Val::Func, vec![ValType::Func];624						1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];625					], {626						Val::Arr(Rc::new(627							arr.iter()628								.cloned()629								.filter(|e| {630									func631										.evaluate_values(context.clone(), &[e.clone()])632										.unwrap()633										.try_cast_bool("filter predicate")634										.unwrap()635								})636								.collect(),637						))638					}),639					("std", "char") => parse_args!(context, "std.char", args, 1, [640						0, n: [Val::Num]!!Val::Num, vec![ValType::Num];641					], {642						let mut out = String::new();643						out.push(std::char::from_u32(n as u32).unwrap());644						Val::Str(out.into())645					}),646					("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [647						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];648					], {649						Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))650					}),651					("std", "md5") => parse_args!(context, "std.md5", args, 1, [652						0, str: [Val::Str]!!Val::Str, vec![ValType::Str];653					], {654						Val::Str(format!("{:x}", md5::compute(str.as_bytes())).into())655					}),656					// faster657					("std", "join") => parse_args!(context, "std.join", args, 2, [658						0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];659						1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];660					], {661						match sep {662							Val::Arr(joiner_items) => {663								let mut out = Vec::new();664665								let mut first = true;666								for item in arr.iter().cloned() {667									if let Val::Arr(items) = item.unwrap_if_lazy()? {668										if !first {669											out.reserve(joiner_items.len());670											out.extend(joiner_items.iter().cloned());671										}672										first = false;673										out.reserve(items.len());674										out.extend(items.iter().cloned());675									} else {676										panic!("all array items should be arrays")677									}678								}679680								Val::Arr(Rc::new(out))681							},682							Val::Str(sep) => {683								let mut out = String::new();684685								let mut first = true;686								for item in arr.iter().cloned() {687									if let Val::Str(item) = item.unwrap_if_lazy()? {688										if !first {689											out += &sep;690										}691										first = false;692										out += &item;693									} else {694										panic!("all array items should be strings")695									}696								}697698								Val::Str(out.into())699							},700							_ => unreachable!()701						}702					}),703					// Faster704					("std", "escapeStringJson") => {705						parse_args!(context, "std.escapeStringJson", args, 1, [706							0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];707						], {708							Val::Str(escape_string_json(&str_).into())709						})710					}711					// Faster712					("std", "manifestJsonEx") => {713						parse_args!(context, "std.manifestJsonEx", args, 2, [714							0, value, vec![];715							1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];716						], {717							Val::Str(manifest_json_ex(&value, &indent)?.into())718						})719					}720					(ns, name) => create_error(crate::error::Error::IntristicNotFound(721						ns.into(),722						name.into(),723					))?,724				},725				Val::Func(f) => {726					let body = || f.evaluate(context, args, *tailstrict);727					if *tailstrict {728						body()?729					} else {730						push(loc, "function call", body)?731					}732				}733				v => create_error(crate::error::Error::OnlyFunctionsCanBeCalledGot(734					v.value_type()?,735				))?,736			}737		}738		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),739		AssertExpr(AssertStmt(value, msg), returned) => {740			let assertion_result = push(&value.1, "assertion condition", || {741				evaluate(context.clone(), &value)?742					.try_cast_bool("assertion condition should be boolean")743			})?;744			if assertion_result {745				evaluate(context, returned)?746			} else if let Some(msg) = msg {747				panic!(748					"assertion failed ({:?}): {}",749					value,750					evaluate(context, msg)?.try_cast_str("assertion message should be string")?751				);752			} else {753				panic!("assertion failed ({:?}): no message", value);754			}755		}756		Error(e) => create_error(crate::Error::RuntimeError(757			evaluate(context, e)?.try_cast_str("error text should be string")?,758		))?,759		IfElse {760			cond,761			cond_then,762			cond_else,763		} => {764			if evaluate(context.clone(), &cond.0)?765				.try_cast_bool("if condition should be boolean")?766			{767				evaluate(context, cond_then)?768			} else {769				match cond_else {770					Some(v) => evaluate(context, v)?,771					None => Val::Null,772				}773			}774		}775		Import(path) => {776			let mut tmp = loc777				.clone()778				.expect("imports can't be used without loc_data")779				.0;780			let import_location = Rc::make_mut(&mut tmp);781			import_location.pop();782			with_state(|s| s.import_file(&import_location, path))?783		}784		ImportStr(path) => {785			let mut tmp = loc786				.clone()787				.expect("imports can't be used without loc_data")788				.0;789			let import_location = Rc::make_mut(&mut tmp);790			import_location.pop();791			Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)792		}793		Literal(LiteralType::Super) => return create_error(crate::error::Error::StandaloneSuper),794	})795}
modifiedcrates/jsonnet-evaluator/src/import.rsdiffbeforeafterboth
--- a/crates/jsonnet-evaluator/src/import.rs
+++ b/crates/jsonnet-evaluator/src/import.rs
@@ -62,8 +62,9 @@
 	}
 }
 
+type ResolutionData = (PathBuf, PathBuf);
 pub struct CachingImportResolver {
-	resolution_cache: RefCell<HashMap<(PathBuf, PathBuf), Result<Rc<PathBuf>>>>,
+	resolution_cache: RefCell<HashMap<ResolutionData, Result<Rc<PathBuf>>>>,
 	loading_cache: RefCell<HashMap<PathBuf, Result<Rc<str>>>>,
 	inner: Box<dyn ImportResolver>,
 }
modifiedcrates/jsonnet-evaluator/src/lib.rsdiffbeforeafterboth
--- a/crates/jsonnet-evaluator/src/lib.rs
+++ b/crates/jsonnet-evaluator/src/lib.rs
@@ -200,7 +200,7 @@
 			self.0
 				.str_files
 				.borrow_mut()
-				.insert(path.clone(), file_str.into());
+				.insert(path.clone(), file_str);
 		}
 		Ok(self.0.str_files.borrow().get(&path).cloned().unwrap())
 	}