1use crate::{2 binding, context_creator, create_error, future_wrapper, lazy_val, push, with_state, Context,3 ContextCreator, FuncDesc, LazyBinding, ObjMember, ObjValue, Result, Val,4};5use closure::closure;6use jsonnet_parser::{7 el, Arg, ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember,8 ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,9 Visibility,10};11use std::{12 collections::{BTreeMap, HashMap},13 rc::Rc,14};1516pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, LazyBinding) {17 let b = b.clone();18 if let Some(params) = &b.params {19 let params = params.clone();20 (21 b.name.clone(),22 LazyBinding::Bindable(Rc::new(move |this, super_obj| {23 Ok(lazy_val!(24 closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(25 context_creator.0(this.clone(), super_obj.clone())?,26 params.clone(),27 b.value.clone(),28 )))29 ))30 })),31 )32 } else {33 (34 b.name.clone(),35 LazyBinding::Bindable(Rc::new(move |this, super_obj| {36 Ok(lazy_val!(closure!(clone context_creator, clone b, ||37 push(b.value.clone(), "thunk".to_owned(), ||{38 evaluate(39 context_creator.0(this.clone(), super_obj.clone())?,40 &b.value41 )42 })43 )))44 })),45 )46 }47}4849pub fn evaluate_method(ctx: Context, params: ParamsDesc, body: LocExpr) -> Val {50 Val::Func(FuncDesc { ctx, params, body })51}5253pub fn evaluate_field_name(54 context: Context,55 field_name: &jsonnet_parser::FieldName,56) -> Result<Option<String>> {57 Ok(match field_name {58 jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),59 jsonnet_parser::FieldName::Dyn(expr) => {60 let value = evaluate(context, expr)?.unwrap_if_lazy()?;61 if matches!(value, Val::Null) {62 None63 } else {64 Some(value.try_cast_str("dynamic field name")?)65 }66 }67 })68}6970pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {71 Ok(match (op, b) {72 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,73 (UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),74 (UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),75 (UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),76 (op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),77 })78}7980pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {81 Ok(match (a, b) {82 (Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),8384 (Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),85 (Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),86 (Val::Str(v1), Val::Bool(v2)) => Val::Str(format!("{}{}", v1, v2)),87 (Val::Bool(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),88 (Val::Str(v1), Val::Null) => Val::Str(format!("{}null", v1)),89 (Val::Null, Val::Str(v2)) => Val::Str(format!("null{}", v2)),9091 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),92 (Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),93 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),94 _ => panic!("can't add: {:?} and {:?}", a, b),95 })96}9798pub fn evaluate_binary_op_special(99 context: Context,100 a: &LocExpr,101 op: BinaryOpType,102 b: &LocExpr,103) -> Result<Val> {104 Ok(105 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {106 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),107 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),108 (a, op, eb) => {109 evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?110 }111 },112 )113}114115pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {116 Ok(match (a, op, b) {117 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,118119 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),120121 122 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),123 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),124125 126 (Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),127 (Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),128 (Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),129 (Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),130131 132 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),133 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {134 if *v2 <= f64::EPSILON {135 create_error(crate::Error::DivisionByZero)?136 }137 Val::Num(v1 / v2)138 }139140 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),141142 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),143 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),144 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),145 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),146147 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {148 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)149 }150 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {151 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)152 }153 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {154 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)155 }156 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {157 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)158 }159 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {160 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)161 }162163 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),164 })165}166167future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);168future_wrapper!(ObjValue, FutureObjValue);169170pub fn evaluate_comp(171 context: Context,172 value: &LocExpr,173 specs: &[CompSpec],174) -> Result<Option<Vec<Val>>> {175 Ok(match specs.get(0) {176 None => Some(vec![evaluate(context, &value)?]),177 Some(CompSpec::IfSpec(IfSpecData(cond))) => {178 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {179 evaluate_comp(context, value, &specs[1..])?180 } else {181 None182 }183 }184 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {185 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {186 Val::Arr(list) => {187 let mut out = Vec::new();188 for item in list {189 let item = item.clone().unwrap_if_lazy()?;190 out.push(evaluate_comp(191 context.with_var(var.clone(), item)?,192 value,193 &specs[1..],194 )?);195 }196 Some(out.iter().flatten().flatten().cloned().collect())197 }198 _ => panic!("for expression evaluated to non-iterable value"),199 }200 }201 })202}203204205pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {206 Ok(match object {207 ObjBody::MemberList(members) => {208 let new_bindings = FutureNewBindings::new();209 let future_this = FutureObjValue::new();210 let context_creator = context_creator!(211 closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {212 Ok(context.clone().extend_unbound(213 new_bindings.clone().unwrap(),214 context.clone().dollar().clone().or_else(||this.clone()),215 Some(this.unwrap()),216 super_obj217 )?)218 })219 );220 {221 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();222 for (n, b) in members223 .iter()224 .filter_map(|m| match m {225 Member::BindStmt(b) => Some(b.clone()),226 _ => None,227 })228 .map(|b| evaluate_binding(&b, context_creator.clone()))229 {230 bindings.insert(n, b);231 }232 new_bindings.fill(bindings);233 }234235 let mut new_members = BTreeMap::new();236 for member in members.into_iter() {237 match member {238 Member::Field(FieldMember {239 name,240 plus,241 params: None,242 visibility,243 value,244 }) => {245 let name = evaluate_field_name(context.clone(), &name)?;246 if name.is_none() {247 continue;248 }249 let name = name.unwrap();250 new_members.insert(251 name.clone(),252 ObjMember {253 add: plus,254 visibility: visibility.clone(),255 invoke: binding!(256 closure!(clone name, clone value, clone context_creator, |this, super_obj| {257 push(value.clone(), "object ".to_owned()+&name+" field", ||{258 let context = context_creator.0(this, super_obj)?;259 evaluate(260 context,261 &value,262 )?.unwrap_if_lazy()263 })264 })265 ),266 },267 );268 }269 Member::Field(FieldMember {270 name,271 params: Some(params),272 value,273 ..274 }) => {275 let name = evaluate_field_name(context.clone(), &name)?;276 if name.is_none() {277 continue;278 }279 let name = name.unwrap();280 new_members.insert(281 name,282 ObjMember {283 add: false,284 visibility: Visibility::Hidden,285 invoke: binding!(286 closure!(clone value, clone context_creator, |this, super_obj| {287 288 Ok(evaluate_method(289 context_creator.0(this, super_obj)?,290 params.clone(),291 value.clone(),292 ))293 })294 ),295 },296 );297 }298 Member::BindStmt(_) => {}299 Member::AssertStmt(_) => {}300 }301 }302 future_this.fill(ObjValue::new(None, Rc::new(new_members)))303 }304 _ => todo!(),305 })306}307308#[inline(always)]309pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {310 use Expr::*;311 let locexpr = expr.clone();312 let LocExpr(expr, loc) = expr;313 Ok(match &**expr {314 Literal(LiteralType::This) => Val::Obj(315 context316 .this()317 .clone()318 .unwrap_or_else(|| panic!("this not found")),319 ),320 Literal(LiteralType::Dollar) => Val::Obj(321 context322 .dollar()323 .clone()324 .unwrap_or_else(|| panic!("dollar not found")),325 ),326 Literal(LiteralType::True) => Val::Bool(true),327 Literal(LiteralType::False) => Val::Bool(false),328 Literal(LiteralType::Null) => Val::Null,329 Parened(e) => evaluate(context, e)?,330 Str(v) => Val::Str(v.clone()),331 Num(v) => Val::Num(*v),332 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,333 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,334 Var(name) => push(locexpr, "var".to_owned(), || {335 Val::Lazy(context.binding(&name)).unwrap_if_lazy()336 })?,337 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {338 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;339 context340 .super_obj()341 .clone()342 .expect("no super found")343 .get_raw(&name, &context.this().clone().expect("no this found"))?344 .expect("value not found")345 }346 Index(value, index) => {347 match (348 evaluate(context.clone(), value)?.unwrap_if_lazy()?,349 evaluate(context, index)?,350 ) {351 (Val::Obj(v), Val::Str(s)) => {352 if let Some(v) = v.get(&s)? {353 v.unwrap_if_lazy()?354 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {355 Val::Intristic(n, s)356 } else {357 create_error(crate::Error::NoSuchField(s))?358 }359 }360 (Val::Arr(v), Val::Num(n)) => {361 if n.fract() > f64::EPSILON {362 create_error(crate::Error::FractionalIndex)?363 }364 v.get(n as usize)365 .unwrap_or_else(|| panic!("out of bounds"))366 .clone()367 .unwrap_if_lazy()?368 }369 (Val::Str(s), Val::Num(n)) => {370 Val::Str(s.chars().skip(n as usize).take(1).collect())371 }372 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),373 }374 }375 LocalExpr(bindings, returned) => {376 let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();377 let future_context = Context::new_future();378379 let context_creator = context_creator!(380 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))381 );382383 for (k, v) in bindings384 .iter()385 .map(|b| evaluate_binding(b, context_creator.clone()))386 {387 new_bindings.insert(k, v);388 }389390 let context = context391 .extend_unbound(new_bindings, None, None, None)?392 .into_future(future_context);393 evaluate(context, &returned.clone())?394 }395 Arr(items) => {396 let mut out = Vec::with_capacity(items.len());397 for item in items {398 out.push(Val::Lazy(lazy_val!(399 closure!(clone context, clone item, || {400 evaluate(context.clone(), &item)401 })402 )));403 }404 Val::Arr(out)405 }406 ArrComp(expr, compspecs) => Val::Arr(407 408 evaluate_comp(context, expr, compspecs)?.unwrap(),409 ),410 Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),411 ObjExtend(s, t) => evaluate_add_op(412 &evaluate(context.clone(), s)?,413 &Val::Obj(evaluate_object(context, t.clone())?),414 )?,415 Apply(value, args, tailstrict) => {416 let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;417 match value {418 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {419 420 ("std", "length") => {421 assert_eq!(args.len(), 1);422 let expr = &args.get(0).unwrap().1;423 match evaluate(context, expr)? {424 Val::Str(n) => Val::Num(n.chars().count() as f64),425 Val::Arr(i) => Val::Num(i.len() as f64),426 Val::Obj(o) => Val::Num(o.fields().len() as f64),427 v => panic!("can't get length of {:?}", v),428 }429 }430 431 ("std", "type") => {432 assert_eq!(args.len(), 1);433 let expr = &args.get(0).unwrap().1;434 Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())435 }436 437 ("std", "makeArray") => {438 assert_eq!(args.len(), 2);439 if let (Val::Num(v), Val::Func(d)) = (440 evaluate(context.clone(), &args[0].1)?,441 evaluate(context, &args[1].1)?,442 ) {443 assert!(v >= 0.0);444 let mut out = Vec::with_capacity(v as usize);445 for i in 0..v as usize {446 let call_ctx =447 Context::new().with_var("v".to_owned(), Val::Num(i as f64))?;448 out.push(d.evaluate(449 call_ctx,450 &ArgsDesc(vec![Arg(None, el!(Expr::Var("v".to_owned())))]),451 true,452 )?)453 }454 Val::Arr(out)455 } else {456 panic!("bad makeArray call");457 }458 }459 460 ("std", "codepoint") => {461 assert_eq!(args.len(), 1);462 if let Val::Str(s) = evaluate(context, &args[0].1)? {463 assert!(464 s.chars().count() == 1,465 "std.codepoint should receive single char string"466 );467 Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)468 } else {469 panic!("bad codepoint call");470 }471 }472 473 ("std", "objectFieldsEx") => {474 assert_eq!(args.len(), 2);475 if let (Val::Obj(body), Val::Bool(_include_hidden)) = (476 evaluate(context.clone(), &args[0].1)?,477 evaluate(context, &args[1].1)?,478 ) {479 480 Val::Arr(body.fields().into_iter().map(Val::Str).collect())481 } else {482 panic!("bad objectFieldsEx call");483 }484 }485 ("std", "primitiveEquals") => {486 assert_eq!(args.len(), 2);487 let (a, b) = (488 evaluate(context.clone(), &args[0].1)?,489 evaluate(context, &args[1].1)?,490 );491 Val::Bool(a == b)492 }493 ("std", "modulo") => {494 assert_eq!(args.len(), 2);495 if let (Val::Num(a), Val::Num(b)) = (496 evaluate(context.clone(), &args[0].1)?,497 evaluate(context, &args[1].1)?,498 ) {499 Val::Num(a % b)500 } else {501 panic!("bad modulo call");502 }503 }504 ("std", "floor") => {505 assert_eq!(args.len(), 1);506 if let Val::Num(a) = evaluate(context, &args[0].1)? {507 Val::Num(a.floor())508 } else {509 panic!("bad floor call");510 }511 }512 ("std", "trace") => {513 assert_eq!(args.len(), 2);514 if let (Val::Str(a), b) = (515 evaluate(context.clone(), &args[0].1)?,516 evaluate(context, &args[1].1)?,517 ) {518 519 println!("TRACE: {}", a);520 b521 } else {522 panic!("bad trace call");523 }524 }525 ("std", "pow") => {526 assert_eq!(args.len(), 2);527 if let (Val::Num(a), Val::Num(b)) = (528 evaluate(context.clone(), &args[0].1)?,529 evaluate(context, &args[1].1)?,530 ) {531 Val::Num(a.powf(b))532 } else {533 panic!("bad pow call");534 }535 }536 ("std", "extVar") => {537 assert_eq!(args.len(), 1);538 if let Val::Str(a) = evaluate(context, &args[0].1)? {539 with_state(|s| s.0.ext_vars.borrow().get(&a).cloned()).ok_or_else(540 || {541 create_error::<()>(crate::Error::UndefinedExternalVariable(a))542 .err()543 .unwrap()544 },545 )?546 } else {547 panic!("bad extVar call");548 }549 }550 (ns, name) => panic!("Intristic not found: {}.{}", ns, name),551 },552 Val::Func(f) => {553 let body = #[inline(always)]554 || f.evaluate(context, args, *tailstrict);555 if *tailstrict {556 body()?557 } else {558 push(locexpr, "function call".to_owned(), body)?559 }560 }561 _ => panic!("{:?} is not a function", value),562 }563 }564 Function(params, body) => evaluate_method(context, params.clone(), body.clone()),565 AssertExpr(AssertStmt(value, msg), returned) => {566 let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {567 evaluate(context.clone(), &value)?568 .try_cast_bool("assertion condition should be boolean")569 })?;570 if assertion_result {571 push(572 returned.clone(),573 "assert 'return' branch".to_owned(),574 || evaluate(context, returned),575 )?576 } else if let Some(msg) = msg {577 panic!(578 "assertion failed ({:?}): {}",579 value,580 evaluate(context, msg)?.try_cast_str("assertion message should be string")?581 );582 } else {583 panic!("assertion failed ({:?}): no message", value);584 }585 }586 Error(e) => create_error(crate::Error::RuntimeError(587 evaluate(context, e)?.try_cast_str("error text should be string")?,588 ))?,589 IfElse {590 cond,591 cond_then,592 cond_else,593 } => {594 let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {595 evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")596 })?;597 if condition_result {598 push(599 cond_then.clone(),600 "if condition 'then' branch".to_owned(),601 || evaluate(context, cond_then),602 )?603 } else {604 match cond_else {605 Some(v) => evaluate(context, v)?,606 None => Val::Null,607 }608 }609 }610 Import(path) => {611 let mut lib_path = loc612 .clone()613 .expect("imports can't be used without loc_data")614 .0615 .clone();616 lib_path.pop();617 lib_path.push(path);618 with_state(|s| s.import_file(&lib_path))?619 }620 _ => panic!(621 "evaluation not implemented: {:?}",622 LocExpr(expr.clone(), loc.clone())623 ),624 })625}