1use crate::{2 binding, context_creator, create_error, function_default, function_rhs, future_wrapper,3 lazy_val, push, Context, ContextCreator, FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue,4 Result, Val,5};6use closure::closure;7use jsonnet_parser::{8 ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember, ForSpecData,9 IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType, 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(args) = &b.params {19 let args = args.clone();20 (21 b.name.clone(),22 LazyBinding::Bindable(Rc::new(move |this, super_obj| {23 Ok(lazy_val!(24 closure!(clone b, clone args, clone context_creator, || Ok(evaluate_method(25 context_creator.0(this.clone(), super_obj.clone())?,26 &b.value,27 args.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, expr: &LocExpr, arg_spec: ParamsDesc) -> Val {50 Val::Func(FuncDesc {51 ctx,52 params: arg_spec,53 eval_rhs: function_rhs!(closure!(clone expr, |ctx| evaluate(ctx, &expr))),54 eval_default: function_default!(closure!(|ctx, default| evaluate(ctx, &default))),55 })56}5758pub fn evaluate_field_name(59 context: Context,60 field_name: &jsonnet_parser::FieldName,61) -> Result<Option<String>> {62 Ok(match field_name {63 jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),64 jsonnet_parser::FieldName::Dyn(expr) => {65 let value = evaluate(context, expr)?.unwrap_if_lazy()?;66 if matches!(value, Val::Null) {67 None68 } else {69 Some(value.try_cast_str("dynamic field name")?)70 }71 }72 })73}7475pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {76 Ok(match (op, b) {77 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,78 (UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),79 (UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),80 (UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),81 (op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),82 })83}8485pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {86 Ok(match (a, b) {87 (Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),8889 (Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),90 (Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),91 (Val::Str(v1), Val::Bool(v2)) => Val::Str(format!("{}{}", v1, v2)),92 (Val::Bool(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),93 (Val::Str(v1), Val::Null) => Val::Str(format!("{}null", v1)),94 (Val::Null, Val::Str(v2)) => Val::Str(format!("null{}", v2)),9596 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),97 (Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),98 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),99 _ => panic!("can't add: {:?} and {:?}", a, b),100 })101}102103pub fn evaluate_binary_op_special(104 context: Context,105 a: &LocExpr,106 op: BinaryOpType,107 b: &LocExpr,108) -> Result<Val> {109 Ok(110 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {111 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),112 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),113 (a, op, eb) => {114 evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?115 }116 },117 )118}119120pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {121 Ok(match (a, op, b) {122 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,123124 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),125126 127 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),128 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),129130 131 (Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),132 (Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),133 (Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),134 (Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),135136 137 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),138 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {139 if *v2 <= f64::EPSILON {140 create_error(crate::Error::DivisionByZero)?141 }142 Val::Num(v1 / v2)143 }144145 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),146147 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),148 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),149 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),150 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),151152 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {153 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)154 }155 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {156 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)157 }158 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {159 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)160 }161 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {162 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)163 }164 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {165 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)166 }167168 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),169 })170}171172future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);173future_wrapper!(ObjValue, FutureObjValue);174175pub fn evaluate_comp(176 context: Context,177 value: &LocExpr,178 specs: &[CompSpec],179) -> Result<Option<Vec<Val>>> {180 Ok(match specs.get(0) {181 None => Some(vec![evaluate(context, &value)?]),182 Some(CompSpec::IfSpec(IfSpecData(cond))) => {183 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {184 evaluate_comp(context, value, &specs[1..])?185 } else {186 None187 }188 }189 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {190 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {191 Val::Arr(list) => {192 let mut out = Vec::new();193 for item in list {194 let item = item.clone().unwrap_if_lazy()?;195 out.push(evaluate_comp(196 context.with_var(var.clone(), item)?,197 value,198 &specs[1..],199 )?);200 }201 Some(out.iter().flatten().flatten().cloned().collect())202 }203 _ => panic!("for expression evaluated to non-iterable value"),204 }205 }206 })207}208209210pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {211 Ok(match object {212 ObjBody::MemberList(members) => {213 let new_bindings = FutureNewBindings::new();214 let future_this = FutureObjValue::new();215 let context_creator = context_creator!(216 closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {217 Ok(context.clone().extend(218 new_bindings.clone().unwrap(),219 context.clone().dollar().clone().or_else(||this.clone()),220 Some(this.unwrap()),221 super_obj222 )?)223 })224 );225 {226 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();227 for (n, b) in members228 .iter()229 .filter_map(|m| match m {230 Member::BindStmt(b) => Some(b.clone()),231 _ => None,232 })233 .map(|b| evaluate_binding(&b, context_creator.clone()))234 {235 bindings.insert(n, b);236 }237 new_bindings.fill(bindings);238 }239240 let mut new_members = BTreeMap::new();241 for member in members.into_iter() {242 match member {243 Member::Field(FieldMember {244 name,245 plus,246 params: None,247 visibility,248 value,249 }) => {250 let name = evaluate_field_name(context.clone(), &name)?;251 if name.is_none() {252 continue;253 }254 let name = name.unwrap();255 new_members.insert(256 name.clone(),257 ObjMember {258 add: plus,259 visibility: visibility.clone(),260 invoke: binding!(261 closure!(clone name, clone value, clone context_creator, |this, super_obj| {262 push(value.clone(), "object ".to_owned()+&name+" field", ||{263 let context = context_creator.0(this, super_obj)?;264 evaluate(265 context,266 &value,267 )?.unwrap_if_lazy()268 })269 })270 ),271 },272 );273 }274 Member::Field(FieldMember {275 name,276 params: Some(params),277 value,278 ..279 }) => {280 let name = evaluate_field_name(context.clone(), &name)?;281 if name.is_none() {282 continue;283 }284 let name = name.unwrap();285 new_members.insert(286 name,287 ObjMember {288 add: false,289 visibility: Visibility::Hidden,290 invoke: binding!(291 closure!(clone value, clone context_creator, |this, super_obj| {292 293 Ok(evaluate_method(294 context_creator.0(this, super_obj)?,295 &value.clone(),296 params.clone(),297 ))298 })299 ),300 },301 );302 }303 Member::BindStmt(_) => {}304 Member::AssertStmt(_) => {}305 }306 }307 future_this.fill(ObjValue::new(None, Rc::new(new_members)))308 }309 _ => todo!(),310 })311}312313#[inline(always)]314pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {315 use Expr::*;316 let locexpr = expr.clone();317 let LocExpr(expr, loc) = expr;318 Ok(match &**expr {319 Literal(LiteralType::This) => Val::Obj(320 context321 .this()322 .clone()323 .unwrap_or_else(|| panic!("this not found")),324 ),325 Literal(LiteralType::Dollar) => Val::Obj(326 context327 .dollar()328 .clone()329 .unwrap_or_else(|| panic!("dollar not found")),330 ),331 Literal(LiteralType::True) => Val::Bool(true),332 Literal(LiteralType::False) => Val::Bool(false),333 Literal(LiteralType::Null) => Val::Null,334 Parened(e) => evaluate(context, e)?,335 Str(v) => Val::Str(v.clone()),336 Num(v) => Val::Num(*v),337 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,338 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,339 Var(name) => push(locexpr, "var".to_owned(), || {340 Val::Lazy(context.binding(&name)).unwrap_if_lazy()341 })?,342 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {343 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;344 context345 .super_obj()346 .clone()347 .expect("no super found")348 .get_raw(&name, &context.this().clone().expect("no this found"))?349 .expect("value not found")350 }351 Index(value, index) => {352 match (353 evaluate(context.clone(), value)?.unwrap_if_lazy()?,354 evaluate(context, index)?,355 ) {356 (Val::Obj(v), Val::Str(s)) => {357 if let Some(v) = v.get(&s)? {358 v.unwrap_if_lazy()?359 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {360 Val::Intristic(n, s)361 } else {362 create_error(crate::Error::NoSuchField(s))?363 }364 }365 (Val::Arr(v), Val::Num(n)) => {366 if n.fract() > f64::EPSILON {367 create_error(crate::Error::FractionalIndex)?368 }369 v.get(n as usize)370 .unwrap_or_else(|| panic!("out of bounds"))371 .clone()372 .unwrap_if_lazy()?373 }374 (Val::Str(s), Val::Num(n)) => {375 Val::Str(s.chars().skip(n as usize).take(1).collect())376 }377 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),378 }379 }380 LocalExpr(bindings, returned) => {381 let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();382 let future_context = Context::new_future();383384 let context_creator = context_creator!(385 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))386 );387388 for (k, v) in bindings389 .iter()390 .map(|b| evaluate_binding(b, context_creator.clone()))391 {392 new_bindings.insert(k, v);393 }394395 let context = context396 .extend(new_bindings, None, None, None)?397 .into_future(future_context);398 evaluate(context, &returned.clone())?399 }400 Arr(items) => {401 let mut out = Vec::with_capacity(items.len());402 for item in items {403 out.push(Val::Lazy(lazy_val!(404 closure!(clone context, clone item, || {405 evaluate(context.clone(), &item)406 })407 )));408 }409 Val::Arr(out)410 }411 ArrComp(expr, compspecs) => Val::Arr(412 413 evaluate_comp(context, expr, compspecs)?.unwrap(),414 ),415 Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),416 ObjExtend(s, t) => evaluate_add_op(417 &evaluate(context.clone(), s)?,418 &Val::Obj(evaluate_object(context, t.clone())?),419 )?,420 Apply(value, ArgsDesc(args), tailstrict) => {421 let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;422 match value {423 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {424 425 ("std", "length") => {426 assert_eq!(args.len(), 1);427 let expr = &args.get(0).unwrap().1;428 match evaluate(context, expr)? {429 Val::Str(n) => Val::Num(n.chars().count() as f64),430 Val::Arr(i) => Val::Num(i.len() as f64),431 Val::Obj(o) => Val::Num(o.fields().len() as f64),432 v => panic!("can't get length of {:?}", v),433 }434 }435 436 ("std", "type") => {437 assert_eq!(args.len(), 1);438 let expr = &args.get(0).unwrap().1;439 Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())440 }441 442 ("std", "makeArray") => {443 assert_eq!(args.len(), 2);444 if let (Val::Num(v), Val::Func(d)) = (445 evaluate(context.clone(), &args[0].1)?,446 evaluate(context, &args[1].1)?,447 ) {448 assert!(v >= 0.0);449 let mut out = Vec::with_capacity(v as usize);450 for i in 0..v as usize {451 out.push(d.evaluate(vec![(None, Val::Num(i as f64))])?)452 }453 Val::Arr(out)454 } else {455 panic!("bad makeArray call");456 }457 }458 459 ("std", "codepoint") => {460 assert_eq!(args.len(), 1);461 if let Val::Str(s) = evaluate(context, &args[0].1)? {462 assert!(463 s.chars().count() == 1,464 "std.codepoint should receive single char string"465 );466 Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)467 } else {468 panic!("bad codepoint call");469 }470 }471 472 ("std", "objectFieldsEx") => {473 assert_eq!(args.len(), 2);474 if let (Val::Obj(body), Val::Bool(_include_hidden)) = (475 evaluate(context.clone(), &args[0].1)?,476 evaluate(context, &args[1].1)?,477 ) {478 479 Val::Arr(body.fields().into_iter().map(Val::Str).collect())480 } else {481 panic!("bad objectFieldsEx call");482 }483 }484 ("std", "primitiveEquals") => {485 assert_eq!(args.len(), 2);486 let (a, b) = (487 evaluate(context.clone(), &args[0].1)?,488 evaluate(context, &args[1].1)?,489 );490 Val::Bool(a == b)491 }492 ("std", "modulo") => {493 assert_eq!(args.len(), 2);494 if let (Val::Num(a), Val::Num(b)) = (495 evaluate(context.clone(), &args[0].1)?,496 evaluate(context, &args[1].1)?,497 ) {498 Val::Num(a % b)499 } else {500 panic!("bad modulo call");501 }502 }503 ("std", "floor") => {504 assert_eq!(args.len(), 1);505 if let Val::Num(a) = evaluate(context, &args[0].1)? {506 Val::Num(a.floor())507 } else {508 panic!("bad floor call");509 }510 }511 ("std", "trace") => {512 assert_eq!(args.len(), 2);513 if let (Val::Str(a), b) = (514 evaluate(context.clone(), &args[0].1)?,515 evaluate(context, &args[1].1)?,516 ) {517 518 println!("TRACE: {}", a);519 b520 } else {521 panic!("bad trace call");522 }523 }524 ("std", "pow") => {525 assert_eq!(args.len(), 2);526 if let (Val::Num(a), Val::Num(b)) = (527 evaluate(context.clone(), &args[0].1)?,528 evaluate(context, &args[1].1)?,529 ) {530 Val::Num(a.powf(b))531 } else {532 panic!("bad pow call");533 }534 }535 (ns, name) => panic!("Intristic not found: {}.{}", ns, name),536 },537 Val::Func(f) => {538 let body = #[inline(always)]539 || {540 f.evaluate(541 args.clone()542 .into_iter()543 .map(544 #[inline(always)]545 move |a| {546 Ok((547 a.clone().0,548 if *tailstrict {549 Val::Lazy(LazyVal::new_resolved(evaluate(550 context.clone(),551 &a.1,552 )?))553 } else {554 Val::Lazy(lazy_val!(555 closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))556 ))557 },558 ))559 },560 )561 .collect::<Result<Vec<_>>>()?,562 )563 };564 if *tailstrict {565 body()?566 } else {567 push(locexpr, "function call".to_owned(), body)?568 }569 }570 _ => panic!("{:?} is not a function", value),571 }572 }573 Function(params, body) => evaluate_method(context, body, params.clone()),574 AssertExpr(AssertStmt(value, msg), returned) => {575 let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {576 evaluate(context.clone(), &value)?577 .try_cast_bool("assertion condition should be boolean")578 })?;579 if assertion_result {580 push(581 returned.clone(),582 "assert 'return' branch".to_owned(),583 || evaluate(context, returned),584 )?585 } else if let Some(msg) = msg {586 panic!(587 "assertion failed ({:?}): {}",588 value,589 evaluate(context, msg)?.try_cast_str("assertion message should be string")?590 );591 } else {592 panic!("assertion failed ({:?}): no message", value);593 }594 }595 Error(e) => create_error(crate::Error::RuntimeError(596 evaluate(context, e)?.try_cast_str("error text should be string")?,597 ))?,598 IfElse {599 cond,600 cond_then,601 cond_else,602 } => {603 let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {604 evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")605 })?;606 if condition_result {607 push(608 cond_then.clone(),609 "if condition 'then' branch".to_owned(),610 || evaluate(context, cond_then),611 )?612 } else {613 match cond_else {614 Some(v) => evaluate(context, v)?,615 None => Val::Null,616 }617 }618 }619 _ => panic!(620 "evaluation not implemented: {:?}",621 LocExpr(expr.clone(), loc.clone())622 ),623 })624}