git.delta.rocks / jrsonnet / refs/commits / 3302a3dd316f

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		(op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),72	})73}7475pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {76	Ok(match (a, b) {77		(Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),78		(Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),79		(Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),80		(Val::Str(v1), v2) => Val::Str(format!("{}{:?}", v1, v2)),81		(Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),82		(Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),83		(Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),84		_ => panic!("can't add: {:?} and {:?}", a, b),85	})86}8788pub fn evaluate_binary_op(context: Context, a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {89	Ok(match (a, op, b) {90		(Val::Lazy(a), o, b) => evaluate_binary_op(context, &a.evaluate()?, o, b)?,91		(a, o, Val::Lazy(b)) => evaluate_binary_op(context, a, o, &b.evaluate()?)?,9293		(a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,9495		(Val::Str(v1), BinaryOpType::Ne, Val::Str(v2)) => Val::Bool(v1 != v2),9697		(Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),98		(Val::Str(format), BinaryOpType::Mod, args) => evaluate(99			context100				.with_var("__tmp__format__".to_owned(), Val::Str(format.to_owned()))?101				.with_var(102					"__tmp__args__".to_owned(),103					match args {104						Val::Arr(v) => Val::Arr(v.clone()),105						v => Val::Arr(vec![v.clone()]),106					},107				)?,108			&el!(Expr::Apply(109				el!(Expr::Index(110					el!(Expr::Var("std".to_owned())),111					el!(Expr::Str("format".to_owned()))112				)),113				ArgsDesc(vec![114					Arg(None, el!(Expr::Var("__tmp__format__".to_owned()))),115					Arg(None, el!(Expr::Var("__tmp__args__".to_owned())))116				])117			)),118		)?,119120		(Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),121		(Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),122123		(Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),124		(Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => Val::Num(v1 / v2),125		(Val::Num(v1), BinaryOpType::Mod, Val::Num(v2)) => Val::Num(v1 % v2),126127		(Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),128129		(Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {130			Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)131		}132		(Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {133			Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)134		}135136		(Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),137		(Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),138		(Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),139		(Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),140141		(Val::Num(v1), BinaryOpType::Eq, Val::Num(v2)) => Val::Bool((v1 - v2).abs() < f64::EPSILON),142		(Val::Num(v1), BinaryOpType::Ne, Val::Num(v2)) => Val::Bool((v1 - v2).abs() > f64::EPSILON),143144		(Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {145			Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)146		}147		(Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {148			Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)149		}150		(Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {151			Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)152		}153		(a, BinaryOpType::Eq, b) => Val::Bool(a == b),154		(a, BinaryOpType::Ne, b) => Val::Bool(a != b),155		_ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),156	})157}158159future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);160future_wrapper!(ObjValue, FutureObjValue);161162pub fn evaluate_comp(163	context: Context,164	value: &LocExpr,165	specs: &[CompSpec],166) -> Result<Option<Vec<Val>>> {167	Ok(match specs.get(0) {168		None => Some(vec![evaluate(context, &value)?]),169		Some(CompSpec::IfSpec(IfSpecData(cond))) => {170			if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {171				evaluate_comp(context, value, &specs[1..])?172			} else {173				None174			}175		}176		Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {177			match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {178				Val::Arr(list) => {179					let mut out = Vec::new();180					for item in list {181						let item = item.clone();182						out.push(evaluate_comp(183							context.with_var(var.clone(), item)?,184							value,185							&specs[1..],186						)?);187					}188					Some(out.iter().flatten().flatten().cloned().collect())189				}190				_ => panic!("for expression evaluated to non-iterable value"),191			}192		}193	})194}195196// TODO: Asserts197pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {198	Ok(match object {199		ObjBody::MemberList(members) => {200			let new_bindings = FutureNewBindings::new();201			let future_this = FutureObjValue::new();202			let context_creator = context_creator!(203				closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {204					Ok(context.clone().extend(205						new_bindings.clone().unwrap(),206						context.clone().dollar().clone().or_else(||this.clone()),207						Some(this.unwrap()),208						super_obj209					)?)210				})211			);212			{213				let mut bindings: HashMap<String, LazyBinding> = HashMap::new();214				for (n, b) in members215					.iter()216					.filter_map(|m| match m {217						Member::BindStmt(b) => Some(b.clone()),218						_ => None,219					})220					.map(|b| evaluate_binding(&b, context_creator.clone()))221				{222					bindings.insert(n, b);223				}224				new_bindings.fill(bindings);225			}226227			let mut new_members = BTreeMap::new();228			for member in members.into_iter() {229				match member {230					Member::Field(FieldMember {231						name,232						plus,233						params: None,234						visibility,235						value,236					}) => {237						let name = evaluate_field_name(context.clone(), &name)?;238						new_members.insert(239							name,240							ObjMember {241								add: plus,242								visibility: visibility.clone(),243								invoke: binding!(244									closure!(clone value, clone context_creator, |this, super_obj| {245										let context = context_creator.0(this, super_obj)?;246										// TODO: Assert247										evaluate(248											context,249											&value,250										)?.unwrap_if_lazy()251									})252								),253							},254						);255					}256					Member::Field(FieldMember {257						name,258						params: Some(params),259						value,260						..261					}) => {262						let name = evaluate_field_name(context.clone(), &name)?;263						new_members.insert(264							name,265							ObjMember {266								add: false,267								visibility: Visibility::Hidden,268								invoke: binding!(269									closure!(clone value, clone context_creator, |this, super_obj| {270										// TODO: Assert271										Ok(evaluate_method(272											context_creator.0(this, super_obj)?,273											&value.clone(),274											params.clone(),275										))276									})277								),278							},279						);280					}281					Member::BindStmt(_) => {}282					Member::AssertStmt(_) => {}283				}284			}285			future_this.fill(ObjValue::new(None, Rc::new(new_members)))286		}287		_ => todo!(),288	})289}290291pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {292	use Expr::*;293	push(expr.clone(), "expr".to_owned(), || {294		let LocExpr(expr, loc) = expr;295		Ok(match &**expr {296			Literal(LiteralType::This) => Val::Obj(297				context298					.this()299					.clone()300					.unwrap_or_else(|| panic!("this not found")),301			),302			Literal(LiteralType::Super) => Val::Obj(303				context304					.super_obj()305					.clone()306					.unwrap_or_else(|| panic!("super not found")),307			),308			Literal(LiteralType::True) => Val::Bool(true),309			Literal(LiteralType::False) => Val::Bool(false),310			Literal(LiteralType::Null) => Val::Null,311			Parened(e) => evaluate(context, e)?,312			Str(v) => Val::Str(v.clone()),313			Num(v) => Val::Num(*v),314			BinaryOp(v1, o, v2) => {315				let a = evaluate(context.clone(), v1)?.unwrap_if_lazy()?;316				let op = *o;317				let b = evaluate(context.clone(), v2)?.unwrap_if_lazy()?;318				evaluate_binary_op(context, &a, op, &b)?319			}320			UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,321			Var(name) => Val::Lazy(context.binding(&name)).unwrap_if_lazy()?,322			Index(value, index) => {323				match (324					evaluate(context.clone(), value)?.unwrap_if_lazy()?,325					evaluate(context.clone(), index)?,326				) {327					(Val::Obj(v), Val::Str(s)) => {328						if let Some(v) = v.get(&s)? {329							v.unwrap_if_lazy()?330						} else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {331							Val::Intristic(n, s)332						} else {333							create_error(crate::Error::NoSuchField(s))?334						}335					}336					(Val::Arr(v), Val::Num(n)) => v337						.get(n as usize)338						.unwrap_or_else(|| panic!("out of bounds"))339						.clone(),340					(Val::Str(s), Val::Num(n)) => {341						Val::Str(s.chars().skip(n as usize).take(1).collect())342					}343					(v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),344				}345			}346			LocalExpr(bindings, returned) => {347				let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();348				let future_context = Context::new_future();349350				let context_creator = context_creator!(351					closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))352				);353354				for (k, v) in bindings355					.iter()356					.map(|b| evaluate_binding(b, context_creator.clone()))357				{358					new_bindings.insert(k, v);359				}360361				let context = context362					.extend(new_bindings, None, None, None)?363					.into_future(future_context);364				evaluate(context, &returned.clone())?365			}366			Arr(items) => {367				let mut out = Vec::with_capacity(items.len());368				for item in items {369					out.push(evaluate(context.clone(), item)?);370				}371				Val::Arr(out)372			}373			ArrComp(expr, compspecs) => Val::Arr(374				// First compspec should be forspec, so no "None" possible here375				evaluate_comp(context, expr, compspecs)?.unwrap(),376			),377			Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),378			Apply(value, ArgsDesc(args)) => {379				let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;380				match value {381					// TODO: Capture context of application382					Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {383						// arr/string/function384						("std", "length") => {385							assert_eq!(args.len(), 1);386							let expr = &args.get(0).unwrap().1;387							match evaluate(context, expr)? {388								Val::Str(n) => Val::Num(n.chars().count() as f64),389								Val::Arr(i) => Val::Num(i.len() as f64),390								v => panic!("can't get length of {:?}", v),391							}392						}393						// any394						("std", "type") => {395							assert_eq!(args.len(), 1);396							let expr = &args.get(0).unwrap().1;397							Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())398						}399						// length, idx=>any400						("std", "makeArray") => {401							assert_eq!(args.len(), 2);402							if let (Val::Num(v), Val::Func(d)) = (403								evaluate(context.clone(), &args[0].1)?,404								evaluate(context, &args[1].1)?,405							) {406								assert!(v > 0.0);407								let mut out = Vec::with_capacity(v as usize);408								for i in 0..v as usize {409									out.push(d.evaluate(vec![(None, Val::Num(i as f64))])?)410								}411								Val::Arr(out)412							} else {413								panic!("bad makeArray call");414							}415						}416						// string417						("std", "codepoint") => {418							assert_eq!(args.len(), 1);419							if let Val::Str(s) = evaluate(context, &args[0].1)? {420								assert!(421									s.chars().count() == 1,422									"std.codepoint should receive single char string"423								);424								Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)425							} else {426								panic!("bad codepoint call");427							}428						}429						// object, includeHidden430						("std", "objectFieldsEx") => {431							assert_eq!(args.len(), 2);432							if let (Val::Obj(body), Val::Bool(_include_hidden)) = (433								evaluate(context.clone(), &args[0].1)?,434								evaluate(context, &args[1].1)?,435							) {436								// TODO: handle visibility (_include_hidden)437								Val::Arr(body.fields().into_iter().map(Val::Str).collect())438							} else {439								panic!("bad objectFieldsEx call");440							}441						}442						(ns, name) => panic!("Intristic not found: {}.{}", ns, name),443					},444					Val::Func(f) => f.evaluate(445						args.clone()446							.into_iter()447							.map(move |a| {448								(449									a.clone().0,450									Val::Lazy(lazy_val!(451										closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))452									)),453								)454							})455							.collect(),456					)?,457					_ => panic!("{:?} is not a function", value),458				}459			}460			Function(params, body) => evaluate_method(context, body, params.clone()),461			AssertExpr(AssertStmt(value, msg), returned) => {462				if evaluate(context.clone(), &value)?463					.try_cast_bool("assertion condition should be boolean")?464				{465					evaluate(context, returned)?466				} else if let Some(msg) = msg {467					panic!(468						"assertion failed ({:?}): {}",469						value,470						evaluate(context, msg)?471							.try_cast_str("assertion message should be string")?472					);473				} else {474					panic!("assertion failed ({:?}): no message", value);475				}476			}477			Error(e) => create_error(crate::Error::RuntimeError(478				evaluate(context, e)?.try_cast_str("error text should be string")?,479			))?,480			IfElse {481				cond,482				cond_then,483				cond_else,484			} => {485				if evaluate(context.clone(), &cond.0)?486					.try_cast_bool("if condition should be boolean")?487				{488					evaluate(context, cond_then)?489				} else {490					match cond_else {491						Some(v) => evaluate(context, v)?,492						None => Val::Bool(false),493					}494				}495			}496			_ => panic!(497				"evaluation not implemented: {:?}",498				LocExpr(expr.clone(), loc.clone())499			),500		})501	})502}