1use crate::{error::Error::*, evaluate, lazy_val, resolved_lazy_val, throw, Context, Result, Val};2use closure::closure;3use jrsonnet_parser::{ArgsDesc, ParamsDesc};4use std::{collections::HashMap, rc::Rc};56const NO_DEFAULT_CONTEXT: &str =7 "no default context set for call with defined default parameter value";8910111213141516pub fn parse_function_call(17 ctx: Context,18 body_ctx: Option<Context>,19 params: &ParamsDesc,20 args: &ArgsDesc,21 tailstrict: bool,22) -> Result<Context> {23 let mut out = HashMap::new();24 let mut positioned_args = vec![None; params.0.len()];25 for (id, arg) in args.iter().enumerate() {26 let idx = if let Some(name) = &arg.0 {27 params28 .iter()29 .position(|p| *p.0 == *name)30 .ok_or_else(|| UnknownFunctionParameter(name.clone()))?31 } else {32 id33 };3435 if idx >= params.len() {36 throw!(TooManyArgsFunctionHas(params.len()));37 }38 if positioned_args[idx].is_some() {39 throw!(BindingParameterASecondTime(params[idx].0.clone()));40 }41 positioned_args[idx] = Some(arg.1.clone());42 }43 44 for (id, p) in params.iter().enumerate() {45 let (ctx, expr) = if let Some(arg) = &positioned_args[id] {46 (ctx.clone(), arg)47 } else if let Some(default) = &p.1 {48 (body_ctx.clone().expect(NO_DEFAULT_CONTEXT), default)49 } else {50 throw!(FunctionParameterNotBoundInCall(p.0.clone()));51 };52 let val = if tailstrict {53 resolved_lazy_val!(evaluate(ctx, expr)?)54 } else {55 lazy_val!(closure!(clone ctx, clone expr, ||evaluate(ctx.clone(), &expr)))56 };57 out.insert(p.0.clone(), val);58 }5960 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)61}6263pub fn parse_function_call_map(64 ctx: Context,65 body_ctx: Option<Context>,66 params: &ParamsDesc,67 args: &HashMap<Rc<str>, Val>,68 tailstrict: bool,69) -> Result<Context> {70 let mut out = HashMap::new();71 let mut positioned_args = vec![None; params.0.len()];72 for (name, val) in args.iter() {73 let idx = params74 .iter()75 .position(|p| *p.0 == **name)76 .ok_or_else(|| UnknownFunctionParameter((&name as &str).to_owned()))?;7778 if idx >= params.len() {79 throw!(TooManyArgsFunctionHas(params.len()));80 }81 if positioned_args[idx].is_some() {82 throw!(BindingParameterASecondTime(params[idx].0.clone()));83 }84 positioned_args[idx] = Some(val.clone());85 }86 87 for (id, p) in params.iter().enumerate() {88 let val = if let Some(arg) = positioned_args[id].take() {89 resolved_lazy_val!(arg)90 } else if let Some(default) = &p.1 {91 if tailstrict {92 resolved_lazy_val!(evaluate(93 body_ctx.clone().expect(NO_DEFAULT_CONTEXT),94 default95 )?)96 } else {97 let body_ctx = body_ctx.clone();98 let default = default.clone();99 lazy_val!(move || {100 evaluate(body_ctx.clone().expect(NO_DEFAULT_CONTEXT), &default)101 })102 }103 } else {104 throw!(FunctionParameterNotBoundInCall(p.0.clone()));105 };106 out.insert(p.0.clone(), val);107 }108109 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)110}111112pub(crate) fn place_args(113 ctx: Context,114 body_ctx: Option<Context>,115 params: &ParamsDesc,116 args: &[Val],117) -> Result<Context> {118 let mut out = HashMap::new();119 let mut positioned_args = vec![None; params.0.len()];120 for (id, arg) in args.iter().enumerate() {121 if id >= params.len() {122 throw!(TooManyArgsFunctionHas(params.len()));123 }124 positioned_args[id] = Some(arg);125 }126 127 for (id, p) in params.iter().enumerate() {128 let val = if let Some(arg) = &positioned_args[id] {129 (*arg).clone()130 } else if let Some(default) = &p.1 {131 evaluate(ctx.clone(), default)?132 } else {133 throw!(FunctionParameterNotBoundInCall(p.0.clone()));134 };135 out.insert(p.0.clone(), resolved_lazy_val!(val));136 }137138 Ok(body_ctx.unwrap_or(ctx).extend(out, None, None, None)?)139}140141#[macro_export]142macro_rules! parse_args {143 ($ctx: expr, $fn_name: expr, $args: expr, $total_args: expr, [144 $($id: expr, $name: ident $(: [$($p: path)|+] $(!! $a: path)?)?, $nt: expr);+ $(;)?145 ], $handler:block) => {{146 use crate::{throw, error::Error::*};147 let args = $args;148 if args.len() > $total_args {149 throw!(TooManyArgsFunctionHas($total_args));150 }151 $(152 if args.len() <= $id {153 throw!(FunctionParameterNotBoundInCall(stringify!($name).into()));154 }155 let $name = &args[$id];156 if $name.0.is_some() {157 if $name.0.as_ref().unwrap() != stringify!($name) {158 throw!(IntristicArgumentReorderingIsNotSupportedYet);159 }160 }161 let $name = evaluate($ctx.clone(), &$name.1)?;162 $(163 match $name {164 $($p(_))|+ => {},165 _ => throw!(TypeMismatch(166 concat!($fn_name, " ", stringify!($id), "nd (", stringify!($name), ") argument"),167 $nt, $name.value_type()?168 )),169 };170 $(171 let $name = match $name {172 $a(v) => v,173 _ =>throw!(TypeMismatch(concat!($fn_name, " ", stringify!($id), "nd (", stringify!($name), ") argument"), $nt, $name.value_type()?)),174 };175 )*176 )*177 )+178 ($handler as crate::Result<_>)179 }};180}181182#[test]183fn test() -> Result<()> {184 use crate::val::ValType;185 use jrsonnet_parser::*;186 let state = crate::EvaluationState::default();187 let evaluator = state.with_stdlib();188 let ctx = evaluator.create_default_context()?;189 evaluator.run_in_state(|| {190 parse_args!(ctx, "test", ArgsDesc(vec![191 Arg(None, el!(Expr::Num(2.0))),192 Arg(Some("b".into()), el!(Expr::Num(1.0))),193 ]), 2, [194 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];195 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];196 ], {197 assert!((a - 2.0).abs() <= f64::EPSILON);198 assert!((b - 1.0).abs() <= f64::EPSILON);199 Ok(())200 })201 .unwrap();202 Ok(())203 })204}