`5df60b8b674f`

source

crates/jrsonnet-evaluator/src/function/parse.rs6.2 KiBsource history
1use jrsonnet_parser::{2	function::{FunctionSignature, ParamName},3	ExprParams,4};5use rustc_hash::FxHashMap;67use super::arglike::ArgsLike;8use crate::{9	bail,10	destructure::destruct,11	error::{ErrorKind::*, Result},12	evaluate_named_param,13	gc::WithCapacityExt as _,14	Context, Pending, Thunk, Val,15};1617/// Creates correct [context](Context) for function body evaluation returning error on invalid call.18///19/// ## Parameters20/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)21/// * `body_ctx`: used for default parameter values' execution and for body execution (if set)22/// * `params`: function parameters' definition23/// * `args`: passed function arguments24/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily25pub fn parse_function_call(26	ctx: Context,27	body_ctx: Context,28	params: &ExprParams,29	args: &dyn ArgsLike,30	tailstrict: bool,31) -> Result<Context> {32	let mut passed_args = FxHashMap::with_capacity(params.binds_len());33	if args.unnamed_len() > params.signature.len() {34		bail!(TooManyArgsFunctionHas(35			params.signature.len(),36			params.signature.clone(),37		))38	}3940	let mut filled_named = 0;41	let mut filled_positionals = 0;4243	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {44		destruct(45			&params.exprs[id].destruct,46			arg,47			Pending::new_filled(ctx.clone()),48			&mut passed_args,49		)?;50		filled_positionals += 1;51		Ok(())52	})?;5354	args.named_iter(ctx, tailstrict, &mut |name, value| {55		// FIXME: O(n) for arg existence check56		if !params.exprs.iter().any(|p| &p.destruct.name() == name) {57			bail!(UnknownFunctionParameter(name.clone()));58		}59		if passed_args.insert(name.clone(), value).is_some() {60			bail!(BindingParameterASecondTime(name.clone()));61		}62		filled_named += 1;63		Ok(())64	})?;6566	if filled_named + filled_positionals < params.len() {67		// Some args are unset, but maybe we have defaults for them68		// Default values should be created in newly created context69		let fctx = Context::new_future();70		let mut defaults =71			FxHashMap::with_capacity(params.binds_len() - filled_named - filled_positionals);7273		for (idx, into, default) in params74			.exprs75			.iter()76			.enumerate()77			.filter_map(|(i, p)| Some((i, &p.destruct, p.default.as_ref()?)))78		{79			if let ParamName::Named(name) = into.name() {80				if passed_args.contains_key(&name) {81					continue;82				}83			} else if idx < filled_positionals {84				continue;85			}8687			destruct(88				into,89				{90					let ctx = fctx.clone();91					let name = into.name();92					let value = default.clone();93					Thunk!(move || evaluate_named_param(ctx.unwrap(), &value, name))94				},95				fctx.clone(),96				&mut defaults,97			)?;98			if into.name().is_named() {99				filled_named += 1;100			} else {101				filled_positionals += 1;102			}103		}104105		// Some args still weren't filled106		if filled_named + filled_positionals != params.len() {107			for param in params.exprs.iter().skip(args.unnamed_len()) {108				let mut found = false;109				args.named_names(&mut |name| {110					if &param.destruct.name() == name {111						found = true;112					}113				});114				if !found {115					bail!(FunctionParameterNotBoundInCall(116						param.destruct.name(),117						params.signature.clone()118					));119				}120			}121			unreachable!();122		}123124		Ok(body_ctx125			.extend_bindings(passed_args)126			.extend_bindings(defaults)127			.into_future(fctx))128	} else {129		let body_ctx = body_ctx.extend_bindings(passed_args);130		Ok(body_ctx)131	}132}133134/// You shouldn't probally use this function, use `jrsonnet_macros::builtin` instead135///136/// ## Parameters137/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)138/// * `params`: function parameters' definition139/// * `args`: passed function arguments140/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily141pub fn parse_builtin_call(142	ctx: Context,143	params: FunctionSignature,144	args: &dyn ArgsLike,145	tailstrict: bool,146) -> Result<Vec<Option<Thunk<Val>>>> {147	let mut passed_args: Vec<Option<Thunk<Val>>> = vec![None; params.len()];148	if args.unnamed_len() > params.len() {149		bail!(TooManyArgsFunctionHas(params.len(), params,))150	}151152	let mut filled_args = 0;153154	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {155		passed_args[id] = Some(arg);156		filled_args += 1;157		Ok(())158	})?;159160	args.named_iter(ctx, tailstrict, &mut |name, arg| {161		// FIXME: O(n) for arg existence check162		let id = params163			.iter()164			.position(|p| p.name() == name)165			.ok_or_else(|| UnknownFunctionParameter(name.clone()))?;166		if passed_args[id].replace(arg).is_some() {167			bail!(BindingParameterASecondTime(name.clone()));168		}169		filled_args += 1;170		Ok(())171	})?;172173	if filled_args < params.len() {174		for (id, _) in params.iter().enumerate().filter(|(_, p)| p.has_default()) {175			if passed_args[id].is_some() {176				continue;177			}178			filled_args += 1;179		}180181		// Some args still wasn't filled182		if filled_args != params.len() {183			for param in params.iter().skip(args.unnamed_len()) {184				let mut found = false;185				args.named_names(&mut |name| {186					if param.name() == name {187						found = true;188					}189				});190				if !found {191					bail!(FunctionParameterNotBoundInCall(192						param.name().clone(),193						params,194					));195				}196			}197			unreachable!();198		}199	}200	Ok(passed_args)201}202203/// Creates Context, which has all argument default values applied204/// and with unbound values causing error to be returned205pub fn parse_default_function_call(body_ctx: Context, params: &ExprParams) -> Result<Context> {206	let fctx = Context::new_future();207208	let mut bindings = FxHashMap::with_capacity(params.binds_len());209210	for param in params.exprs.iter() {211		if let Some(v) = &param.default {212			destruct(213				&param.destruct.clone(),214				{215					let ctx = fctx.clone();216					let name = param.destruct.name();217					let value = v.clone();218					Thunk!(move || evaluate_named_param(ctx.unwrap(), &value, name))219				},220				fctx.clone(),221				&mut bindings,222			)?;223		} else {224			destruct(225				&param.destruct,226				{227					let param_name = param.destruct.name();228					let params = params.clone();229					Thunk!(move || Err(FunctionParameterNotBoundInCall(230						param_name,231						params.signature232					)233					.into()))234				},235				fctx.clone(),236				&mut bindings,237			)?;238		}239	}240241	Ok(body_ctx.extend_bindings(bindings).into_future(fctx))242}