`a9a2382fcb8f`

source

crates/jrsonnet-evaluator/src/function/parse.rs6.9 KiBsource history
1use std::mem::replace;23use jrsonnet_gcmodule::Trace;4use jrsonnet_interner::IStr;5use jrsonnet_parser::{LocExpr, ParamsDesc};67use super::{arglike::ArgsLike, builtin::BuiltinParam};8use crate::{9	bail,10	destructure::destruct,11	error::{ErrorKind::*, Result},12	evaluate_named,13	function::builtin::ParamDefault,14	gc::GcHashMap,15	Context, Pending, Thunk, Val,16};1718/// Creates correct [context](Context) for function body evaluation returning error on invalid call.19///20/// ## Parameters21/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)22/// * `body_ctx`: used for default parameter values' execution and for body execution (if set)23/// * `params`: function parameters' definition24/// * `args`: passed function arguments25/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily26pub fn parse_function_call(27	ctx: Context,28	body_ctx: Context,29	params: &ParamsDesc,30	args: &dyn ArgsLike,31	tailstrict: bool,32) -> Result<Context> {33	let mut passed_args =34		GcHashMap::with_capacity(params.iter().map(|p| p.0.capacity_hint()).sum());35	if args.unnamed_len() > params.len() {36		bail!(TooManyArgsFunctionHas(37			params.len(),38			params39				.iter()40				.map(|p| (p.0.name(), ParamDefault::exists(p.1.is_some())))41				.collect()42		))43	}4445	let mut filled_named = 0;46	let mut filled_positionals = 0;4748	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {49		let name = params[id].0.clone();50		destruct(51			&name,52			arg,53			Pending::new_filled(ctx.clone()),54			&mut passed_args,55		)?;56		filled_positionals += 1;57		Ok(())58	})?;5960	args.named_iter(ctx, tailstrict, &mut |name, value| {61		// FIXME: O(n) for arg existence check62		if !params.iter().any(|p| p.0.name().as_ref() == Some(name)) {63			bail!(UnknownFunctionParameter((name as &str).to_owned()));64		}65		if passed_args.insert(name.clone(), value).is_some() {66			bail!(BindingParameterASecondTime(name.clone()));67		}68		filled_named += 1;69		Ok(())70	})?;7172	if filled_named + filled_positionals < params.len() {73		// Some args are unset, but maybe we have defaults for them74		// Default values should be created in newly created context75		let fctx = Context::new_future();76		let mut defaults = GcHashMap::with_capacity(77			params.iter().map(|p| p.0.capacity_hint()).sum::<usize>()78				- filled_named79				- filled_positionals,80		);8182		for (idx, param) in params.iter().enumerate().filter(|p| p.1 .1.is_some()) {83			if let Some(name) = param.0.name() {84				if passed_args.contains_key(&name) {85					continue;86				}87			} else if idx < filled_positionals {88				continue;89			}9091			destruct(92				&param.0,93				{94					let ctx = fctx.clone();95					let name = param.0.name().unwrap_or_else(|| "<destruct>".into());96					let value = param.1.clone().expect("default exists");97					Thunk!(move || evaluate_named(ctx.unwrap(), &value, name))98				},99				fctx.clone(),100				&mut defaults,101			)?;102			if param.0.name().is_some() {103				filled_named += 1;104			} else {105				filled_positionals += 1;106			}107		}108109		// Some args still weren't filled110		if filled_named + filled_positionals != params.len() {111			for param in params.iter().skip(args.unnamed_len()) {112				let mut found = false;113				args.named_names(&mut |name| {114					if Some(name) == param.0.name().as_ref() {115						found = true;116					}117				});118				if !found {119					bail!(FunctionParameterNotBoundInCall(120						param.0.clone().name(),121						params122							.iter()123							.map(|p| (p.0.name(), ParamDefault::exists(p.1.is_some())))124							.collect()125					));126				}127			}128			unreachable!();129		}130131		Ok(body_ctx132			.extend(passed_args, None, None, None)133			.extend(defaults, None, None, None)134			.into_future(fctx))135	} else {136		let body_ctx = body_ctx.extend(passed_args, None, None, None);137		Ok(body_ctx)138	}139}140141/// You shouldn't probally use this function, use `jrsonnet_macros::builtin` instead142///143/// ## Parameters144/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)145/// * `params`: function parameters' definition146/// * `args`: passed function arguments147/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily148pub fn parse_builtin_call(149	ctx: Context,150	params: &[BuiltinParam],151	args: &dyn ArgsLike,152	tailstrict: bool,153) -> Result<Vec<Option<Thunk<Val>>>> {154	let mut passed_args: Vec<Option<Thunk<Val>>> = vec![None; params.len()];155	if args.unnamed_len() > params.len() {156		bail!(TooManyArgsFunctionHas(157			params.len(),158			params159				.iter()160				.map(|p| (p.name().as_str().map(IStr::from), p.default()))161				.collect()162		))163	}164165	let mut filled_args = 0;166167	args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {168		passed_args[id] = Some(arg);169		filled_args += 1;170		Ok(())171	})?;172173	args.named_iter(ctx, tailstrict, &mut |name, arg| {174		// FIXME: O(n) for arg existence check175		let id = params176			.iter()177			.position(|p| p.name() == name)178			.ok_or_else(|| UnknownFunctionParameter((name as &str).to_owned()))?;179		if replace(&mut passed_args[id], Some(arg)).is_some() {180			bail!(BindingParameterASecondTime(name.clone()));181		}182		filled_args += 1;183		Ok(())184	})?;185186	if filled_args < params.len() {187		for (id, _) in params.iter().enumerate().filter(|(_, p)| p.has_default()) {188			if passed_args[id].is_some() {189				continue;190			}191			filled_args += 1;192		}193194		// Some args still wasn't filled195		if filled_args != params.len() {196			for param in params.iter().skip(args.unnamed_len()) {197				let mut found = false;198				args.named_names(&mut |name| {199					if param.name() == name {200						found = true;201					}202				});203				if !found {204					bail!(FunctionParameterNotBoundInCall(205						param.name().as_str().map(IStr::from),206						params207							.iter()208							.map(|p| (p.name().as_str().map(IStr::from), p.default()))209							.collect()210					));211				}212			}213			unreachable!();214		}215	}216	Ok(passed_args)217}218219/// Creates Context, which has all argument default values applied220/// and with unbound values causing error to be returned221pub fn parse_default_function_call(body_ctx: Context, params: &ParamsDesc) -> Result<Context> {222	let fctx = Context::new_future();223224	let mut bindings = GcHashMap::with_capacity(params.iter().map(|p| p.0.capacity_hint()).sum());225226	for param in params.iter() {227		if let Some(v) = &param.1 {228			destruct(229				&param.0.clone(),230				{231					let ctx = fctx.clone();232					let name = param.0.name().unwrap_or_else(|| "<destruct>".into());233					let value = v.clone();234					Thunk!(move || evaluate_named(ctx.unwrap(), &value, name))235				},236				fctx.clone(),237				&mut bindings,238			)?;239		} else {240			destruct(241				&param.0,242				{243					let param_name = param.0.name().unwrap_or_else(|| "<destruct>".into());244					let params = params.clone();245					Thunk!(move || Err(FunctionParameterNotBoundInCall(246						Some(param_name),247						params248							.iter()249							.map(|p| (p.0.name(), ParamDefault::exists(p.1.is_some())))250							.collect(),251					)252					.into()))253				},254				fctx.clone(),255				&mut bindings,256			)?;257		}258	}259260	Ok(body_ctx261		.extend(bindings, None, None, None)262		.into_future(fctx))263}