`bcda254909e4`

source

crates/jsonnet-evaluator/src/function.rs4.6 KiBsource history
1use crate::{create_error, evaluate, lazy_val, resolved_lazy_val, Context, Error, Result, Val};2use closure::closure;3use jsonnet_parser::{ArgsDesc, ParamsDesc};4use std::collections::HashMap;56/// Creates correct [context](Context) for function body evaluation, returning error on invalid call7///8/// * `ctx` used for passed argument expressions execution, and for body execution (if `body_ctx` is not set)9/// * `body_ctx` used for default parameter values execution, and for body execution (if set)10/// * `params` function parameters definition11/// * `args` passed function arguments12/// * `tailstruct` if true - function arguments is eager executed, otherwise - lazy13pub fn parse_function_call(14	ctx: Context,15	body_ctx: Option<Context>,16	params: &ParamsDesc,17	args: &ArgsDesc,18	tailstrict: bool,19) -> Result<Context> {20	let mut out = HashMap::new();21	let mut positioned_args = vec![None; params.0.len()];22	for (id, arg) in args.iter().enumerate() {23		let idx = if let Some(name) = &arg.0 {24			params.iter().position(|p| *p.0 == *name).ok_or_else(|| {25				create_error::<()>(Error::UnknownFunctionParameter(name.clone()))26					.err()27					.unwrap()28			})?29		} else {30			id31		};3233		if idx >= params.len() {34			create_error(Error::TooManyArgsFunctionHas(params.len()))?;35		}36		if positioned_args[idx].is_some() {37			create_error(Error::BindingParameterASecondTime(params[idx].0.clone()))?;38		}39		positioned_args[idx] = Some(arg.1.clone());40	}41	// Fill defaults42	for (id, p) in params.iter().enumerate() {43		let (ctx, expr) = if let Some(arg) = &positioned_args[id] {44			(ctx.clone(), arg)45		} else if let Some(default) = &p.1 {46			(47				body_ctx48					.clone()49					.expect("no default context set for call with defined default parameter value"),50				default,51			)52		} else {53			create_error(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;54			unreachable!()55		};56		let val = if tailstrict {57			resolved_lazy_val!(evaluate(ctx, expr)?)58		} else {59			lazy_val!(closure!(clone ctx, clone expr, ||evaluate(ctx.clone(), &expr)))60		};61		out.insert(p.0.clone(), val);62	}6364	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)65}6667pub(crate) fn place_args(68	ctx: Context,69	body_ctx: Option<Context>,70	params: &ParamsDesc,71	args: &[Val],72) -> Result<Context> {73	let mut out = HashMap::new();74	let mut positioned_args = vec![None; params.0.len()];75	for (id, arg) in args.iter().enumerate() {76		if id >= params.len() {77			create_error(Error::TooManyArgsFunctionHas(params.len()))?;78		}79		positioned_args[id] = Some(arg);80	}81	// Fill defaults82	for (id, p) in params.iter().enumerate() {83		let val = if let Some(arg) = &positioned_args[id] {84			(*arg).clone()85		} else if let Some(default) = &p.1 {86			evaluate(ctx.clone(), default)?87		} else {88			create_error(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;89			unreachable!()90		};91		out.insert(p.0.clone(), resolved_lazy_val!(val));92	}9394	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)95}9697#[macro_export]98macro_rules! parse_args {99	($ctx: expr, $fn_name: expr, $args: expr, $total_args: expr, [100		$($id: expr, $name: ident $(: [$($p: path)|+] $(!! $a: path)?)?, $nt: expr);+ $(;)?101	], $handler:block) => {{102		use crate::error::Error;103		let args = $args;104		if args.len() > $total_args {105			create_error(Error::TooManyArgsFunctionHas($total_args))?;106		}107		$(108			if args.len() <= $id {109				create_error(Error::FunctionParameterNotBoundInCall(stringify!($name).into()))?;110			}111			let $name = &args[$id];112			if $name.0.is_some() {113				if $name.0.as_ref().unwrap() != stringify!($name) {114					create_error(Error::IntristicArgumentReorderingIsNotSupportedYet)?;115				}116			}117			let $name = evaluate($ctx.clone(), &$name.1)?;118			$(119				match $name {120					$($p(_))|+ => {},121					_ => create_error(Error::TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd argument"), $nt, $name.value_type()?))?,122				};123				$(124					let $name = match $name {125						$a(v) => v,126						_ => create_error(Error::TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd argument"), $nt, $name.value_type()?))?,127					};128				)*129			)*130		)+131		$handler132	}};133}134135#[test]136fn test() -> Result<()> {137	use jsonnet_parser::*;138	use crate::val::ValType;139	let state = crate::EvaluationState::default();140	let evaluator = state.with_stdlib();141	let ctx = evaluator.create_default_context()?;142	evaluator.run_in_state(|| {143		parse_args!(ctx, "test", ArgsDesc(vec![144			Arg(None, el!(Expr::Num(2.0))),145			Arg(Some("b".into()), el!(Expr::Num(1.0))),146		]), 2, [147			0, a: [Val::Num]!!Val::Num, vec![ValType::Num];148			1, b: [Val::Num]!!Val::Num, vec![ValType::Num];149		], {150			assert!((a - 2.0).abs() <= f64::EPSILON);151			assert!((b - 1.0).abs() <= f64::EPSILON);152		});153		Ok(())154	})155}