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

difftreelog

feat trace function names

Лач2020-07-16parent: #0dad2a6.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	ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, ExprLocation, FieldMember,9	ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,10	Visibility,11};12use std::{collections::HashMap, rc::Rc};1314pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (Rc<str>, LazyBinding) {15	let b = b.clone();16	if let Some(params) = &b.params {17		let params = params.clone();18		(19			b.name.clone(),20			LazyBinding::Bindable(Rc::new(move |this, super_obj| {21				Ok(lazy_val!(22					closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(23						context_creator.0(this.clone(), super_obj.clone())?,24						params.clone(),25						b.value.clone(),26					)))27				))28			})),29		)30	} else {31		(32			b.name.clone(),33			LazyBinding::Bindable(Rc::new(move |this, super_obj| {34				Ok(lazy_val!(closure!(clone context_creator, clone b, ||35					push(&b.value.1, "thunk", ||{36						evaluate(37							context_creator.0(this.clone(), super_obj.clone())?,38							&b.value39						)40					})41				)))42			})),43		)44	}45}4647pub fn evaluate_method(ctx: Context, params: ParamsDesc, body: LocExpr) -> Val {48	Val::Func(FuncDesc { ctx, params, body })49}5051pub fn evaluate_field_name(52	context: Context,53	field_name: &jrsonnet_parser::FieldName,54) -> Result<Option<Rc<str>>> {55	Ok(match field_name {56		jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),57		jrsonnet_parser::FieldName::Dyn(expr) => {58			let lazy = evaluate(context, expr)?;59			let value = lazy.unwrap_if_lazy()?;60			if matches!(value, Val::Null) {61				None62			} else {63				Some(value.try_cast_str("dynamic field name")?)64			}65		}66	})67}6869pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {70	Ok(match (op, b) {71		(o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,72		(UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),73		(UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),74		(UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),75		(op, o) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(76			op,77			o.value_type()?,78		))?,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::new_checked_num(v1 + v2)?,96		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(97			BinaryOpType::Add,98			a.value_type()?,99			b.value_type()?,100		))?,101	})102}103104pub fn evaluate_binary_op_special(105	context: Context,106	a: &LocExpr,107	op: BinaryOpType,108	b: &LocExpr,109) -> Result<Val> {110	Ok(111		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {112			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),113			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),114			(a, op, eb) => {115				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?116			}117		},118	)119}120121pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {122	Ok(match (a, op, b) {123		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,124125		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),126127		// Bool X Bool128		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),129		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),130131		// Str X Str132		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),133		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),134		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),135		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),136137		// Num X Num138		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::new_checked_num(v1 * v2)?,139		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {140			if *v2 <= f64::EPSILON {141				create_error_result(crate::Error::DivisionByZero)?142			}143			Val::new_checked_num(v1 / v2)?144		}145146		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::new_checked_num(v1 - v2)?,147148		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),149		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),150		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),151		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),152153		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {154			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)155		}156		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {157			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)158		}159		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {160			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)161		}162		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {163			if *v2 < 0.0 {164				create_error_result(Error::RuntimeError("shift by negative exponent".into()))?165			}166			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)167		}168		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {169			if *v2 < 0.0 {170				create_error_result(Error::RuntimeError("shift by negative exponent".into()))?171			}172			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)173		}174175		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(176			op,177			a.value_type()?,178			b.value_type()?,179		))?,180	})181}182183future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);184future_wrapper!(ObjValue, FutureObjValue);185186pub fn evaluate_comp<T>(187	context: Context,188	value: &impl Fn(Context) -> Result<T>,189	specs: &[CompSpec],190) -> Result<Option<Vec<T>>> {191	Ok(match specs.get(0) {192		None => Some(vec![value(context)?]),193		Some(CompSpec::IfSpec(IfSpecData(cond))) => {194			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {195				evaluate_comp(context, value, &specs[1..])?196			} else {197				None198			}199		}200		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {201			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {202				Val::Arr(list) => {203					let mut out = Vec::new();204					for item in list.iter() {205						let item = item.unwrap_if_lazy()?;206						out.push(evaluate_comp(207							context.with_var(var.clone(), item.clone())?,208							value,209							&specs[1..],210						)?);211					}212					Some(out.into_iter().flatten().flatten().collect())213				}214				_ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,215			}216		}217	})218}219220pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {221	let new_bindings = FutureNewBindings::new();222	let future_this = FutureObjValue::new();223	let context_creator = context_creator!(224		closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {225			Ok(context.extend_unbound(226				new_bindings.clone().unwrap(),227				context.dollar().clone().or_else(||this.clone()),228				Some(this.unwrap()),229				super_obj230			)?)231		})232	);233	{234		let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();235		for (n, b) in members236			.iter()237			.filter_map(|m| match m {238				Member::BindStmt(b) => Some(b.clone()),239				_ => None,240			})241			.map(|b| evaluate_binding(&b, context_creator.clone()))242		{243			bindings.insert(n, b);244		}245		new_bindings.fill(bindings);246	}247248	let mut new_members = HashMap::new();249	for member in members.iter() {250		match member {251			Member::Field(FieldMember {252				name,253				plus,254				params: None,255				visibility,256				value,257			}) => {258				let name = evaluate_field_name(context.clone(), &name)?;259				if name.is_none() {260					continue;261				}262				let name = name.unwrap();263				new_members.insert(264					name.clone(),265					ObjMember {266						add: *plus,267						visibility: *visibility,268						invoke: LazyBinding::Bindable(Rc::new(269							closure!(clone name, clone value, clone context_creator, |this, super_obj| {270								Ok(LazyVal::new_resolved(evaluate(271									context_creator.0(this, super_obj)?,272									&value,273								)?))274							}),275						)),276						location: value.1.clone(),277					},278				);279			}280			Member::Field(FieldMember {281				name,282				params: Some(params),283				value,284				..285			}) => {286				let name = evaluate_field_name(context.clone(), &name)?;287				if name.is_none() {288					continue;289				}290				let name = name.unwrap();291				new_members.insert(292					name,293					ObjMember {294						add: false,295						visibility: Visibility::Hidden,296						invoke: LazyBinding::Bindable(Rc::new(297							closure!(clone value, clone context_creator, clone params, |this, super_obj| {298								// TODO: Assert299								Ok(LazyVal::new_resolved(evaluate_method(300									context_creator.0(this, super_obj)?,301									params.clone(),302									value.clone(),303								)))304							}),305						)),306						location: value.1.clone(),307					},308				);309			}310			Member::BindStmt(_) => {}311			Member::AssertStmt(_) => {}312		}313	}314	Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))315}316317pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {318	Ok(match object {319		ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,320		ObjBody::ObjComp(obj) => {321			let future_this = FutureObjValue::new();322			let mut new_members = HashMap::new();323			for (k, v) in evaluate_comp(324				context.clone(),325				&|ctx| {326					let new_bindings = FutureNewBindings::new();327					let context_creator = context_creator!(328						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {329							Ok(context.extend_unbound(330								new_bindings.clone().unwrap(),331								context.dollar().clone().or_else(||this.clone()),332								None,333								super_obj334							)?)335						})336					);337					let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();338					for (n, b) in obj339						.pre_locals340						.iter()341						.chain(obj.post_locals.iter())342						.map(|b| evaluate_binding(b, context_creator.clone()))343					{344						bindings.insert(n, b);345					}346					let bindings = new_bindings.fill(bindings);347					let ctx = ctx.extend_unbound(bindings, None, None, None)?;348					let key = evaluate(ctx.clone(), &obj.key)?;349					let value = LazyBinding::Bindable(Rc::new(350						closure!(clone ctx, clone obj.value, |this, _super_obj| {351							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))352						}),353					));354355					Ok((key, value))356				},357				&obj.compspecs,358			)?359			.unwrap()360			{361				match k {362					Val::Null => {}363					Val::Str(n) => {364						new_members.insert(365							n,366							ObjMember {367								add: false,368								visibility: Visibility::Normal,369								invoke: v,370								location: obj.value.1.clone(),371							},372						);373					}374					v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,375				}376			}377378			future_this.fill(ObjValue::new(None, Rc::new(new_members)))379		}380	})381}382383/// Extracts code block and disables inlining for them384/// Fixes WASM to java bytecode compilation failing because of very large method385macro_rules! noinline {386	($e:expr) => {387		(#[inline(never)]388		move || $e)()389	};390}391392pub fn evaluate_apply(393	context: Context,394	value: &LocExpr,395	args: &ArgsDesc,396	loc: &Option<ExprLocation>,397	tailstrict: bool,398) -> Result<Val> {399	let lazy = evaluate(context.clone(), value)?;400	let value = lazy.unwrap_if_lazy()?;401	Ok(match value {402		Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {403			// arr/string/function404			("std", "length") => noinline!(parse_args!(context, "std.length", args, 1, [405				0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];406			], {407				Ok(match x {408					Val::Str(n) => Val::Num(n.chars().count() as f64),409					Val::Arr(i) => Val::Num(i.len() as f64),410					Val::Obj(o) => Val::Num(411						o.fields_visibility()412							.into_iter()413							.filter(|(_k, v)| *v)414							.count() as f64,415					),416					_ => unreachable!(),417				})418			}))?,419			// any420			("std", "type") => parse_args!(context, "std.type", args, 1, [421				0, x, vec![];422			], {423				Val::Str(x.value_type()?.name().into())424			}),425			// length, idx=>any426			("std", "makeArray") => noinline!(parse_args!(context, "std.makeArray", args, 2, [427				0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];428				1, func: [Val::Func]!!Val::Func, vec![ValType::Func];429			], {430				if sz < 0.0 {431					create_error_result(crate::error::Error::RuntimeError(format!("makeArray requires size >= 0, got {}", sz).into()))?;432				}433				let mut out = Vec::with_capacity(sz as usize);434				for i in 0..sz as usize {435					out.push(func.evaluate_values(436						Context::new(),437						&[Val::Num(i as f64)]438					)?)439				}440				Ok(Val::Arr(Rc::new(out)))441			}))?,442			// string443			("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [444				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];445			], {446				assert!(447					str.chars().count() == 1,448					"std.codepoint should receive single char string"449				);450				Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)451			}),452			// object, includeHidden453			("std", "objectFieldsEx") => {454				noinline!(parse_args!(context, "std.objectFieldsEx",args, 2, [455					0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];456					1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];457				], {458					let mut out = obj.fields_visibility()459						.into_iter()460						.filter(|(_k, v)| *v || inc_hidden)461						.map(|(k, _v)|k)462						.collect::<Vec<_>>();463					out.sort();464					Ok(Val::Arr(Rc::new(out.into_iter().map(Val::Str).collect())))465				}))?466			}467			// object, field, includeHidden468			("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [469				0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];470				1, f: [Val::Str]!!Val::Str, vec![ValType::Str];471				2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];472			], {473				Val::Bool(474					obj.fields_visibility()475						.into_iter()476						.filter(|(_k, v)| *v || inc_hidden)477						.any(|(k, _v)| *k == *f),478				)479			}),480			("std", "primitiveEquals") => parse_args!(context, "std.primitiveEquals", args, 2, [481				0, a, vec![];482				1, b, vec![];483			], {484				Val::Bool(primitive_equals(&a, &b)?)485			}),486			// faster487			("std", "equals") => parse_args!(context, "std.equals", args, 2, [488				0, a, vec![];489				1, b, vec![];490			], {491				Val::Bool(equals(&a, &b)?)492			}),493			("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [494				0, a: [Val::Num]!!Val::Num, vec![ValType::Num];495				1, b: [Val::Num]!!Val::Num, vec![ValType::Num];496			], {497				Val::Num(a % b)498			}),499			("std", "floor") => parse_args!(context, "std.floor", args, 1, [500				0, x: [Val::Num]!!Val::Num, vec![ValType::Num];501			], {502				Val::Num(x.floor())503			}),504			("std", "trace") => parse_args!(context, "std.trace", args, 2, [505				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];506				1, rest, vec![];507			], {508				eprint!("TRACE: ");509				if let Some(loc) = loc {510					with_state(|s|{511						let locs = s.map_source_locations(&loc.0, &[loc.1]);512						eprint!("{}:{} ", loc.0.display(), locs[0].line);513					});514				}515				eprintln!("{}", str);516				rest517			}),518			("std", "pow") => parse_args!(context, "std.modulo", args, 2, [519				0, x: [Val::Num]!!Val::Num, vec![ValType::Num];520				1, n: [Val::Num]!!Val::Num, vec![ValType::Num];521			], {522				Val::Num(x.powf(n))523			}),524			("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [525				0, x: [Val::Str]!!Val::Str, vec![ValType::Str];526			], {527				with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(528					|| create_error(crate::Error::UndefinedExternalVariable(x)),529				)?530			}),531			("std", "filter") => noinline!(parse_args!(context, "std.filter", args, 2, [532				0, func: [Val::Func]!!Val::Func, vec![ValType::Func];533				1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];534			], {535				Ok(Val::Arr(Rc::new(536					arr.iter()537						.cloned()538						.filter(|e| {539							func540								.evaluate_values(context.clone(), &[e.clone()])541								.unwrap()542								.try_cast_bool("filter predicate")543								.unwrap()544						})545						.collect(),546				)))547			}))?,548			("std", "char") => parse_args!(context, "std.char", args, 1, [549				0, n: [Val::Num]!!Val::Num, vec![ValType::Num];550			], {551				let mut out = String::new();552				out.push(std::char::from_u32(n as u32).unwrap());553				Val::Str(out.into())554			}),555			("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [556				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];557			], {558				Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))559			}),560			("std", "md5") => noinline!(parse_args!(context, "std.md5", args, 1, [561				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];562			], {563				Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))564			}))?,565			// faster566			("std", "base64") => parse_args!(context, "std.base64", args, 1, [567				0, input: [Val::Str | Val::Arr], vec![ValType::Arr, ValType::Str];568			], {569				Val::Str(match input {570					Val::Str(s) => {571						base64::encode(s.bytes().collect::<Vec<_>>()).into()572					},573					Val::Arr(a) => {574						base64::encode(a.iter().map(|v| {575							Ok(v.clone().try_cast_num("base64 array")? as u8)576						}).collect::<Result<Vec<_>>>()?).into()577					},578					_ => unreachable!()579				})580			}),581			// faster582			("std", "join") => noinline!(parse_args!(context, "std.join", args, 2, [583				0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];584				1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];585			], {586				Ok(match sep {587					Val::Arr(joiner_items) => {588						let mut out = Vec::new();589590						let mut first = true;591						for item in arr.iter().cloned() {592							if let Val::Arr(items) = item.unwrap_if_lazy()? {593								if !first {594									out.reserve(joiner_items.len());595									out.extend(joiner_items.iter().cloned());596								}597								first = false;598								out.reserve(items.len());599								out.extend(items.iter().cloned());600							} else {601								create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;602							}603						}604605						Val::Arr(Rc::new(out))606					},607					Val::Str(sep) => {608						let mut out = String::new();609610						let mut first = true;611						for item in arr.iter().cloned() {612							if let Val::Str(item) = item.unwrap_if_lazy()? {613								if !first {614									out += &sep;615								}616								first = false;617								out += &item;618							} else {619								create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;620							}621						}622623						Val::Str(out.into())624					},625					_ => unreachable!()626				})627			}))?,628			// Faster629			("std", "escapeStringJson") => parse_args!(context, "std.escapeStringJson", args, 1, [630				0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];631			], {632				Val::Str(escape_string_json(&str_).into())633			}),634			// Faster635			("std", "manifestJsonEx") => parse_args!(context, "std.manifestJsonEx", args, 2, [636				0, value, vec![];637				1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];638			], {639				Val::Str(manifest_json_ex(&value, &indent)?.into())640			}),641			(ns, name) => {642				create_error_result(crate::Error::IntristicNotFound(ns.into(), name.into()))?643			}644		},645		Val::Func(f) => {646			let body = || f.evaluate(context, args, tailstrict);647			if tailstrict {648				body()?649			} else {650				push(loc, "function call", body)?651			}652		}653		v => create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?,654	})655}656657pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {658	use Expr::*;659	let LocExpr(expr, loc) = expr;660	Ok(match &**expr {661		Literal(LiteralType::This) => Val::Obj(662			context663				.this()664				.clone()665				.ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,666		),667		Literal(LiteralType::Dollar) => Val::Obj(668			context669				.dollar()670				.clone()671				.ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,672		),673		Literal(LiteralType::True) => Val::Bool(true),674		Literal(LiteralType::False) => Val::Bool(false),675		Literal(LiteralType::Null) => Val::Null,676		Parened(e) => evaluate(context, e)?,677		Str(v) => Val::Str(v.clone()),678		Num(v) => Val::new_checked_num(*v)?,679		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,680		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,681		Var(name) => push(682			loc,683			|| "var".to_owned(),684			|| Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?),685		)?,686		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {687			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;688			context689				.super_obj()690				.clone()691				.expect("no super found")692				.get_raw(&name, &context.this().clone().expect("no this found"))?693				.expect("value not found")694		}695		Index(value, index) => {696			match (697				evaluate(context.clone(), value)?.unwrap_if_lazy()?,698				evaluate(context, index)?,699			) {700				(Val::Obj(v), Val::Str(s)) => {701					if let Some(v) = v.get(s.clone())? {702						v.unwrap_if_lazy()?703					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {704						Val::Intristic(n, s)705					} else {706						create_error_result(crate::Error::NoSuchField(s))?707					}708				}709				(Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(710					ValType::Obj,711					ValType::Str,712					n.value_type()?,713				))?,714715				(Val::Arr(v), Val::Num(n)) => {716					if n.fract() > f64::EPSILON {717						create_error_result(crate::Error::FractionalIndex)?718					}719					v.get(n as usize)720						.ok_or_else(|| {721							create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))722						})?723						.clone()724						.unwrap_if_lazy()?725				}726				(Val::Arr(_), Val::Str(n)) => {727					create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?728				}729				(Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(730					ValType::Arr,731					ValType::Num,732					n.value_type()?,733				))?,734735				(Val::Str(s), Val::Num(n)) => Val::Str(736					s.chars()737						.skip(n as usize)738						.take(1)739						.collect::<String>()740						.into(),741				),742				(Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(743					ValType::Str,744					ValType::Num,745					n.value_type()?,746				))?,747748				(v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,749			}750		}751		LocalExpr(bindings, returned) => {752			let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();753			let future_context = Context::new_future();754755			let context_creator = context_creator!(756				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))757			);758759			for (k, v) in bindings760				.iter()761				.map(|b| evaluate_binding(b, context_creator.clone()))762			{763				new_bindings.insert(k, v);764			}765766			let context = context767				.extend_unbound(new_bindings, None, None, None)?768				.into_future(future_context);769			evaluate(context, &returned.clone())?770		}771		Arr(items) => {772			let mut out = Vec::with_capacity(items.len());773			for item in items {774				out.push(Val::Lazy(lazy_val!(775					closure!(clone context, clone item, || {776						evaluate(context.clone(), &item)777					})778				)));779			}780			Val::Arr(Rc::new(out))781		}782		ArrComp(expr, compspecs) => Val::Arr(783			// First compspec should be forspec, so no "None" possible here784			Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),785		),786		Obj(body) => Val::Obj(evaluate_object(context, body)?),787		ObjExtend(s, t) => evaluate_add_op(788			&evaluate(context.clone(), s)?,789			&Val::Obj(evaluate_object(context, t)?),790		)?,791		Apply(value, args, tailstrict) => evaluate_apply(context, value, args, loc, *tailstrict)?,792		Function(params, body) => evaluate_method(context, params.clone(), body.clone()),793		AssertExpr(AssertStmt(value, msg), returned) => {794			let assertion_result = push(795				&value.1,796				|| "assertion condition".to_owned(),797				|| {798					evaluate(context.clone(), &value)?799						.try_cast_bool("assertion condition should be boolean")800				},801			)?;802			if assertion_result {803				evaluate(context, returned)?804			} else if let Some(msg) = msg {805				create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?806			} else {807				create_error_result(crate::Error::AssertionFailed(Val::Null))?808			}809		}810		Error(e) => push(811			&loc,812			|| "error statement".to_owned(),813			|| {814				create_error_result(crate::Error::RuntimeError(815					evaluate(context, e)?.try_cast_str("error text should be string")?,816				))?817			},818		)?,819		IfElse {820			cond,821			cond_then,822			cond_else,823		} => {824			if evaluate(context.clone(), &cond.0)?825				.try_cast_bool("if condition should be boolean")?826			{827				evaluate(context, cond_then)?828			} else {829				match cond_else {830					Some(v) => evaluate(context, v)?,831					None => Val::Null,832				}833			}834		}835		Import(path) => {836			let mut tmp = loc837				.clone()838				.expect("imports can't be used without loc_data")839				.0;840			let import_location = Rc::make_mut(&mut tmp);841			import_location.pop();842			with_state(|s| s.import_file(&import_location, path))?843		}844		ImportStr(path) => {845			let mut tmp = loc846				.clone()847				.expect("imports can't be used without loc_data")848				.0;849			let import_location = Rc::make_mut(&mut tmp);850			import_location.pop();851			Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)852		}853		Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),854	})855}
after · crates/jrsonnet-evaluator/src/evaluate.rs
1use crate::{2	context_creator, create_error, create_error_result, equals, escape_string_json, future_wrapper,3	lazy_val, manifest_json_ex, parse_args, primitive_equals, push, with_state, Context,4	ContextCreator, Error, FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue, Result, Val,5	ValType,6};7use closure::closure;8use jrsonnet_parser::{9	ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, ExprLocation, FieldMember,10	ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,11	Visibility,12};13use std::{collections::HashMap, rc::Rc};1415pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (Rc<str>, LazyBinding) {16	let b = b.clone();17	if let Some(params) = &b.params {18		let params = params.clone();19		(20			b.name.clone(),21			LazyBinding::Bindable(Rc::new(move |this, super_obj| {22				Ok(lazy_val!(23					closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(24						context_creator.0(this.clone(), super_obj.clone())?,25						b.name.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						evaluate_named(38							context_creator.0(this.clone(), super_obj.clone())?,39							&b.value,40							b.name.clone()41						)42				)))43			})),44		)45	}46}4748pub fn evaluate_method(ctx: Context, name: Rc<str>, params: ParamsDesc, body: LocExpr) -> Val {49	Val::Func(Rc::new(FuncDesc {50		name,51		ctx,52		params,53		body,54	}))55}5657pub fn evaluate_field_name(58	context: Context,59	field_name: &jrsonnet_parser::FieldName,60) -> Result<Option<Rc<str>>> {61	Ok(match field_name {62		jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),63		jrsonnet_parser::FieldName::Dyn(expr) => {64			let lazy = evaluate(context, expr)?;65			let value = lazy.unwrap_if_lazy()?;66			if matches!(value, Val::Null) {67				None68			} else {69				Some(value.try_cast_str("dynamic field name")?)70			}71		}72	})73}7475pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {76	Ok(match (op, b) {77		(o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,78		(UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),79		(UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),80		(UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),81		(op, o) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(82			op,83			o.value_type()?,84		))?,85	})86}8788pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {89	Ok(match (a, b) {90		(Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),9192		// Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)93		(Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),94		(Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),9596		(Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),97		(o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9899		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),100		(Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),101		(Val::Num(v1), Val::Num(v2)) => Val::new_checked_num(v1 + v2)?,102		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(103			BinaryOpType::Add,104			a.value_type()?,105			b.value_type()?,106		))?,107	})108}109110pub fn evaluate_binary_op_special(111	context: Context,112	a: &LocExpr,113	op: BinaryOpType,114	b: &LocExpr,115) -> Result<Val> {116	Ok(117		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {118			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),119			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),120			(a, op, eb) => {121				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?122			}123		},124	)125}126127pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {128	Ok(match (a, op, b) {129		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,130131		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),132133		// Bool X Bool134		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),135		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),136137		// Str X Str138		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),139		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),140		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),141		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),142143		// Num X Num144		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::new_checked_num(v1 * v2)?,145		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {146			if *v2 <= f64::EPSILON {147				create_error_result(crate::Error::DivisionByZero)?148			}149			Val::new_checked_num(v1 / v2)?150		}151152		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::new_checked_num(v1 - v2)?,153154		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),155		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),156		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),157		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),158159		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {160			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)161		}162		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {163			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)164		}165		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {166			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)167		}168		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {169			if *v2 < 0.0 {170				create_error_result(Error::RuntimeError("shift by negative exponent".into()))?171			}172			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)173		}174		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {175			if *v2 < 0.0 {176				create_error_result(Error::RuntimeError("shift by negative exponent".into()))?177			}178			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)179		}180181		_ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(182			op,183			a.value_type()?,184			b.value_type()?,185		))?,186	})187}188189future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);190future_wrapper!(ObjValue, FutureObjValue);191192pub fn evaluate_comp<T>(193	context: Context,194	value: &impl Fn(Context) -> Result<T>,195	specs: &[CompSpec],196) -> Result<Option<Vec<T>>> {197	Ok(match specs.get(0) {198		None => Some(vec![value(context)?]),199		Some(CompSpec::IfSpec(IfSpecData(cond))) => {200			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {201				evaluate_comp(context, value, &specs[1..])?202			} else {203				None204			}205		}206		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {207			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {208				Val::Arr(list) => {209					let mut out = Vec::new();210					for item in list.iter() {211						let item = item.unwrap_if_lazy()?;212						out.push(evaluate_comp(213							context.with_var(var.clone(), item.clone())?,214							value,215							&specs[1..],216						)?);217					}218					Some(out.into_iter().flatten().flatten().collect())219				}220				_ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,221			}222		}223	})224}225226pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {227	let new_bindings = FutureNewBindings::new();228	let future_this = FutureObjValue::new();229	let context_creator = context_creator!(230		closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {231			Ok(context.extend_unbound(232				new_bindings.clone().unwrap(),233				context.dollar().clone().or_else(||this.clone()),234				Some(this.unwrap()),235				super_obj236			)?)237		})238	);239	{240		let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();241		for (n, b) in members242			.iter()243			.filter_map(|m| match m {244				Member::BindStmt(b) => Some(b.clone()),245				_ => None,246			})247			.map(|b| evaluate_binding(&b, context_creator.clone()))248		{249			bindings.insert(n, b);250		}251		new_bindings.fill(bindings);252	}253254	let mut new_members = HashMap::new();255	for member in members.iter() {256		match member {257			Member::Field(FieldMember {258				name,259				plus,260				params: None,261				visibility,262				value,263			}) => {264				let name = evaluate_field_name(context.clone(), &name)?;265				if name.is_none() {266					continue;267				}268				let name = name.unwrap();269				new_members.insert(270					name.clone(),271					ObjMember {272						add: *plus,273						visibility: *visibility,274						invoke: LazyBinding::Bindable(Rc::new(275							closure!(clone name, clone value, clone context_creator, |this, super_obj| {276								Ok(LazyVal::new_resolved(evaluate(277									context_creator.0(this, super_obj)?,278									&value,279								)?))280							}),281						)),282						location: value.1.clone(),283					},284				);285			}286			Member::Field(FieldMember {287				name,288				params: Some(params),289				value,290				..291			}) => {292				let name = evaluate_field_name(context.clone(), &name)?;293				if name.is_none() {294					continue;295				}296				let name = name.unwrap();297				new_members.insert(298					name.clone(),299					ObjMember {300						add: false,301						visibility: Visibility::Hidden,302						invoke: LazyBinding::Bindable(Rc::new(303							closure!(clone value, clone context_creator, clone params, clone name, |this, super_obj| {304								// TODO: Assert305								Ok(LazyVal::new_resolved(evaluate_method(306									context_creator.0(this, super_obj)?,307									name.clone(),308									params.clone(),309									value.clone(),310								)))311							}),312						)),313						location: value.1.clone(),314					},315				);316			}317			Member::BindStmt(_) => {}318			Member::AssertStmt(_) => {}319		}320	}321	Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))322}323324pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {325	Ok(match object {326		ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,327		ObjBody::ObjComp(obj) => {328			let future_this = FutureObjValue::new();329			let mut new_members = HashMap::new();330			for (k, v) in evaluate_comp(331				context.clone(),332				&|ctx| {333					let new_bindings = FutureNewBindings::new();334					let context_creator = context_creator!(335						closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {336							Ok(context.extend_unbound(337								new_bindings.clone().unwrap(),338								context.dollar().clone().or_else(||this.clone()),339								None,340								super_obj341							)?)342						})343					);344					let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();345					for (n, b) in obj346						.pre_locals347						.iter()348						.chain(obj.post_locals.iter())349						.map(|b| evaluate_binding(b, context_creator.clone()))350					{351						bindings.insert(n, b);352					}353					let bindings = new_bindings.fill(bindings);354					let ctx = ctx.extend_unbound(bindings, None, None, None)?;355					let key = evaluate(ctx.clone(), &obj.key)?;356					let value = LazyBinding::Bindable(Rc::new(357						closure!(clone ctx, clone obj.value, |this, _super_obj| {358							Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))359						}),360					));361362					Ok((key, value))363				},364				&obj.compspecs,365			)?366			.unwrap()367			{368				match k {369					Val::Null => {}370					Val::Str(n) => {371						new_members.insert(372							n,373							ObjMember {374								add: false,375								visibility: Visibility::Normal,376								invoke: v,377								location: obj.value.1.clone(),378							},379						);380					}381					v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,382				}383			}384385			future_this.fill(ObjValue::new(None, Rc::new(new_members)))386		}387	})388}389390/// Extracts code block and disables inlining for them391/// Fixes WASM to java bytecode compilation failing because of very large method392macro_rules! noinline {393	($e:expr) => {394		(#[inline(never)]395		move || $e)()396	};397}398399pub fn evaluate_apply(400	context: Context,401	value: &LocExpr,402	args: &ArgsDesc,403	loc: &Option<ExprLocation>,404	tailstrict: bool,405) -> Result<Val> {406	let lazy = evaluate(context.clone(), value)?;407	let value = lazy.unwrap_if_lazy()?;408	Ok(match value {409		Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {410			// arr/string/function411			("std", "length") => noinline!(parse_args!(context, "std.length", args, 1, [412				0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];413			], {414				Ok(match x {415					Val::Str(n) => Val::Num(n.chars().count() as f64),416					Val::Arr(i) => Val::Num(i.len() as f64),417					Val::Obj(o) => Val::Num(418						o.fields_visibility()419							.into_iter()420							.filter(|(_k, v)| *v)421							.count() as f64,422					),423					_ => unreachable!(),424				})425			}))?,426			// any427			("std", "type") => parse_args!(context, "std.type", args, 1, [428				0, x, vec![];429			], {430				Val::Str(x.value_type()?.name().into())431			}),432			// length, idx=>any433			("std", "makeArray") => noinline!(parse_args!(context, "std.makeArray", args, 2, [434				0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];435				1, func: [Val::Func]!!Val::Func, vec![ValType::Func];436			], {437				if sz < 0.0 {438					create_error_result(crate::error::Error::RuntimeError(format!("makeArray requires size >= 0, got {}", sz).into()))?;439				}440				let mut out = Vec::with_capacity(sz as usize);441				for i in 0..sz as usize {442					out.push(func.evaluate_values(443						Context::new(),444						&[Val::Num(i as f64)]445					)?)446				}447				Ok(Val::Arr(Rc::new(out)))448			}))?,449			// string450			("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [451				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];452			], {453				assert!(454					str.chars().count() == 1,455					"std.codepoint should receive single char string"456				);457				Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)458			}),459			// object, includeHidden460			("std", "objectFieldsEx") => {461				noinline!(parse_args!(context, "std.objectFieldsEx",args, 2, [462					0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];463					1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];464				], {465					let mut out = obj.fields_visibility()466						.into_iter()467						.filter(|(_k, v)| *v || inc_hidden)468						.map(|(k, _v)|k)469						.collect::<Vec<_>>();470					out.sort();471					Ok(Val::Arr(Rc::new(out.into_iter().map(Val::Str).collect())))472				}))?473			}474			// object, field, includeHidden475			("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [476				0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];477				1, f: [Val::Str]!!Val::Str, vec![ValType::Str];478				2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];479			], {480				Val::Bool(481					obj.fields_visibility()482						.into_iter()483						.filter(|(_k, v)| *v || inc_hidden)484						.any(|(k, _v)| *k == *f),485				)486			}),487			("std", "primitiveEquals") => parse_args!(context, "std.primitiveEquals", args, 2, [488				0, a, vec![];489				1, b, vec![];490			], {491				Val::Bool(primitive_equals(&a, &b)?)492			}),493			// faster494			("std", "equals") => parse_args!(context, "std.equals", args, 2, [495				0, a, vec![];496				1, b, vec![];497			], {498				Val::Bool(equals(&a, &b)?)499			}),500			("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [501				0, a: [Val::Num]!!Val::Num, vec![ValType::Num];502				1, b: [Val::Num]!!Val::Num, vec![ValType::Num];503			], {504				Val::Num(a % b)505			}),506			("std", "floor") => parse_args!(context, "std.floor", args, 1, [507				0, x: [Val::Num]!!Val::Num, vec![ValType::Num];508			], {509				Val::Num(x.floor())510			}),511			("std", "trace") => parse_args!(context, "std.trace", args, 2, [512				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];513				1, rest, vec![];514			], {515				eprint!("TRACE: ");516				if let Some(loc) = loc {517					with_state(|s|{518						let locs = s.map_source_locations(&loc.0, &[loc.1]);519						eprint!("{}:{} ", loc.0.display(), locs[0].line);520					});521				}522				eprintln!("{}", str);523				rest524			}),525			("std", "pow") => parse_args!(context, "std.modulo", args, 2, [526				0, x: [Val::Num]!!Val::Num, vec![ValType::Num];527				1, n: [Val::Num]!!Val::Num, vec![ValType::Num];528			], {529				Val::Num(x.powf(n))530			}),531			("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [532				0, x: [Val::Str]!!Val::Str, vec![ValType::Str];533			], {534				with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(535					|| create_error(crate::Error::UndefinedExternalVariable(x)),536				)?537			}),538			("std", "filter") => noinline!(parse_args!(context, "std.filter", args, 2, [539				0, func: [Val::Func]!!Val::Func, vec![ValType::Func];540				1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];541			], {542				Ok(Val::Arr(Rc::new(543					arr.iter()544						.cloned()545						.filter(|e| {546							func547								.evaluate_values(context.clone(), &[e.clone()])548								.unwrap()549								.try_cast_bool("filter predicate")550								.unwrap()551						})552						.collect(),553				)))554			}))?,555			("std", "char") => parse_args!(context, "std.char", args, 1, [556				0, n: [Val::Num]!!Val::Num, vec![ValType::Num];557			], {558				let mut out = String::new();559				out.push(std::char::from_u32(n as u32).unwrap());560				Val::Str(out.into())561			}),562			("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [563				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];564			], {565				Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))566			}),567			("std", "md5") => noinline!(parse_args!(context, "std.md5", args, 1, [568				0, str: [Val::Str]!!Val::Str, vec![ValType::Str];569			], {570				Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))571			}))?,572			// faster573			("std", "base64") => parse_args!(context, "std.base64", args, 1, [574				0, input: [Val::Str | Val::Arr], vec![ValType::Arr, ValType::Str];575			], {576				Val::Str(match input {577					Val::Str(s) => {578						base64::encode(s.bytes().collect::<Vec<_>>()).into()579					},580					Val::Arr(a) => {581						base64::encode(a.iter().map(|v| {582							Ok(v.clone().try_cast_num("base64 array")? as u8)583						}).collect::<Result<Vec<_>>>()?).into()584					},585					_ => unreachable!()586				})587			}),588			// faster589			("std", "join") => noinline!(parse_args!(context, "std.join", args, 2, [590				0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];591				1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];592			], {593				Ok(match sep {594					Val::Arr(joiner_items) => {595						let mut out = Vec::new();596597						let mut first = true;598						for item in arr.iter().cloned() {599							if let Val::Arr(items) = item.unwrap_if_lazy()? {600								if !first {601									out.reserve(joiner_items.len());602									out.extend(joiner_items.iter().cloned());603								}604								first = false;605								out.reserve(items.len());606								out.extend(items.iter().cloned());607							} else {608								create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;609							}610						}611612						Val::Arr(Rc::new(out))613					},614					Val::Str(sep) => {615						let mut out = String::new();616617						let mut first = true;618						for item in arr.iter().cloned() {619							if let Val::Str(item) = item.unwrap_if_lazy()? {620								if !first {621									out += &sep;622								}623								first = false;624								out += &item;625							} else {626								create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;627							}628						}629630						Val::Str(out.into())631					},632					_ => unreachable!()633				})634			}))?,635			// Faster636			("std", "escapeStringJson") => parse_args!(context, "std.escapeStringJson", args, 1, [637				0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];638			], {639				Val::Str(escape_string_json(&str_).into())640			}),641			// Faster642			("std", "manifestJsonEx") => parse_args!(context, "std.manifestJsonEx", args, 2, [643				0, value, vec![];644				1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];645			], {646				Val::Str(manifest_json_ex(&value, &indent)?.into())647			}),648			(ns, name) => {649				create_error_result(crate::Error::IntristicNotFound(ns.into(), name.into()))?650			}651		},652		Val::Func(f) => {653			let body = || f.evaluate(context, args, tailstrict);654			if tailstrict {655				body()?656			} else {657				push(loc, || format!("function <{}> call", f.name), body)?658			}659		}660		v => create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?,661	})662}663664pub fn evaluate_named(context: Context, lexpr: &LocExpr, name: Rc<str>) -> Result<Val> {665	use Expr::*;666	let LocExpr(expr, _loc) = lexpr;667	Ok(match &**expr {668		Function(params, body) => evaluate_method(context, name, params.clone(), body.clone()),669		_ => evaluate(context, lexpr)?,670	})671}672673pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {674	use Expr::*;675	let LocExpr(expr, loc) = expr;676	Ok(match &**expr {677		Literal(LiteralType::This) => Val::Obj(678			context679				.this()680				.clone()681				.ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,682		),683		Literal(LiteralType::Dollar) => Val::Obj(684			context685				.dollar()686				.clone()687				.ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,688		),689		Literal(LiteralType::True) => Val::Bool(true),690		Literal(LiteralType::False) => Val::Bool(false),691		Literal(LiteralType::Null) => Val::Null,692		Parened(e) => evaluate(context, e)?,693		Str(v) => Val::Str(v.clone()),694		Num(v) => Val::new_checked_num(*v)?,695		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,696		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,697		Var(name) => push(698			loc,699			|| format!("variable <{}>", name),700			|| Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?),701		)?,702		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {703			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;704			context705				.super_obj()706				.clone()707				.expect("no super found")708				.get_raw(&name, &context.this().clone().expect("no this found"))?709				.expect("value not found")710		}711		Index(value, index) => {712			match (713				evaluate(context.clone(), value)?.unwrap_if_lazy()?,714				evaluate(context, index)?,715			) {716				(Val::Obj(v), Val::Str(s)) => {717					if let Some(v) = v.get(s.clone())? {718						v.unwrap_if_lazy()?719					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {720						Val::Intristic(n, s)721					} else {722						create_error_result(crate::Error::NoSuchField(s))?723					}724				}725				(Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(726					ValType::Obj,727					ValType::Str,728					n.value_type()?,729				))?,730731				(Val::Arr(v), Val::Num(n)) => {732					if n.fract() > f64::EPSILON {733						create_error_result(crate::Error::FractionalIndex)?734					}735					v.get(n as usize)736						.ok_or_else(|| {737							create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))738						})?739						.clone()740						.unwrap_if_lazy()?741				}742				(Val::Arr(_), Val::Str(n)) => {743					create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?744				}745				(Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(746					ValType::Arr,747					ValType::Num,748					n.value_type()?,749				))?,750751				(Val::Str(s), Val::Num(n)) => Val::Str(752					s.chars()753						.skip(n as usize)754						.take(1)755						.collect::<String>()756						.into(),757				),758				(Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(759					ValType::Str,760					ValType::Num,761					n.value_type()?,762				))?,763764				(v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,765			}766		}767		LocalExpr(bindings, returned) => {768			let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();769			let future_context = Context::new_future();770771			let context_creator = context_creator!(772				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))773			);774775			for (k, v) in bindings776				.iter()777				.map(|b| evaluate_binding(b, context_creator.clone()))778			{779				new_bindings.insert(k, v);780			}781782			let context = context783				.extend_unbound(new_bindings, None, None, None)?784				.into_future(future_context);785			evaluate(context, &returned.clone())?786		}787		Arr(items) => {788			let mut out = Vec::with_capacity(items.len());789			for item in items {790				out.push(Val::Lazy(lazy_val!(791					closure!(clone context, clone item, || {792						evaluate(context.clone(), &item)793					})794				)));795			}796			Val::Arr(Rc::new(out))797		}798		ArrComp(expr, compspecs) => Val::Arr(799			// First compspec should be forspec, so no "None" possible here800			Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),801		),802		Obj(body) => Val::Obj(evaluate_object(context, body)?),803		ObjExtend(s, t) => evaluate_add_op(804			&evaluate(context.clone(), s)?,805			&Val::Obj(evaluate_object(context, t)?),806		)?,807		Apply(value, args, tailstrict) => evaluate_apply(context, value, args, loc, *tailstrict)?,808		Function(params, body) => {809			evaluate_method(context, "anonymous".into(), params.clone(), body.clone())810		}811		AssertExpr(AssertStmt(value, msg), returned) => {812			let assertion_result = push(813				&value.1,814				|| "assertion condition".to_owned(),815				|| {816					evaluate(context.clone(), &value)?817						.try_cast_bool("assertion condition should be boolean")818				},819			)?;820			if assertion_result {821				evaluate(context, returned)?822			} else if let Some(msg) = msg {823				create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?824			} else {825				create_error_result(crate::Error::AssertionFailed(Val::Null))?826			}827		}828		Error(e) => push(829			&loc,830			|| "error statement".to_owned(),831			|| {832				create_error_result(crate::Error::RuntimeError(833					evaluate(context, e)?.try_cast_str("error text should be string")?,834				))?835			},836		)?,837		IfElse {838			cond,839			cond_then,840			cond_else,841		} => {842			if evaluate(context.clone(), &cond.0)?843				.try_cast_bool("if condition should be boolean")?844			{845				evaluate(context, cond_then)?846			} else {847				match cond_else {848					Some(v) => evaluate(context, v)?,849					None => Val::Null,850				}851			}852		}853		Import(path) => {854			let mut tmp = loc855				.clone()856				.expect("imports can't be used without loc_data")857				.0;858			let import_location = Rc::make_mut(&mut tmp);859			import_location.pop();860			with_state(|s| s.import_file(&import_location, path))?861		}862		ImportStr(path) => {863			let mut tmp = loc864				.clone()865				.expect("imports can't be used without loc_data")866				.0;867			let import_location = Rc::make_mut(&mut tmp);868			import_location.pop();869			Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)870		}871		Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),872	})873}
modifiedcrates/jrsonnet-evaluator/src/val.rsdiffbeforeafterboth
--- a/crates/jrsonnet-evaluator/src/val.rs
+++ b/crates/jrsonnet-evaluator/src/val.rs
@@ -58,6 +58,7 @@
 
 #[derive(Debug, PartialEq)]
 pub struct FuncDesc {
+	pub name: Rc<str>,
 	pub ctx: Context,
 	pub params: ParamsDesc,
 	pub body: LocExpr,