git.delta.rocks / jrsonnet / refs/commits / 1bfba233fc03

1use std::rc::Rc;23use jrsonnet_gcmodule::{Cc, Trace};4use jrsonnet_interner::IStr;5use jrsonnet_ir::{6	ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, ExprParams, FieldMember,7	FieldName, ForSpecData, IfSpecData, ImportKind, LiteralType, ObjBody, ObjMembers, Spanned,8	function::ParamName,9};10use jrsonnet_types::ValType;11use rustc_hash::FxHashMap;1213use self::destructure::destruct;14use crate::{15	Context, Error, ObjValue, ObjValueBuilder, ObjectAssertion, Pending, Result, ResultExt,16	SupThis, Unbound, Val,17	arr::ArrValue,18	bail,19	destructure::evaluate_dest,20	error::{ErrorKind::*, suggest_object_fields},21	evaluate::operator::{evaluate_binary_op_special, evaluate_unary_op},22	function::{CallLocation, FuncDesc, FuncVal},23	gc::WithCapacityExt as _,24	in_frame,25	typed::{FromUntyped, IntoUntyped as _, Typed},26	val::{CachedUnbound, IndexableVal, NumValue, StrValue, Thunk},27	with_state,28};29pub mod destructure;30pub mod operator;3132// This is the amount of bytes that need to be left on the stack before increasing the size.33// It must be at least as large as the stack required by any code that does not call34// `ensure_sufficient_stack`.35const RED_ZONE: usize = 100 * 1024; // 100k3637// Only the first stack that is pushed, grows exponentially (2^n * STACK_PER_RECURSION) from then38// on. This flag has performance relevant characteristics. Don't set it too high.39const STACK_PER_RECURSION: usize = 1024 * 1024; // 1MB4041/// Grows the stack on demand to prevent stack overflow. Call this in strategic locations42/// to "break up" recursive calls. E.g. almost any call to `visit_expr` or equivalent can benefit43/// from this.44///45/// Should not be sprinkled around carelessly, as it causes a little bit of overhead.46#[inline]47pub fn ensure_sufficient_stack<R>(f: impl FnOnce() -> R) -> R {48	stacker::maybe_grow(RED_ZONE, STACK_PER_RECURSION, f)49}5051pub fn evaluate_trivial(expr: &Expr) -> Option<Val> {52	fn is_trivial(expr: &Expr) -> bool {53		match expr {54			Expr::Str(_)55			| Expr::Num(_)56			| Expr::Literal(LiteralType::False | LiteralType::True | LiteralType::Null) => true,57			Expr::Arr(a) => a.iter().all(is_trivial),58			_ => false,59		}60	}61	Some(match expr {62		Expr::Str(s) => Val::string(s.clone()),63		Expr::Num(n) => {64			Val::Num(NumValue::new(*n).expect("parser will not allow non-finite values"))65		}66		Expr::Literal(LiteralType::False) => Val::Bool(false),67		Expr::Literal(LiteralType::True) => Val::Bool(true),68		Expr::Literal(LiteralType::Null) => Val::Null,69		Expr::Arr(n) => {70			if n.iter().any(|e| !is_trivial(e)) {71				return None;72			}73			Val::Arr(ArrValue::eager(74				n.iter()75					.map(evaluate_trivial)76					.map(|e| e.expect("checked trivial"))77					.collect(),78			))79		}80		_ => return None,81	})82}8384pub fn evaluate_method(ctx: Context, name: IStr, params: ExprParams, body: Rc<Expr>) -> Val {85	Val::Func(FuncVal::Normal(Cc::new(FuncDesc {86		name,87		ctx,88		params,89		body,90	})))91}9293pub fn evaluate_field_name(ctx: Context, field_name: &Spanned<FieldName>) -> Result<Option<IStr>> {94	Ok(match &field_name.value {95		FieldName::Fixed(n) => Some(n.clone()),96		FieldName::Dyn(expr) => in_frame(97			CallLocation::new(&field_name.span),98			|| "evaluating field name".to_string(),99			|| {100				let v = evaluate(ctx, expr)?;101				Ok(if matches!(v, Val::Null) {102					None103				} else {104					Some(IStr::from_untyped(v)?)105				})106			},107		)?,108	})109}110111pub fn evaluate_comp(112	ctx: Context,113	specs: &[CompSpec],114	callback: &mut impl FnMut(Context) -> Result<()>,115) -> Result<()> {116	match specs.first() {117		None => callback(ctx)?,118		Some(CompSpec::IfSpec(IfSpecData { cond, span: _ })) => {119			if bool::from_untyped(evaluate(ctx.clone(), cond)?)? {120				evaluate_comp(ctx, &specs[1..], callback)?;121			}122		}123		Some(CompSpec::ForSpec(ForSpecData {124			destruct: into,125			over,126		})) => {127			match evaluate(ctx.clone(), over)? {128				Val::Arr(list) => {129					for item in list.iter_lazy() {130						let fctx = Pending::new();131						let mut new_bindings = FxHashMap::with_capacity(into.binds_len());132						destruct(into, item, fctx.clone(), &mut new_bindings)?;133						let ctx = ctx.clone().extend_bindings(new_bindings).into_future(fctx);134135						evaluate_comp(ctx, &specs[1..], callback)?;136					}137				}138				#[cfg(feature = "exp-object-iteration")]139				Val::Obj(obj) => {140					for field in obj.fields(141						// TODO: Should there be ability to preserve iteration order?142						#[cfg(feature = "exp-preserve-order")]143						false,144					) {145						let fctx = Pending::new();146						let mut new_bindings = FxHashMap::with_capacity(into.binds_len());147						let obj = obj.clone();148						let value = Thunk::evaluated(Val::Arr(ArrValue::lazy(vec![149							Thunk::evaluated(Val::string(field.clone())),150							Thunk!(move || obj.get(field).transpose().expect(151								"field exists, as field name was obtained from object.fields()",152							)),153						])));154						destruct(into, value, fctx.clone(), &mut new_bindings)?;155						let ctx = ctx.clone().extend_bindings(new_bindings).into_future(fctx);156157						evaluate_comp(ctx, &specs[1..], callback)?;158					}159				}160				_ => bail!(InComprehensionCanOnlyIterateOverArray),161			}162		}163	}164	Ok(())165}166167trait CloneableUnbound<T>: Unbound<Bound = T> + Clone {}168impl<V, T> CloneableUnbound<T> for V where V: Unbound<Bound = T> + Clone {}169170fn evaluate_object_locals(171	fctx: Context,172	locals: Rc<Vec<BindSpec>>,173) -> impl CloneableUnbound<Context> {174	#[derive(Trace, Clone)]175	struct UnboundLocals {176		fctx: Context,177		locals: Rc<Vec<BindSpec>>,178	}179	impl Unbound for UnboundLocals {180		type Bound = Context;181182		fn bind(&self, sup_this: SupThis) -> Result<Context> {183			let fctx = Context::new_future();184			let mut new_bindings =185				FxHashMap::with_capacity(self.locals.iter().map(BindSpec::binds_len).sum());186			for b in self.locals.iter() {187				evaluate_dest(b, fctx.clone(), &mut new_bindings)?;188			}189190			let ctx = self.fctx.clone();191192			let ctx = ctx193				.extend_bindings_sup_this(new_bindings, sup_this)194				.into_future(fctx);195196			Ok(ctx)197		}198	}199200	UnboundLocals { fctx, locals }201}202203pub fn evaluate_field_member<B: Unbound<Bound = Context> + Clone>(204	builder: &mut ObjValueBuilder,205	ctx: Context,206	uctx: B,207	field: &FieldMember,208) -> Result<()> {209	let name = evaluate_field_name(ctx, &field.name)?;210	let Some(name) = name else {211		return Ok(());212	};213214	match field {215		FieldMember {216			plus,217			params: None,218			visibility,219			value,220			..221		} => {222			#[derive(Trace)]223			struct UnboundValue<B: Trace> {224				uctx: B,225				value: Rc<Expr>,226				name: IStr,227			}228			impl<B: Unbound<Bound = Context>> Unbound for UnboundValue<B> {229				type Bound = Val;230				fn bind(&self, sup_this: SupThis) -> Result<Val> {231					evaluate_named(self.uctx.bind(sup_this)?, &self.value, self.name.clone())232				}233			}234235			builder236				.field(name.clone())237				.with_add(*plus)238				.with_visibility(*visibility)239				.with_location(field.name.span.clone())240				.bindable(UnboundValue {241					uctx,242					value: value.clone(),243					name,244				})?;245		}246		FieldMember {247			params: Some(params),248			visibility,249			value,250			..251		} => {252			#[derive(Trace)]253			struct UnboundMethod<B: Trace> {254				uctx: B,255				value: Rc<Expr>,256				params: ExprParams,257				name: IStr,258			}259			impl<B: Unbound<Bound = Context>> Unbound for UnboundMethod<B> {260				type Bound = Val;261				fn bind(&self, sup_this: SupThis) -> Result<Val> {262					Ok(evaluate_method(263						self.uctx.bind(sup_this)?,264						self.name.clone(),265						self.params.clone(),266						self.value.clone(),267					))268				}269			}270271			builder272				.field(name.clone())273				.with_visibility(*visibility)274				// .with_location(value.span())275				.bindable(UnboundMethod {276					uctx,277					value: value.clone(),278					params: params.clone(),279					name,280				})?;281		}282	}283	Ok(())284}285286#[derive(Trace, Clone)]287struct DirectUnbound(Context);288impl Unbound for DirectUnbound {289	type Bound = Context;290	fn bind(&self, sup_this: SupThis) -> Result<Context> {291		Ok(self292			.0293			.clone()294			.extend_bindings_sup_this(FxHashMap::new(), sup_this))295	}296}297298#[allow(clippy::too_many_lines)]299pub fn evaluate_member_list_object(300	super_obj: Option<ObjValue>,301	ctx: Context,302	members: &ObjMembers,303) -> Result<ObjValue> {304	#[derive(Trace)]305	struct ObjectAssert<B: Trace> {306		uctx: B,307		asserts: Rc<Vec<AssertStmt>>,308	}309	impl<B: Unbound<Bound = Context>> ObjectAssertion for ObjectAssert<B> {310		fn run(&self, sup_this: SupThis) -> Result<()> {311			let ctx = self.uctx.bind(sup_this)?;312			for assert in &*self.asserts {313				evaluate_assert(ctx.clone(), assert)?;314			}315			Ok(())316		}317	}318319	let mut builder = ObjValueBuilder::new();320	if let Some(super_obj) = super_obj {321		builder.with_super(super_obj);322	}323324	if members.locals.is_empty() {325		// We can use the same context for all field evaluation, it doesn't depends on locals, only on this/super326		let uctx = DirectUnbound(ctx.clone());327		for field in &members.fields {328			evaluate_field_member(&mut builder, ctx.clone(), uctx.clone(), field)?;329		}330		if !members.asserts.is_empty() {331			builder.assert(ObjectAssert {332				uctx,333				asserts: members.asserts.clone(),334			});335		}336	} else {337		let locals = members.locals.clone();338		// We have single context for all fields, so we can cache them together339		let uctx = CachedUnbound::new(evaluate_object_locals(ctx.clone(), locals));340		for field in &members.fields {341			evaluate_field_member(&mut builder, ctx.clone(), uctx.clone(), field)?;342		}343		if !members.asserts.is_empty() {344			builder.assert(ObjectAssert {345				uctx,346				asserts: members.asserts.clone(),347			});348		}349	}350351	Ok(builder.build())352}353354pub fn evaluate_object(355	super_obj: Option<ObjValue>,356	ctx: Context,357	object: &ObjBody,358) -> Result<ObjValue> {359	Ok(match object {360		ObjBody::MemberList(members) => evaluate_member_list_object(super_obj, ctx, members)?,361		ObjBody::ObjComp(obj) => {362			let mut builder = ObjValueBuilder::new();363			if let Some(super_obj) = super_obj {364				builder.with_super(super_obj);365			}366			let locals = obj.locals.clone();367			evaluate_comp(ctx, &obj.compspecs, &mut |ctx| {368				let uctx = evaluate_object_locals(ctx.clone(), locals.clone());369370				evaluate_field_member(&mut builder, ctx, uctx, &obj.field)371			})?;372373			builder.build()374		}375	})376}377378pub fn evaluate_apply(379	ctx: Context,380	value: &Expr,381	args: &ArgsDesc,382	loc: CallLocation<'_>,383	tailstrict: bool,384) -> Result<Val> {385	let value = evaluate(ctx.clone(), value)?;386	Ok(match value {387		Val::Func(f) => {388			let body = || f.evaluate(ctx, loc, args, tailstrict);389			if tailstrict {390				body()?391			} else {392				in_frame(loc, || format!("function <{}> call", f.name()), body)?393			}394		}395		v => bail!(OnlyFunctionsCanBeCalledGot(v.value_type())),396	})397}398399pub fn evaluate_assert(ctx: Context, assertion: &AssertStmt) -> Result<()> {400	let value = &assertion.0;401	let msg = &assertion.1;402	let assertion_result = in_frame(403		CallLocation::new(&value.span),404		|| "assertion condition".to_owned(),405		|| bool::from_untyped(evaluate(ctx.clone(), value)?),406	)?;407	if !assertion_result {408		in_frame(409			CallLocation::new(&value.span),410			|| "assertion failure".to_owned(),411			|| {412				if let Some(msg) = msg {413					bail!(AssertionFailed(evaluate(ctx, msg)?.to_string()?));414				}415				bail!(AssertionFailed(Val::Null.to_string()?));416			},417		)?;418	}419	Ok(())420}421422pub fn evaluate_named_param(ctx: Context, expr: &Expr, name: ParamName) -> Result<Val> {423	match name {424		ParamName::Named(name) => evaluate_named(ctx, expr, name),425		ParamName::Unnamed => evaluate(ctx, expr),426	}427}428429pub fn evaluate_named(ctx: Context, expr: &Expr, name: IStr) -> Result<Val> {430	use Expr::*;431	Ok(match expr {432		Function(params, body) => evaluate_method(ctx, name, params.clone(), body.clone()),433		_ => evaluate(ctx, expr)?,434	})435}436437#[allow(clippy::too_many_lines)]438pub fn evaluate(ctx: Context, expr: &Expr) -> Result<Val> {439	use Expr::*;440441	Ok(match expr {442		Literal(LiteralType::This) => Val::Obj(ctx.try_this()?),443		Literal(LiteralType::Super) => Val::Obj(ctx.try_sup_this()?.standalone_super()?),444		Literal(LiteralType::Dollar) => Val::Obj(ctx.try_dollar()?),445		Literal(LiteralType::True) => Val::Bool(true),446		Literal(LiteralType::False) => Val::Bool(false),447		Literal(LiteralType::Null) => Val::Null,448		Str(v) => Val::string(v.clone()),449		Num(v) => Val::try_num(*v)?,450		// I have tried to remove special behavior from super by implementing standalone-super451		// expresion, but looks like this case still needs special treatment.452		//453		// Note that other jsonnet implementations will fail on `if value in (super)` expression,454		// because the standalone super literal is not supported, that is because in other455		// implementations `in super` treated differently from `in smth_else`.456		BinaryOp(bin)457			if matches!(&bin.rhs, Expr::Literal(LiteralType::Super))458				&& bin.op == BinaryOpType::In =>459		{460			let sup_this = ctx.try_sup_this()?;461			// In jsonnet, "field" in e is eager, LHS expression is always executed regardless of super existence.462			// In jrsonnet, however, this wasn't true, this was kept here for compatibility.463			if !sup_this.has_super() {464				return Ok(Val::Bool(false));465			}466			let field = evaluate(ctx, &bin.lhs)?;467			Val::Bool(sup_this.field_in_super(field.to_string()?))468		}469		BinaryOp(bin) => evaluate_binary_op_special(ctx, &bin.lhs, bin.op, &bin.rhs)?,470		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(ctx, v)?)?,471		Var(name) => in_frame(472			CallLocation::new(&name.span),473			|| format!("local <{}> access", &**name),474			|| ctx.binding((**name).clone())?.evaluate(),475		)?,476		Index { indexable, parts } => ensure_sufficient_stack(|| {477			let mut parts = parts.iter();478			let mut indexable = if matches!(&**indexable, Expr::Literal(LiteralType::Super)) {479				let part = parts.next().expect("at least part should exist");480				// sup_this existence check might also be skipped here for null-coalesce...481				// But I believe this might cause errors.482				let sup_this = ctx.try_sup_this()?;483				if !sup_this.has_super() {484					#[cfg(feature = "exp-null-coaelse")]485					if part.null_coaelse {486						return Ok(Val::Null);487					}488					bail!(NoSuperFound)489				}490				let name = evaluate(ctx.clone(), &part.value)?;491492				let Val::Str(name) = name else {493					bail!(ValueIndexMustBeTypeGot(494						ValType::Obj,495						ValType::Str,496						name.value_type(),497					))498				};499500				let name = name.into_flat();501				match sup_this502					.get_super(name.clone())503					.with_description_src(&part.span, || format!("field <{name}> access"))?504				{505					Some(v) => v,506					#[cfg(feature = "exp-null-coaelse")]507					None if part.null_coaelse => return Ok(Val::Null),508					None => {509						let suggestions = suggest_object_fields(510							&sup_this.standalone_super().expect("super exists"),511							name.clone(),512						);513514						bail!(NoSuchField(name, suggestions))515					}516				}517			} else {518				evaluate(ctx.clone(), indexable)?519			};520521			for part in parts {522				indexable = match (indexable, evaluate(ctx.clone(), &part.value)?) {523					(Val::Obj(v), Val::Str(key)) => match v524						.get(key.clone().into_flat())525						.with_description_src(&part.span, || format!("field <{key}> access"))?526					{527						Some(v) => v,528						#[cfg(feature = "exp-null-coaelse")]529						None if part.null_coaelse => return Ok(Val::Null),530						None => {531							let suggestions = suggest_object_fields(&v, key.clone().into_flat());532533							return Err(Error::from(NoSuchField(534								key.clone().into_flat(),535								suggestions,536							)))537							.with_description_src(&part.span, || format!("field <{key}> access"));538						}539					},540					(Val::Obj(_), n) => bail!(ValueIndexMustBeTypeGot(541						ValType::Obj,542						ValType::Str,543						n.value_type(),544					)),545					(Val::Arr(v), Val::Num(n)) => {546						let n = n.get();547						if n.fract() > f64::EPSILON {548							bail!(FractionalIndex)549						}550						if n < 0.0 {551							bail!(ArrayBoundsError(n as isize, v.len()));552						}553						v.get(n as usize)?554							.ok_or_else(|| ArrayBoundsError(n as isize, v.len()))?555					}556					(Val::Arr(_), Val::Str(n)) => {557						bail!(AttemptedIndexAnArrayWithString(n.into_flat()))558					}559					(Val::Arr(_), n) => bail!(ValueIndexMustBeTypeGot(560						ValType::Arr,561						ValType::Num,562						n.value_type(),563					)),564565					(Val::Str(s), Val::Num(n)) => Val::Str({566						let n = n.get();567						if n.fract() > f64::EPSILON {568							bail!(FractionalIndex)569						}570						if n < 0.0 {571							bail!(ArrayBoundsError(n as isize, s.into_flat().chars().count()));572						}573						let v: IStr = s574							.clone()575							.into_flat()576							.chars()577							.skip(n as usize)578							.take(1)579							.collect::<String>()580							.into();581						if v.is_empty() {582							bail!(StringBoundsError(n as usize, s.into_flat().chars().count()))583						}584						StrValue::Flat(v)585					}),586					(Val::Str(_), n) => bail!(ValueIndexMustBeTypeGot(587						ValType::Str,588						ValType::Num,589						n.value_type(),590					)),591					#[cfg(feature = "exp-null-coaelse")]592					(Val::Null, _) if part.null_coaelse => return Ok(Val::Null),593					(v, _) => bail!(CantIndexInto(v.value_type())),594				};595			}596			Ok(indexable)597		})?,598		LocalExpr(bindings, returned) => {599			let mut new_bindings: FxHashMap<IStr, Thunk<Val>> =600				FxHashMap::with_capacity(bindings.iter().map(BindSpec::binds_len).sum());601			let fctx = Context::new_future();602			for b in bindings {603				evaluate_dest(b, fctx.clone(), &mut new_bindings)?;604			}605			let ctx = ctx.extend_bindings(new_bindings).into_future(fctx);606			evaluate(ctx, returned)?607		}608		Arr(items) => {609			if items.is_empty() {610				Val::Arr(ArrValue::empty())611			} else {612				Val::Arr(ArrValue::expr(ctx, items.clone()))613			}614		}615		ArrComp(expr, comp_specs) => {616			let mut out = Vec::new();617			evaluate_comp(ctx, comp_specs, &mut |ctx| {618				let expr = expr.clone();619				out.push(Thunk!(move || evaluate(ctx, &expr)));620				Ok(())621			})?;622			Val::Arr(ArrValue::lazy(out))623		}624		Obj(body) => Val::Obj(evaluate_object(None, ctx, body)?),625		ObjExtend(a, b) => {626			let base = evaluate(ctx.clone(), a)?;627			match base {628				Val::Obj(base_obj) => Val::Obj(evaluate_object(Some(base_obj), ctx, b)?),629				_ => bail!("ObjExtend lhs should be an object value"),630			}631		}632		Apply(value, args, tailstrict) => ensure_sufficient_stack(|| {633			evaluate_apply(ctx, value, args, CallLocation::new(&args.span), *tailstrict)634		})?,635		Function(params, body) => {636			evaluate_method(ctx, "anonymous".into(), params.clone(), body.clone())637		}638		AssertExpr(assert) => {639			evaluate_assert(ctx.clone(), &assert.assert)?;640			evaluate(ctx, &assert.rest)?641		}642		ErrorStmt(s, e) => in_frame(643			CallLocation::new(s),644			|| "error statement".to_owned(),645			|| bail!(RuntimeError(evaluate(ctx, e)?.to_string()?,)),646		)?,647		IfElse(if_else) => {648			if in_frame(649				CallLocation::new(&if_else.cond.span),650				|| "if condition".to_owned(),651				|| bool::from_untyped(evaluate(ctx.clone(), &if_else.cond.cond)?),652			)? {653				evaluate(ctx, &if_else.cond_then)?654			} else {655				match &if_else.cond_else {656					Some(v) => evaluate(ctx, v)?,657					None => Val::Null,658				}659			}660		}661		Slice(slice) => {662			fn parse_idx<T: Typed + FromUntyped>(663				ctx: Context,664				expr: Option<&Spanned<Expr>>,665				desc: &'static str,666			) -> Result<Option<T>> {667				if let Some(value) = expr {668					Ok(in_frame(669						CallLocation::new(&value.span),670						|| format!("slice {desc}"),671						|| <Option<T>>::from_untyped(evaluate(ctx, value)?),672					)?)673				} else {674					Ok(None)675				}676			}677678			let indexable = evaluate(ctx.clone(), &slice.value)?;679680			let start = parse_idx(ctx.clone(), slice.slice.start.as_ref(), "start")?;681			let end = parse_idx(ctx.clone(), slice.slice.end.as_ref(), "end")?;682			let step = parse_idx(ctx, slice.slice.step.as_ref(), "step")?;683684			IndexableVal::into_untyped(indexable.into_indexable()?.slice(start, end, step)?)?685		}686		Import(kind, path) => {687			let Expr::Str(path) = &**path else {688				bail!("computed imports are not supported")689			};690			with_state(|s| {691				let span = &kind.span;692				let resolved_path = s.resolve_from(span.0.source_path(), path)?;693				Ok(match &**kind {694					ImportKind::Normal => in_frame(695						CallLocation::new(span),696						|| format!("import {:?}", path.clone()),697						|| s.import_resolved(resolved_path),698					)?,699					ImportKind::Str => Val::string(s.import_resolved_str(resolved_path)?),700					ImportKind::Bin => {701						Val::Arr(ArrValue::bytes(s.import_resolved_bin(resolved_path)?))702					}703				}) as Result<Val>704			})?705		}706	})707}