git.delta.rocks / jrsonnet / refs/commits / 3d527a7bcaaf

1use crate::{2	builtin::{3		call_builtin,4		manifest::{manifest_json_ex, ManifestJsonOptions, ManifestType},5	},6	error::Error::*,7	evaluate,8	function::{parse_function_call, parse_function_call_map, place_args},9	native::NativeCallback,10	throw, with_state, Context, ObjValue, Result,11};12use jrsonnet_interner::IStr;13use jrsonnet_parser::{el, Arg, ArgsDesc, Expr, ExprLocation, LiteralType, LocExpr, ParamsDesc};14use jrsonnet_types::ValType;15use std::{cell::RefCell, collections::HashMap, fmt::Debug, rc::Rc};1617enum LazyValInternals {18	Computed(Val),19	Waiting(Box<dyn Fn() -> Result<Val>>),20}21#[derive(Clone)]22pub struct LazyVal(Rc<RefCell<LazyValInternals>>);23impl LazyVal {24	pub fn new(f: Box<dyn Fn() -> Result<Val>>) -> Self {25		Self(Rc::new(RefCell::new(LazyValInternals::Waiting(f))))26	}27	pub fn new_resolved(val: Val) -> Self {28		Self(Rc::new(RefCell::new(LazyValInternals::Computed(val))))29	}30	pub fn evaluate(&self) -> Result<Val> {31		let new_value = match &*self.0.borrow() {32			LazyValInternals::Computed(v) => return Ok(v.clone()),33			LazyValInternals::Waiting(f) => f()?,34		};35		*self.0.borrow_mut() = LazyValInternals::Computed(new_value.clone());36		Ok(new_value)37	}38}3940#[macro_export]41macro_rules! lazy_val {42	($f: expr) => {43		$crate::LazyVal::new(Box::new($f))44	};45}46#[macro_export]47macro_rules! resolved_lazy_val {48	($f: expr) => {49		$crate::LazyVal::new_resolved($f)50	};51}52impl Debug for LazyVal {53	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {54		write!(f, "Lazy")55	}56}57impl PartialEq for LazyVal {58	fn eq(&self, other: &Self) -> bool {59		Rc::ptr_eq(&self.0, &other.0)60	}61}6263#[derive(Debug, PartialEq)]64pub struct FuncDesc {65	pub name: IStr,66	pub ctx: Context,67	pub params: ParamsDesc,68	pub body: LocExpr,69}7071#[derive(Debug)]72pub enum FuncVal {73	/// Plain function implemented in jsonnet74	Normal(FuncDesc),75	/// Standard library function76	Intrinsic(IStr),77	/// Library functions implemented in native78	NativeExt(IStr, Rc<NativeCallback>),79}8081impl PartialEq for FuncVal {82	fn eq(&self, other: &Self) -> bool {83		match (self, other) {84			(Self::Normal(a), Self::Normal(b)) => a == b,85			(Self::Intrinsic(an), Self::Intrinsic(bn)) => an == bn,86			(Self::NativeExt(an, _), Self::NativeExt(bn, _)) => an == bn,87			(..) => false,88		}89	}90}91impl FuncVal {92	pub fn is_ident(&self) -> bool {93		matches!(&self, Self::Intrinsic(n) if n as &str == "id")94	}95	pub fn name(&self) -> IStr {96		match self {97			Self::Normal(normal) => normal.name.clone(),98			Self::Intrinsic(name) => format!("std.{}", name).into(),99			Self::NativeExt(n, _) => format!("native.{}", n).into(),100		}101	}102	pub fn evaluate(103		&self,104		call_ctx: Context,105		loc: &Option<ExprLocation>,106		args: &ArgsDesc,107		tailstrict: bool,108	) -> Result<Val> {109		match self {110			Self::Normal(func) => {111				let ctx = parse_function_call(112					call_ctx,113					Some(func.ctx.clone()),114					&func.params,115					args,116					tailstrict,117				)?;118				evaluate(ctx, &func.body)119			}120			Self::Intrinsic(name) => call_builtin(call_ctx, loc, name, args),121			Self::NativeExt(_name, handler) => {122				let args = parse_function_call(call_ctx, None, &handler.params, args, true)?;123				let mut out_args = Vec::with_capacity(handler.params.len());124				for p in handler.params.0.iter() {125					out_args.push(args.binding(p.0.clone())?.evaluate()?);126				}127				Ok(handler.call(&out_args)?)128			}129		}130	}131132	pub fn evaluate_map(133		&self,134		call_ctx: Context,135		args: &HashMap<IStr, Val>,136		tailstrict: bool,137	) -> Result<Val> {138		match self {139			Self::Normal(func) => {140				let ctx = parse_function_call_map(141					call_ctx,142					Some(func.ctx.clone()),143					&func.params,144					args,145					tailstrict,146				)?;147				evaluate(ctx, &func.body)148			}149			Self::Intrinsic(_) => todo!(),150			Self::NativeExt(_, _) => todo!(),151		}152	}153154	pub fn evaluate_values(&self, call_ctx: Context, args: &[Val]) -> Result<Val> {155		match self {156			Self::Normal(func) => {157				let ctx = place_args(call_ctx, Some(func.ctx.clone()), &func.params, args)?;158				evaluate(ctx, &func.body)159			}160			Self::Intrinsic(_) => todo!(),161			Self::NativeExt(_, _) => todo!(),162		}163	}164}165166#[derive(Clone)]167pub enum ManifestFormat {168	YamlStream(Box<ManifestFormat>),169	Yaml(usize),170	Json(usize),171	ToString,172	String,173}174175#[derive(Debug, Clone)]176pub enum ArrValue {177	Lazy(Rc<Vec<LazyVal>>),178	Eager(Rc<Vec<Val>>),179}180impl ArrValue {181	pub fn len(&self) -> usize {182		match self {183			ArrValue::Lazy(l) => l.len(),184			ArrValue::Eager(e) => e.len(),185		}186	}187188	pub fn is_empty(&self) -> bool {189		self.len() == 0190	}191192	pub fn get(&self, index: usize) -> Result<Option<Val>> {193		match self {194			ArrValue::Lazy(vec) => {195				if let Some(v) = vec.get(index) {196					Ok(Some(v.evaluate()?))197				} else {198					Ok(None)199				}200			}201			ArrValue::Eager(vec) => Ok(vec.get(index).cloned()),202		}203	}204205	pub fn get_lazy(&self, index: usize) -> Option<LazyVal> {206		match self {207			ArrValue::Lazy(vec) => vec.get(index).cloned(),208			ArrValue::Eager(vec) => vec209				.get(index)210				.cloned()211				.map(|val| LazyVal::new_resolved(val)),212		}213	}214215	pub fn evaluated(&self) -> Result<Rc<Vec<Val>>> {216		Ok(match self {217			ArrValue::Lazy(vec) => {218				let mut out = Vec::with_capacity(vec.len());219				for item in vec.iter() {220					out.push(item.evaluate()?);221				}222				Rc::new(out)223			}224			ArrValue::Eager(vec) => vec.clone(),225		})226	}227228	pub fn iter(&self) -> impl DoubleEndedIterator<Item = Result<Val>> + '_ {229		(0..self.len()).map(move |idx| match self {230			ArrValue::Lazy(l) => l[idx].evaluate(),231			ArrValue::Eager(e) => Ok(e[idx].clone()),232		})233	}234235	pub fn iter_lazy(&self) -> impl DoubleEndedIterator<Item = LazyVal> + '_ {236		(0..self.len()).map(move |idx| match self {237			ArrValue::Lazy(l) => l[idx].clone(),238			ArrValue::Eager(e) => LazyVal::new_resolved(e[idx].clone()),239		})240	}241242	pub fn reversed(self) -> Self {243		match self {244			ArrValue::Lazy(vec) => {245				let mut out = (&vec as &Vec<_>).clone();246				out.reverse();247				Self::Lazy(Rc::new(out))248			}249			ArrValue::Eager(vec) => {250				let mut out = (&vec as &Vec<_>).clone();251				out.reverse();252				Self::Eager(Rc::new(out))253			}254		}255	}256257	pub fn ptr_eq(a: &ArrValue, b: &ArrValue) -> bool {258		match (a, b) {259			(ArrValue::Lazy(a), ArrValue::Lazy(b)) => Rc::ptr_eq(a, b),260			(ArrValue::Eager(a), ArrValue::Eager(b)) => Rc::ptr_eq(a, b),261			_ => false,262		}263	}264}265266impl From<Vec<LazyVal>> for ArrValue {267	fn from(v: Vec<LazyVal>) -> Self {268		Self::Lazy(Rc::new(v))269	}270}271272impl From<Vec<Val>> for ArrValue {273	fn from(v: Vec<Val>) -> Self {274		Self::Eager(Rc::new(v))275	}276}277278#[derive(Debug, Clone)]279pub enum Val {280	Bool(bool),281	Null,282	Str(IStr),283	Num(f64),284	Arr(ArrValue),285	Obj(ObjValue),286	Func(Rc<FuncVal>),287}288289macro_rules! matches_unwrap {290	($e: expr, $p: pat, $r: expr) => {291		match $e {292			$p => $r,293			_ => panic!("no match"),294			}295	};296}297impl Val {298	/// Creates `Val::Num` after checking for numeric overflow.299	/// As numbers are `f64`, we can just check for their finity.300	pub fn new_checked_num(num: f64) -> Result<Self> {301		if num.is_finite() {302			Ok(Self::Num(num))303		} else {304			throw!(RuntimeError("overflow".into()))305		}306	}307308	pub fn assert_type(&self, context: &'static str, val_type: ValType) -> Result<()> {309		let this_type = self.value_type();310		if this_type != val_type {311			throw!(TypeMismatch(context, vec![val_type], this_type))312		} else {313			Ok(())314		}315	}316	pub fn unwrap_num(self) -> Result<f64> {317		Ok(matches_unwrap!(self, Self::Num(v), v))318	}319	pub fn unwrap_func(self) -> Result<Rc<FuncVal>> {320		Ok(matches_unwrap!(self, Self::Func(v), v))321	}322	pub fn try_cast_bool(self, context: &'static str) -> Result<bool> {323		self.assert_type(context, ValType::Bool)?;324		Ok(matches_unwrap!(self, Self::Bool(v), v))325	}326	pub fn try_cast_str(self, context: &'static str) -> Result<IStr> {327		self.assert_type(context, ValType::Str)?;328		Ok(matches_unwrap!(self, Self::Str(v), v))329	}330	pub fn try_cast_num(self, context: &'static str) -> Result<f64> {331		self.assert_type(context, ValType::Num)?;332		self.unwrap_num()333	}334	pub fn value_type(&self) -> ValType {335		match self {336			Self::Str(..) => ValType::Str,337			Self::Num(..) => ValType::Num,338			Self::Arr(..) => ValType::Arr,339			Self::Obj(..) => ValType::Obj,340			Self::Bool(_) => ValType::Bool,341			Self::Null => ValType::Null,342			Self::Func(..) => ValType::Func,343		}344	}345346	pub fn to_string(&self) -> Result<IStr> {347		Ok(match self {348			Self::Bool(true) => "true".into(),349			Self::Bool(false) => "false".into(),350			Self::Null => "null".into(),351			Self::Str(s) => s.clone(),352			v => manifest_json_ex(353				&v,354				&ManifestJsonOptions {355					padding: "",356					mtype: ManifestType::ToString,357				},358			)?359			.into(),360		})361	}362363	/// Expects value to be object, outputs (key, manifested value) pairs364	pub fn manifest_multi(&self, ty: &ManifestFormat) -> Result<Vec<(IStr, IStr)>> {365		let obj = match self {366			Self::Obj(obj) => obj,367			_ => throw!(MultiManifestOutputIsNotAObject),368		};369		let keys = obj.visible_fields();370		let mut out = Vec::with_capacity(keys.len());371		for key in keys {372			let value = obj373				.get(key.clone())?374				.expect("item in object")375				.manifest(ty)?;376			out.push((key, value));377		}378		Ok(out)379	}380381	/// Expects value to be array, outputs manifested values382	pub fn manifest_stream(&self, ty: &ManifestFormat) -> Result<Vec<IStr>> {383		let arr = match self {384			Self::Arr(a) => a,385			_ => throw!(StreamManifestOutputIsNotAArray),386		};387		let mut out = Vec::with_capacity(arr.len());388		for i in arr.iter() {389			out.push(i?.manifest(ty)?);390		}391		Ok(out)392	}393394	pub fn manifest(&self, ty: &ManifestFormat) -> Result<IStr> {395		Ok(match ty {396			ManifestFormat::YamlStream(format) => {397				let arr = match self {398					Self::Arr(a) => a,399					_ => throw!(StreamManifestOutputIsNotAArray),400				};401				let mut out = String::new();402403				match format as &ManifestFormat {404					ManifestFormat::YamlStream(_) => throw!(StreamManifestOutputCannotBeRecursed),405					ManifestFormat::String => throw!(StreamManifestCannotNestString),406					_ => {}407				};408409				if !arr.is_empty() {410					for v in arr.iter() {411						out.push_str("---\n");412						out.push_str(&v?.manifest(format)?);413						out.push('\n');414					}415					out.push_str("...");416				}417418				out.into()419			}420			ManifestFormat::Yaml(padding) => self.to_yaml(*padding)?,421			ManifestFormat::Json(padding) => self.to_json(*padding)?,422			ManifestFormat::ToString => self.to_string()?,423			ManifestFormat::String => match self {424				Self::Str(s) => s.clone(),425				_ => throw!(StringManifestOutputIsNotAString),426			},427		})428	}429430	/// For manifestification431	pub fn to_json(&self, padding: usize) -> Result<IStr> {432		manifest_json_ex(433			self,434			&ManifestJsonOptions {435				padding: &" ".repeat(padding),436				mtype: if padding == 0 {437					ManifestType::Minify438				} else {439					ManifestType::Manifest440				},441			},442		)443		.map(|s| s.into())444	}445446	/// Calls `std.manifestJson`447	#[cfg(feature = "faster")]448	pub fn to_std_json(&self, padding: usize) -> Result<Rc<str>> {449		manifest_json_ex(450			self,451			&ManifestJsonOptions {452				padding: &" ".repeat(padding),453				mtype: ManifestType::Std,454			},455		)456		.map(|s| s.into())457	}458459	/// Calls `std.manifestJson`460	#[cfg(not(feature = "faster"))]461	pub fn to_std_json(&self, padding: usize) -> Result<Rc<str>> {462		with_state(|s| {463			let ctx = s464				.create_default_context()?465				.with_var("__tmp__to_json__".into(), self.clone())?;466			Ok(evaluate(467				ctx,468				&el!(Expr::Apply(469					el!(Expr::Index(470						el!(Expr::Var("std".into())),471						el!(Expr::Str("manifestJsonEx".into()))472					)),473					ArgsDesc(vec![474						Arg(None, el!(Expr::Var("__tmp__to_json__".into()))),475						Arg(None, el!(Expr::Str(" ".repeat(padding).into())))476					]),477					false478				)),479			)?480			.try_cast_str("to json")?)481		})482	}483	pub fn to_yaml(&self, padding: usize) -> Result<IStr> {484		with_state(|s| {485			let ctx = s486				.create_default_context()?487				.with_var("__tmp__to_json__".into(), self.clone());488			Ok(evaluate(489				ctx,490				&el!(Expr::Apply(491					el!(Expr::Index(492						el!(Expr::Var("std".into())),493						el!(Expr::Str("manifestYamlDoc".into()))494					)),495					ArgsDesc(vec![496						Arg(None, el!(Expr::Var("__tmp__to_json__".into()))),497						Arg(498							None,499							el!(Expr::Literal(if padding != 0 {500								LiteralType::True501							} else {502								LiteralType::False503							}))504						)505					]),506					false507				)),508			)?509			.try_cast_str("to json")?)510		})511	}512}513514const fn is_function_like(val: &Val) -> bool {515	matches!(val, Val::Func(_))516}517518/// Native implementation of `std.primitiveEquals`519pub fn primitive_equals(val_a: &Val, val_b: &Val) -> Result<bool> {520	Ok(match (val_a, val_b) {521		(Val::Bool(a), Val::Bool(b)) => a == b,522		(Val::Null, Val::Null) => true,523		(Val::Str(a), Val::Str(b)) => a == b,524		(Val::Num(a), Val::Num(b)) => (a - b).abs() <= f64::EPSILON,525		(Val::Arr(_), Val::Arr(_)) => throw!(RuntimeError(526			"primitiveEquals operates on primitive types, got array".into(),527		)),528		(Val::Obj(_), Val::Obj(_)) => throw!(RuntimeError(529			"primitiveEquals operates on primitive types, got object".into(),530		)),531		(a, b) if is_function_like(&a) && is_function_like(&b) => {532			throw!(RuntimeError("cannot test equality of functions".into()))533		}534		(_, _) => false,535	})536}537538/// Native implementation of `std.equals`539pub fn equals(val_a: &Val, val_b: &Val) -> Result<bool> {540	if val_a.value_type() != val_b.value_type() {541		return Ok(false);542	}543	match (val_a, val_b) {544		(Val::Arr(a), Val::Arr(b)) => {545			if ArrValue::ptr_eq(a, b) {546				return Ok(true);547			}548			if a.len() != b.len() {549				return Ok(false);550			}551			for (a, b) in a.iter().zip(b.iter()) {552				if !equals(&a?, &b?)? {553					return Ok(false);554				}555			}556			Ok(true)557		}558		(Val::Obj(a), Val::Obj(b)) => {559			if ObjValue::ptr_eq(a, b) {560				return Ok(true);561			}562			let fields = a.visible_fields();563			if fields != b.visible_fields() {564				return Ok(false);565			}566			for field in fields {567				if !equals(&a.get(field.clone())?.unwrap(), &b.get(field)?.unwrap())? {568					return Ok(false);569				}570			}571			Ok(true)572		}573		(a, b) => Ok(primitive_equals(&a, &b)?),574	}575}