1use crate::{2 create_error, create_error_result, evaluate, lazy_val, resolved_lazy_val, Context, Error,3 Result, Val,4};5use closure::closure;6use jrsonnet_parser::{ArgsDesc, ParamsDesc};7use std::{collections::HashMap, rc::Rc};89const NO_DEFAULT_CONTEXT: &str =10 "no default context set for call with defined default parameter value";111213141516171819pub fn parse_function_call(20 ctx: Context,21 body_ctx: Option<Context>,22 params: &ParamsDesc,23 args: &ArgsDesc,24 tailstrict: bool,25) -> Result<Context> {26 let mut out = HashMap::new();27 let mut positioned_args = vec![None; params.0.len()];28 for (id, arg) in args.iter().enumerate() {29 let idx = if let Some(name) = &arg.0 {30 params31 .iter()32 .position(|p| *p.0 == *name)33 .ok_or_else(|| create_error(Error::UnknownFunctionParameter(name.clone())))?34 } else {35 id36 };3738 if idx >= params.len() {39 create_error_result(Error::TooManyArgsFunctionHas(params.len()))?;40 }41 if positioned_args[idx].is_some() {42 create_error_result(Error::BindingParameterASecondTime(params[idx].0.clone()))?;43 }44 positioned_args[idx] = Some(arg.1.clone());45 }46 47 for (id, p) in params.iter().enumerate() {48 let (ctx, expr) = if let Some(arg) = &positioned_args[id] {49 (ctx.clone(), arg)50 } else if let Some(default) = &p.1 {51 (body_ctx.clone().expect(NO_DEFAULT_CONTEXT), default)52 } else {53 create_error_result(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;54 unreachable!()55 };56 let val = if tailstrict {57 resolved_lazy_val!(evaluate(ctx, expr)?)58 } else {59 lazy_val!(closure!(clone ctx, clone expr, ||evaluate(ctx.clone(), &expr)))60 };61 out.insert(p.0.clone(), val);62 }6364 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)65}6667pub fn parse_function_call_map(68 ctx: Context,69 body_ctx: Option<Context>,70 params: &ParamsDesc,71 args: &HashMap<Rc<str>, Val>,72 tailstrict: bool,73) -> Result<Context> {74 let mut out = HashMap::new();75 let mut positioned_args = vec![None; params.0.len()];76 for (name, val) in args.iter() {77 let idx = params.iter().position(|p| *p.0 == **name).ok_or_else(|| {78 create_error(Error::UnknownFunctionParameter((&name as &str).to_owned()))79 })?;8081 if idx >= params.len() {82 create_error_result(Error::TooManyArgsFunctionHas(params.len()))?;83 }84 if positioned_args[idx].is_some() {85 create_error_result(Error::BindingParameterASecondTime(params[idx].0.clone()))?;86 }87 positioned_args[idx] = Some(val.clone());88 }89 90 for (id, p) in params.iter().enumerate() {91 let val = if let Some(arg) = positioned_args[id].take() {92 resolved_lazy_val!(arg)93 } else if let Some(default) = &p.1 {94 if tailstrict {95 resolved_lazy_val!(evaluate(96 body_ctx.clone().expect(NO_DEFAULT_CONTEXT),97 default98 )?)99 } else {100 let body_ctx = body_ctx.clone();101 let default = default.clone();102 lazy_val!(move || {103 evaluate(body_ctx.clone().expect(NO_DEFAULT_CONTEXT), &default)104 })105 }106 } else {107 create_error_result(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;108 unreachable!()109 };110 out.insert(p.0.clone(), val);111 }112113 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)114}115116pub(crate) fn place_args(117 ctx: Context,118 body_ctx: Option<Context>,119 params: &ParamsDesc,120 args: &[Val],121) -> Result<Context> {122 let mut out = HashMap::new();123 let mut positioned_args = vec![None; params.0.len()];124 for (id, arg) in args.iter().enumerate() {125 if id >= params.len() {126 create_error_result(Error::TooManyArgsFunctionHas(params.len()))?;127 }128 positioned_args[id] = Some(arg);129 }130 131 for (id, p) in params.iter().enumerate() {132 let val = if let Some(arg) = &positioned_args[id] {133 (*arg).clone()134 } else if let Some(default) = &p.1 {135 evaluate(ctx.clone(), default)?136 } else {137 create_error_result(Error::FunctionParameterNotBoundInCall(p.0.clone()))?;138 unreachable!()139 };140 out.insert(p.0.clone(), resolved_lazy_val!(val));141 }142143 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)144}145146#[macro_export]147macro_rules! parse_args {148 ($ctx: expr, $fn_name: expr, $args: expr, $total_args: expr, [149 $($id: expr, $name: ident $(: [$($p: path)|+] $(!! $a: path)?)?, $nt: expr);+ $(;)?150 ], $handler:block) => {{151 use crate::Error;152 let args = $args;153 if args.len() > $total_args {154 create_error_result(Error::TooManyArgsFunctionHas($total_args))?;155 }156 $(157 if args.len() <= $id {158 create_error_result(Error::FunctionParameterNotBoundInCall(stringify!($name).into()))?;159 }160 let $name = &args[$id];161 if $name.0.is_some() {162 if $name.0.as_ref().unwrap() != stringify!($name) {163 create_error_result(Error::IntristicArgumentReorderingIsNotSupportedYet)?;164 }165 }166 let $name = evaluate($ctx.clone(), &$name.1)?;167 $(168 match $name {169 $($p(_))|+ => {},170 _ => create_error_result(Error::TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd (", stringify!($name), ") argument"), $nt, $name.value_type()?))?,171 };172 $(173 let $name = match $name {174 $a(v) => v,175 _ => create_error_result(Error::TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd (", stringify!($name), ") argument"), $nt, $name.value_type()?))?,176 };177 )*178 )*179 )+180 $handler181 }};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 });202 Ok(())203 })204}