difftreelog
refactor(evaluator) extract Apply evaluation
in: master
2 files changed
crates/jrsonnet-evaluator/src/evaluate.rsdiffbeforeafterboth1use crate::{2 context_creator, create_error, create_error_result, escape_string_json, future_wrapper,3 lazy_val, manifest_json_ex, parse_args, push, with_state, Context, ContextCreator, Error,4 FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue, Result, Val, ValType,5};6use closure::closure;7use jrsonnet_parser::{8 AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember, ForSpecData, IfSpecData,9 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) -> (Rc<str>, 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.1, "thunk", ||{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: &jrsonnet_parser::FieldName,56) -> Result<Option<Rc<str>>> {57 Ok(match field_name {58 jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),59 jrsonnet_parser::FieldName::Dyn(expr) => {60 let lazy = evaluate(context, expr)?;61 let value = lazy.unwrap_if_lazy()?;62 if matches!(value, Val::Null) {63 None64 } else {65 Some(value.try_cast_str("dynamic field name")?)66 }67 }68 })69}7071pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {72 Ok(match (op, b) {73 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,74 (UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),75 (UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),76 (UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),77 (op, o) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(78 op,79 o.value_type()?,80 ))?,81 })82}8384pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {85 Ok(match (a, b) {86 (Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),8788 // Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)89 (Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),90 (Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),9192 (Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),93 (o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9495 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),96 (Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),97 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),98 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(99 BinaryOpType::Add,100 a.value_type()?,101 b.value_type()?,102 ))?,103 })104}105106pub fn evaluate_binary_op_special(107 context: Context,108 a: &LocExpr,109 op: BinaryOpType,110 b: &LocExpr,111) -> Result<Val> {112 Ok(113 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {114 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),115 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),116 (a, op, eb) => {117 evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?118 }119 },120 )121}122123pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {124 Ok(match (a, op, b) {125 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,126127 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),128129 // Bool X Bool130 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),131 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),132133 // Str X Str134 (Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),135 (Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),136 (Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),137 (Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),138139 // Num X Num140 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),141 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {142 if *v2 <= f64::EPSILON {143 create_error_result(crate::Error::DivisionByZero)?144 }145 Val::Num(v1 / v2)146 }147148 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),149150 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),151 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),152 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),153 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),154155 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {156 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)157 }158 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {159 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)160 }161 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {162 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)163 }164 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {165 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)166 }167 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {168 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)169 }170171 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(172 op,173 a.value_type()?,174 b.value_type()?,175 ))?,176 })177}178179future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);180future_wrapper!(ObjValue, FutureObjValue);181182pub fn evaluate_comp<T>(183 context: Context,184 value: &impl Fn(Context) -> Result<T>,185 specs: &[CompSpec],186) -> Result<Option<Vec<T>>> {187 Ok(match specs.get(0) {188 None => Some(vec![value(context)?]),189 Some(CompSpec::IfSpec(IfSpecData(cond))) => {190 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {191 evaluate_comp(context, value, &specs[1..])?192 } else {193 None194 }195 }196 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {197 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {198 Val::Arr(list) => {199 let mut out = Vec::new();200 for item in list.iter() {201 let item = item.unwrap_if_lazy()?;202 out.push(evaluate_comp(203 context.with_var(var.clone(), item.clone())?,204 value,205 &specs[1..],206 )?);207 }208 Some(out.into_iter().flatten().flatten().collect())209 }210 _ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,211 }212 }213 })214}215216pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {217 let new_bindings = FutureNewBindings::new();218 let future_this = FutureObjValue::new();219 let context_creator = context_creator!(220 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {221 Ok(context.extend_unbound(222 new_bindings.clone().unwrap(),223 context.dollar().clone().or_else(||this.clone()),224 Some(this.unwrap()),225 super_obj226 )?)227 })228 );229 {230 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();231 for (n, b) in members232 .iter()233 .filter_map(|m| match m {234 Member::BindStmt(b) => Some(b.clone()),235 _ => None,236 })237 .map(|b| evaluate_binding(&b, context_creator.clone()))238 {239 bindings.insert(n, b);240 }241 new_bindings.fill(bindings);242 }243244 let mut new_members = BTreeMap::new();245 for member in members.iter() {246 match member {247 Member::Field(FieldMember {248 name,249 plus,250 params: None,251 visibility,252 value,253 }) => {254 let name = evaluate_field_name(context.clone(), &name)?;255 if name.is_none() {256 continue;257 }258 let name = name.unwrap();259 new_members.insert(260 name.clone(),261 ObjMember {262 add: *plus,263 visibility: *visibility,264 invoke: LazyBinding::Bindable(Rc::new(265 closure!(clone name, clone value, clone context_creator, |this, super_obj| {266 Ok(LazyVal::new_resolved(push(&value.1, "object field", ||{267 let context = context_creator.0(this, super_obj)?;268 evaluate(269 context,270 &value,271 )272 })?))273 }),274 )),275 },276 );277 }278 Member::Field(FieldMember {279 name,280 params: Some(params),281 value,282 ..283 }) => {284 let name = evaluate_field_name(context.clone(), &name)?;285 if name.is_none() {286 continue;287 }288 let name = name.unwrap();289 new_members.insert(290 name,291 ObjMember {292 add: false,293 visibility: Visibility::Hidden,294 invoke: LazyBinding::Bindable(Rc::new(295 closure!(clone value, clone context_creator, clone params, |this, super_obj| {296 // TODO: Assert297 Ok(LazyVal::new_resolved(evaluate_method(298 context_creator.0(this, super_obj)?,299 params.clone(),300 value.clone(),301 )))302 }),303 )),304 },305 );306 }307 Member::BindStmt(_) => {}308 Member::AssertStmt(_) => {}309 }310 }311 Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))312}313314pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {315 Ok(match object {316 ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,317 ObjBody::ObjComp(obj) => {318 let future_this = FutureObjValue::new();319 let mut new_members = BTreeMap::new();320 for (k, v) in evaluate_comp(321 context.clone(),322 &|ctx| {323 let new_bindings = FutureNewBindings::new();324 let context_creator = context_creator!(325 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {326 Ok(context.extend_unbound(327 new_bindings.clone().unwrap(),328 context.dollar().clone().or_else(||this.clone()),329 None,330 super_obj331 )?)332 })333 );334 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();335 for (n, b) in obj336 .pre_locals337 .iter()338 .chain(obj.post_locals.iter())339 .map(|b| evaluate_binding(b, context_creator.clone()))340 {341 bindings.insert(n, b);342 }343 let bindings = new_bindings.fill(bindings);344 let ctx = ctx.extend_unbound(bindings, None, None, None)?;345 let key = evaluate(ctx.clone(), &obj.key)?;346 let value = LazyBinding::Bindable(Rc::new(347 closure!(clone ctx, clone obj.value, |this, _super_obj| {348 Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))349 }),350 ));351352 Ok((key, value))353 },354 &obj.compspecs,355 )?356 .unwrap()357 {358 match k {359 Val::Null => {}360 Val::Str(n) => {361 new_members.insert(362 n,363 ObjMember {364 add: false,365 visibility: Visibility::Normal,366 invoke: v,367 },368 );369 }370 v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,371 }372 }373374 future_this.fill(ObjValue::new(None, Rc::new(new_members)))375 }376 })377}378379pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {380 use Expr::*;381 let LocExpr(expr, loc) = expr;382 Ok(match &**expr {383 Literal(LiteralType::This) => Val::Obj(384 context385 .this()386 .clone()387 .ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,388 ),389 Literal(LiteralType::Dollar) => Val::Obj(390 context391 .dollar()392 .clone()393 .ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,394 ),395 Literal(LiteralType::True) => Val::Bool(true),396 Literal(LiteralType::False) => Val::Bool(false),397 Literal(LiteralType::Null) => Val::Null,398 Parened(e) => evaluate(context, e)?,399 Str(v) => Val::Str(v.clone()),400 Num(v) => Val::Num(*v),401 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,402 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,403 Var(name) => push(loc, "var", || {404 Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?)405 })?,406 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {407 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;408 context409 .super_obj()410 .clone()411 .expect("no super found")412 .get_raw(&name, &context.this().clone().expect("no this found"))?413 .expect("value not found")414 }415 Index(value, index) => {416 match (417 evaluate(context.clone(), value)?.unwrap_if_lazy()?,418 evaluate(context, index)?,419 ) {420 (Val::Obj(v), Val::Str(s)) => {421 if let Some(v) = v.get(s.clone())? {422 v.unwrap_if_lazy()?423 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {424 Val::Intristic(n, s)425 } else {426 create_error_result(crate::Error::NoSuchField(s))?427 }428 }429 (Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(430 ValType::Obj,431 ValType::Str,432 n.value_type()?,433 ))?,434435 (Val::Arr(v), Val::Num(n)) => {436 if n.fract() > f64::EPSILON {437 create_error_result(crate::Error::FractionalIndex)?438 }439 v.get(n as usize)440 .ok_or_else(|| {441 create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))442 })?443 .clone()444 .unwrap_if_lazy()?445 }446 (Val::Arr(_), Val::Str(n)) => {447 create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?448 }449 (Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(450 ValType::Arr,451 ValType::Num,452 n.value_type()?,453 ))?,454455 (Val::Str(s), Val::Num(n)) => Val::Str(456 s.chars()457 .skip(n as usize)458 .take(1)459 .collect::<String>()460 .into(),461 ),462 (Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(463 ValType::Str,464 ValType::Num,465 n.value_type()?,466 ))?,467468 (v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,469 }470 }471 LocalExpr(bindings, returned) => {472 let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();473 let future_context = Context::new_future();474475 let context_creator = context_creator!(476 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))477 );478479 for (k, v) in bindings480 .iter()481 .map(|b| evaluate_binding(b, context_creator.clone()))482 {483 new_bindings.insert(k, v);484 }485486 let context = context487 .extend_unbound(new_bindings, None, None, None)?488 .into_future(future_context);489 evaluate(context, &returned.clone())?490 }491 Arr(items) => {492 let mut out = Vec::with_capacity(items.len());493 for item in items {494 out.push(Val::Lazy(lazy_val!(495 closure!(clone context, clone item, || {496 evaluate(context.clone(), &item)497 })498 )));499 }500 Val::Arr(Rc::new(out))501 }502 ArrComp(expr, compspecs) => Val::Arr(503 // First compspec should be forspec, so no "None" possible here504 Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),505 ),506 Obj(body) => Val::Obj(evaluate_object(context, body)?),507 ObjExtend(s, t) => evaluate_add_op(508 &evaluate(context.clone(), s)?,509 &Val::Obj(evaluate_object(context, t)?),510 )?,511 Apply(value, args, tailstrict) => {512 let lazy = evaluate(context.clone(), value)?;513 let value = lazy.unwrap_if_lazy()?;514 match value {515 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {516 // arr/string/function517 ("std", "length") => parse_args!(context, "std.length", args, 1, [518 0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];519 ], {520 match x {521 Val::Str(n) => Val::Num(n.chars().count() as f64),522 Val::Arr(i) => Val::Num(i.len() as f64),523 Val::Obj(o) => Val::Num(524 o.fields_visibility()525 .into_iter()526 .filter(|(_k, v)| *v)527 .count() as f64,528 ),529 _ => unreachable!(),530 }531 }),532 // any533 ("std", "type") => parse_args!(context, "std.type", args, 1, [534 0, x, vec![];535 ], {536 Val::Str(x.value_type()?.name().into())537 }),538 // length, idx=>any539 ("std", "makeArray") => parse_args!(context, "std.makeArray", args, 2, [540 0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];541 1, func: [Val::Func]!!Val::Func, vec![ValType::Func];542 ], {543 assert!(sz >= 0.0);544 let mut out = Vec::with_capacity(sz as usize);545 for i in 0..sz as usize {546 out.push(func.evaluate_values(547 Context::new(),548 &[Val::Num(i as f64)]549 )?)550 }551 Val::Arr(Rc::new(out))552 }),553 // string554 ("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [555 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];556 ], {557 assert!(558 str.chars().count() == 1,559 "std.codepoint should receive single char string"560 );561 Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)562 }),563 // object, includeHidden564 ("std", "objectFieldsEx") => {565 parse_args!(context, "std.objectFieldsEx",args, 2, [566 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];567 1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];568 ], {569 Val::Arr(Rc::new(570 obj.fields_visibility()571 .into_iter()572 .filter(|(_k, v)| *v || inc_hidden)573 .map(|(k, _v)| Val::Str(k))574 .collect(),575 ))576 })577 }578 // object, field, includeHidden579 ("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [580 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];581 1, f: [Val::Str]!!Val::Str, vec![ValType::Str];582 2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];583 ], {584 Val::Bool(585 obj.fields_visibility()586 .into_iter()587 .filter(|(_k, v)| *v || inc_hidden)588 .any(|(k, _v)| *k == *f),589 )590 }),591 ("std", "primitiveEquals") => {592 parse_args!(context, "std.primitiveEquals", args, 2, [593 0, a, vec![];594 1, b, vec![];595 ], {596 Val::Bool(a == b)597 })598 }599 ("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [600 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];601 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];602 ], {603 Val::Num(a % b)604 }),605 ("std", "floor") => parse_args!(context, "std.floor", args, 1, [606 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];607 ], {608 Val::Num(x.floor())609 }),610 ("std", "trace") => parse_args!(context, "std.trace", args, 2, [611 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];612 1, rest, vec![];613 ], {614 // TODO: Line numbers as in original jsonnet615 println!("TRACE: {}", str);616 rest617 }),618 ("std", "pow") => parse_args!(context, "std.modulo", args, 2, [619 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];620 1, n: [Val::Num]!!Val::Num, vec![ValType::Num];621 ], {622 Val::Num(x.powf(n))623 }),624 ("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [625 0, x: [Val::Str]!!Val::Str, vec![ValType::Str];626 ], {627 with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(628 || create_error(crate::Error::UndefinedExternalVariable(x)),629 )?630 }),631 ("std", "filter") => parse_args!(context, "std.filter", args, 2, [632 0, func: [Val::Func]!!Val::Func, vec![ValType::Func];633 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];634 ], {635 Val::Arr(Rc::new(636 arr.iter()637 .cloned()638 .filter(|e| {639 func640 .evaluate_values(context.clone(), &[e.clone()])641 .unwrap()642 .try_cast_bool("filter predicate")643 .unwrap()644 })645 .collect(),646 ))647 }),648 ("std", "char") => parse_args!(context, "std.char", args, 1, [649 0, n: [Val::Num]!!Val::Num, vec![ValType::Num];650 ], {651 let mut out = String::new();652 out.push(std::char::from_u32(n as u32).unwrap());653 Val::Str(out.into())654 }),655 ("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [656 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];657 ], {658 Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))659 }),660 ("std", "md5") => parse_args!(context, "std.md5", args, 1, [661 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];662 ], {663 Val::Str(format!("{:x}", md5::compute(str.as_bytes())).into())664 }),665 // faster666 ("std", "join") => parse_args!(context, "std.join", args, 2, [667 0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];668 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];669 ], {670 match sep {671 Val::Arr(joiner_items) => {672 let mut out = Vec::new();673674 let mut first = true;675 for item in arr.iter().cloned() {676 if let Val::Arr(items) = item.unwrap_if_lazy()? {677 if !first {678 out.reserve(joiner_items.len());679 out.extend(joiner_items.iter().cloned());680 }681 first = false;682 out.reserve(items.len());683 out.extend(items.iter().cloned());684 } else {685 create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;686 }687 }688689 Val::Arr(Rc::new(out))690 },691 Val::Str(sep) => {692 let mut out = String::new();693694 let mut first = true;695 for item in arr.iter().cloned() {696 if let Val::Str(item) = item.unwrap_if_lazy()? {697 if !first {698 out += &sep;699 }700 first = false;701 out += &item;702 } else {703 create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;704 }705 }706707 Val::Str(out.into())708 },709 _ => unreachable!()710 }711 }),712 // Faster713 ("std", "escapeStringJson") => {714 parse_args!(context, "std.escapeStringJson", args, 1, [715 0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];716 ], {717 Val::Str(escape_string_json(&str_).into())718 })719 }720 // Faster721 ("std", "manifestJsonEx") => {722 parse_args!(context, "std.manifestJsonEx", args, 2, [723 0, value, vec![];724 1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];725 ], {726 Val::Str(manifest_json_ex(&value, &indent)?.into())727 })728 }729 (ns, name) => create_error_result(crate::Error::IntristicNotFound(730 ns.into(),731 name.into(),732 ))?,733 },734 Val::Func(f) => {735 let body = || f.evaluate(context, args, *tailstrict);736 if *tailstrict {737 body()?738 } else {739 push(loc, "function call", body)?740 }741 }742 v => {743 create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?744 }745 }746 }747 Function(params, body) => evaluate_method(context, params.clone(), body.clone()),748 AssertExpr(AssertStmt(value, msg), returned) => {749 let assertion_result = push(&value.1, "assertion condition", || {750 evaluate(context.clone(), &value)?751 .try_cast_bool("assertion condition should be boolean")752 })?;753 if assertion_result {754 evaluate(context, returned)?755 } else if let Some(msg) = msg {756 create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?757 } else {758 create_error_result(crate::Error::AssertionFailed(Val::Null))?759 }760 }761 Error(e) => create_error_result(crate::Error::RuntimeError(762 evaluate(context, e)?.try_cast_str("error text should be string")?,763 ))?,764 IfElse {765 cond,766 cond_then,767 cond_else,768 } => {769 if evaluate(context.clone(), &cond.0)?770 .try_cast_bool("if condition should be boolean")?771 {772 evaluate(context, cond_then)?773 } else {774 match cond_else {775 Some(v) => evaluate(context, v)?,776 None => Val::Null,777 }778 }779 }780 Import(path) => {781 let mut tmp = loc782 .clone()783 .expect("imports can't be used without loc_data")784 .0;785 let import_location = Rc::make_mut(&mut tmp);786 import_location.pop();787 with_state(|s| s.import_file(&import_location, path))?788 }789 ImportStr(path) => {790 let mut tmp = loc791 .clone()792 .expect("imports can't be used without loc_data")793 .0;794 let import_location = Rc::make_mut(&mut tmp);795 import_location.pop();796 Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)797 }798 Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),799 })800}1use crate::{2 context_creator, create_error, create_error_result, escape_string_json, future_wrapper,3 lazy_val, manifest_json_ex, parse_args, push, with_state, Context, ContextCreator, Error,4 FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue, Result, Val, ValType,5};6use closure::closure;7use jrsonnet_parser::{8 ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, ExprLocation, FieldMember,9 ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,10 Visibility,11};12use std::{13 collections::{BTreeMap, HashMap},14 rc::Rc,15};1617pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (Rc<str>, LazyBinding) {18 let b = b.clone();19 if let Some(params) = &b.params {20 let params = params.clone();21 (22 b.name.clone(),23 LazyBinding::Bindable(Rc::new(move |this, super_obj| {24 Ok(lazy_val!(25 closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(26 context_creator.0(this.clone(), super_obj.clone())?,27 params.clone(),28 b.value.clone(),29 )))30 ))31 })),32 )33 } else {34 (35 b.name.clone(),36 LazyBinding::Bindable(Rc::new(move |this, super_obj| {37 Ok(lazy_val!(closure!(clone context_creator, clone b, ||38 push(&b.value.1, "thunk", ||{39 evaluate(40 context_creator.0(this.clone(), super_obj.clone())?,41 &b.value42 )43 })44 )))45 })),46 )47 }48}4950pub fn evaluate_method(ctx: Context, params: ParamsDesc, body: LocExpr) -> Val {51 Val::Func(FuncDesc { ctx, params, body })52}5354pub fn evaluate_field_name(55 context: Context,56 field_name: &jrsonnet_parser::FieldName,57) -> Result<Option<Rc<str>>> {58 Ok(match field_name {59 jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),60 jrsonnet_parser::FieldName::Dyn(expr) => {61 let lazy = evaluate(context, expr)?;62 let value = lazy.unwrap_if_lazy()?;63 if matches!(value, Val::Null) {64 None65 } else {66 Some(value.try_cast_str("dynamic field name")?)67 }68 }69 })70}7172pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {73 Ok(match (op, b) {74 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,75 (UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),76 (UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),77 (UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),78 (op, o) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(79 op,80 o.value_type()?,81 ))?,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).into()),8889 // Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)90 (Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),91 (Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),9293 (Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),94 (o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9596 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),97 (Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),98 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),99 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(100 BinaryOpType::Add,101 a.value_type()?,102 b.value_type()?,103 ))?,104 })105}106107pub fn evaluate_binary_op_special(108 context: Context,109 a: &LocExpr,110 op: BinaryOpType,111 b: &LocExpr,112) -> Result<Val> {113 Ok(114 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {115 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),116 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),117 (a, op, eb) => {118 evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?119 }120 },121 )122}123124pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {125 Ok(match (a, op, b) {126 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,127128 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),129130 // Bool X Bool131 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),132 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),133134 // Str X Str135 (Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),136 (Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),137 (Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),138 (Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),139140 // Num X Num141 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),142 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {143 if *v2 <= f64::EPSILON {144 create_error_result(crate::Error::DivisionByZero)?145 }146 Val::Num(v1 / v2)147 }148149 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),150151 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),152 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),153 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),154 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),155156 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {157 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)158 }159 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {160 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)161 }162 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {163 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)164 }165 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {166 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)167 }168 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {169 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)170 }171172 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(173 op,174 a.value_type()?,175 b.value_type()?,176 ))?,177 })178}179180future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);181future_wrapper!(ObjValue, FutureObjValue);182183pub fn evaluate_comp<T>(184 context: Context,185 value: &impl Fn(Context) -> Result<T>,186 specs: &[CompSpec],187) -> Result<Option<Vec<T>>> {188 Ok(match specs.get(0) {189 None => Some(vec![value(context)?]),190 Some(CompSpec::IfSpec(IfSpecData(cond))) => {191 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {192 evaluate_comp(context, value, &specs[1..])?193 } else {194 None195 }196 }197 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {198 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {199 Val::Arr(list) => {200 let mut out = Vec::new();201 for item in list.iter() {202 let item = item.unwrap_if_lazy()?;203 out.push(evaluate_comp(204 context.with_var(var.clone(), item.clone())?,205 value,206 &specs[1..],207 )?);208 }209 Some(out.into_iter().flatten().flatten().collect())210 }211 _ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,212 }213 }214 })215}216217pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {218 let new_bindings = FutureNewBindings::new();219 let future_this = FutureObjValue::new();220 let context_creator = context_creator!(221 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {222 Ok(context.extend_unbound(223 new_bindings.clone().unwrap(),224 context.dollar().clone().or_else(||this.clone()),225 Some(this.unwrap()),226 super_obj227 )?)228 })229 );230 {231 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();232 for (n, b) in members233 .iter()234 .filter_map(|m| match m {235 Member::BindStmt(b) => Some(b.clone()),236 _ => None,237 })238 .map(|b| evaluate_binding(&b, context_creator.clone()))239 {240 bindings.insert(n, b);241 }242 new_bindings.fill(bindings);243 }244245 let mut new_members = BTreeMap::new();246 for member in members.iter() {247 match member {248 Member::Field(FieldMember {249 name,250 plus,251 params: None,252 visibility,253 value,254 }) => {255 let name = evaluate_field_name(context.clone(), &name)?;256 if name.is_none() {257 continue;258 }259 let name = name.unwrap();260 new_members.insert(261 name.clone(),262 ObjMember {263 add: *plus,264 visibility: *visibility,265 invoke: LazyBinding::Bindable(Rc::new(266 closure!(clone name, clone value, clone context_creator, |this, super_obj| {267 Ok(LazyVal::new_resolved(push(&value.1, "object field", ||{268 let context = context_creator.0(this, super_obj)?;269 evaluate(270 context,271 &value,272 )273 })?))274 }),275 )),276 },277 );278 }279 Member::Field(FieldMember {280 name,281 params: Some(params),282 value,283 ..284 }) => {285 let name = evaluate_field_name(context.clone(), &name)?;286 if name.is_none() {287 continue;288 }289 let name = name.unwrap();290 new_members.insert(291 name,292 ObjMember {293 add: false,294 visibility: Visibility::Hidden,295 invoke: LazyBinding::Bindable(Rc::new(296 closure!(clone value, clone context_creator, clone params, |this, super_obj| {297 // TODO: Assert298 Ok(LazyVal::new_resolved(evaluate_method(299 context_creator.0(this, super_obj)?,300 params.clone(),301 value.clone(),302 )))303 }),304 )),305 },306 );307 }308 Member::BindStmt(_) => {}309 Member::AssertStmt(_) => {}310 }311 }312 Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))313}314315pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {316 Ok(match object {317 ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,318 ObjBody::ObjComp(obj) => {319 let future_this = FutureObjValue::new();320 let mut new_members = BTreeMap::new();321 for (k, v) in evaluate_comp(322 context.clone(),323 &|ctx| {324 let new_bindings = FutureNewBindings::new();325 let context_creator = context_creator!(326 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {327 Ok(context.extend_unbound(328 new_bindings.clone().unwrap(),329 context.dollar().clone().or_else(||this.clone()),330 None,331 super_obj332 )?)333 })334 );335 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();336 for (n, b) in obj337 .pre_locals338 .iter()339 .chain(obj.post_locals.iter())340 .map(|b| evaluate_binding(b, context_creator.clone()))341 {342 bindings.insert(n, b);343 }344 let bindings = new_bindings.fill(bindings);345 let ctx = ctx.extend_unbound(bindings, None, None, None)?;346 let key = evaluate(ctx.clone(), &obj.key)?;347 let value = LazyBinding::Bindable(Rc::new(348 closure!(clone ctx, clone obj.value, |this, _super_obj| {349 Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))350 }),351 ));352353 Ok((key, value))354 },355 &obj.compspecs,356 )?357 .unwrap()358 {359 match k {360 Val::Null => {}361 Val::Str(n) => {362 new_members.insert(363 n,364 ObjMember {365 add: false,366 visibility: Visibility::Normal,367 invoke: v,368 },369 );370 }371 v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,372 }373 }374375 future_this.fill(ObjValue::new(None, Rc::new(new_members)))376 }377 })378}379380/// Extracts code block and disables inlining for them381/// Fixes WASM to java bytecode compilation failing because of very large method382macro_rules! noinline {383 ($e:expr) => {384 (#[inline(never)]385 move || $e)()386 };387}388389pub fn evaluate_apply(390 context: Context,391 value: &LocExpr,392 args: &ArgsDesc,393 loc: &Option<ExprLocation>,394 tailstrict: bool,395) -> Result<Val> {396 let lazy = evaluate(context.clone(), value)?;397 let value = lazy.unwrap_if_lazy()?;398 Ok(match value {399 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {400 // arr/string/function401 ("std", "length") => noinline!(parse_args!(context, "std.length", args, 1, [402 0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];403 ], {404 Ok(match x {405 Val::Str(n) => Val::Num(n.chars().count() as f64),406 Val::Arr(i) => Val::Num(i.len() as f64),407 Val::Obj(o) => Val::Num(408 o.fields_visibility()409 .into_iter()410 .filter(|(_k, v)| *v)411 .count() as f64,412 ),413 _ => unreachable!(),414 })415 }))?,416 // any417 ("std", "type") => parse_args!(context, "std.type", args, 1, [418 0, x, vec![];419 ], {420 Val::Str(x.value_type()?.name().into())421 }),422 // length, idx=>any423 ("std", "makeArray") => noinline!(parse_args!(context, "std.makeArray", args, 2, [424 0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];425 1, func: [Val::Func]!!Val::Func, vec![ValType::Func];426 ], {427 assert!(sz >= 0.0);428 let mut out = Vec::with_capacity(sz as usize);429 for i in 0..sz as usize {430 out.push(func.evaluate_values(431 Context::new(),432 &[Val::Num(i as f64)]433 )?)434 }435 Ok(Val::Arr(Rc::new(out)))436 }))?,437 // string438 ("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [439 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];440 ], {441 assert!(442 str.chars().count() == 1,443 "std.codepoint should receive single char string"444 );445 Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)446 }),447 // object, includeHidden448 ("std", "objectFieldsEx") => {449 noinline!(parse_args!(context, "std.objectFieldsEx",args, 2, [450 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];451 1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];452 ], {453 Ok(Val::Arr(Rc::new(454 obj.fields_visibility()455 .into_iter()456 .filter(|(_k, v)| *v || inc_hidden)457 .map(|(k, _v)| Val::Str(k))458 .collect(),459 )))460 }))?461 }462 // object, field, includeHidden463 ("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [464 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];465 1, f: [Val::Str]!!Val::Str, vec![ValType::Str];466 2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];467 ], {468 Val::Bool(469 obj.fields_visibility()470 .into_iter()471 .filter(|(_k, v)| *v || inc_hidden)472 .any(|(k, _v)| *k == *f),473 )474 }),475 ("std", "primitiveEquals") => parse_args!(context, "std.primitiveEquals", args, 2, [476 0, a, vec![];477 1, b, vec![];478 ], {479 Val::Bool(a == b)480 }),481 ("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [482 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];483 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];484 ], {485 Val::Num(a % b)486 }),487 ("std", "floor") => parse_args!(context, "std.floor", args, 1, [488 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];489 ], {490 Val::Num(x.floor())491 }),492 ("std", "trace") => parse_args!(context, "std.trace", args, 2, [493 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];494 1, rest, vec![];495 ], {496 // TODO: Line numbers as in original jsonnet497 println!("TRACE: {}", str);498 rest499 }),500 ("std", "pow") => parse_args!(context, "std.modulo", args, 2, [501 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];502 1, n: [Val::Num]!!Val::Num, vec![ValType::Num];503 ], {504 Val::Num(x.powf(n))505 }),506 ("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [507 0, x: [Val::Str]!!Val::Str, vec![ValType::Str];508 ], {509 with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(510 || create_error(crate::Error::UndefinedExternalVariable(x)),511 )?512 }),513 ("std", "filter") => noinline!(parse_args!(context, "std.filter", args, 2, [514 0, func: [Val::Func]!!Val::Func, vec![ValType::Func];515 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];516 ], {517 Ok(Val::Arr(Rc::new(518 arr.iter()519 .cloned()520 .filter(|e| {521 func522 .evaluate_values(context.clone(), &[e.clone()])523 .unwrap()524 .try_cast_bool("filter predicate")525 .unwrap()526 })527 .collect(),528 )))529 }))?,530 ("std", "char") => parse_args!(context, "std.char", args, 1, [531 0, n: [Val::Num]!!Val::Num, vec![ValType::Num];532 ], {533 let mut out = String::new();534 out.push(std::char::from_u32(n as u32).unwrap());535 Val::Str(out.into())536 }),537 ("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [538 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];539 ], {540 Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))541 }),542 ("std", "md5") => noinline!(parse_args!(context, "std.md5", args, 1, [543 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];544 ], {545 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))546 }))?,547 // faster548 ("std", "join") => noinline!(parse_args!(context, "std.join", args, 2, [549 0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];550 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];551 ], {552 Ok(match sep {553 Val::Arr(joiner_items) => {554 let mut out = Vec::new();555556 let mut first = true;557 for item in arr.iter().cloned() {558 if let Val::Arr(items) = item.unwrap_if_lazy()? {559 if !first {560 out.reserve(joiner_items.len());561 out.extend(joiner_items.iter().cloned());562 }563 first = false;564 out.reserve(items.len());565 out.extend(items.iter().cloned());566 } else {567 create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;568 }569 }570571 Val::Arr(Rc::new(out))572 },573 Val::Str(sep) => {574 let mut out = String::new();575576 let mut first = true;577 for item in arr.iter().cloned() {578 if let Val::Str(item) = item.unwrap_if_lazy()? {579 if !first {580 out += &sep;581 }582 first = false;583 out += &item;584 } else {585 create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;586 }587 }588589 Val::Str(out.into())590 },591 _ => unreachable!()592 })593 }))?,594 // Faster595 ("std", "escapeStringJson") => parse_args!(context, "std.escapeStringJson", args, 1, [596 0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];597 ], {598 Val::Str(escape_string_json(&str_).into())599 }),600 // Faster601 ("std", "manifestJsonEx") => parse_args!(context, "std.manifestJsonEx", args, 2, [602 0, value, vec![];603 1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];604 ], {605 Val::Str(manifest_json_ex(&value, &indent)?.into())606 }),607 (ns, name) => {608 create_error_result(crate::Error::IntristicNotFound(ns.into(), name.into()))?609 }610 },611 Val::Func(f) => {612 let body = || f.evaluate(context, args, tailstrict);613 if tailstrict {614 body()?615 } else {616 push(loc, "function call", body)?617 }618 }619 v => create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?,620 })621}622623pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {624 use Expr::*;625 let LocExpr(expr, loc) = expr;626 Ok(match &**expr {627 Literal(LiteralType::This) => Val::Obj(628 context629 .this()630 .clone()631 .ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,632 ),633 Literal(LiteralType::Dollar) => Val::Obj(634 context635 .dollar()636 .clone()637 .ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,638 ),639 Literal(LiteralType::True) => Val::Bool(true),640 Literal(LiteralType::False) => Val::Bool(false),641 Literal(LiteralType::Null) => Val::Null,642 Parened(e) => evaluate(context, e)?,643 Str(v) => Val::Str(v.clone()),644 Num(v) => Val::Num(*v),645 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,646 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,647 Var(name) => push(loc, "var", || {648 Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?)649 })?,650 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {651 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;652 context653 .super_obj()654 .clone()655 .expect("no super found")656 .get_raw(&name, &context.this().clone().expect("no this found"))?657 .expect("value not found")658 }659 Index(value, index) => {660 match (661 evaluate(context.clone(), value)?.unwrap_if_lazy()?,662 evaluate(context, index)?,663 ) {664 (Val::Obj(v), Val::Str(s)) => {665 if let Some(v) = v.get(s.clone())? {666 v.unwrap_if_lazy()?667 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {668 Val::Intristic(n, s)669 } else {670 create_error_result(crate::Error::NoSuchField(s))?671 }672 }673 (Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(674 ValType::Obj,675 ValType::Str,676 n.value_type()?,677 ))?,678679 (Val::Arr(v), Val::Num(n)) => {680 if n.fract() > f64::EPSILON {681 create_error_result(crate::Error::FractionalIndex)?682 }683 v.get(n as usize)684 .ok_or_else(|| {685 create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))686 })?687 .clone()688 .unwrap_if_lazy()?689 }690 (Val::Arr(_), Val::Str(n)) => {691 create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?692 }693 (Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(694 ValType::Arr,695 ValType::Num,696 n.value_type()?,697 ))?,698699 (Val::Str(s), Val::Num(n)) => Val::Str(700 s.chars()701 .skip(n as usize)702 .take(1)703 .collect::<String>()704 .into(),705 ),706 (Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(707 ValType::Str,708 ValType::Num,709 n.value_type()?,710 ))?,711712 (v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,713 }714 }715 LocalExpr(bindings, returned) => {716 let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();717 let future_context = Context::new_future();718719 let context_creator = context_creator!(720 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))721 );722723 for (k, v) in bindings724 .iter()725 .map(|b| evaluate_binding(b, context_creator.clone()))726 {727 new_bindings.insert(k, v);728 }729730 let context = context731 .extend_unbound(new_bindings, None, None, None)?732 .into_future(future_context);733 evaluate(context, &returned.clone())?734 }735 Arr(items) => {736 let mut out = Vec::with_capacity(items.len());737 for item in items {738 out.push(Val::Lazy(lazy_val!(739 closure!(clone context, clone item, || {740 evaluate(context.clone(), &item)741 })742 )));743 }744 Val::Arr(Rc::new(out))745 }746 ArrComp(expr, compspecs) => Val::Arr(747 // First compspec should be forspec, so no "None" possible here748 Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),749 ),750 Obj(body) => Val::Obj(evaluate_object(context, body)?),751 ObjExtend(s, t) => evaluate_add_op(752 &evaluate(context.clone(), s)?,753 &Val::Obj(evaluate_object(context, t)?),754 )?,755 Apply(value, args, tailstrict) => evaluate_apply(context, value, args, loc, *tailstrict)?,756 Function(params, body) => evaluate_method(context, params.clone(), body.clone()),757 AssertExpr(AssertStmt(value, msg), returned) => {758 let assertion_result = push(&value.1, "assertion condition", || {759 evaluate(context.clone(), &value)?760 .try_cast_bool("assertion condition should be boolean")761 })?;762 if assertion_result {763 evaluate(context, returned)?764 } else if let Some(msg) = msg {765 create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?766 } else {767 create_error_result(crate::Error::AssertionFailed(Val::Null))?768 }769 }770 Error(e) => create_error_result(crate::Error::RuntimeError(771 evaluate(context, e)?.try_cast_str("error text should be string")?,772 ))?,773 IfElse {774 cond,775 cond_then,776 cond_else,777 } => {778 if evaluate(context.clone(), &cond.0)?779 .try_cast_bool("if condition should be boolean")?780 {781 evaluate(context, cond_then)?782 } else {783 match cond_else {784 Some(v) => evaluate(context, v)?,785 None => Val::Null,786 }787 }788 }789 Import(path) => {790 let mut tmp = loc791 .clone()792 .expect("imports can't be used without loc_data")793 .0;794 let import_location = Rc::make_mut(&mut tmp);795 import_location.pop();796 with_state(|s| s.import_file(&import_location, path))?797 }798 ImportStr(path) => {799 let mut tmp = loc800 .clone()801 .expect("imports can't be used without loc_data")802 .0;803 let import_location = Rc::make_mut(&mut tmp);804 import_location.pop();805 Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)806 }807 Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),808 })809}crates/jrsonnet-evaluator/src/lib.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/lib.rs
+++ b/crates/jrsonnet-evaluator/src/lib.rs
@@ -2,6 +2,7 @@
#![feature(type_alias_impl_trait)]
#![feature(debug_non_exhaustive)]
#![feature(test)]
+#![feature(stmt_expr_attributes)]
#![allow(macro_expanded_macro_exports_accessed_by_absolute_paths)]
extern crate test;