git.delta.rocks / jrsonnet / refs/commits / 7288e541bb59

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	inline_parse_function_call(ctx, body_ctx, params, args, tailstrict)21}2223/// See [parse_function_call](parse_function_call)24///25/// ## Notes26/// This function is always inlined for tailstrict27#[inline(always)]28pub(crate) fn inline_parse_function_call(29	ctx: Context,30	body_ctx: Option<Context>,31	params: &ParamsDesc,32	args: &ArgsDesc,33	tailstrict: bool,34) -> Result<Context> {35	let mut out = HashMap::new();36	let mut positioned_args = vec![None; params.0.len()];37	for (id, arg) in args.iter().enumerate() {38		let idx = if let Some(name) = &arg.0 {39			params.iter().position(|p| &p.0 == name).ok_or_else(|| {40				create_error::<()>(Error::UnknownFunctionParameter(name.clone()))41					.err()42					.unwrap()43			})?44		} else {45			id46		};4748		if idx >= params.len() {49			create_error(Error::TooManyArgsFunctionHas(params.len()))?;50		}51		if positioned_args[idx].is_some() {52			create_error(Error::BindingParameterASecondTime(params[idx].0.clone()))?;53		}54		positioned_args[idx] = Some(arg.1.clone());55	}56	// Fill defaults57	for (id, p) in params.iter().enumerate() {58		let (ctx, expr) = if let Some(arg) = &positioned_args[id] {59			(ctx.clone(), arg)60		} else if let Some(default) = &p.1 {61			(62				body_ctx63					.clone()64					.expect("no default context set for call with defined default parameter value"),65				default,66			)67		} else {68			create_error(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;69			unreachable!()70		};71		let val = if tailstrict {72			resolved_lazy_val!(evaluate(ctx, expr)?)73		} else {74			lazy_val!(closure!(clone ctx, clone expr, ||evaluate(ctx.clone(), &expr)))75		};76		out.insert(p.0.clone(), val);77	}7879	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)80}8182#[inline(always)]83pub(crate) fn place_args(84	ctx: Context,85	body_ctx: Option<Context>,86	params: &ParamsDesc,87	args: &[Val],88) -> Result<Context> {89	let mut out = HashMap::new();90	let mut positioned_args = vec![None; params.0.len()];91	for (id, arg) in args.iter().enumerate() {92		if id >= params.len() {93			create_error(Error::TooManyArgsFunctionHas(params.len()))?;94		}95		positioned_args[id] = Some(arg);96	}97	// Fill defaults98	for (id, p) in params.iter().enumerate() {99		let val = if let Some(arg) = &positioned_args[id] {100			(*arg).clone()101		} else if let Some(default) = &p.1 {102			evaluate(ctx.clone(), default)?103		} else {104			create_error(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;105			unreachable!()106		};107		out.insert(p.0.clone(), resolved_lazy_val!(val));108	}109110	Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)111}