`6baa18de6571`

source

crates/jrsonnet-evaluator/src/function/parse.rs6.9 KiBsource history
1use std::mem::replace;23use jrsonnet_interner::IStr;4use jrsonnet_parser::ParamsDesc;56use super::{arglike::ArgsLike, builtin::BuiltinParam};7use crate::{8	bail,9	destructure::destruct,10	error::{ErrorKind::*, Result},11	evaluate_named,12	function::builtin::ParamDefault,13	gc::GcHashMap,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: &ParamsDesc,29	args: &dyn ArgsLike,30	tailstrict: bool,31) -> Result<Context> {32	let mut passed_args =33		GcHashMap::with_capacity(params.iter().map(|p| p.0.capacity_hint()).sum());34	if args.unnamed_len() > params.len() {35		bail!(TooManyArgsFunctionHas(36			params.len(),37			params38				.iter()39				.map(|p| (p.0.name(), ParamDefault::exists(p.1.is_some())))40				.collect()41		))42	}4344	let mut filled_named = 0;45	let mut filled_positionals = 0;4647	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {48		let name = params[id].0.clone();49		destruct(50			&name,51			arg,52			Pending::new_filled(ctx.clone()),53			&mut passed_args,54		)?;55		filled_positionals += 1;56		Ok(())57	})?;5859	args.named_iter(ctx, tailstrict, &mut |name, value| {60		// FIXME: O(n) for arg existence check61		if !params.iter().any(|p| p.0.name().as_ref() == Some(name)) {62			bail!(UnknownFunctionParameter((name as &str).to_owned()));63		}64		if passed_args.insert(name.clone(), value).is_some() {65			bail!(BindingParameterASecondTime(name.clone()));66		}67		filled_named += 1;68		Ok(())69	})?;7071	if filled_named + filled_positionals < params.len() {72		// Some args are unset, but maybe we have defaults for them73		// Default values should be created in newly created context74		let fctx = Context::new_future();75		let mut defaults = GcHashMap::with_capacity(76			params.iter().map(|p| p.0.capacity_hint()).sum::<usize>()77				- filled_named78				- filled_positionals,79		);8081		for (idx, param) in params.iter().enumerate().filter(|p| p.1 .1.is_some()) {82			if let Some(name) = param.0.name() {83				if passed_args.contains_key(&name) {84					continue;85				}86			} else if idx < filled_positionals {87				continue;88			}8990			destruct(91				&param.0,92				{93					let ctx = fctx.clone();94					let name = param.0.name().unwrap_or_else(|| "<destruct>".into());95					let value = param.1.clone().expect("default exists");96					Thunk!(move || evaluate_named(ctx.unwrap(), &value, name))97				},98				fctx.clone(),99				&mut defaults,100			)?;101			if param.0.name().is_some() {102				filled_named += 1;103			} else {104				filled_positionals += 1;105			}106		}107108		// Some args still weren't filled109		if filled_named + filled_positionals != params.len() {110			for param in params.iter().skip(args.unnamed_len()) {111				let mut found = false;112				args.named_names(&mut |name| {113					if Some(name) == param.0.name().as_ref() {114						found = true;115					}116				});117				if !found {118					bail!(FunctionParameterNotBoundInCall(119						param.0.clone().name(),120						params121							.iter()122							.map(|p| (p.0.name(), ParamDefault::exists(p.1.is_some())))123							.collect()124					));125				}126			}127			unreachable!();128		}129130		Ok(body_ctx131			.extend(passed_args, None, None, None)132			.extend(defaults, None, None, None)133			.into_future(fctx))134	} else {135		let body_ctx = body_ctx.extend(passed_args, None, None, None);136		Ok(body_ctx)137	}138}139140/// You shouldn't probally use this function, use `jrsonnet_macros::builtin` instead141///142/// ## Parameters143/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)144/// * `params`: function parameters' definition145/// * `args`: passed function arguments146/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily147pub fn parse_builtin_call(148	ctx: Context,149	params: &[BuiltinParam],150	args: &dyn ArgsLike,151	tailstrict: bool,152) -> Result<Vec<Option<Thunk<Val>>>> {153	let mut passed_args: Vec<Option<Thunk<Val>>> = vec![None; params.len()];154	if args.unnamed_len() > params.len() {155		bail!(TooManyArgsFunctionHas(156			params.len(),157			params158				.iter()159				.map(|p| (p.name().as_str().map(IStr::from), p.default()))160				.collect()161		))162	}163164	let mut filled_args = 0;165166	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {167		passed_args[id] = Some(arg);168		filled_args += 1;169		Ok(())170	})?;171172	args.named_iter(ctx, tailstrict, &mut |name, arg| {173		// FIXME: O(n) for arg existence check174		let id = params175			.iter()176			.position(|p| p.name() == name)177			.ok_or_else(|| UnknownFunctionParameter((name as &str).to_owned()))?;178		if replace(&mut passed_args[id], Some(arg)).is_some() {179			bail!(BindingParameterASecondTime(name.clone()));180		}181		filled_args += 1;182		Ok(())183	})?;184185	if filled_args < params.len() {186		for (id, _) in params.iter().enumerate().filter(|(_, p)| p.has_default()) {187			if passed_args[id].is_some() {188				continue;189			}190			filled_args += 1;191		}192193		// Some args still wasn't filled194		if filled_args != params.len() {195			for param in params.iter().skip(args.unnamed_len()) {196				let mut found = false;197				args.named_names(&mut |name| {198					if param.name() == name {199						found = true;200					}201				});202				if !found {203					bail!(FunctionParameterNotBoundInCall(204						param.name().as_str().map(IStr::from),205						params206							.iter()207							.map(|p| (p.name().as_str().map(IStr::from), p.default()))208							.collect()209					));210				}211			}212			unreachable!();213		}214	}215	Ok(passed_args)216}217218/// Creates Context, which has all argument default values applied219/// and with unbound values causing error to be returned220pub fn parse_default_function_call(body_ctx: Context, params: &ParamsDesc) -> Result<Context> {221	let fctx = Context::new_future();222223	let mut bindings = GcHashMap::with_capacity(params.iter().map(|p| p.0.capacity_hint()).sum());224225	for param in params.iter() {226		if let Some(v) = &param.1 {227			destruct(228				&param.0.clone(),229				{230					let ctx = fctx.clone();231					let name = param.0.name().unwrap_or_else(|| "<destruct>".into());232					let value = v.clone();233					Thunk!(move || evaluate_named(ctx.unwrap(), &value, name))234				},235				fctx.clone(),236				&mut bindings,237			)?;238		} else {239			destruct(240				&param.0,241				{242					let param_name = param.0.name().unwrap_or_else(|| "<destruct>".into());243					let params = params.clone();244					Thunk!(move || Err(FunctionParameterNotBoundInCall(245						Some(param_name),246						params247							.iter()248							.map(|p| (p.0.name(), ParamDefault::exists(p.1.is_some())))249							.collect(),250					)251					.into()))252				},253				fctx.clone(),254				&mut bindings,255			)?;256		}257	}258259	Ok(body_ctx260		.extend(bindings, None, None, None)261		.into_future(fctx))262}