difftreelog
feat obj.objectHasEx support
in: master
1 file changed
crates/jsonnet-evaluator/src/evaluate.rsdiffbeforeafterboth1use crate::{2 context_creator, create_error, future_wrapper, lazy_val, push, with_state, Context,3 ContextCreator, Error, FuncDesc, LazyBinding, LazyVal, 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 // Bool X Bool122 (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 // Str X Str126 (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 // Num X Num132 (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);169170#[inline(always)]171pub fn evaluate_comp<T>(172 context: Context,173 value: &impl Fn(Context) -> Result<T>,174 specs: &[CompSpec],175) -> Result<Option<Vec<T>>> {176 Ok(match specs.get(0) {177 None => Some(vec![value(context)?]),178 Some(CompSpec::IfSpec(IfSpecData(cond))) => {179 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {180 evaluate_comp(context, value, &specs[1..])?181 } else {182 None183 }184 }185 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {186 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {187 Val::Arr(list) => {188 let mut out = Vec::new();189 for item in list {190 let item = item.clone().unwrap_if_lazy()?;191 out.push(evaluate_comp(192 context.with_var(var.clone(), item)?,193 value,194 &specs[1..],195 )?);196 }197 Some(out.into_iter().flatten().flatten().collect())198 }199 _ => panic!("for expression evaluated to non-iterable value"),200 }201 }202 })203}204205// TODO: Asserts206pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {207 Ok(match object {208 ObjBody::MemberList(members) => {209 let new_bindings = FutureNewBindings::new();210 let future_this = FutureObjValue::new();211 let context_creator = context_creator!(212 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {213 Ok(context.clone().extend_unbound(214 new_bindings.clone().unwrap(),215 context.clone().dollar().clone().or_else(||this.clone()),216 Some(this.unwrap()),217 super_obj218 )?)219 })220 );221 {222 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();223 for (n, b) in members224 .iter()225 .filter_map(|m| match m {226 Member::BindStmt(b) => Some(b.clone()),227 _ => None,228 })229 .map(|b| evaluate_binding(&b, context_creator.clone()))230 {231 bindings.insert(n, b);232 }233 new_bindings.fill(bindings);234 }235236 let mut new_members = BTreeMap::new();237 for member in members.into_iter() {238 match member {239 Member::Field(FieldMember {240 name,241 plus,242 params: None,243 visibility,244 value,245 }) => {246 let name = evaluate_field_name(context.clone(), &name)?;247 if name.is_none() {248 continue;249 }250 let name = name.unwrap();251 new_members.insert(252 name.clone(),253 ObjMember {254 add: plus,255 visibility: visibility.clone(),256 invoke: LazyBinding::Bindable(Rc::new(257 closure!(clone name, clone value, clone context_creator, |this, super_obj| {258 Ok(LazyVal::new_resolved(push(value.clone(), "object ".to_owned()+&name+" field", ||{259 let context = context_creator.0(this, super_obj)?;260 evaluate(261 context,262 &value,263 )?.unwrap_if_lazy()264 })?))265 }),266 )),267 },268 );269 }270 Member::Field(FieldMember {271 name,272 params: Some(params),273 value,274 ..275 }) => {276 let name = evaluate_field_name(context.clone(), &name)?;277 if name.is_none() {278 continue;279 }280 let name = name.unwrap();281 new_members.insert(282 name,283 ObjMember {284 add: false,285 visibility: Visibility::Hidden,286 invoke: LazyBinding::Bindable(Rc::new(287 closure!(clone value, clone context_creator, |this, super_obj| {288 // TODO: Assert289 Ok(LazyVal::new_resolved(evaluate_method(290 context_creator.0(this, super_obj)?,291 params.clone(),292 value.clone(),293 )))294 }),295 )),296 },297 );298 }299 Member::BindStmt(_) => {}300 Member::AssertStmt(_) => {}301 }302 }303 future_this.fill(ObjValue::new(None, Rc::new(new_members)))304 }305 ObjBody::ObjComp {306 pre_locals,307 key,308 value,309 post_locals,310 compspecs,311 } => {312 let future_this = FutureObjValue::new();313 let mut new_members = BTreeMap::new();314 for (k, v) in evaluate_comp(315 context.clone(),316 &|ctx| {317 let new_bindings = FutureNewBindings::new();318 let context_creator = context_creator!(319 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {320 Ok(context.clone().extend_unbound(321 new_bindings.clone().unwrap(),322 context.clone().dollar().clone().or_else(||this.clone()),323 None,324 super_obj325 )?)326 })327 );328 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();329 for (n, b) in pre_locals330 .iter()331 .chain(post_locals.iter())332 .map(|b| evaluate_binding(b, context_creator.clone()))333 {334 bindings.insert(n, b);335 }336 let bindings = new_bindings.fill(bindings);337 let ctx = ctx.extend_unbound(bindings, None, None, None)?;338 let key = evaluate(ctx.clone(), &key)?;339 let value = LazyBinding::Bindable(Rc::new(340 closure!(clone ctx, clone value, |this, _super_obj| {341 Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))342 }),343 ));344345 Ok((key, value))346 },347 &compspecs,348 )?349 .unwrap()350 {351 match k {352 Val::Null => {}353 Val::Str(n) => {354 new_members.insert(355 n,356 ObjMember {357 add: false,358 visibility: Visibility::Normal,359 invoke: v,360 },361 );362 }363 v => create_error(Error::FieldMustBeStringGot(v.value_type()?))?,364 }365 }366367 future_this.fill(ObjValue::new(None, Rc::new(new_members)))368 }369 })370}371372#[inline(always)]373pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {374 use Expr::*;375 let locexpr = expr.clone();376 let LocExpr(expr, loc) = expr;377 Ok(match &**expr {378 Literal(LiteralType::This) => Val::Obj(379 context380 .this()381 .clone()382 .unwrap_or_else(|| panic!("this not found")),383 ),384 Literal(LiteralType::Dollar) => Val::Obj(385 context386 .dollar()387 .clone()388 .unwrap_or_else(|| panic!("dollar not found")),389 ),390 Literal(LiteralType::True) => Val::Bool(true),391 Literal(LiteralType::False) => Val::Bool(false),392 Literal(LiteralType::Null) => Val::Null,393 Parened(e) => evaluate(context, e)?,394 Str(v) => Val::Str(v.clone()),395 Num(v) => Val::Num(*v),396 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,397 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,398 Var(name) => push(locexpr, "var".to_owned(), || {399 Val::Lazy(context.binding(&name)).unwrap_if_lazy()400 })?,401 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {402 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;403 context404 .super_obj()405 .clone()406 .expect("no super found")407 .get_raw(&name, &context.this().clone().expect("no this found"))?408 .expect("value not found")409 }410 Index(value, index) => {411 match (412 evaluate(context.clone(), value)?.unwrap_if_lazy()?,413 evaluate(context, index)?,414 ) {415 (Val::Obj(v), Val::Str(s)) => {416 if let Some(v) = v.get(&s)? {417 v.unwrap_if_lazy()?418 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {419 Val::Intristic(n, s)420 } else {421 create_error(crate::Error::NoSuchField(s))?422 }423 }424 (Val::Arr(v), Val::Num(n)) => {425 if n.fract() > f64::EPSILON {426 create_error(crate::Error::FractionalIndex)?427 }428 v.get(n as usize)429 .unwrap_or_else(|| panic!("out of bounds"))430 .clone()431 .unwrap_if_lazy()?432 }433 (Val::Str(s), Val::Num(n)) => {434 Val::Str(s.chars().skip(n as usize).take(1).collect())435 }436 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),437 }438 }439 LocalExpr(bindings, returned) => {440 let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();441 let future_context = Context::new_future();442443 let context_creator = context_creator!(444 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))445 );446447 for (k, v) in bindings448 .iter()449 .map(|b| evaluate_binding(b, context_creator.clone()))450 {451 new_bindings.insert(k, v);452 }453454 let context = context455 .extend_unbound(new_bindings, None, None, None)?456 .into_future(future_context);457 evaluate(context, &returned.clone())?458 }459 Arr(items) => {460 let mut out = Vec::with_capacity(items.len());461 for item in items {462 out.push(Val::Lazy(lazy_val!(463 closure!(clone context, clone item, || {464 evaluate(context.clone(), &item)465 })466 )));467 }468 Val::Arr(out)469 }470 ArrComp(expr, compspecs) => Val::Arr(471 // First compspec should be forspec, so no "None" possible here472 evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap(),473 ),474 Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),475 ObjExtend(s, t) => evaluate_add_op(476 &evaluate(context.clone(), s)?,477 &Val::Obj(evaluate_object(context, t.clone())?),478 )?,479 Apply(value, args, tailstrict) => {480 let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;481 match value {482 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {483 // arr/string/function484 ("std", "length") => {485 assert_eq!(args.len(), 1);486 let expr = &args.get(0).unwrap().1;487 match evaluate(context, expr)? {488 Val::Str(n) => Val::Num(n.chars().count() as f64),489 Val::Arr(i) => Val::Num(i.len() as f64),490 Val::Obj(o) => Val::Num(491 o.fields_visibility()492 .into_iter()493 .filter(|(_k, v)| *v)494 .count() as f64,495 ),496 v => panic!("can't get length of {:?}", v),497 }498 }499 // any500 ("std", "type") => {501 assert_eq!(args.len(), 1);502 let expr = &args.get(0).unwrap().1;503 Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())504 }505 // length, idx=>any506 ("std", "makeArray") => {507 assert_eq!(args.len(), 2);508 if let (Val::Num(v), Val::Func(d)) = (509 evaluate(context.clone(), &args[0].1)?,510 evaluate(context, &args[1].1)?,511 ) {512 assert!(v >= 0.0);513 let mut out = Vec::with_capacity(v as usize);514 for i in 0..v as usize {515 let call_ctx =516 Context::new().with_var("v".to_owned(), Val::Num(i as f64))?;517 out.push(d.evaluate(518 call_ctx,519 &ArgsDesc(vec![Arg(None, el!(Expr::Var("v".to_owned())))]),520 true,521 )?)522 }523 Val::Arr(out)524 } else {525 panic!("bad makeArray call");526 }527 }528 // string529 ("std", "codepoint") => {530 assert_eq!(args.len(), 1);531 if let Val::Str(s) = evaluate(context, &args[0].1)? {532 assert!(533 s.chars().count() == 1,534 "std.codepoint should receive single char string"535 );536 Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)537 } else {538 panic!("bad codepoint call");539 }540 }541 // object, includeHidden542 ("std", "objectFieldsEx") => {543 assert_eq!(args.len(), 2);544 if let (Val::Obj(body), Val::Bool(include_hidden)) = (545 evaluate(context.clone(), &args[0].1)?,546 evaluate(context, &args[1].1)?,547 ) {548 Val::Arr(549 body.fields_visibility()550 .into_iter()551 .filter(|(_k, v)| *v || include_hidden)552 .map(|(k, _v)| Val::Str(k))553 .collect(),554 )555 } else {556 panic!("bad objectFieldsEx call");557 }558 }559 ("std", "primitiveEquals") => {560 assert_eq!(args.len(), 2);561 let (a, b) = (562 evaluate(context.clone(), &args[0].1)?,563 evaluate(context, &args[1].1)?,564 );565 Val::Bool(a == b)566 }567 ("std", "modulo") => {568 assert_eq!(args.len(), 2);569 if let (Val::Num(a), Val::Num(b)) = (570 evaluate(context.clone(), &args[0].1)?,571 evaluate(context, &args[1].1)?,572 ) {573 Val::Num(a % b)574 } else {575 panic!("bad modulo call");576 }577 }578 ("std", "floor") => {579 assert_eq!(args.len(), 1);580 if let Val::Num(a) = evaluate(context, &args[0].1)? {581 Val::Num(a.floor())582 } else {583 panic!("bad floor call");584 }585 }586 ("std", "trace") => {587 assert_eq!(args.len(), 2);588 if let (Val::Str(a), b) = (589 evaluate(context.clone(), &args[0].1)?,590 evaluate(context, &args[1].1)?,591 ) {592 // TODO: Line numbers as in original jsonnet593 println!("TRACE: {}", a);594 b595 } else {596 panic!("bad trace call");597 }598 }599 ("std", "pow") => {600 assert_eq!(args.len(), 2);601 if let (Val::Num(a), Val::Num(b)) = (602 evaluate(context.clone(), &args[0].1)?,603 evaluate(context, &args[1].1)?,604 ) {605 Val::Num(a.powf(b))606 } else {607 panic!("bad pow call");608 }609 }610 ("std", "extVar") => {611 assert_eq!(args.len(), 1);612 if let Val::Str(a) = evaluate(context, &args[0].1)? {613 with_state(|s| s.0.ext_vars.borrow().get(&a).cloned()).ok_or_else(614 || {615 create_error::<()>(crate::Error::UndefinedExternalVariable(a))616 .err()617 .unwrap()618 },619 )?620 } else {621 panic!("bad extVar call");622 }623 }624 (ns, name) => panic!("Intristic not found: {}.{}", ns, name),625 },626 Val::Func(f) => {627 let body = #[inline(always)]628 || f.evaluate(context, args, *tailstrict);629 if *tailstrict {630 body()?631 } else {632 push(locexpr, "function call".to_owned(), body)?633 }634 }635 _ => panic!("{:?} is not a function", value),636 }637 }638 Function(params, body) => evaluate_method(context, params.clone(), body.clone()),639 AssertExpr(AssertStmt(value, msg), returned) => {640 let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {641 evaluate(context.clone(), &value)?642 .try_cast_bool("assertion condition should be boolean")643 })?;644 if assertion_result {645 push(646 returned.clone(),647 "assert 'return' branch".to_owned(),648 || evaluate(context, returned),649 )?650 } else if let Some(msg) = msg {651 panic!(652 "assertion failed ({:?}): {}",653 value,654 evaluate(context, msg)?.try_cast_str("assertion message should be string")?655 );656 } else {657 panic!("assertion failed ({:?}): no message", value);658 }659 }660 Error(e) => create_error(crate::Error::RuntimeError(661 evaluate(context, e)?.try_cast_str("error text should be string")?,662 ))?,663 IfElse {664 cond,665 cond_then,666 cond_else,667 } => {668 let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {669 evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")670 })?;671 if condition_result {672 push(673 cond_then.clone(),674 "if condition 'then' branch".to_owned(),675 || evaluate(context, cond_then),676 )?677 } else {678 match cond_else {679 Some(v) => evaluate(context, v)?,680 None => Val::Null,681 }682 }683 }684 Import(path) => {685 let mut lib_path = loc686 .clone()687 .expect("imports can't be used without loc_data")688 .0689 .clone();690 lib_path.pop();691 lib_path.push(path);692 with_state(|s| s.import_file(&lib_path))?693 }694 _ => panic!(695 "evaluation not implemented: {:?}",696 LocExpr(expr.clone(), loc.clone())697 ),698 })699}