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

1use crate::{2	binding, context_creator, create_error, function_default, function_rhs, future_wrapper,3	lazy_binding, lazy_val, push, Context, ContextCreator, FuncDesc, LazyBinding, ObjMember,4	ObjValue, Result, Val,5};6use closure::closure;7use jsonnet_parser::{8	el, Arg, ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember,9	ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,10	Visibility,11};12use std::{13	collections::{BTreeMap, HashMap},14	rc::Rc,15};1617pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, LazyBinding) {18	let b = b.clone();19	if let Some(args) = &b.params {20		let args = args.clone();21		(22			b.name.clone(),23			lazy_binding!(move |this, super_obj| 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	} else {32		(33			b.name.clone(),34			lazy_binding!(move |this, super_obj| {35				Ok(lazy_val!(36					closure!(clone context_creator, clone b, || evaluate(37						context_creator.0(this.clone(), super_obj.clone())?,38						&b.value39					))40				))41			}),42		)43	}44}4546pub fn evaluate_method(ctx: Context, expr: &LocExpr, arg_spec: ParamsDesc) -> Val {47	Val::Func(FuncDesc {48		ctx,49		params: arg_spec,50		eval_rhs: function_rhs!(closure!(clone expr, |ctx| evaluate(ctx, &expr))),51		eval_default: function_default!(closure!(|ctx, default| evaluate(ctx, &default))),52	})53}5455pub fn evaluate_field_name(56	context: Context,57	field_name: &jsonnet_parser::FieldName,58) -> Result<String> {59	Ok(match field_name {60		jsonnet_parser::FieldName::Fixed(n) => n.clone(),61		jsonnet_parser::FieldName::Dyn(expr) => {62			evaluate(context, expr)?.try_cast_str("dynamic field name")?63		}64	})65}6667pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {68	Ok(match (op, b) {69		(o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,70		(UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),71		(UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),72		(UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),73		(op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),74	})75}7677pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {78	Ok(match (a, b) {79		(Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),8081		(Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),82		(Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),83		(Val::Str(v1), Val::Bool(v2)) => Val::Str(format!("{}{}", v1, v2)),84		(Val::Bool(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),85		(Val::Str(v1), Val::Null) => Val::Str(format!("{}null", v1)),86		(Val::Null, Val::Str(v2)) => Val::Str(format!("null{}", v2)),8788		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),89		(Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),90		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),91		_ => panic!("can't add: {:?} and {:?}", a, b),92	})93}9495pub fn evaluate_binary_op_special(96	context: Context,97	a: &LocExpr,98	op: BinaryOpType,99	b: &LocExpr,100) -> Result<Val> {101	Ok(102		match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {103			(Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),104			(Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),105			(a, op, eb) => evaluate_binary_op_normal(106				context.clone(),107				&a,108				op,109				&evaluate(context, eb)?.unwrap_if_lazy()?,110			)?,111		},112	)113}114115pub fn evaluate_binary_op_normal(116	context: Context,117	a: &Val,118	op: BinaryOpType,119	b: &Val,120) -> Result<Val> {121	Ok(match (a, op, b) {122		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,123124		(Val::Str(v1), BinaryOpType::Eq, Val::Str(v2)) => Val::Bool(v1 == v2),125		(Val::Str(v1), BinaryOpType::Ne, Val::Str(v2)) => Val::Bool(v1 != v2),126127		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),128		(Val::Str(format), BinaryOpType::Mod, args) => evaluate(129			context130				.with_var("__tmp__format__".to_owned(), Val::Str(format.to_owned()))?131				.with_var(132					"__tmp__args__".to_owned(),133					match args {134						Val::Arr(v) => Val::Arr(v.clone()),135						v => Val::Arr(vec![v.clone()]),136					},137				)?,138			&el!(Expr::Apply(139				el!(Expr::Index(140					el!(Expr::Var("std".to_owned())),141					el!(Expr::Str("format".to_owned()))142				)),143				ArgsDesc(vec![144					Arg(None, el!(Expr::Var("__tmp__format__".to_owned()))),145					Arg(None, el!(Expr::Var("__tmp__args__".to_owned())))146				])147			)),148		)?,149150		// Bool X Bool151		(Val::Bool(a), BinaryOpType::Eq, Val::Bool(b)) => Val::Bool(a == b),152		(Val::Bool(a), BinaryOpType::Ne, Val::Bool(b)) => Val::Bool(a != b),153154		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),155		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),156157		// Str X Str158		(Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),159		(Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),160		(Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),161		(Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),162163		// Num X Num164		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),165		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => Val::Num(v1 / v2),166		(Val::Num(v1), BinaryOpType::Mod, Val::Num(v2)) => Val::Num(v1 % v2),167168		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),169170		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),171		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),172		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),173		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),174175		(Val::Num(v1), BinaryOpType::Eq, Val::Num(v2)) => Val::Bool((v1 - v2).abs() < f64::EPSILON),176		(Val::Num(v1), BinaryOpType::Ne, Val::Num(v2)) => Val::Bool((v1 - v2).abs() > f64::EPSILON),177178		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {179			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)180		}181		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {182			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)183		}184		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {185			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)186		}187		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {188			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)189		}190		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {191			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)192		}193194		// Arr X Arr195		(Val::Arr(a), BinaryOpType::Eq, Val::Arr(b)) => {196			if a.len() != b.len() {197				Val::Bool(false)198			} else {199				for i in 0..a.len() {200					if let Val::Bool(v) = evaluate_binary_op_normal(201						context.clone(),202						&a[i].clone().unwrap_if_lazy()?,203						op,204						&b[i].clone().unwrap_if_lazy()?,205					)? {206						if !v {207							return Ok(Val::Bool(false));208						}209					} else {210						unreachable!()211					}212				}213				return Ok(Val::Bool(true));214			}215		}216		_ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),217	})218}219220future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);221future_wrapper!(ObjValue, FutureObjValue);222223pub fn evaluate_comp(224	context: Context,225	value: &LocExpr,226	specs: &[CompSpec],227) -> Result<Option<Vec<Val>>> {228	Ok(match specs.get(0) {229		None => Some(vec![evaluate(context, &value)?]),230		Some(CompSpec::IfSpec(IfSpecData(cond))) => {231			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {232				evaluate_comp(context, value, &specs[1..])?233			} else {234				None235			}236		}237		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {238			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {239				Val::Arr(list) => {240					let mut out = Vec::new();241					for item in list {242						let item = item.clone().unwrap_if_lazy()?;243						out.push(evaluate_comp(244							context.with_var(var.clone(), item)?,245							value,246							&specs[1..],247						)?);248					}249					Some(out.iter().flatten().flatten().cloned().collect())250				}251				_ => panic!("for expression evaluated to non-iterable value"),252			}253		}254	})255}256257// TODO: Asserts258pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {259	Ok(match object {260		ObjBody::MemberList(members) => {261			let new_bindings = FutureNewBindings::new();262			let future_this = FutureObjValue::new();263			let context_creator = context_creator!(264				closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {265					Ok(context.clone().extend(266						new_bindings.clone().unwrap(),267						context.clone().dollar().clone().or_else(||this.clone()),268						Some(this.unwrap()),269						super_obj270					)?)271				})272			);273			{274				let mut bindings: HashMap<String, LazyBinding> = HashMap::new();275				for (n, b) in members276					.iter()277					.filter_map(|m| match m {278						Member::BindStmt(b) => Some(b.clone()),279						_ => None,280					})281					.map(|b| evaluate_binding(&b, context_creator.clone()))282				{283					bindings.insert(n, b);284				}285				new_bindings.fill(bindings);286			}287288			let mut new_members = BTreeMap::new();289			for member in members.into_iter() {290				match member {291					Member::Field(FieldMember {292						name,293						plus,294						params: None,295						visibility,296						value,297					}) => {298						let name = evaluate_field_name(context.clone(), &name)?;299						new_members.insert(300							name.clone(),301							ObjMember {302								add: plus,303								visibility: visibility.clone(),304								invoke: binding!(305									closure!(clone name, clone value, clone context_creator, |this, super_obj| {306										push(value.clone(), "object ".to_owned()+&name+" field", ||{307											let context = context_creator.0(this, super_obj)?;308											evaluate(309												context,310												&value,311											)?.unwrap_if_lazy()312										})313									})314								),315							},316						);317					}318					Member::Field(FieldMember {319						name,320						params: Some(params),321						value,322						..323					}) => {324						let name = evaluate_field_name(context.clone(), &name)?;325						new_members.insert(326							name,327							ObjMember {328								add: false,329								visibility: Visibility::Hidden,330								invoke: binding!(331									closure!(clone value, clone context_creator, |this, super_obj| {332										// TODO: Assert333										Ok(evaluate_method(334											context_creator.0(this, super_obj)?,335											&value.clone(),336											params.clone(),337										))338									})339								),340							},341						);342					}343					Member::BindStmt(_) => {}344					Member::AssertStmt(_) => {}345				}346			}347			future_this.fill(ObjValue::new(None, Rc::new(new_members)))348		}349		_ => todo!(),350	})351}352353pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {354	use Expr::*;355	let locexpr = expr.clone();356	let LocExpr(expr, loc) = expr;357	Ok(match &**expr {358		Literal(LiteralType::This) => Val::Obj(359			context360				.this()361				.clone()362				.unwrap_or_else(|| panic!("this not found")),363		),364		Literal(LiteralType::Super) => Val::Obj(365			context366				.super_obj()367				.clone()368				.unwrap_or_else(|| panic!("super not found")),369		),370		Literal(LiteralType::Dollar) => Val::Obj(371			context372				.dollar()373				.clone()374				.unwrap_or_else(|| panic!("dollar not found")),375		),376		Literal(LiteralType::True) => Val::Bool(true),377		Literal(LiteralType::False) => Val::Bool(false),378		Literal(LiteralType::Null) => Val::Null,379		Parened(e) => evaluate(context, e)?,380		Str(v) => Val::Str(v.clone()),381		Num(v) => Val::Num(*v),382		BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,383		UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,384		Var(name) => Val::Lazy(context.binding(&name)).unwrap_if_lazy()?,385		Index(value, index) => {386			match (387				evaluate(context.clone(), value)?.unwrap_if_lazy()?,388				evaluate(context.clone(), index)?,389			) {390				(Val::Obj(v), Val::Str(s)) => {391					if let Some(v) = v.get(&s)? {392						v.unwrap_if_lazy()?393					} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {394						Val::Intristic(n, s)395					} else {396						create_error(crate::Error::NoSuchField(s))?397					}398				}399				(Val::Arr(v), Val::Num(n)) => v400					.get(n as usize)401					.unwrap_or_else(|| panic!("out of bounds"))402					.clone()403					.unwrap_if_lazy()?,404				(Val::Str(s), Val::Num(n)) => {405					Val::Str(s.chars().skip(n as usize).take(1).collect())406				}407				(v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),408			}409		}410		LocalExpr(bindings, returned) => {411			let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();412			let future_context = Context::new_future();413414			let context_creator = context_creator!(415				closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))416			);417418			for (k, v) in bindings419				.iter()420				.map(|b| evaluate_binding(b, context_creator.clone()))421			{422				new_bindings.insert(k, v);423			}424425			let context = context426				.extend(new_bindings, None, None, None)?427				.into_future(future_context);428			evaluate(context, &returned.clone())?429		}430		Arr(items) => {431			let mut out = Vec::with_capacity(items.len());432			for item in items {433				out.push(Val::Lazy(lazy_val!(434					closure!(clone context, clone item, || {435						evaluate(context.clone(), &item)436					})437				)));438			}439			Val::Arr(out)440		}441		ArrComp(expr, compspecs) => Val::Arr(442			// First compspec should be forspec, so no "None" possible here443			evaluate_comp(context, expr, compspecs)?.unwrap(),444		),445		Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),446		Apply(value, ArgsDesc(args)) => {447			let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;448			match value {449				Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {450					// arr/string/function451					("std", "length") => {452						assert_eq!(args.len(), 1);453						let expr = &args.get(0).unwrap().1;454						match evaluate(context, expr)? {455							Val::Str(n) => Val::Num(n.chars().count() as f64),456							Val::Arr(i) => Val::Num(i.len() as f64),457							v => panic!("can't get length of {:?}", v),458						}459					}460					// any461					("std", "type") => {462						assert_eq!(args.len(), 1);463						let expr = &args.get(0).unwrap().1;464						Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())465					}466					// length, idx=>any467					("std", "makeArray") => {468						assert_eq!(args.len(), 2);469						if let (Val::Num(v), Val::Func(d)) = (470							evaluate(context.clone(), &args[0].1)?,471							evaluate(context, &args[1].1)?,472						) {473							assert!(v > 0.0);474							let mut out = Vec::with_capacity(v as usize);475							for i in 0..v as usize {476								out.push(d.evaluate(vec![(None, Val::Num(i as f64))])?)477							}478							Val::Arr(out)479						} else {480							panic!("bad makeArray call");481						}482					}483					// string484					("std", "codepoint") => {485						assert_eq!(args.len(), 1);486						if let Val::Str(s) = evaluate(context, &args[0].1)? {487							assert!(488								s.chars().count() == 1,489								"std.codepoint should receive single char string"490							);491							Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)492						} else {493							panic!("bad codepoint call");494						}495					}496					// object, includeHidden497					("std", "objectFieldsEx") => {498						assert_eq!(args.len(), 2);499						if let (Val::Obj(body), Val::Bool(_include_hidden)) = (500							evaluate(context.clone(), &args[0].1)?,501							evaluate(context, &args[1].1)?,502						) {503							// TODO: handle visibility (_include_hidden)504							Val::Arr(body.fields().into_iter().map(Val::Str).collect())505						} else {506							panic!("bad objectFieldsEx call");507						}508					}509					(ns, name) => panic!("Intristic not found: {}.{}", ns, name),510				},511				Val::Func(f) => push(locexpr.clone(), "function call".to_owned(), || {512					f.evaluate(513						args.clone()514							.into_iter()515							.map(move |a| {516								(517									a.clone().0,518									Val::Lazy(lazy_val!(519										closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))520									)),521								)522							})523							.collect(),524					)525				})?,526				_ => panic!("{:?} is not a function", value),527			}528		}529		Function(params, body) => evaluate_method(context, body, params.clone()),530		AssertExpr(AssertStmt(value, msg), returned) => {531			if push(value.clone(), "assertion condition".to_owned(), || {532				evaluate(context.clone(), &value)?533					.try_cast_bool("assertion condition should be boolean")534			})? {535				push(536					returned.clone(),537					"assert 'return' branch".to_owned(),538					|| evaluate(context, returned),539				)?540			} else if let Some(msg) = msg {541				panic!(542					"assertion failed ({:?}): {}",543					value,544					evaluate(context, msg)?.try_cast_str("assertion message should be string")?545				);546			} else {547				panic!("assertion failed ({:?}): no message", value);548			}549		}550		Error(e) => create_error(crate::Error::RuntimeError(551			evaluate(context, e)?.try_cast_str("error text should be string")?,552		))?,553		IfElse {554			cond,555			cond_then,556			cond_else,557		} => {558			if push(cond.0.clone(), "if condition".to_owned(), || {559				evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")560			})? {561				push(562					cond_then.clone(),563					"if condition 'then' branch".to_owned(),564					|| evaluate(context, cond_then),565				)?566			} else {567				match cond_else {568					Some(v) => push(v.clone(), "if condition 'else' branch".to_owned(), || {569						evaluate(context, v)570					})?,571					None => Val::Bool(false),572				}573			}574		}575		_ => panic!(576			"evaluation not implemented: {:?}",577			LocExpr(expr.clone(), loc.clone())578		),579	})580}