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

1use crate::{2	binding, context_creator, create_error, function_default, function_rhs, future_wrapper,3	lazy_val, push, Context, ContextCreator, FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue,4	Result, Val,5};6use closure::closure;7use jsonnet_parser::{8	ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember, ForSpecData,9	IfSpecData, 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) -> (String, LazyBinding) {17	let b = b.clone();18	if let Some(args) = &b.params {19		let args = args.clone();20		(21			b.name.clone(),22			LazyBinding::Bindable(Rc::new(move |this, super_obj| {23				Ok(lazy_val!(24					closure!(clone b, clone args, clone context_creator, || Ok(evaluate_method(25						context_creator.0(this.clone(), super_obj.clone())?,26						&b.value,27						args.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.clone(), "thunk".to_owned(), ||{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, expr: &LocExpr, arg_spec: ParamsDesc) -> Val {50	Val::Func(FuncDesc {51		ctx,52		params: arg_spec,53		eval_rhs: function_rhs!(closure!(clone expr, |ctx| evaluate(ctx, &expr))),54		eval_default: function_default!(closure!(|ctx, default| evaluate(ctx, &default))),55	})56}5758pub fn evaluate_field_name(59	context: Context,60	field_name: &jsonnet_parser::FieldName,61) -> Result<Option<String>> {62	Ok(match field_name {63		jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),64		jsonnet_parser::FieldName::Dyn(expr) => {65			let value = evaluate(context, expr)?.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) => panic!("unary op not implemented: {:?} {:?}", op, o),82	})83}8485pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {86	Ok(match (a, b) {87		(Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),8889		(Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),90		(Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),91		(Val::Str(v1), Val::Bool(v2)) => Val::Str(format!("{}{}", v1, v2)),92		(Val::Bool(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),93		(Val::Str(v1), Val::Null) => Val::Str(format!("{}null", v1)),94		(Val::Null, Val::Str(v2)) => Val::Str(format!("null{}", v2)),9596		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),97		(Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),98		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),99		_ => panic!("can't add: {:?} and {:?}", a, b),100	})101}102103pub fn evaluate_binary_op_special(104	context: Context,105	a: &LocExpr,106	op: BinaryOpType,107	b: &LocExpr,108) -> Result<Val> {109	Ok(110		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {111			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),112			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),113			(a, op, eb) => {114				evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?115			}116		},117	)118}119120pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {121	Ok(match (a, op, b) {122		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,123124		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),125126		// Bool X Bool127		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),128		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),129130		// Str X Str131		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),132		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),133		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),134		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),135136		// Num X Num137		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),138		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {139			if *v2 <= f64::EPSILON {140				create_error(crate::Error::DivisionByZero)?141			}142			Val::Num(v1 / v2)143		}144145		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),146147		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),148		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),149		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),150		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),151152		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {153			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)154		}155		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {156			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)157		}158		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {159			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)160		}161		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {162			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)163		}164		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {165			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)166		}167168		_ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),169	})170}171172future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);173future_wrapper!(ObjValue, FutureObjValue);174175pub fn evaluate_comp(176	context: Context,177	value: &LocExpr,178	specs: &[CompSpec],179) -> Result<Option<Vec<Val>>> {180	Ok(match specs.get(0) {181		None => Some(vec![evaluate(context, &value)?]),182		Some(CompSpec::IfSpec(IfSpecData(cond))) => {183			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {184				evaluate_comp(context, value, &specs[1..])?185			} else {186				None187			}188		}189		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {190			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {191				Val::Arr(list) => {192					let mut out = Vec::new();193					for item in list {194						let item = item.clone().unwrap_if_lazy()?;195						out.push(evaluate_comp(196							context.with_var(var.clone(), item)?,197							value,198							&specs[1..],199						)?);200					}201					Some(out.iter().flatten().flatten().cloned().collect())202				}203				_ => panic!("for expression evaluated to non-iterable value"),204			}205		}206	})207}208209// TODO: Asserts210pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {211	Ok(match object {212		ObjBody::MemberList(members) => {213			let new_bindings = FutureNewBindings::new();214			let future_this = FutureObjValue::new();215			let context_creator = context_creator!(216				closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {217					Ok(context.clone().extend(218						new_bindings.clone().unwrap(),219						context.clone().dollar().clone().or_else(||this.clone()),220						Some(this.unwrap()),221						super_obj222					)?)223				})224			);225			{226				let mut bindings: HashMap<String, LazyBinding> = HashMap::new();227				for (n, b) in members228					.iter()229					.filter_map(|m| match m {230						Member::BindStmt(b) => Some(b.clone()),231						_ => None,232					})233					.map(|b| evaluate_binding(&b, context_creator.clone()))234				{235					bindings.insert(n, b);236				}237				new_bindings.fill(bindings);238			}239240			let mut new_members = BTreeMap::new();241			for member in members.into_iter() {242				match member {243					Member::Field(FieldMember {244						name,245						plus,246						params: None,247						visibility,248						value,249					}) => {250						let name = evaluate_field_name(context.clone(), &name)?;251						if name.is_none() {252							continue;253						}254						let name = name.unwrap();255						new_members.insert(256							name.clone(),257							ObjMember {258								add: plus,259								visibility: visibility.clone(),260								invoke: binding!(261									closure!(clone name, clone value, clone context_creator, |this, super_obj| {262										push(value.clone(), "object ".to_owned()+&name+" field", ||{263											let context = context_creator.0(this, super_obj)?;264											evaluate(265												context,266												&value,267											)?.unwrap_if_lazy()268										})269									})270								),271							},272						);273					}274					Member::Field(FieldMember {275						name,276						params: Some(params),277						value,278						..279					}) => {280						let name = evaluate_field_name(context.clone(), &name)?;281						if name.is_none() {282							continue;283						}284						let name = name.unwrap();285						new_members.insert(286							name,287							ObjMember {288								add: false,289								visibility: Visibility::Hidden,290								invoke: binding!(291									closure!(clone value, clone context_creator, |this, super_obj| {292										// TODO: Assert293										Ok(evaluate_method(294											context_creator.0(this, super_obj)?,295											&value.clone(),296											params.clone(),297										))298									})299								),300							},301						);302					}303					Member::BindStmt(_) => {}304					Member::AssertStmt(_) => {}305				}306			}307			future_this.fill(ObjValue::new(None, Rc::new(new_members)))308		}309		_ => todo!(),310	})311}312313#[inline(always)]314pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {315	use Expr::*;316	let locexpr = expr.clone();317	let LocExpr(expr, loc) = expr;318	Ok(match &**expr {319		Literal(LiteralType::This) => Val::Obj(320			context321				.this()322				.clone()323				.unwrap_or_else(|| panic!("this not found")),324		),325		Literal(LiteralType::Dollar) => Val::Obj(326			context327				.dollar()328				.clone()329				.unwrap_or_else(|| panic!("dollar not found")),330		),331		Literal(LiteralType::True) => Val::Bool(true),332		Literal(LiteralType::False) => Val::Bool(false),333		Literal(LiteralType::Null) => Val::Null,334		Parened(e) => evaluate(context, e)?,335		Str(v) => Val::Str(v.clone()),336		Num(v) => Val::Num(*v),337		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,338		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,339		Var(name) => push(locexpr, "var".to_owned(), || {340			Val::Lazy(context.binding(&name)).unwrap_if_lazy()341		})?,342		Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {343			let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;344			context345				.super_obj()346				.clone()347				.expect("no super found")348				.get_raw(&name, &context.this().clone().expect("no this found"))?349				.expect("value not found")350		}351		Index(value, index) => {352			match (353				evaluate(context.clone(), value)?.unwrap_if_lazy()?,354				evaluate(context, index)?,355			) {356				(Val::Obj(v), Val::Str(s)) => {357					if let Some(v) = v.get(&s)? {358						v.unwrap_if_lazy()?359					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {360						Val::Intristic(n, s)361					} else {362						create_error(crate::Error::NoSuchField(s))?363					}364				}365				(Val::Arr(v), Val::Num(n)) => {366					if n.fract() > f64::EPSILON {367						create_error(crate::Error::FractionalIndex)?368					}369					v.get(n as usize)370						.unwrap_or_else(|| panic!("out of bounds"))371						.clone()372						.unwrap_if_lazy()?373				}374				(Val::Str(s), Val::Num(n)) => {375					Val::Str(s.chars().skip(n as usize).take(1).collect())376				}377				(v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),378			}379		}380		LocalExpr(bindings, returned) => {381			let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();382			let future_context = Context::new_future();383384			let context_creator = context_creator!(385				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))386			);387388			for (k, v) in bindings389				.iter()390				.map(|b| evaluate_binding(b, context_creator.clone()))391			{392				new_bindings.insert(k, v);393			}394395			let context = context396				.extend(new_bindings, None, None, None)?397				.into_future(future_context);398			evaluate(context, &returned.clone())?399		}400		Arr(items) => {401			let mut out = Vec::with_capacity(items.len());402			for item in items {403				out.push(Val::Lazy(lazy_val!(404					closure!(clone context, clone item, || {405						evaluate(context.clone(), &item)406					})407				)));408			}409			Val::Arr(out)410		}411		ArrComp(expr, compspecs) => Val::Arr(412			// First compspec should be forspec, so no "None" possible here413			evaluate_comp(context, expr, compspecs)?.unwrap(),414		),415		Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),416		ObjExtend(s, t) => evaluate_add_op(417			&evaluate(context.clone(), s)?,418			&Val::Obj(evaluate_object(context, t.clone())?),419		)?,420		Apply(value, ArgsDesc(args), tailstrict) => {421			let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;422			match value {423				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {424					// arr/string/function425					("std", "length") => {426						assert_eq!(args.len(), 1);427						let expr = &args.get(0).unwrap().1;428						match evaluate(context, expr)? {429							Val::Str(n) => Val::Num(n.chars().count() as f64),430							Val::Arr(i) => Val::Num(i.len() as f64),431							Val::Obj(o) => Val::Num(o.fields().len() as f64),432							v => panic!("can't get length of {:?}", v),433						}434					}435					// any436					("std", "type") => {437						assert_eq!(args.len(), 1);438						let expr = &args.get(0).unwrap().1;439						Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())440					}441					// length, idx=>any442					("std", "makeArray") => {443						assert_eq!(args.len(), 2);444						if let (Val::Num(v), Val::Func(d)) = (445							evaluate(context.clone(), &args[0].1)?,446							evaluate(context, &args[1].1)?,447						) {448							assert!(v >= 0.0);449							let mut out = Vec::with_capacity(v as usize);450							for i in 0..v as usize {451								out.push(d.evaluate(vec![(None, Val::Num(i as f64))])?)452							}453							Val::Arr(out)454						} else {455							panic!("bad makeArray call");456						}457					}458					// string459					("std", "codepoint") => {460						assert_eq!(args.len(), 1);461						if let Val::Str(s) = evaluate(context, &args[0].1)? {462							assert!(463								s.chars().count() == 1,464								"std.codepoint should receive single char string"465							);466							Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)467						} else {468							panic!("bad codepoint call");469						}470					}471					// object, includeHidden472					("std", "objectFieldsEx") => {473						assert_eq!(args.len(), 2);474						if let (Val::Obj(body), Val::Bool(_include_hidden)) = (475							evaluate(context.clone(), &args[0].1)?,476							evaluate(context, &args[1].1)?,477						) {478							// TODO: handle visibility (_include_hidden)479							Val::Arr(body.fields().into_iter().map(Val::Str).collect())480						} else {481							panic!("bad objectFieldsEx call");482						}483					}484					("std", "primitiveEquals") => {485						assert_eq!(args.len(), 2);486						let (a, b) = (487							evaluate(context.clone(), &args[0].1)?,488							evaluate(context, &args[1].1)?,489						);490						Val::Bool(a == b)491					}492					("std", "modulo") => {493						assert_eq!(args.len(), 2);494						if let (Val::Num(a), Val::Num(b)) = (495							evaluate(context.clone(), &args[0].1)?,496							evaluate(context, &args[1].1)?,497						) {498							Val::Num(a % b)499						} else {500							panic!("bad modulo call");501						}502					}503					("std", "floor") => {504						assert_eq!(args.len(), 1);505						if let Val::Num(a) = evaluate(context, &args[0].1)? {506							Val::Num(a.floor())507						} else {508							panic!("bad floor call");509						}510					}511					("std", "trace") => {512						assert_eq!(args.len(), 2);513						if let (Val::Str(a), b) = (514							evaluate(context.clone(), &args[0].1)?,515							evaluate(context, &args[1].1)?,516						) {517							// TODO: Line numbers as in original jsonnet518							println!("TRACE: {}", a);519							b520						} else {521							panic!("bad trace call");522						}523					}524					("std", "pow") => {525						assert_eq!(args.len(), 2);526						if let (Val::Num(a), Val::Num(b)) = (527							evaluate(context.clone(), &args[0].1)?,528							evaluate(context, &args[1].1)?,529						) {530							Val::Num(a.powf(b))531						} else {532							panic!("bad pow call");533						}534					}535					(ns, name) => panic!("Intristic not found: {}.{}", ns, name),536				},537				Val::Func(f) => {538					let body = #[inline(always)]539					|| {540						f.evaluate(541							args.clone()542								.into_iter()543								.map(544									#[inline(always)]545									move |a| {546										Ok((547											a.clone().0,548											if *tailstrict {549												Val::Lazy(LazyVal::new_resolved(evaluate(550													context.clone(),551													&a.1,552												)?))553											} else {554												Val::Lazy(lazy_val!(555													closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))556												))557											},558										))559									},560								)561								.collect::<Result<Vec<_>>>()?,562						)563					};564					if *tailstrict {565						body()?566					} else {567						push(locexpr, "function call".to_owned(), body)?568					}569				}570				_ => panic!("{:?} is not a function", value),571			}572		}573		Function(params, body) => evaluate_method(context, body, params.clone()),574		AssertExpr(AssertStmt(value, msg), returned) => {575			let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {576				evaluate(context.clone(), &value)?577					.try_cast_bool("assertion condition should be boolean")578			})?;579			if assertion_result {580				push(581					returned.clone(),582					"assert 'return' branch".to_owned(),583					|| evaluate(context, returned),584				)?585			} else if let Some(msg) = msg {586				panic!(587					"assertion failed ({:?}): {}",588					value,589					evaluate(context, msg)?.try_cast_str("assertion message should be string")?590				);591			} else {592				panic!("assertion failed ({:?}): no message", value);593			}594		}595		Error(e) => create_error(crate::Error::RuntimeError(596			evaluate(context, e)?.try_cast_str("error text should be string")?,597		))?,598		IfElse {599			cond,600			cond_then,601			cond_else,602		} => {603			let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {604				evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")605			})?;606			if condition_result {607				push(608					cond_then.clone(),609					"if condition 'then' branch".to_owned(),610					|| evaluate(context, cond_then),611				)?612			} else {613				match cond_else {614					Some(v) => evaluate(context, v)?,615					None => Val::Null,616				}617			}618		}619		_ => panic!(620			"evaluation not implemented: {:?}",621			LocExpr(expr.clone(), loc.clone())622		),623	})624}