`b55d14a56b86`

source

crates/jrsonnet-evaluator/src/function.rs5.2 KiBsource history
1use crate::{error::Error::*, evaluate, lazy_val, resolved_lazy_val, throw, Context, Result, Val};2use closure::closure;3use jrsonnet_parser::{ArgsDesc, ParamsDesc};4use rustc_hash::FxHashMap;5use std::{collections::HashMap, hash::BuildHasherDefault, rc::Rc};67const NO_DEFAULT_CONTEXT: &str =8	"no default context set for call with defined default parameter value";910/// Creates correct [context](Context) for function body evaluation returning error on invalid call.11///12/// ## Parameters13/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)14/// * `body_ctx`: used for default parameter values' execution and for body execution (if set)15/// * `params`: function parameters' definition16/// * `args`: passed function arguments17/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily18pub fn parse_function_call(19	ctx: Context,20	body_ctx: Option<Context>,21	params: &ParamsDesc,22	args: &ArgsDesc,23	tailstrict: bool,24) -> Result<Context> {25	let mut out = HashMap::with_capacity_and_hasher(params.len(), BuildHasherDefault::default());26	let mut positioned_args = vec![None; params.0.len()];27	for (id, arg) in args.iter().enumerate() {28		let idx = if let Some(name) = &arg.0 {29			params30				.iter()31				.position(|p| *p.0 == *name)32				.ok_or_else(|| UnknownFunctionParameter(name.clone()))?33		} else {34			id35		};3637		if idx >= params.len() {38			throw!(TooManyArgsFunctionHas(params.len()));39		}40		if positioned_args[idx].is_some() {41			throw!(BindingParameterASecondTime(params[idx].0.clone()));42		}43		positioned_args[idx] = Some(arg.1.clone());44	}45	// Fill defaults46	for (id, p) in params.iter().enumerate() {47		let (ctx, expr) = if let Some(arg) = &positioned_args[id] {48			(ctx.clone(), arg)49		} else if let Some(default) = &p.1 {50			(body_ctx.clone().expect(NO_DEFAULT_CONTEXT), default)51		} else {52			throw!(FunctionParameterNotBoundInCall(p.0.clone()));53		};54		let val = if tailstrict {55			resolved_lazy_val!(evaluate(ctx, expr)?)56		} else {57			lazy_val!(closure!(clone ctx, clone expr, ||evaluate(ctx.clone(), &expr)))58		};59		out.insert(p.0.clone(), val);60	}6162	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None))63}6465pub fn parse_function_call_map(66	ctx: Context,67	body_ctx: Option<Context>,68	params: &ParamsDesc,69	args: &HashMap<Rc<str>, Val>,70	tailstrict: bool,71) -> Result<Context> {72	let mut out = FxHashMap::with_capacity_and_hasher(params.len(), BuildHasherDefault::default());73	let mut positioned_args = vec![None; params.0.len()];74	for (name, val) in args.iter() {75		let idx = params76			.iter()77			.position(|p| *p.0 == **name)78			.ok_or_else(|| UnknownFunctionParameter((name as &str).to_owned()))?;7980		if idx >= params.len() {81			throw!(TooManyArgsFunctionHas(params.len()));82		}83		if positioned_args[idx].is_some() {84			throw!(BindingParameterASecondTime(params[idx].0.clone()));85		}86		positioned_args[idx] = Some(val.clone());87	}88	// Fill defaults89	for (id, p) in params.iter().enumerate() {90		let val = if let Some(arg) = positioned_args[id].take() {91			resolved_lazy_val!(arg)92		} else if let Some(default) = &p.1 {93			if tailstrict {94				resolved_lazy_val!(evaluate(95					body_ctx.clone().expect(NO_DEFAULT_CONTEXT),96					default97				)?)98			} else {99				let body_ctx = body_ctx.clone();100				let default = default.clone();101				lazy_val!(move || {102					evaluate(body_ctx.clone().expect(NO_DEFAULT_CONTEXT), &default)103				})104			}105		} else {106			throw!(FunctionParameterNotBoundInCall(p.0.clone()));107		};108		out.insert(p.0.clone(), val);109	}110111	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None))112}113114pub fn place_args(115	ctx: Context,116	body_ctx: Option<Context>,117	params: &ParamsDesc,118	args: &[Val],119) -> Result<Context> {120	let mut out = FxHashMap::with_capacity_and_hasher(params.len(), BuildHasherDefault::default());121	let mut positioned_args = vec![None; params.0.len()];122	for (id, arg) in args.iter().enumerate() {123		if id >= params.len() {124			throw!(TooManyArgsFunctionHas(params.len()));125		}126		positioned_args[id] = Some(arg);127	}128	// Fill defaults129	for (id, p) in params.iter().enumerate() {130		let val = if let Some(arg) = &positioned_args[id] {131			(*arg).clone()132		} else if let Some(default) = &p.1 {133			evaluate(ctx.clone(), default)?134		} else {135			throw!(FunctionParameterNotBoundInCall(p.0.clone()));136		};137		out.insert(p.0.clone(), resolved_lazy_val!(val));138	}139140	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None))141}142143#[macro_export]144macro_rules! parse_args {145	($ctx: expr, $fn_name: expr, $args: expr, $total_args: expr, [146		$($id: expr, $name: ident: $ty: expr $(=>$match: path)?);+ $(;)?147	], $handler:block) => {{148		let args = $args;149		if args.len() > $total_args {150			throw!(TooManyArgsFunctionHas($total_args));151		}152		$(153			if args.len() <= $id {154				throw!(FunctionParameterNotBoundInCall(stringify!($name).into()));155			}156			let $name = &args[$id];157			if $name.0.is_some() {158				if $name.0.as_ref().unwrap() != stringify!($name) {159					throw!(IntrinsicArgumentReorderingIsNotSupportedYet);160				}161			}162			let $name = push(&None, || format!("evaluating argument"), || {163				let value = evaluate($ctx.clone(), &$name.1)?;164				$ty.check(&value)?;165				Ok(value)166			})?;167			$(168				let $name = if let $match(v) = $name {169					v170				} else {171					unreachable!();172				};173			)?174		)+175		($handler as crate::Result<_>)176	}};177}