1use crate::{error::Error::*, evaluate, lazy_val, resolved_lazy_val, throw, Context, Result, Val};2use closure::closure;3use jrsonnet_parser::{ArgsDesc, ParamsDesc};4use rustc_hash::FxHashMap;5use std::{collections::HashMap, hash::BuildHasherDefault, rc::Rc};67const NO_DEFAULT_CONTEXT: &str =8 "no default context set for call with defined default parameter value";9101112131415161718pub fn parse_function_call(19 ctx: Context,20 body_ctx: Option<Context>,21 params: &ParamsDesc,22 args: &ArgsDesc,23 tailstrict: bool,24) -> Result<Context> {25 let mut out = HashMap::with_capacity_and_hasher(params.len(), BuildHasherDefault::default());26 let mut positioned_args = vec![None; params.0.len()];27 for (id, arg) in args.iter().enumerate() {28 let idx = if let Some(name) = &arg.0 {29 params30 .iter()31 .position(|p| *p.0 == *name)32 .ok_or_else(|| UnknownFunctionParameter(name.clone()))?33 } else {34 id35 };3637 if idx >= params.len() {38 throw!(TooManyArgsFunctionHas(params.len()));39 }40 if positioned_args[idx].is_some() {41 throw!(BindingParameterASecondTime(params[idx].0.clone()));42 }43 positioned_args[idx] = Some(arg.1.clone());44 }45 46 for (id, p) in params.iter().enumerate() {47 let (ctx, expr) = if let Some(arg) = &positioned_args[id] {48 (ctx.clone(), arg)49 } else if let Some(default) = &p.1 {50 (body_ctx.clone().expect(NO_DEFAULT_CONTEXT), default)51 } else {52 throw!(FunctionParameterNotBoundInCall(p.0.clone()));53 };54 let val = if tailstrict {55 resolved_lazy_val!(evaluate(ctx, expr)?)56 } else {57 lazy_val!(closure!(clone ctx, clone expr, ||evaluate(ctx.clone(), &expr)))58 };59 out.insert(p.0.clone(), val);60 }6162 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None))63}6465pub fn parse_function_call_map(66 ctx: Context,67 body_ctx: Option<Context>,68 params: &ParamsDesc,69 args: &HashMap<Rc<str>, Val>,70 tailstrict: bool,71) -> Result<Context> {72 let mut out = FxHashMap::with_capacity_and_hasher(params.len(), BuildHasherDefault::default());73 let mut positioned_args = vec![None; params.0.len()];74 for (name, val) in args.iter() {75 let idx = params76 .iter()77 .position(|p| *p.0 == **name)78 .ok_or_else(|| UnknownFunctionParameter((name as &str).to_owned()))?;7980 if idx >= params.len() {81 throw!(TooManyArgsFunctionHas(params.len()));82 }83 if positioned_args[idx].is_some() {84 throw!(BindingParameterASecondTime(params[idx].0.clone()));85 }86 positioned_args[idx] = Some(val.clone());87 }88 89 for (id, p) in params.iter().enumerate() {90 let val = if let Some(arg) = positioned_args[id].take() {91 resolved_lazy_val!(arg)92 } else if let Some(default) = &p.1 {93 if tailstrict {94 resolved_lazy_val!(evaluate(95 body_ctx.clone().expect(NO_DEFAULT_CONTEXT),96 default97 )?)98 } else {99 let body_ctx = body_ctx.clone();100 let default = default.clone();101 lazy_val!(move || {102 evaluate(body_ctx.clone().expect(NO_DEFAULT_CONTEXT), &default)103 })104 }105 } else {106 throw!(FunctionParameterNotBoundInCall(p.0.clone()));107 };108 out.insert(p.0.clone(), val);109 }110111 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None))112}113114pub fn place_args(115 ctx: Context,116 body_ctx: Option<Context>,117 params: &ParamsDesc,118 args: &[Val],119) -> Result<Context> {120 let mut out = FxHashMap::with_capacity_and_hasher(params.len(), BuildHasherDefault::default());121 let mut positioned_args = vec![None; params.0.len()];122 for (id, arg) in args.iter().enumerate() {123 if id >= params.len() {124 throw!(TooManyArgsFunctionHas(params.len()));125 }126 positioned_args[id] = Some(arg);127 }128 129 for (id, p) in params.iter().enumerate() {130 let val = if let Some(arg) = &positioned_args[id] {131 (*arg).clone()132 } else if let Some(default) = &p.1 {133 evaluate(ctx.clone(), default)?134 } else {135 throw!(FunctionParameterNotBoundInCall(p.0.clone()));136 };137 out.insert(p.0.clone(), resolved_lazy_val!(val));138 }139140 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None))141}142143#[macro_export]144macro_rules! parse_args {145 ($ctx: expr, $fn_name: expr, $args: expr, $total_args: expr, [146 $($id: expr, $name: ident $(: [$($p: path)|+] $(!! $a: path)?)?, $nt: expr);+ $(;)?147 ], $handler:block) => {{148 use crate::{throw, error::Error::*};149 let args = $args;150 if args.len() > $total_args {151 throw!(TooManyArgsFunctionHas($total_args));152 }153 $(154 if args.len() <= $id {155 throw!(FunctionParameterNotBoundInCall(stringify!($name).into()));156 }157 let $name = &args[$id];158 if $name.0.is_some() {159 if $name.0.as_ref().unwrap() != stringify!($name) {160 throw!(IntrinsicArgumentReorderingIsNotSupportedYet);161 }162 }163 let $name = evaluate($ctx.clone(), &$name.1)?;164 $(165 match $name {166 $($p(_))|+ => {},167 _ => throw!(TypeMismatch(168 concat!($fn_name, " ", stringify!($id), "nd (", stringify!($name), ") argument"),169 $nt, $name.value_type()?170 )),171 };172 $(173 let $name = match $name {174 $a(v) => v,175 _ =>throw!(TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd (", stringify!($name), ") argument"), $nt, $name.value_type()?)),176 };177 )*178 )*179 )+180 ($handler as crate::Result<_>)181 }};182}183184#[test]185fn test() -> Result<()> {186 use crate::val::ValType;187 use jrsonnet_parser::*;188 let state = crate::EvaluationState::default();189 let evaluator = state.with_stdlib();190 let ctx = evaluator.create_default_context()?;191 evaluator.run_in_state(|| {192 parse_args!(ctx, "test", ArgsDesc(vec![193 Arg(None, el!(Expr::Num(2.0))),194 Arg(Some("b".into()), el!(Expr::Num(1.0))),195 ]), 2, [196 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];197 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];198 ], {199 assert!((a - 2.0).abs() <= f64::EPSILON);200 assert!((b - 1.0).abs() <= f64::EPSILON);201 Ok(())202 })203 .unwrap();204 Ok(())205 })206}