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

1use crate::{2	create_error, create_error_result, evaluate, lazy_val, resolved_lazy_val, Context, Error,3	Result, Val,4};5use closure::closure;6use jrsonnet_parser::{ArgsDesc, ParamsDesc};7use std::{collections::HashMap, rc::Rc};89const NO_DEFAULT_CONTEXT: &str =10	"no default context set for call with defined default parameter value";1112/// Creates correct [context](Context) for function body evaluation, returning error on invalid call13///14/// * `ctx` used for passed argument expressions execution, and for body execution (if `body_ctx` is not set)15/// * `body_ctx` used for default parameter values execution, and for body execution (if set)16/// * `params` function parameters definition17/// * `args` passed function arguments18/// * `tailstruct` if true - function arguments is eager executed, otherwise - lazy19pub fn parse_function_call(20	ctx: Context,21	body_ctx: Option<Context>,22	params: &ParamsDesc,23	args: &ArgsDesc,24	tailstrict: bool,25) -> Result<Context> {26	let mut out = HashMap::new();27	let mut positioned_args = vec![None; params.0.len()];28	for (id, arg) in args.iter().enumerate() {29		let idx = if let Some(name) = &arg.0 {30			params31				.iter()32				.position(|p| *p.0 == *name)33				.ok_or_else(|| create_error(Error::UnknownFunctionParameter(name.clone())))?34		} else {35			id36		};3738		if idx >= params.len() {39			create_error_result(Error::TooManyArgsFunctionHas(params.len()))?;40		}41		if positioned_args[idx].is_some() {42			create_error_result(Error::BindingParameterASecondTime(params[idx].0.clone()))?;43		}44		positioned_args[idx] = Some(arg.1.clone());45	}46	// Fill defaults47	for (id, p) in params.iter().enumerate() {48		let (ctx, expr) = if let Some(arg) = &positioned_args[id] {49			(ctx.clone(), arg)50		} else if let Some(default) = &p.1 {51			(body_ctx.clone().expect(NO_DEFAULT_CONTEXT), default)52		} else {53			create_error_result(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 fn parse_function_call_map(68	ctx: Context,69	body_ctx: Option<Context>,70	params: &ParamsDesc,71	args: &HashMap<Rc<str>, Val>,72	tailstrict: bool,73) -> Result<Context> {74	let mut out = HashMap::new();75	let mut positioned_args = vec![None; params.0.len()];76	for (name, val) in args.iter() {77		let idx = params.iter().position(|p| *p.0 == **name).ok_or_else(|| {78			create_error(Error::UnknownFunctionParameter((&name as &str).to_owned()))79		})?;8081		if idx >= params.len() {82			create_error_result(Error::TooManyArgsFunctionHas(params.len()))?;83		}84		if positioned_args[idx].is_some() {85			create_error_result(Error::BindingParameterASecondTime(params[idx].0.clone()))?;86		}87		positioned_args[idx] = Some(val.clone());88	}89	// Fill defaults90	for (id, p) in params.iter().enumerate() {91		let val = if let Some(arg) = positioned_args[id].take() {92			resolved_lazy_val!(arg)93		} else if let Some(default) = &p.1 {94			if tailstrict {95				resolved_lazy_val!(evaluate(96					body_ctx.clone().expect(NO_DEFAULT_CONTEXT),97					default98				)?)99			} else {100				let body_ctx = body_ctx.clone();101				let default = default.clone();102				lazy_val!(move || {103					evaluate(body_ctx.clone().expect(NO_DEFAULT_CONTEXT), &default)104				})105			}106		} else {107			create_error_result(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;108			unreachable!()109		};110		out.insert(p.0.clone(), val);111	}112113	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)114}115116pub(crate) fn place_args(117	ctx: Context,118	body_ctx: Option<Context>,119	params: &ParamsDesc,120	args: &[Val],121) -> Result<Context> {122	let mut out = HashMap::new();123	let mut positioned_args = vec![None; params.0.len()];124	for (id, arg) in args.iter().enumerate() {125		if id >= params.len() {126			create_error_result(Error::TooManyArgsFunctionHas(params.len()))?;127		}128		positioned_args[id] = Some(arg);129	}130	// Fill defaults131	for (id, p) in params.iter().enumerate() {132		let val = if let Some(arg) = &positioned_args[id] {133			(*arg).clone()134		} else if let Some(default) = &p.1 {135			evaluate(ctx.clone(), default)?136		} else {137			create_error_result(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;138			unreachable!()139		};140		out.insert(p.0.clone(), resolved_lazy_val!(val));141	}142143	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)144}145146#[macro_export]147macro_rules! parse_args {148	($ctx: expr, $fn_name: expr, $args: expr, $total_args: expr, [149		$($id: expr, $name: ident $(: [$($p: path)|+] $(!! $a: path)?)?, $nt: expr);+ $(;)?150	], $handler:block) => {{151		use crate::Error;152		let args = $args;153		if args.len() > $total_args {154			create_error_result(Error::TooManyArgsFunctionHas($total_args))?;155		}156		$(157			if args.len() <= $id {158				create_error_result(Error::FunctionParameterNotBoundInCall(stringify!($name).into()))?;159			}160			let $name = &args[$id];161			if $name.0.is_some() {162				if $name.0.as_ref().unwrap() != stringify!($name) {163					create_error_result(Error::IntristicArgumentReorderingIsNotSupportedYet)?;164				}165			}166			let $name = evaluate($ctx.clone(), &$name.1)?;167			$(168				match $name {169					$($p(_))|+ => {},170					_ => create_error_result(Error::TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd argument"), $nt, $name.value_type()?))?,171				};172				$(173					let $name = match $name {174						$a(v) => v,175						_ => create_error_result(Error::TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd argument"), $nt, $name.value_type()?))?,176					};177				)*178			)*179		)+180		$handler181	}};182}183184#[test]185fn test() -> Result<()> {186	use crate::val::ValType;187	use jrsonnet_parser::*;188	let state = crate::EvaluationState::default();189	let evaluator = state.with_stdlib();190	let ctx = evaluator.create_default_context()?;191	evaluator.run_in_state(|| {192		parse_args!(ctx, "test", ArgsDesc(vec![193			Arg(None, el!(Expr::Num(2.0))),194			Arg(Some("b".into()), el!(Expr::Num(1.0))),195		]), 2, [196			0, a: [Val::Num]!!Val::Num, vec![ValType::Num];197			1, b: [Val::Num]!!Val::Num, vec![ValType::Num];198		], {199			assert!((a - 2.0).abs() <= f64::EPSILON);200			assert!((b - 1.0).abs() <= f64::EPSILON);201		});202		Ok(())203	})204}