`b3f009bb7b2e`

source

crates/jrsonnet-evaluator/src/function/parse.rs6.2 KiBsource history
1use std::mem::replace;23use jrsonnet_parser::{4	function::{FunctionSignature, ParamName},5	ExprParams,6};7use rustc_hash::FxHashMap;89use super::arglike::ArgsLike;10use crate::{11	bail,12	destructure::destruct,13	error::{ErrorKind::*, Result},14	evaluate_named_param,15	gc::WithCapacityExt as _,16	Context, Pending, Thunk, Val,17};1819/// Creates correct [context](Context) for function body evaluation returning error on invalid call.20///21/// ## Parameters22/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)23/// * `body_ctx`: used for default parameter values' execution and for body execution (if set)24/// * `params`: function parameters' definition25/// * `args`: passed function arguments26/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily27pub fn parse_function_call(28	ctx: Context,29	body_ctx: Context,30	params: &ExprParams,31	args: &dyn ArgsLike,32	tailstrict: bool,33) -> Result<Context> {34	let mut passed_args = FxHashMap::with_capacity(params.binds_len());35	if args.unnamed_len() > params.signature.len() {36		bail!(TooManyArgsFunctionHas(37			params.signature.len(),38			params.signature.clone(),39		))40	}4142	let mut filled_named = 0;43	let mut filled_positionals = 0;4445	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {46		destruct(47			&params.exprs[id].destruct,48			arg,49			Pending::new_filled(ctx.clone()),50			&mut passed_args,51		)?;52		filled_positionals += 1;53		Ok(())54	})?;5556	args.named_iter(ctx, tailstrict, &mut |name, value| {57		// FIXME: O(n) for arg existence check58		if !params.exprs.iter().any(|p| &p.destruct.name() == name) {59			bail!(UnknownFunctionParameter(name.clone()));60		}61		if passed_args.insert(name.clone(), value).is_some() {62			bail!(BindingParameterASecondTime(name.clone()));63		}64		filled_named += 1;65		Ok(())66	})?;6768	if filled_named + filled_positionals < params.len() {69		// Some args are unset, but maybe we have defaults for them70		// Default values should be created in newly created context71		let fctx = Context::new_future();72		let mut defaults =73			FxHashMap::with_capacity(params.binds_len() - filled_named - filled_positionals);7475		for (idx, into, default) in params76			.exprs77			.iter()78			.enumerate()79			.filter_map(|(i, p)| Some((i, &p.destruct, p.default.as_ref()?)))80		{81			if let ParamName::Named(name) = into.name() {82				if passed_args.contains_key(&name) {83					continue;84				}85			} else if idx < filled_positionals {86				continue;87			}8889			destruct(90				&into,91				{92					let ctx = fctx.clone();93					let name = into.name();94					let value = default.clone();95					Thunk!(move || evaluate_named_param(ctx.unwrap(), &value, name))96				},97				fctx.clone(),98				&mut defaults,99			)?;100			if !into.name().is_anonymous() {101				filled_named += 1;102			} else {103				filled_positionals += 1;104			}105		}106107		// Some args still weren't filled108		if filled_named + filled_positionals != params.len() {109			for param in params.exprs.iter().skip(args.unnamed_len()) {110				let mut found = false;111				args.named_names(&mut |name| {112					if &param.destruct.name() == name {113						found = true;114					}115				});116				if !found {117					bail!(FunctionParameterNotBoundInCall(118						param.destruct.name(),119						params.signature.clone()120					));121				}122			}123			unreachable!();124		}125126		Ok(body_ctx127			.extend_bindings(passed_args)128			.extend_bindings(defaults)129			.into_future(fctx))130	} else {131		let body_ctx = body_ctx.extend_bindings(passed_args);132		Ok(body_ctx)133	}134}135136/// You shouldn't probally use this function, use `jrsonnet_macros::builtin` instead137///138/// ## Parameters139/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)140/// * `params`: function parameters' definition141/// * `args`: passed function arguments142/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily143pub fn parse_builtin_call(144	ctx: Context,145	params: FunctionSignature,146	args: &dyn ArgsLike,147	tailstrict: bool,148) -> Result<Vec<Option<Thunk<Val>>>> {149	let mut passed_args: Vec<Option<Thunk<Val>>> = vec![None; params.len()];150	if args.unnamed_len() > params.len() {151		bail!(TooManyArgsFunctionHas(params.len(), params,))152	}153154	let mut filled_args = 0;155156	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {157		passed_args[id] = Some(arg);158		filled_args += 1;159		Ok(())160	})?;161162	args.named_iter(ctx, tailstrict, &mut |name, arg| {163		// FIXME: O(n) for arg existence check164		let id = params165			.iter()166			.position(|p| p.name() == name)167			.ok_or_else(|| UnknownFunctionParameter(name.clone()))?;168		if replace(&mut passed_args[id], Some(arg)).is_some() {169			bail!(BindingParameterASecondTime(name.clone()));170		}171		filled_args += 1;172		Ok(())173	})?;174175	if filled_args < params.len() {176		for (id, _) in params.iter().enumerate().filter(|(_, p)| p.has_default()) {177			if passed_args[id].is_some() {178				continue;179			}180			filled_args += 1;181		}182183		// Some args still wasn't filled184		if filled_args != params.len() {185			for param in params.iter().skip(args.unnamed_len()) {186				let mut found = false;187				args.named_names(&mut |name| {188					if param.name() == name {189						found = true;190					}191				});192				if !found {193					bail!(FunctionParameterNotBoundInCall(194						param.name().clone(),195						params,196					));197				}198			}199			unreachable!();200		}201	}202	Ok(passed_args)203}204205/// Creates Context, which has all argument default values applied206/// and with unbound values causing error to be returned207pub fn parse_default_function_call(body_ctx: Context, params: &ExprParams) -> Result<Context> {208	let fctx = Context::new_future();209210	let mut bindings = FxHashMap::with_capacity(params.binds_len());211212	for param in params.exprs.iter() {213		if let Some(v) = &param.default {214			destruct(215				&param.destruct.clone(),216				{217					let ctx = fctx.clone();218					let name = param.destruct.name();219					let value = v.clone();220					Thunk!(move || evaluate_named_param(ctx.unwrap(), &value, name))221				},222				fctx.clone(),223				&mut bindings,224			)?;225		} else {226			destruct(227				&param.destruct,228				{229					let param_name = param.destruct.name();230					let params = params.clone();231					Thunk!(move || Err(FunctionParameterNotBoundInCall(232						param_name,233						params.signature.clone()234					)235					.into()))236				},237				fctx.clone(),238				&mut bindings,239			)?;240		}241	}242243	Ok(body_ctx.extend_bindings(bindings).into_future(fctx))244}