difftreelog
fix(evaluator) do not panic in bad operations
in: master
2 files changed
crates/jsonnet-evaluator/src/error.rsdiffbeforeafterboth--- a/crates/jsonnet-evaluator/src/error.rs
+++ b/crates/jsonnet-evaluator/src/error.rs
@@ -1,5 +1,5 @@
use crate::ValType;
-use jsonnet_parser::ExprLocation;
+use jsonnet_parser::{BinaryOpType, ExprLocation, UnaryOpType};
use std::{path::PathBuf, rc::Rc};
#[derive(Debug, Clone)]
@@ -7,6 +7,9 @@
IntristicNotFound(Rc<str>, Rc<str>),
IntristicArgumentReorderingIsNotSupportedYet,
+ UnaryOperatorDoesNotOperateOnType(UnaryOpType, ValType),
+ BinaryOperatorDoesNotOperateOnValues(BinaryOpType, ValType, ValType),
+
VariableIsNotDefined(String),
TypeMismatch(&'static str, Vec<ValType>, ValType),
NoSuchField(Rc<str>),
crates/jsonnet-evaluator/src/evaluate.rsdiffbeforeafterboth1use crate::{2 context_creator, create_error, escape_string_json, future_wrapper, lazy_val, manifest_json_ex,3 parse_args, push, with_state, Context, ContextCreator, Error, FuncDesc, LazyBinding, LazyVal,4 ObjMember, ObjValue, Result, Val, ValType,5};6use closure::closure;7use jsonnet_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: &jsonnet_parser::FieldName,56) -> Result<Option<Rc<str>>> {57 Ok(match field_name {58 jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),59 jsonnet_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) => panic!("unary op not implemented: {:?} {:?}", op, o),78 })79}8081pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {82 Ok(match (a, b) {83 (Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),8485 // Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)86 (Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),87 (Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),8889 (Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),90 (o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9192 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),93 (Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),94 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),95 _ => panic!("can't add: {:?} and {:?}", a, b),96 })97}9899pub fn evaluate_binary_op_special(100 context: Context,101 a: &LocExpr,102 op: BinaryOpType,103 b: &LocExpr,104) -> Result<Val> {105 Ok(106 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {107 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),108 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),109 (a, op, eb) => {110 evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?111 }112 },113 )114}115116pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {117 Ok(match (a, op, b) {118 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,119120 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),121122 // Bool X Bool123 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),124 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),125126 // Str X Str127 (Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),128 (Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),129 (Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),130 (Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),131132 // Num X Num133 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),134 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {135 if *v2 <= f64::EPSILON {136 create_error(crate::Error::DivisionByZero)?137 }138 Val::Num(v1 / v2)139 }140141 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),142143 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),144 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),145 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),146 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),147148 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {149 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)150 }151 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {152 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)153 }154 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {155 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)156 }157 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {158 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)159 }160 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {161 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)162 }163164 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),165 })166}167168future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);169future_wrapper!(ObjValue, FutureObjValue);170171pub 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.iter() {190 let item = item.unwrap_if_lazy()?;191 out.push(evaluate_comp(192 context.with_var(var.clone(), item.clone())?,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}204205pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {206 let new_bindings = FutureNewBindings::new();207 let future_this = FutureObjValue::new();208 let context_creator = context_creator!(209 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {210 Ok(context.extend_unbound(211 new_bindings.clone().unwrap(),212 context.dollar().clone().or_else(||this.clone()),213 Some(this.unwrap()),214 super_obj215 )?)216 })217 );218 {219 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();220 for (n, b) in members221 .iter()222 .filter_map(|m| match m {223 Member::BindStmt(b) => Some(b.clone()),224 _ => None,225 })226 .map(|b| evaluate_binding(&b, context_creator.clone()))227 {228 bindings.insert(n, b);229 }230 new_bindings.fill(bindings);231 }232233 let mut new_members = BTreeMap::new();234 for member in members.iter() {235 match member {236 Member::Field(FieldMember {237 name,238 plus,239 params: None,240 visibility,241 value,242 }) => {243 let name = evaluate_field_name(context.clone(), &name)?;244 if name.is_none() {245 continue;246 }247 let name = name.unwrap();248 new_members.insert(249 name.clone(),250 ObjMember {251 add: *plus,252 visibility: *visibility,253 invoke: LazyBinding::Bindable(Rc::new(254 closure!(clone name, clone value, clone context_creator, |this, super_obj| {255 Ok(LazyVal::new_resolved(push(&value.1, "object field", ||{256 let context = context_creator.0(this, super_obj)?;257 evaluate(258 context,259 &value,260 )261 })?))262 }),263 )),264 },265 );266 }267 Member::Field(FieldMember {268 name,269 params: Some(params),270 value,271 ..272 }) => {273 let name = evaluate_field_name(context.clone(), &name)?;274 if name.is_none() {275 continue;276 }277 let name = name.unwrap();278 new_members.insert(279 name,280 ObjMember {281 add: false,282 visibility: Visibility::Hidden,283 invoke: LazyBinding::Bindable(Rc::new(284 closure!(clone value, clone context_creator, clone params, |this, super_obj| {285 // TODO: Assert286 Ok(LazyVal::new_resolved(evaluate_method(287 context_creator.0(this, super_obj)?,288 params.clone(),289 value.clone(),290 )))291 }),292 )),293 },294 );295 }296 Member::BindStmt(_) => {}297 Member::AssertStmt(_) => {}298 }299 }300 Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))301}302303pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {304 Ok(match object {305 ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,306 ObjBody::ObjComp(obj) => {307 let future_this = FutureObjValue::new();308 let mut new_members = BTreeMap::new();309 for (k, v) in evaluate_comp(310 context.clone(),311 &|ctx| {312 let new_bindings = FutureNewBindings::new();313 let context_creator = context_creator!(314 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {315 Ok(context.extend_unbound(316 new_bindings.clone().unwrap(),317 context.dollar().clone().or_else(||this.clone()),318 None,319 super_obj320 )?)321 })322 );323 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();324 for (n, b) in obj325 .pre_locals326 .iter()327 .chain(obj.post_locals.iter())328 .map(|b| evaluate_binding(b, context_creator.clone()))329 {330 bindings.insert(n, b);331 }332 let bindings = new_bindings.fill(bindings);333 let ctx = ctx.extend_unbound(bindings, None, None, None)?;334 let key = evaluate(ctx.clone(), &obj.key)?;335 let value = LazyBinding::Bindable(Rc::new(336 closure!(clone ctx, clone obj.value, |this, _super_obj| {337 Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))338 }),339 ));340341 Ok((key, value))342 },343 &obj.compspecs,344 )?345 .unwrap()346 {347 match k {348 Val::Null => {}349 Val::Str(n) => {350 new_members.insert(351 n,352 ObjMember {353 add: false,354 visibility: Visibility::Normal,355 invoke: v,356 },357 );358 }359 v => create_error(Error::FieldMustBeStringGot(v.value_type()?))?,360 }361 }362363 future_this.fill(ObjValue::new(None, Rc::new(new_members)))364 }365 })366}367368pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {369 use Expr::*;370 let LocExpr(expr, loc) = expr;371 Ok(match &**expr {372 Literal(LiteralType::This) => Val::Obj(373 context374 .this()375 .clone()376 .unwrap_or_else(|| panic!("this not found")),377 ),378 Literal(LiteralType::Dollar) => Val::Obj(379 context380 .dollar()381 .clone()382 .unwrap_or_else(|| panic!("dollar not found")),383 ),384 Literal(LiteralType::True) => Val::Bool(true),385 Literal(LiteralType::False) => Val::Bool(false),386 Literal(LiteralType::Null) => Val::Null,387 Parened(e) => evaluate(context, e)?,388 Str(v) => Val::Str(v.clone()),389 Num(v) => Val::Num(*v),390 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,391 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,392 Var(name) => push(loc, "var", || {393 Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?)394 })?,395 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {396 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;397 context398 .super_obj()399 .clone()400 .expect("no super found")401 .get_raw(&name, &context.this().clone().expect("no this found"))?402 .expect("value not found")403 }404 Index(value, index) => {405 match (406 evaluate(context.clone(), value)?.unwrap_if_lazy()?,407 evaluate(context, index)?,408 ) {409 (Val::Obj(v), Val::Str(s)) => {410 if let Some(v) = v.get(s.clone())? {411 v.unwrap_if_lazy()?412 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {413 Val::Intristic(n, s)414 } else {415 create_error(crate::Error::NoSuchField(s))?416 }417 }418 (Val::Obj(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(419 ValType::Obj,420 ValType::Str,421 n.value_type()?,422 ))?,423424 (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::Arr(_), Val::Str(n)) => {434 create_error(crate::Error::AttemptedIndexAnArrayWithString(n))?435 }436 (Val::Arr(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(437 ValType::Arr,438 ValType::Num,439 n.value_type()?,440 ))?,441442 (Val::Str(s), Val::Num(n)) => Val::Str(443 s.chars()444 .skip(n as usize)445 .take(1)446 .collect::<String>()447 .into(),448 ),449 (Val::Str(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(450 ValType::Str,451 ValType::Num,452 n.value_type()?,453 ))?,454455 (v, _) => create_error(crate::Error::CantIndexInto(v.value_type()?))?,456 }457 }458 LocalExpr(bindings, returned) => {459 let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();460 let future_context = Context::new_future();461462 let context_creator = context_creator!(463 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))464 );465466 for (k, v) in bindings467 .iter()468 .map(|b| evaluate_binding(b, context_creator.clone()))469 {470 new_bindings.insert(k, v);471 }472473 let context = context474 .extend_unbound(new_bindings, None, None, None)?475 .into_future(future_context);476 evaluate(context, &returned.clone())?477 }478 Arr(items) => {479 let mut out = Vec::with_capacity(items.len());480 for item in items {481 out.push(Val::Lazy(lazy_val!(482 closure!(clone context, clone item, || {483 evaluate(context.clone(), &item)484 })485 )));486 }487 Val::Arr(Rc::new(out))488 }489 ArrComp(expr, compspecs) => Val::Arr(490 // First compspec should be forspec, so no "None" possible here491 Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),492 ),493 Obj(body) => Val::Obj(evaluate_object(context, body)?),494 ObjExtend(s, t) => evaluate_add_op(495 &evaluate(context.clone(), s)?,496 &Val::Obj(evaluate_object(context, t)?),497 )?,498 Apply(value, args, tailstrict) => {499 let lazy = evaluate(context.clone(), value)?;500 let value = lazy.unwrap_if_lazy()?;501 match value {502 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {503 // arr/string/function504 ("std", "length") => parse_args!(context, "std.length", args, 1, [505 0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];506 ], {507 match x {508 Val::Str(n) => Val::Num(n.chars().count() as f64),509 Val::Arr(i) => Val::Num(i.len() as f64),510 Val::Obj(o) => Val::Num(511 o.fields_visibility()512 .into_iter()513 .filter(|(_k, v)| *v)514 .count() as f64,515 ),516 _ => unreachable!(),517 }518 }),519 // any520 ("std", "type") => parse_args!(context, "std.type", args, 1, [521 0, x, vec![];522 ], {523 Val::Str(x.value_type()?.name().into())524 }),525 // length, idx=>any526 ("std", "makeArray") => parse_args!(context, "std.makeArray", args, 2, [527 0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];528 1, func: [Val::Func]!!Val::Func, vec![ValType::Func];529 ], {530 assert!(sz >= 0.0);531 let mut out = Vec::with_capacity(sz as usize);532 for i in 0..sz as usize {533 out.push(func.evaluate_values(534 Context::new(),535 &[Val::Num(i as f64)]536 )?)537 }538 Val::Arr(Rc::new(out))539 }),540 // string541 ("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [542 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];543 ], {544 assert!(545 str.chars().count() == 1,546 "std.codepoint should receive single char string"547 );548 Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)549 }),550 // object, includeHidden551 ("std", "objectFieldsEx") => {552 parse_args!(context, "std.objectFieldsEx",args, 2, [553 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];554 1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];555 ], {556 Val::Arr(Rc::new(557 obj.fields_visibility()558 .into_iter()559 .filter(|(_k, v)| *v || inc_hidden)560 .map(|(k, _v)| Val::Str(k))561 .collect(),562 ))563 })564 }565 // object, field, includeHidden566 ("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [567 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];568 1, f: [Val::Str]!!Val::Str, vec![ValType::Str];569 2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];570 ], {571 Val::Bool(572 obj.fields_visibility()573 .into_iter()574 .filter(|(_k, v)| *v || inc_hidden)575 .any(|(k, _v)| *k == *f),576 )577 }),578 ("std", "primitiveEquals") => {579 parse_args!(context, "std.primitiveEquals", args, 2, [580 0, a, vec![];581 1, b, vec![];582 ], {583 Val::Bool(a == b)584 })585 }586 ("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [587 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];588 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];589 ], {590 Val::Num(a % b)591 }),592 ("std", "floor") => parse_args!(context, "std.floor", args, 1, [593 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];594 ], {595 Val::Num(x.floor())596 }),597 ("std", "trace") => parse_args!(context, "std.trace", args, 2, [598 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];599 1, rest, vec![];600 ], {601 // TODO: Line numbers as in original jsonnet602 println!("TRACE: {}", str);603 rest604 }),605 ("std", "pow") => parse_args!(context, "std.modulo", args, 2, [606 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];607 1, n: [Val::Num]!!Val::Num, vec![ValType::Num];608 ], {609 Val::Num(x.powf(n))610 }),611 ("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [612 0, x: [Val::Str]!!Val::Str, vec![ValType::Str];613 ], {614 with_state(|s| s.0.ext_vars.borrow().get(&x).cloned()).ok_or_else(615 || {616 create_error::<()>(crate::Error::UndefinedExternalVariable(x))617 .err()618 .unwrap()619 },620 )?621 }),622 ("std", "filter") => parse_args!(context, "std.filter", args, 2, [623 0, func: [Val::Func]!!Val::Func, vec![ValType::Func];624 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];625 ], {626 Val::Arr(Rc::new(627 arr.iter()628 .cloned()629 .filter(|e| {630 func631 .evaluate_values(context.clone(), &[e.clone()])632 .unwrap()633 .try_cast_bool("filter predicate")634 .unwrap()635 })636 .collect(),637 ))638 }),639 ("std", "char") => parse_args!(context, "std.char", args, 1, [640 0, n: [Val::Num]!!Val::Num, vec![ValType::Num];641 ], {642 let mut out = String::new();643 out.push(std::char::from_u32(n as u32).unwrap());644 Val::Str(out.into())645 }),646 ("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [647 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];648 ], {649 Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))650 }),651 ("std", "md5") => parse_args!(context, "std.md5", args, 1, [652 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];653 ], {654 Val::Str(format!("{:x}", md5::compute(str.as_bytes())).into())655 }),656 // faster657 ("std", "join") => parse_args!(context, "std.join", args, 2, [658 0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];659 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];660 ], {661 match sep {662 Val::Arr(joiner_items) => {663 let mut out = Vec::new();664665 let mut first = true;666 for item in arr.iter().cloned() {667 if let Val::Arr(items) = item.unwrap_if_lazy()? {668 if !first {669 out.reserve(joiner_items.len());670 out.extend(joiner_items.iter().cloned());671 }672 first = false;673 out.reserve(items.len());674 out.extend(items.iter().cloned());675 } else {676 panic!("all array items should be arrays")677 }678 }679680 Val::Arr(Rc::new(out))681 },682 Val::Str(sep) => {683 let mut out = String::new();684685 let mut first = true;686 for item in arr.iter().cloned() {687 if let Val::Str(item) = item.unwrap_if_lazy()? {688 if !first {689 out += &sep;690 }691 first = false;692 out += &item;693 } else {694 panic!("all array items should be strings")695 }696 }697698 Val::Str(out.into())699 },700 _ => unreachable!()701 }702 }),703 // Faster704 ("std", "escapeStringJson") => {705 parse_args!(context, "std.escapeStringJson", args, 1, [706 0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];707 ], {708 Val::Str(escape_string_json(&str_).into())709 })710 }711 // Faster712 ("std", "manifestJsonEx") => {713 parse_args!(context, "std.manifestJsonEx", args, 2, [714 0, value, vec![];715 1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];716 ], {717 Val::Str(manifest_json_ex(&value, &indent)?.into())718 })719 }720 (ns, name) => create_error(crate::error::Error::IntristicNotFound(721 ns.into(),722 name.into(),723 ))?,724 },725 Val::Func(f) => {726 let body = || f.evaluate(context, args, *tailstrict);727 if *tailstrict {728 body()?729 } else {730 push(loc, "function call", body)?731 }732 }733 v => create_error(crate::error::Error::OnlyFunctionsCanBeCalledGot(734 v.value_type()?,735 ))?,736 }737 }738 Function(params, body) => evaluate_method(context, params.clone(), body.clone()),739 AssertExpr(AssertStmt(value, msg), returned) => {740 let assertion_result = push(&value.1, "assertion condition", || {741 evaluate(context.clone(), &value)?742 .try_cast_bool("assertion condition should be boolean")743 })?;744 if assertion_result {745 evaluate(context, returned)?746 } else if let Some(msg) = msg {747 panic!(748 "assertion failed ({:?}): {}",749 value,750 evaluate(context, msg)?.try_cast_str("assertion message should be string")?751 );752 } else {753 panic!("assertion failed ({:?}): no message", value);754 }755 }756 Error(e) => create_error(crate::Error::RuntimeError(757 evaluate(context, e)?.try_cast_str("error text should be string")?,758 ))?,759 IfElse {760 cond,761 cond_then,762 cond_else,763 } => {764 if evaluate(context.clone(), &cond.0)?765 .try_cast_bool("if condition should be boolean")?766 {767 evaluate(context, cond_then)?768 } else {769 match cond_else {770 Some(v) => evaluate(context, v)?,771 None => Val::Null,772 }773 }774 }775 Import(path) => {776 let mut tmp = loc777 .clone()778 .expect("imports can't be used without loc_data")779 .0;780 let import_location = Rc::make_mut(&mut tmp);781 import_location.pop();782 with_state(|s| s.import_file(&import_location, path))?783 }784 ImportStr(path) => {785 let mut tmp = loc786 .clone()787 .expect("imports can't be used without loc_data")788 .0;789 let import_location = Rc::make_mut(&mut tmp);790 import_location.pop();791 Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)792 }793 Literal(LiteralType::Super) => return create_error(crate::error::Error::StandaloneSuper),794 })795}1use crate::{2 context_creator, create_error, escape_string_json, future_wrapper, lazy_val, manifest_json_ex,3 parse_args, push, with_state, Context, ContextCreator, Error, FuncDesc, LazyBinding, LazyVal,4 ObjMember, ObjValue, Result, Val, ValType,5};6use closure::closure;7use jsonnet_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: &jsonnet_parser::FieldName,56) -> Result<Option<Rc<str>>> {57 Ok(match field_name {58 jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),59 jsonnet_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(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(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(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(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 _ => panic!("for expression evaluated to non-iterable value"),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(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 .unwrap_or_else(|| panic!("this not found")),388 ),389 Literal(LiteralType::Dollar) => Val::Obj(390 context391 .dollar()392 .clone()393 .unwrap_or_else(|| panic!("dollar not found")),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(crate::Error::NoSuchField(s))?427 }428 }429 (Val::Obj(_), n) => create_error(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(crate::Error::FractionalIndex)?438 }439 v.get(n as usize)440 .unwrap_or_else(|| panic!("out of bounds"))441 .clone()442 .unwrap_if_lazy()?443 }444 (Val::Arr(_), Val::Str(n)) => {445 create_error(crate::Error::AttemptedIndexAnArrayWithString(n))?446 }447 (Val::Arr(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(448 ValType::Arr,449 ValType::Num,450 n.value_type()?,451 ))?,452453 (Val::Str(s), Val::Num(n)) => Val::Str(454 s.chars()455 .skip(n as usize)456 .take(1)457 .collect::<String>()458 .into(),459 ),460 (Val::Str(_), n) => create_error(crate::Error::ValueIndexMustBeTypeGot(461 ValType::Str,462 ValType::Num,463 n.value_type()?,464 ))?,465466 (v, _) => create_error(crate::Error::CantIndexInto(v.value_type()?))?,467 }468 }469 LocalExpr(bindings, returned) => {470 let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();471 let future_context = Context::new_future();472473 let context_creator = context_creator!(474 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))475 );476477 for (k, v) in bindings478 .iter()479 .map(|b| evaluate_binding(b, context_creator.clone()))480 {481 new_bindings.insert(k, v);482 }483484 let context = context485 .extend_unbound(new_bindings, None, None, None)?486 .into_future(future_context);487 evaluate(context, &returned.clone())?488 }489 Arr(items) => {490 let mut out = Vec::with_capacity(items.len());491 for item in items {492 out.push(Val::Lazy(lazy_val!(493 closure!(clone context, clone item, || {494 evaluate(context.clone(), &item)495 })496 )));497 }498 Val::Arr(Rc::new(out))499 }500 ArrComp(expr, compspecs) => Val::Arr(501 // First compspec should be forspec, so no "None" possible here502 Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),503 ),504 Obj(body) => Val::Obj(evaluate_object(context, body)?),505 ObjExtend(s, t) => evaluate_add_op(506 &evaluate(context.clone(), s)?,507 &Val::Obj(evaluate_object(context, t)?),508 )?,509 Apply(value, args, tailstrict) => {510 let lazy = evaluate(context.clone(), value)?;511 let value = lazy.unwrap_if_lazy()?;512 match value {513 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {514 // arr/string/function515 ("std", "length") => parse_args!(context, "std.length", args, 1, [516 0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];517 ], {518 match x {519 Val::Str(n) => Val::Num(n.chars().count() as f64),520 Val::Arr(i) => Val::Num(i.len() as f64),521 Val::Obj(o) => Val::Num(522 o.fields_visibility()523 .into_iter()524 .filter(|(_k, v)| *v)525 .count() as f64,526 ),527 _ => unreachable!(),528 }529 }),530 // any531 ("std", "type") => parse_args!(context, "std.type", args, 1, [532 0, x, vec![];533 ], {534 Val::Str(x.value_type()?.name().into())535 }),536 // length, idx=>any537 ("std", "makeArray") => parse_args!(context, "std.makeArray", args, 2, [538 0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];539 1, func: [Val::Func]!!Val::Func, vec![ValType::Func];540 ], {541 assert!(sz >= 0.0);542 let mut out = Vec::with_capacity(sz as usize);543 for i in 0..sz as usize {544 out.push(func.evaluate_values(545 Context::new(),546 &[Val::Num(i as f64)]547 )?)548 }549 Val::Arr(Rc::new(out))550 }),551 // string552 ("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [553 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];554 ], {555 assert!(556 str.chars().count() == 1,557 "std.codepoint should receive single char string"558 );559 Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)560 }),561 // object, includeHidden562 ("std", "objectFieldsEx") => {563 parse_args!(context, "std.objectFieldsEx",args, 2, [564 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];565 1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];566 ], {567 Val::Arr(Rc::new(568 obj.fields_visibility()569 .into_iter()570 .filter(|(_k, v)| *v || inc_hidden)571 .map(|(k, _v)| Val::Str(k))572 .collect(),573 ))574 })575 }576 // object, field, includeHidden577 ("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [578 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];579 1, f: [Val::Str]!!Val::Str, vec![ValType::Str];580 2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];581 ], {582 Val::Bool(583 obj.fields_visibility()584 .into_iter()585 .filter(|(_k, v)| *v || inc_hidden)586 .any(|(k, _v)| *k == *f),587 )588 }),589 ("std", "primitiveEquals") => {590 parse_args!(context, "std.primitiveEquals", args, 2, [591 0, a, vec![];592 1, b, vec![];593 ], {594 Val::Bool(a == b)595 })596 }597 ("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [598 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];599 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];600 ], {601 Val::Num(a % b)602 }),603 ("std", "floor") => parse_args!(context, "std.floor", args, 1, [604 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];605 ], {606 Val::Num(x.floor())607 }),608 ("std", "trace") => parse_args!(context, "std.trace", args, 2, [609 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];610 1, rest, vec![];611 ], {612 // TODO: Line numbers as in original jsonnet613 println!("TRACE: {}", str);614 rest615 }),616 ("std", "pow") => parse_args!(context, "std.modulo", args, 2, [617 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];618 1, n: [Val::Num]!!Val::Num, vec![ValType::Num];619 ], {620 Val::Num(x.powf(n))621 }),622 ("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [623 0, x: [Val::Str]!!Val::Str, vec![ValType::Str];624 ], {625 with_state(|s| s.0.ext_vars.borrow().get(&x).cloned()).ok_or_else(626 || {627 create_error::<()>(crate::Error::UndefinedExternalVariable(x))628 .err()629 .unwrap()630 },631 )?632 }),633 ("std", "filter") => parse_args!(context, "std.filter", args, 2, [634 0, func: [Val::Func]!!Val::Func, vec![ValType::Func];635 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];636 ], {637 Val::Arr(Rc::new(638 arr.iter()639 .cloned()640 .filter(|e| {641 func642 .evaluate_values(context.clone(), &[e.clone()])643 .unwrap()644 .try_cast_bool("filter predicate")645 .unwrap()646 })647 .collect(),648 ))649 }),650 ("std", "char") => parse_args!(context, "std.char", args, 1, [651 0, n: [Val::Num]!!Val::Num, vec![ValType::Num];652 ], {653 let mut out = String::new();654 out.push(std::char::from_u32(n as u32).unwrap());655 Val::Str(out.into())656 }),657 ("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [658 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];659 ], {660 Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))661 }),662 ("std", "md5") => parse_args!(context, "std.md5", args, 1, [663 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];664 ], {665 Val::Str(format!("{:x}", md5::compute(str.as_bytes())).into())666 }),667 // faster668 ("std", "join") => parse_args!(context, "std.join", args, 2, [669 0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];670 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];671 ], {672 match sep {673 Val::Arr(joiner_items) => {674 let mut out = Vec::new();675676 let mut first = true;677 for item in arr.iter().cloned() {678 if let Val::Arr(items) = item.unwrap_if_lazy()? {679 if !first {680 out.reserve(joiner_items.len());681 out.extend(joiner_items.iter().cloned());682 }683 first = false;684 out.reserve(items.len());685 out.extend(items.iter().cloned());686 } else {687 panic!("all array items should be arrays")688 }689 }690691 Val::Arr(Rc::new(out))692 },693 Val::Str(sep) => {694 let mut out = String::new();695696 let mut first = true;697 for item in arr.iter().cloned() {698 if let Val::Str(item) = item.unwrap_if_lazy()? {699 if !first {700 out += &sep;701 }702 first = false;703 out += &item;704 } else {705 panic!("all array items should be strings")706 }707 }708709 Val::Str(out.into())710 },711 _ => unreachable!()712 }713 }),714 // Faster715 ("std", "escapeStringJson") => {716 parse_args!(context, "std.escapeStringJson", args, 1, [717 0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];718 ], {719 Val::Str(escape_string_json(&str_).into())720 })721 }722 // Faster723 ("std", "manifestJsonEx") => {724 parse_args!(context, "std.manifestJsonEx", args, 2, [725 0, value, vec![];726 1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];727 ], {728 Val::Str(manifest_json_ex(&value, &indent)?.into())729 })730 }731 (ns, name) => create_error(crate::error::Error::IntristicNotFound(732 ns.into(),733 name.into(),734 ))?,735 },736 Val::Func(f) => {737 let body = || f.evaluate(context, args, *tailstrict);738 if *tailstrict {739 body()?740 } else {741 push(loc, "function call", body)?742 }743 }744 v => create_error(crate::error::Error::OnlyFunctionsCanBeCalledGot(745 v.value_type()?,746 ))?,747 }748 }749 Function(params, body) => evaluate_method(context, params.clone(), body.clone()),750 AssertExpr(AssertStmt(value, msg), returned) => {751 let assertion_result = push(&value.1, "assertion condition", || {752 evaluate(context.clone(), &value)?753 .try_cast_bool("assertion condition should be boolean")754 })?;755 if assertion_result {756 evaluate(context, returned)?757 } else if let Some(msg) = msg {758 panic!(759 "assertion failed ({:?}): {}",760 value,761 evaluate(context, msg)?.try_cast_str("assertion message should be string")?762 );763 } else {764 panic!("assertion failed ({:?}): no message", value);765 }766 }767 Error(e) => create_error(crate::Error::RuntimeError(768 evaluate(context, e)?.try_cast_str("error text should be string")?,769 ))?,770 IfElse {771 cond,772 cond_then,773 cond_else,774 } => {775 if evaluate(context.clone(), &cond.0)?776 .try_cast_bool("if condition should be boolean")?777 {778 evaluate(context, cond_then)?779 } else {780 match cond_else {781 Some(v) => evaluate(context, v)?,782 None => Val::Null,783 }784 }785 }786 Import(path) => {787 let mut tmp = loc788 .clone()789 .expect("imports can't be used without loc_data")790 .0;791 let import_location = Rc::make_mut(&mut tmp);792 import_location.pop();793 with_state(|s| s.import_file(&import_location, path))?794 }795 ImportStr(path) => {796 let mut tmp = loc797 .clone()798 .expect("imports can't be used without loc_data")799 .0;800 let import_location = Rc::make_mut(&mut tmp);801 import_location.pop();802 Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)803 }804 Literal(LiteralType::Super) => return create_error(crate::error::Error::StandaloneSuper),805 })806}