difftreelog
feat trace function names
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 ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, ExprLocation, FieldMember,9 ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,10 Visibility,11};12use std::{collections::HashMap, rc::Rc};1314pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (Rc<str>, LazyBinding) {15 let b = b.clone();16 if let Some(params) = &b.params {17 let params = params.clone();18 (19 b.name.clone(),20 LazyBinding::Bindable(Rc::new(move |this, super_obj| {21 Ok(lazy_val!(22 closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(23 context_creator.0(this.clone(), super_obj.clone())?,24 params.clone(),25 b.value.clone(),26 )))27 ))28 })),29 )30 } else {31 (32 b.name.clone(),33 LazyBinding::Bindable(Rc::new(move |this, super_obj| {34 Ok(lazy_val!(closure!(clone context_creator, clone b, ||35 push(&b.value.1, "thunk", ||{36 evaluate(37 context_creator.0(this.clone(), super_obj.clone())?,38 &b.value39 )40 })41 )))42 })),43 )44 }45}4647pub fn evaluate_method(ctx: Context, params: ParamsDesc, body: LocExpr) -> Val {48 Val::Func(FuncDesc { ctx, params, body })49}5051pub fn evaluate_field_name(52 context: Context,53 field_name: &jrsonnet_parser::FieldName,54) -> Result<Option<Rc<str>>> {55 Ok(match field_name {56 jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),57 jrsonnet_parser::FieldName::Dyn(expr) => {58 let lazy = evaluate(context, expr)?;59 let value = lazy.unwrap_if_lazy()?;60 if matches!(value, Val::Null) {61 None62 } else {63 Some(value.try_cast_str("dynamic field name")?)64 }65 }66 })67}6869pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {70 Ok(match (op, b) {71 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,72 (UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),73 (UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),74 (UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),75 (op, o) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(76 op,77 o.value_type()?,78 ))?,79 })80}8182pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {83 Ok(match (a, b) {84 (Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),8586 // Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)87 (Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),88 (Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),8990 (Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),91 (o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9293 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),94 (Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),95 (Val::Num(v1), Val::Num(v2)) => Val::new_checked_num(v1 + v2)?,96 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(97 BinaryOpType::Add,98 a.value_type()?,99 b.value_type()?,100 ))?,101 })102}103104pub fn evaluate_binary_op_special(105 context: Context,106 a: &LocExpr,107 op: BinaryOpType,108 b: &LocExpr,109) -> Result<Val> {110 Ok(111 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {112 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),113 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),114 (a, op, eb) => {115 evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?116 }117 },118 )119}120121pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {122 Ok(match (a, op, b) {123 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,124125 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),126127 // Bool X Bool128 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),129 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),130131 // Str X Str132 (Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),133 (Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),134 (Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),135 (Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),136137 // Num X Num138 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::new_checked_num(v1 * v2)?,139 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {140 if *v2 <= f64::EPSILON {141 create_error_result(crate::Error::DivisionByZero)?142 }143 Val::new_checked_num(v1 / v2)?144 }145146 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::new_checked_num(v1 - v2)?,147148 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),149 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),150 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),151 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),152153 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {154 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)155 }156 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {157 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)158 }159 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {160 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)161 }162 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {163 if *v2 < 0.0 {164 create_error_result(Error::RuntimeError("shift by negative exponent".into()))?165 }166 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)167 }168 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {169 if *v2 < 0.0 {170 create_error_result(Error::RuntimeError("shift by negative exponent".into()))?171 }172 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)173 }174175 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(176 op,177 a.value_type()?,178 b.value_type()?,179 ))?,180 })181}182183future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);184future_wrapper!(ObjValue, FutureObjValue);185186pub fn evaluate_comp<T>(187 context: Context,188 value: &impl Fn(Context) -> Result<T>,189 specs: &[CompSpec],190) -> Result<Option<Vec<T>>> {191 Ok(match specs.get(0) {192 None => Some(vec![value(context)?]),193 Some(CompSpec::IfSpec(IfSpecData(cond))) => {194 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {195 evaluate_comp(context, value, &specs[1..])?196 } else {197 None198 }199 }200 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {201 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {202 Val::Arr(list) => {203 let mut out = Vec::new();204 for item in list.iter() {205 let item = item.unwrap_if_lazy()?;206 out.push(evaluate_comp(207 context.with_var(var.clone(), item.clone())?,208 value,209 &specs[1..],210 )?);211 }212 Some(out.into_iter().flatten().flatten().collect())213 }214 _ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,215 }216 }217 })218}219220pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {221 let new_bindings = FutureNewBindings::new();222 let future_this = FutureObjValue::new();223 let context_creator = context_creator!(224 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {225 Ok(context.extend_unbound(226 new_bindings.clone().unwrap(),227 context.dollar().clone().or_else(||this.clone()),228 Some(this.unwrap()),229 super_obj230 )?)231 })232 );233 {234 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();235 for (n, b) in members236 .iter()237 .filter_map(|m| match m {238 Member::BindStmt(b) => Some(b.clone()),239 _ => None,240 })241 .map(|b| evaluate_binding(&b, context_creator.clone()))242 {243 bindings.insert(n, b);244 }245 new_bindings.fill(bindings);246 }247248 let mut new_members = HashMap::new();249 for member in members.iter() {250 match member {251 Member::Field(FieldMember {252 name,253 plus,254 params: None,255 visibility,256 value,257 }) => {258 let name = evaluate_field_name(context.clone(), &name)?;259 if name.is_none() {260 continue;261 }262 let name = name.unwrap();263 new_members.insert(264 name.clone(),265 ObjMember {266 add: *plus,267 visibility: *visibility,268 invoke: LazyBinding::Bindable(Rc::new(269 closure!(clone name, clone value, clone context_creator, |this, super_obj| {270 Ok(LazyVal::new_resolved(evaluate(271 context_creator.0(this, super_obj)?,272 &value,273 )?))274 }),275 )),276 location: value.1.clone(),277 },278 );279 }280 Member::Field(FieldMember {281 name,282 params: Some(params),283 value,284 ..285 }) => {286 let name = evaluate_field_name(context.clone(), &name)?;287 if name.is_none() {288 continue;289 }290 let name = name.unwrap();291 new_members.insert(292 name,293 ObjMember {294 add: false,295 visibility: Visibility::Hidden,296 invoke: LazyBinding::Bindable(Rc::new(297 closure!(clone value, clone context_creator, clone params, |this, super_obj| {298 // TODO: Assert299 Ok(LazyVal::new_resolved(evaluate_method(300 context_creator.0(this, super_obj)?,301 params.clone(),302 value.clone(),303 )))304 }),305 )),306 location: value.1.clone(),307 },308 );309 }310 Member::BindStmt(_) => {}311 Member::AssertStmt(_) => {}312 }313 }314 Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))315}316317pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {318 Ok(match object {319 ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,320 ObjBody::ObjComp(obj) => {321 let future_this = FutureObjValue::new();322 let mut new_members = HashMap::new();323 for (k, v) in evaluate_comp(324 context.clone(),325 &|ctx| {326 let new_bindings = FutureNewBindings::new();327 let context_creator = context_creator!(328 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {329 Ok(context.extend_unbound(330 new_bindings.clone().unwrap(),331 context.dollar().clone().or_else(||this.clone()),332 None,333 super_obj334 )?)335 })336 );337 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();338 for (n, b) in obj339 .pre_locals340 .iter()341 .chain(obj.post_locals.iter())342 .map(|b| evaluate_binding(b, context_creator.clone()))343 {344 bindings.insert(n, b);345 }346 let bindings = new_bindings.fill(bindings);347 let ctx = ctx.extend_unbound(bindings, None, None, None)?;348 let key = evaluate(ctx.clone(), &obj.key)?;349 let value = LazyBinding::Bindable(Rc::new(350 closure!(clone ctx, clone obj.value, |this, _super_obj| {351 Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))352 }),353 ));354355 Ok((key, value))356 },357 &obj.compspecs,358 )?359 .unwrap()360 {361 match k {362 Val::Null => {}363 Val::Str(n) => {364 new_members.insert(365 n,366 ObjMember {367 add: false,368 visibility: Visibility::Normal,369 invoke: v,370 location: obj.value.1.clone(),371 },372 );373 }374 v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,375 }376 }377378 future_this.fill(ObjValue::new(None, Rc::new(new_members)))379 }380 })381}382383/// Extracts code block and disables inlining for them384/// Fixes WASM to java bytecode compilation failing because of very large method385macro_rules! noinline {386 ($e:expr) => {387 (#[inline(never)]388 move || $e)()389 };390}391392pub fn evaluate_apply(393 context: Context,394 value: &LocExpr,395 args: &ArgsDesc,396 loc: &Option<ExprLocation>,397 tailstrict: bool,398) -> Result<Val> {399 let lazy = evaluate(context.clone(), value)?;400 let value = lazy.unwrap_if_lazy()?;401 Ok(match value {402 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {403 // arr/string/function404 ("std", "length") => noinline!(parse_args!(context, "std.length", args, 1, [405 0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];406 ], {407 Ok(match x {408 Val::Str(n) => Val::Num(n.chars().count() as f64),409 Val::Arr(i) => Val::Num(i.len() as f64),410 Val::Obj(o) => Val::Num(411 o.fields_visibility()412 .into_iter()413 .filter(|(_k, v)| *v)414 .count() as f64,415 ),416 _ => unreachable!(),417 })418 }))?,419 // any420 ("std", "type") => parse_args!(context, "std.type", args, 1, [421 0, x, vec![];422 ], {423 Val::Str(x.value_type()?.name().into())424 }),425 // length, idx=>any426 ("std", "makeArray") => noinline!(parse_args!(context, "std.makeArray", args, 2, [427 0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];428 1, func: [Val::Func]!!Val::Func, vec![ValType::Func];429 ], {430 if sz < 0.0 {431 create_error_result(crate::error::Error::RuntimeError(format!("makeArray requires size >= 0, got {}", sz).into()))?;432 }433 let mut out = Vec::with_capacity(sz as usize);434 for i in 0..sz as usize {435 out.push(func.evaluate_values(436 Context::new(),437 &[Val::Num(i as f64)]438 )?)439 }440 Ok(Val::Arr(Rc::new(out)))441 }))?,442 // string443 ("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [444 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];445 ], {446 assert!(447 str.chars().count() == 1,448 "std.codepoint should receive single char string"449 );450 Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)451 }),452 // object, includeHidden453 ("std", "objectFieldsEx") => {454 noinline!(parse_args!(context, "std.objectFieldsEx",args, 2, [455 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];456 1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];457 ], {458 let mut out = obj.fields_visibility()459 .into_iter()460 .filter(|(_k, v)| *v || inc_hidden)461 .map(|(k, _v)|k)462 .collect::<Vec<_>>();463 out.sort();464 Ok(Val::Arr(Rc::new(out.into_iter().map(Val::Str).collect())))465 }))?466 }467 // object, field, includeHidden468 ("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [469 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];470 1, f: [Val::Str]!!Val::Str, vec![ValType::Str];471 2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];472 ], {473 Val::Bool(474 obj.fields_visibility()475 .into_iter()476 .filter(|(_k, v)| *v || inc_hidden)477 .any(|(k, _v)| *k == *f),478 )479 }),480 ("std", "primitiveEquals") => parse_args!(context, "std.primitiveEquals", args, 2, [481 0, a, vec![];482 1, b, vec![];483 ], {484 Val::Bool(primitive_equals(&a, &b)?)485 }),486 // faster487 ("std", "equals") => parse_args!(context, "std.equals", args, 2, [488 0, a, vec![];489 1, b, vec![];490 ], {491 Val::Bool(equals(&a, &b)?)492 }),493 ("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [494 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];495 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];496 ], {497 Val::Num(a % b)498 }),499 ("std", "floor") => parse_args!(context, "std.floor", args, 1, [500 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];501 ], {502 Val::Num(x.floor())503 }),504 ("std", "trace") => parse_args!(context, "std.trace", args, 2, [505 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];506 1, rest, vec![];507 ], {508 eprint!("TRACE: ");509 if let Some(loc) = loc {510 with_state(|s|{511 let locs = s.map_source_locations(&loc.0, &[loc.1]);512 eprint!("{}:{} ", loc.0.display(), locs[0].line);513 });514 }515 eprintln!("{}", str);516 rest517 }),518 ("std", "pow") => parse_args!(context, "std.modulo", args, 2, [519 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];520 1, n: [Val::Num]!!Val::Num, vec![ValType::Num];521 ], {522 Val::Num(x.powf(n))523 }),524 ("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [525 0, x: [Val::Str]!!Val::Str, vec![ValType::Str];526 ], {527 with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(528 || create_error(crate::Error::UndefinedExternalVariable(x)),529 )?530 }),531 ("std", "filter") => noinline!(parse_args!(context, "std.filter", args, 2, [532 0, func: [Val::Func]!!Val::Func, vec![ValType::Func];533 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];534 ], {535 Ok(Val::Arr(Rc::new(536 arr.iter()537 .cloned()538 .filter(|e| {539 func540 .evaluate_values(context.clone(), &[e.clone()])541 .unwrap()542 .try_cast_bool("filter predicate")543 .unwrap()544 })545 .collect(),546 )))547 }))?,548 ("std", "char") => parse_args!(context, "std.char", args, 1, [549 0, n: [Val::Num]!!Val::Num, vec![ValType::Num];550 ], {551 let mut out = String::new();552 out.push(std::char::from_u32(n as u32).unwrap());553 Val::Str(out.into())554 }),555 ("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [556 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];557 ], {558 Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))559 }),560 ("std", "md5") => noinline!(parse_args!(context, "std.md5", args, 1, [561 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];562 ], {563 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))564 }))?,565 // faster566 ("std", "base64") => parse_args!(context, "std.base64", args, 1, [567 0, input: [Val::Str | Val::Arr], vec![ValType::Arr, ValType::Str];568 ], {569 Val::Str(match input {570 Val::Str(s) => {571 base64::encode(s.bytes().collect::<Vec<_>>()).into()572 },573 Val::Arr(a) => {574 base64::encode(a.iter().map(|v| {575 Ok(v.clone().try_cast_num("base64 array")? as u8)576 }).collect::<Result<Vec<_>>>()?).into()577 },578 _ => unreachable!()579 })580 }),581 // faster582 ("std", "join") => noinline!(parse_args!(context, "std.join", args, 2, [583 0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];584 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];585 ], {586 Ok(match sep {587 Val::Arr(joiner_items) => {588 let mut out = Vec::new();589590 let mut first = true;591 for item in arr.iter().cloned() {592 if let Val::Arr(items) = item.unwrap_if_lazy()? {593 if !first {594 out.reserve(joiner_items.len());595 out.extend(joiner_items.iter().cloned());596 }597 first = false;598 out.reserve(items.len());599 out.extend(items.iter().cloned());600 } else {601 create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;602 }603 }604605 Val::Arr(Rc::new(out))606 },607 Val::Str(sep) => {608 let mut out = String::new();609610 let mut first = true;611 for item in arr.iter().cloned() {612 if let Val::Str(item) = item.unwrap_if_lazy()? {613 if !first {614 out += &sep;615 }616 first = false;617 out += &item;618 } else {619 create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;620 }621 }622623 Val::Str(out.into())624 },625 _ => unreachable!()626 })627 }))?,628 // Faster629 ("std", "escapeStringJson") => parse_args!(context, "std.escapeStringJson", args, 1, [630 0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];631 ], {632 Val::Str(escape_string_json(&str_).into())633 }),634 // Faster635 ("std", "manifestJsonEx") => parse_args!(context, "std.manifestJsonEx", args, 2, [636 0, value, vec![];637 1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];638 ], {639 Val::Str(manifest_json_ex(&value, &indent)?.into())640 }),641 (ns, name) => {642 create_error_result(crate::Error::IntristicNotFound(ns.into(), name.into()))?643 }644 },645 Val::Func(f) => {646 let body = || f.evaluate(context, args, tailstrict);647 if tailstrict {648 body()?649 } else {650 push(loc, "function call", body)?651 }652 }653 v => create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?,654 })655}656657pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {658 use Expr::*;659 let LocExpr(expr, loc) = expr;660 Ok(match &**expr {661 Literal(LiteralType::This) => Val::Obj(662 context663 .this()664 .clone()665 .ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,666 ),667 Literal(LiteralType::Dollar) => Val::Obj(668 context669 .dollar()670 .clone()671 .ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,672 ),673 Literal(LiteralType::True) => Val::Bool(true),674 Literal(LiteralType::False) => Val::Bool(false),675 Literal(LiteralType::Null) => Val::Null,676 Parened(e) => evaluate(context, e)?,677 Str(v) => Val::Str(v.clone()),678 Num(v) => Val::new_checked_num(*v)?,679 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,680 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,681 Var(name) => push(682 loc,683 || "var".to_owned(),684 || Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?),685 )?,686 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {687 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;688 context689 .super_obj()690 .clone()691 .expect("no super found")692 .get_raw(&name, &context.this().clone().expect("no this found"))?693 .expect("value not found")694 }695 Index(value, index) => {696 match (697 evaluate(context.clone(), value)?.unwrap_if_lazy()?,698 evaluate(context, index)?,699 ) {700 (Val::Obj(v), Val::Str(s)) => {701 if let Some(v) = v.get(s.clone())? {702 v.unwrap_if_lazy()?703 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {704 Val::Intristic(n, s)705 } else {706 create_error_result(crate::Error::NoSuchField(s))?707 }708 }709 (Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(710 ValType::Obj,711 ValType::Str,712 n.value_type()?,713 ))?,714715 (Val::Arr(v), Val::Num(n)) => {716 if n.fract() > f64::EPSILON {717 create_error_result(crate::Error::FractionalIndex)?718 }719 v.get(n as usize)720 .ok_or_else(|| {721 create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))722 })?723 .clone()724 .unwrap_if_lazy()?725 }726 (Val::Arr(_), Val::Str(n)) => {727 create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?728 }729 (Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(730 ValType::Arr,731 ValType::Num,732 n.value_type()?,733 ))?,734735 (Val::Str(s), Val::Num(n)) => Val::Str(736 s.chars()737 .skip(n as usize)738 .take(1)739 .collect::<String>()740 .into(),741 ),742 (Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(743 ValType::Str,744 ValType::Num,745 n.value_type()?,746 ))?,747748 (v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,749 }750 }751 LocalExpr(bindings, returned) => {752 let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();753 let future_context = Context::new_future();754755 let context_creator = context_creator!(756 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))757 );758759 for (k, v) in bindings760 .iter()761 .map(|b| evaluate_binding(b, context_creator.clone()))762 {763 new_bindings.insert(k, v);764 }765766 let context = context767 .extend_unbound(new_bindings, None, None, None)?768 .into_future(future_context);769 evaluate(context, &returned.clone())?770 }771 Arr(items) => {772 let mut out = Vec::with_capacity(items.len());773 for item in items {774 out.push(Val::Lazy(lazy_val!(775 closure!(clone context, clone item, || {776 evaluate(context.clone(), &item)777 })778 )));779 }780 Val::Arr(Rc::new(out))781 }782 ArrComp(expr, compspecs) => Val::Arr(783 // First compspec should be forspec, so no "None" possible here784 Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),785 ),786 Obj(body) => Val::Obj(evaluate_object(context, body)?),787 ObjExtend(s, t) => evaluate_add_op(788 &evaluate(context.clone(), s)?,789 &Val::Obj(evaluate_object(context, t)?),790 )?,791 Apply(value, args, tailstrict) => evaluate_apply(context, value, args, loc, *tailstrict)?,792 Function(params, body) => evaluate_method(context, params.clone(), body.clone()),793 AssertExpr(AssertStmt(value, msg), returned) => {794 let assertion_result = push(795 &value.1,796 || "assertion condition".to_owned(),797 || {798 evaluate(context.clone(), &value)?799 .try_cast_bool("assertion condition should be boolean")800 },801 )?;802 if assertion_result {803 evaluate(context, returned)?804 } else if let Some(msg) = msg {805 create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?806 } else {807 create_error_result(crate::Error::AssertionFailed(Val::Null))?808 }809 }810 Error(e) => push(811 &loc,812 || "error statement".to_owned(),813 || {814 create_error_result(crate::Error::RuntimeError(815 evaluate(context, e)?.try_cast_str("error text should be string")?,816 ))?817 },818 )?,819 IfElse {820 cond,821 cond_then,822 cond_else,823 } => {824 if evaluate(context.clone(), &cond.0)?825 .try_cast_bool("if condition should be boolean")?826 {827 evaluate(context, cond_then)?828 } else {829 match cond_else {830 Some(v) => evaluate(context, v)?,831 None => Val::Null,832 }833 }834 }835 Import(path) => {836 let mut tmp = loc837 .clone()838 .expect("imports can't be used without loc_data")839 .0;840 let import_location = Rc::make_mut(&mut tmp);841 import_location.pop();842 with_state(|s| s.import_file(&import_location, path))?843 }844 ImportStr(path) => {845 let mut tmp = loc846 .clone()847 .expect("imports can't be used without loc_data")848 .0;849 let import_location = Rc::make_mut(&mut tmp);850 import_location.pop();851 Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)852 }853 Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),854 })855}1use crate::{2 context_creator, create_error, create_error_result, equals, escape_string_json, future_wrapper,3 lazy_val, manifest_json_ex, parse_args, primitive_equals, push, with_state, Context,4 ContextCreator, Error, FuncDesc, LazyBinding, LazyVal, ObjMember, ObjValue, Result, Val,5 ValType,6};7use closure::closure;8use jrsonnet_parser::{9 ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, ExprLocation, FieldMember,10 ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,11 Visibility,12};13use std::{collections::HashMap, rc::Rc};1415pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (Rc<str>, LazyBinding) {16 let b = b.clone();17 if let Some(params) = &b.params {18 let params = params.clone();19 (20 b.name.clone(),21 LazyBinding::Bindable(Rc::new(move |this, super_obj| {22 Ok(lazy_val!(23 closure!(clone b, clone params, clone context_creator, || Ok(evaluate_method(24 context_creator.0(this.clone(), super_obj.clone())?,25 b.name.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 evaluate_named(38 context_creator.0(this.clone(), super_obj.clone())?,39 &b.value,40 b.name.clone()41 )42 )))43 })),44 )45 }46}4748pub fn evaluate_method(ctx: Context, name: Rc<str>, params: ParamsDesc, body: LocExpr) -> Val {49 Val::Func(Rc::new(FuncDesc {50 name,51 ctx,52 params,53 body,54 }))55}5657pub fn evaluate_field_name(58 context: Context,59 field_name: &jrsonnet_parser::FieldName,60) -> Result<Option<Rc<str>>> {61 Ok(match field_name {62 jrsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),63 jrsonnet_parser::FieldName::Dyn(expr) => {64 let lazy = evaluate(context, expr)?;65 let value = lazy.unwrap_if_lazy()?;66 if matches!(value, Val::Null) {67 None68 } else {69 Some(value.try_cast_str("dynamic field name")?)70 }71 }72 })73}7475pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {76 Ok(match (op, b) {77 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,78 (UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),79 (UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),80 (UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),81 (op, o) => create_error_result(Error::UnaryOperatorDoesNotOperateOnType(82 op,83 o.value_type()?,84 ))?,85 })86}8788pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {89 Ok(match (a, b) {90 (Val::Str(v1), Val::Str(v2)) => Val::Str(((**v1).to_owned() + &v2).into()),9192 // Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)93 (Val::Num(n), Val::Str(o)) => Val::Str(format!("{}{}", n, o).into()),94 (Val::Str(o), Val::Num(n)) => Val::Str(format!("{}{}", o, n).into()),9596 (Val::Str(s), o) => Val::Str(format!("{}{}", s, o.clone().into_json(0)?).into()),97 (o, Val::Str(s)) => Val::Str(format!("{}{}", o.clone().into_json(0)?, s).into()),9899 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),100 (Val::Arr(a), Val::Arr(b)) => Val::Arr(Rc::new([&a[..], &b[..]].concat())),101 (Val::Num(v1), Val::Num(v2)) => Val::new_checked_num(v1 + v2)?,102 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(103 BinaryOpType::Add,104 a.value_type()?,105 b.value_type()?,106 ))?,107 })108}109110pub fn evaluate_binary_op_special(111 context: Context,112 a: &LocExpr,113 op: BinaryOpType,114 b: &LocExpr,115) -> Result<Val> {116 Ok(117 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {118 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),119 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),120 (a, op, eb) => {121 evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?.unwrap_if_lazy()?)?122 }123 },124 )125}126127pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {128 Ok(match (a, op, b) {129 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,130131 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize).into()),132133 // Bool X Bool134 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),135 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),136137 // Str X Str138 (Val::Str(v1), BinaryOpType::Lt, Val::Str(v2)) => Val::Bool(v1 < v2),139 (Val::Str(v1), BinaryOpType::Gt, Val::Str(v2)) => Val::Bool(v1 > v2),140 (Val::Str(v1), BinaryOpType::Lte, Val::Str(v2)) => Val::Bool(v1 <= v2),141 (Val::Str(v1), BinaryOpType::Gte, Val::Str(v2)) => Val::Bool(v1 >= v2),142143 // Num X Num144 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::new_checked_num(v1 * v2)?,145 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => {146 if *v2 <= f64::EPSILON {147 create_error_result(crate::Error::DivisionByZero)?148 }149 Val::new_checked_num(v1 / v2)?150 }151152 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::new_checked_num(v1 - v2)?,153154 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),155 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),156 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),157 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),158159 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {160 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)161 }162 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {163 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)164 }165 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {166 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)167 }168 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {169 if *v2 < 0.0 {170 create_error_result(Error::RuntimeError("shift by negative exponent".into()))?171 }172 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)173 }174 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {175 if *v2 < 0.0 {176 create_error_result(Error::RuntimeError("shift by negative exponent".into()))?177 }178 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)179 }180181 _ => create_error_result(Error::BinaryOperatorDoesNotOperateOnValues(182 op,183 a.value_type()?,184 b.value_type()?,185 ))?,186 })187}188189future_wrapper!(HashMap<Rc<str>, LazyBinding>, FutureNewBindings);190future_wrapper!(ObjValue, FutureObjValue);191192pub fn evaluate_comp<T>(193 context: Context,194 value: &impl Fn(Context) -> Result<T>,195 specs: &[CompSpec],196) -> Result<Option<Vec<T>>> {197 Ok(match specs.get(0) {198 None => Some(vec![value(context)?]),199 Some(CompSpec::IfSpec(IfSpecData(cond))) => {200 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {201 evaluate_comp(context, value, &specs[1..])?202 } else {203 None204 }205 }206 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {207 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {208 Val::Arr(list) => {209 let mut out = Vec::new();210 for item in list.iter() {211 let item = item.unwrap_if_lazy()?;212 out.push(evaluate_comp(213 context.with_var(var.clone(), item.clone())?,214 value,215 &specs[1..],216 )?);217 }218 Some(out.into_iter().flatten().flatten().collect())219 }220 _ => create_error_result(Error::InComprehensionCanOnlyIterateOverArray)?,221 }222 }223 })224}225226pub fn evaluate_member_list_object(context: Context, members: &[Member]) -> Result<ObjValue> {227 let new_bindings = FutureNewBindings::new();228 let future_this = FutureObjValue::new();229 let context_creator = context_creator!(230 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {231 Ok(context.extend_unbound(232 new_bindings.clone().unwrap(),233 context.dollar().clone().or_else(||this.clone()),234 Some(this.unwrap()),235 super_obj236 )?)237 })238 );239 {240 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();241 for (n, b) in members242 .iter()243 .filter_map(|m| match m {244 Member::BindStmt(b) => Some(b.clone()),245 _ => None,246 })247 .map(|b| evaluate_binding(&b, context_creator.clone()))248 {249 bindings.insert(n, b);250 }251 new_bindings.fill(bindings);252 }253254 let mut new_members = HashMap::new();255 for member in members.iter() {256 match member {257 Member::Field(FieldMember {258 name,259 plus,260 params: None,261 visibility,262 value,263 }) => {264 let name = evaluate_field_name(context.clone(), &name)?;265 if name.is_none() {266 continue;267 }268 let name = name.unwrap();269 new_members.insert(270 name.clone(),271 ObjMember {272 add: *plus,273 visibility: *visibility,274 invoke: LazyBinding::Bindable(Rc::new(275 closure!(clone name, clone value, clone context_creator, |this, super_obj| {276 Ok(LazyVal::new_resolved(evaluate(277 context_creator.0(this, super_obj)?,278 &value,279 )?))280 }),281 )),282 location: value.1.clone(),283 },284 );285 }286 Member::Field(FieldMember {287 name,288 params: Some(params),289 value,290 ..291 }) => {292 let name = evaluate_field_name(context.clone(), &name)?;293 if name.is_none() {294 continue;295 }296 let name = name.unwrap();297 new_members.insert(298 name.clone(),299 ObjMember {300 add: false,301 visibility: Visibility::Hidden,302 invoke: LazyBinding::Bindable(Rc::new(303 closure!(clone value, clone context_creator, clone params, clone name, |this, super_obj| {304 // TODO: Assert305 Ok(LazyVal::new_resolved(evaluate_method(306 context_creator.0(this, super_obj)?,307 name.clone(),308 params.clone(),309 value.clone(),310 )))311 }),312 )),313 location: value.1.clone(),314 },315 );316 }317 Member::BindStmt(_) => {}318 Member::AssertStmt(_) => {}319 }320 }321 Ok(future_this.fill(ObjValue::new(None, Rc::new(new_members))))322}323324pub fn evaluate_object(context: Context, object: &ObjBody) -> Result<ObjValue> {325 Ok(match object {326 ObjBody::MemberList(members) => evaluate_member_list_object(context, &members)?,327 ObjBody::ObjComp(obj) => {328 let future_this = FutureObjValue::new();329 let mut new_members = HashMap::new();330 for (k, v) in evaluate_comp(331 context.clone(),332 &|ctx| {333 let new_bindings = FutureNewBindings::new();334 let context_creator = context_creator!(335 closure!(clone context, clone new_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {336 Ok(context.extend_unbound(337 new_bindings.clone().unwrap(),338 context.dollar().clone().or_else(||this.clone()),339 None,340 super_obj341 )?)342 })343 );344 let mut bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();345 for (n, b) in obj346 .pre_locals347 .iter()348 .chain(obj.post_locals.iter())349 .map(|b| evaluate_binding(b, context_creator.clone()))350 {351 bindings.insert(n, b);352 }353 let bindings = new_bindings.fill(bindings);354 let ctx = ctx.extend_unbound(bindings, None, None, None)?;355 let key = evaluate(ctx.clone(), &obj.key)?;356 let value = LazyBinding::Bindable(Rc::new(357 closure!(clone ctx, clone obj.value, |this, _super_obj| {358 Ok(LazyVal::new_resolved(evaluate(ctx.extend(HashMap::new(), None, this, None)?, &value)?))359 }),360 ));361362 Ok((key, value))363 },364 &obj.compspecs,365 )?366 .unwrap()367 {368 match k {369 Val::Null => {}370 Val::Str(n) => {371 new_members.insert(372 n,373 ObjMember {374 add: false,375 visibility: Visibility::Normal,376 invoke: v,377 location: obj.value.1.clone(),378 },379 );380 }381 v => create_error_result(Error::FieldMustBeStringGot(v.value_type()?))?,382 }383 }384385 future_this.fill(ObjValue::new(None, Rc::new(new_members)))386 }387 })388}389390/// Extracts code block and disables inlining for them391/// Fixes WASM to java bytecode compilation failing because of very large method392macro_rules! noinline {393 ($e:expr) => {394 (#[inline(never)]395 move || $e)()396 };397}398399pub fn evaluate_apply(400 context: Context,401 value: &LocExpr,402 args: &ArgsDesc,403 loc: &Option<ExprLocation>,404 tailstrict: bool,405) -> Result<Val> {406 let lazy = evaluate(context.clone(), value)?;407 let value = lazy.unwrap_if_lazy()?;408 Ok(match value {409 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {410 // arr/string/function411 ("std", "length") => noinline!(parse_args!(context, "std.length", args, 1, [412 0, x: [Val::Str|Val::Arr|Val::Obj], vec![ValType::Str, ValType::Arr, ValType::Obj];413 ], {414 Ok(match x {415 Val::Str(n) => Val::Num(n.chars().count() as f64),416 Val::Arr(i) => Val::Num(i.len() as f64),417 Val::Obj(o) => Val::Num(418 o.fields_visibility()419 .into_iter()420 .filter(|(_k, v)| *v)421 .count() as f64,422 ),423 _ => unreachable!(),424 })425 }))?,426 // any427 ("std", "type") => parse_args!(context, "std.type", args, 1, [428 0, x, vec![];429 ], {430 Val::Str(x.value_type()?.name().into())431 }),432 // length, idx=>any433 ("std", "makeArray") => noinline!(parse_args!(context, "std.makeArray", args, 2, [434 0, sz: [Val::Num]!!Val::Num, vec![ValType::Num];435 1, func: [Val::Func]!!Val::Func, vec![ValType::Func];436 ], {437 if sz < 0.0 {438 create_error_result(crate::error::Error::RuntimeError(format!("makeArray requires size >= 0, got {}", sz).into()))?;439 }440 let mut out = Vec::with_capacity(sz as usize);441 for i in 0..sz as usize {442 out.push(func.evaluate_values(443 Context::new(),444 &[Val::Num(i as f64)]445 )?)446 }447 Ok(Val::Arr(Rc::new(out)))448 }))?,449 // string450 ("std", "codepoint") => parse_args!(context, "std.codepoint", args, 1, [451 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];452 ], {453 assert!(454 str.chars().count() == 1,455 "std.codepoint should receive single char string"456 );457 Val::Num(str.chars().take(1).next().unwrap() as u32 as f64)458 }),459 // object, includeHidden460 ("std", "objectFieldsEx") => {461 noinline!(parse_args!(context, "std.objectFieldsEx",args, 2, [462 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];463 1, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];464 ], {465 let mut out = obj.fields_visibility()466 .into_iter()467 .filter(|(_k, v)| *v || inc_hidden)468 .map(|(k, _v)|k)469 .collect::<Vec<_>>();470 out.sort();471 Ok(Val::Arr(Rc::new(out.into_iter().map(Val::Str).collect())))472 }))?473 }474 // object, field, includeHidden475 ("std", "objectHasEx") => parse_args!(context, "std.objectHasEx", args, 3, [476 0, obj: [Val::Obj]!!Val::Obj, vec![ValType::Obj];477 1, f: [Val::Str]!!Val::Str, vec![ValType::Str];478 2, inc_hidden: [Val::Bool]!!Val::Bool, vec![ValType::Bool];479 ], {480 Val::Bool(481 obj.fields_visibility()482 .into_iter()483 .filter(|(_k, v)| *v || inc_hidden)484 .any(|(k, _v)| *k == *f),485 )486 }),487 ("std", "primitiveEquals") => parse_args!(context, "std.primitiveEquals", args, 2, [488 0, a, vec![];489 1, b, vec![];490 ], {491 Val::Bool(primitive_equals(&a, &b)?)492 }),493 // faster494 ("std", "equals") => parse_args!(context, "std.equals", args, 2, [495 0, a, vec![];496 1, b, vec![];497 ], {498 Val::Bool(equals(&a, &b)?)499 }),500 ("std", "modulo") => parse_args!(context, "std.modulo", args, 2, [501 0, a: [Val::Num]!!Val::Num, vec![ValType::Num];502 1, b: [Val::Num]!!Val::Num, vec![ValType::Num];503 ], {504 Val::Num(a % b)505 }),506 ("std", "floor") => parse_args!(context, "std.floor", args, 1, [507 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];508 ], {509 Val::Num(x.floor())510 }),511 ("std", "trace") => parse_args!(context, "std.trace", args, 2, [512 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];513 1, rest, vec![];514 ], {515 eprint!("TRACE: ");516 if let Some(loc) = loc {517 with_state(|s|{518 let locs = s.map_source_locations(&loc.0, &[loc.1]);519 eprint!("{}:{} ", loc.0.display(), locs[0].line);520 });521 }522 eprintln!("{}", str);523 rest524 }),525 ("std", "pow") => parse_args!(context, "std.modulo", args, 2, [526 0, x: [Val::Num]!!Val::Num, vec![ValType::Num];527 1, n: [Val::Num]!!Val::Num, vec![ValType::Num];528 ], {529 Val::Num(x.powf(n))530 }),531 ("std", "extVar") => parse_args!(context, "std.extVar", args, 2, [532 0, x: [Val::Str]!!Val::Str, vec![ValType::Str];533 ], {534 with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or_else(535 || create_error(crate::Error::UndefinedExternalVariable(x)),536 )?537 }),538 ("std", "filter") => noinline!(parse_args!(context, "std.filter", args, 2, [539 0, func: [Val::Func]!!Val::Func, vec![ValType::Func];540 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];541 ], {542 Ok(Val::Arr(Rc::new(543 arr.iter()544 .cloned()545 .filter(|e| {546 func547 .evaluate_values(context.clone(), &[e.clone()])548 .unwrap()549 .try_cast_bool("filter predicate")550 .unwrap()551 })552 .collect(),553 )))554 }))?,555 ("std", "char") => parse_args!(context, "std.char", args, 1, [556 0, n: [Val::Num]!!Val::Num, vec![ValType::Num];557 ], {558 let mut out = String::new();559 out.push(std::char::from_u32(n as u32).unwrap());560 Val::Str(out.into())561 }),562 ("std", "encodeUTF8") => parse_args!(context, "std.encodeUtf8", args, 1, [563 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];564 ], {565 Val::Arr(Rc::new(str.bytes().map(|b| Val::Num(b as f64)).collect()))566 }),567 ("std", "md5") => noinline!(parse_args!(context, "std.md5", args, 1, [568 0, str: [Val::Str]!!Val::Str, vec![ValType::Str];569 ], {570 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))571 }))?,572 // faster573 ("std", "base64") => parse_args!(context, "std.base64", args, 1, [574 0, input: [Val::Str | Val::Arr], vec![ValType::Arr, ValType::Str];575 ], {576 Val::Str(match input {577 Val::Str(s) => {578 base64::encode(s.bytes().collect::<Vec<_>>()).into()579 },580 Val::Arr(a) => {581 base64::encode(a.iter().map(|v| {582 Ok(v.clone().try_cast_num("base64 array")? as u8)583 }).collect::<Result<Vec<_>>>()?).into()584 },585 _ => unreachable!()586 })587 }),588 // faster589 ("std", "join") => noinline!(parse_args!(context, "std.join", args, 2, [590 0, sep: [Val::Str|Val::Arr], vec![ValType::Str, ValType::Arr];591 1, arr: [Val::Arr]!!Val::Arr, vec![ValType::Arr];592 ], {593 Ok(match sep {594 Val::Arr(joiner_items) => {595 let mut out = Vec::new();596597 let mut first = true;598 for item in arr.iter().cloned() {599 if let Val::Arr(items) = item.unwrap_if_lazy()? {600 if !first {601 out.reserve(joiner_items.len());602 out.extend(joiner_items.iter().cloned());603 }604 first = false;605 out.reserve(items.len());606 out.extend(items.iter().cloned());607 } else {608 create_error_result(crate::Error::RuntimeError("in std.join all items should be arrays".into()))?;609 }610 }611612 Val::Arr(Rc::new(out))613 },614 Val::Str(sep) => {615 let mut out = String::new();616617 let mut first = true;618 for item in arr.iter().cloned() {619 if let Val::Str(item) = item.unwrap_if_lazy()? {620 if !first {621 out += &sep;622 }623 first = false;624 out += &item;625 } else {626 create_error_result(crate::Error::RuntimeError("in std.join all items should be strings".into()))?;627 }628 }629630 Val::Str(out.into())631 },632 _ => unreachable!()633 })634 }))?,635 // Faster636 ("std", "escapeStringJson") => parse_args!(context, "std.escapeStringJson", args, 1, [637 0, str_: [Val::Str]!!Val::Str, vec![ValType::Str];638 ], {639 Val::Str(escape_string_json(&str_).into())640 }),641 // Faster642 ("std", "manifestJsonEx") => parse_args!(context, "std.manifestJsonEx", args, 2, [643 0, value, vec![];644 1, indent: [Val::Str]!!Val::Str, vec![ValType::Str];645 ], {646 Val::Str(manifest_json_ex(&value, &indent)?.into())647 }),648 (ns, name) => {649 create_error_result(crate::Error::IntristicNotFound(ns.into(), name.into()))?650 }651 },652 Val::Func(f) => {653 let body = || f.evaluate(context, args, tailstrict);654 if tailstrict {655 body()?656 } else {657 push(loc, || format!("function <{}> call", f.name), body)?658 }659 }660 v => create_error_result(crate::Error::OnlyFunctionsCanBeCalledGot(v.value_type()?))?,661 })662}663664pub fn evaluate_named(context: Context, lexpr: &LocExpr, name: Rc<str>) -> Result<Val> {665 use Expr::*;666 let LocExpr(expr, _loc) = lexpr;667 Ok(match &**expr {668 Function(params, body) => evaluate_method(context, name, params.clone(), body.clone()),669 _ => evaluate(context, lexpr)?,670 })671}672673pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {674 use Expr::*;675 let LocExpr(expr, loc) = expr;676 Ok(match &**expr {677 Literal(LiteralType::This) => Val::Obj(678 context679 .this()680 .clone()681 .ok_or_else(|| create_error(crate::Error::CantUseSelfOutsideOfObject))?,682 ),683 Literal(LiteralType::Dollar) => Val::Obj(684 context685 .dollar()686 .clone()687 .ok_or_else(|| create_error(crate::Error::NoTopLevelObjectFound))?,688 ),689 Literal(LiteralType::True) => Val::Bool(true),690 Literal(LiteralType::False) => Val::Bool(false),691 Literal(LiteralType::Null) => Val::Null,692 Parened(e) => evaluate(context, e)?,693 Str(v) => Val::Str(v.clone()),694 Num(v) => Val::new_checked_num(*v)?,695 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,696 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,697 Var(name) => push(698 loc,699 || format!("variable <{}>", name),700 || Ok(Val::Lazy(context.binding(name.clone())?).unwrap_if_lazy()?),701 )?,702 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {703 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;704 context705 .super_obj()706 .clone()707 .expect("no super found")708 .get_raw(&name, &context.this().clone().expect("no this found"))?709 .expect("value not found")710 }711 Index(value, index) => {712 match (713 evaluate(context.clone(), value)?.unwrap_if_lazy()?,714 evaluate(context, index)?,715 ) {716 (Val::Obj(v), Val::Str(s)) => {717 if let Some(v) = v.get(s.clone())? {718 v.unwrap_if_lazy()?719 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__".into())? {720 Val::Intristic(n, s)721 } else {722 create_error_result(crate::Error::NoSuchField(s))?723 }724 }725 (Val::Obj(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(726 ValType::Obj,727 ValType::Str,728 n.value_type()?,729 ))?,730731 (Val::Arr(v), Val::Num(n)) => {732 if n.fract() > f64::EPSILON {733 create_error_result(crate::Error::FractionalIndex)?734 }735 v.get(n as usize)736 .ok_or_else(|| {737 create_error(crate::Error::ArrayBoundsError(n as usize, v.len()))738 })?739 .clone()740 .unwrap_if_lazy()?741 }742 (Val::Arr(_), Val::Str(n)) => {743 create_error_result(crate::Error::AttemptedIndexAnArrayWithString(n))?744 }745 (Val::Arr(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(746 ValType::Arr,747 ValType::Num,748 n.value_type()?,749 ))?,750751 (Val::Str(s), Val::Num(n)) => Val::Str(752 s.chars()753 .skip(n as usize)754 .take(1)755 .collect::<String>()756 .into(),757 ),758 (Val::Str(_), n) => create_error_result(crate::Error::ValueIndexMustBeTypeGot(759 ValType::Str,760 ValType::Num,761 n.value_type()?,762 ))?,763764 (v, _) => create_error_result(crate::Error::CantIndexInto(v.value_type()?))?,765 }766 }767 LocalExpr(bindings, returned) => {768 let mut new_bindings: HashMap<Rc<str>, LazyBinding> = HashMap::new();769 let future_context = Context::new_future();770771 let context_creator = context_creator!(772 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))773 );774775 for (k, v) in bindings776 .iter()777 .map(|b| evaluate_binding(b, context_creator.clone()))778 {779 new_bindings.insert(k, v);780 }781782 let context = context783 .extend_unbound(new_bindings, None, None, None)?784 .into_future(future_context);785 evaluate(context, &returned.clone())?786 }787 Arr(items) => {788 let mut out = Vec::with_capacity(items.len());789 for item in items {790 out.push(Val::Lazy(lazy_val!(791 closure!(clone context, clone item, || {792 evaluate(context.clone(), &item)793 })794 )));795 }796 Val::Arr(Rc::new(out))797 }798 ArrComp(expr, compspecs) => Val::Arr(799 // First compspec should be forspec, so no "None" possible here800 Rc::new(evaluate_comp(context, &|ctx| evaluate(ctx, expr), compspecs)?.unwrap()),801 ),802 Obj(body) => Val::Obj(evaluate_object(context, body)?),803 ObjExtend(s, t) => evaluate_add_op(804 &evaluate(context.clone(), s)?,805 &Val::Obj(evaluate_object(context, t)?),806 )?,807 Apply(value, args, tailstrict) => evaluate_apply(context, value, args, loc, *tailstrict)?,808 Function(params, body) => {809 evaluate_method(context, "anonymous".into(), params.clone(), body.clone())810 }811 AssertExpr(AssertStmt(value, msg), returned) => {812 let assertion_result = push(813 &value.1,814 || "assertion condition".to_owned(),815 || {816 evaluate(context.clone(), &value)?817 .try_cast_bool("assertion condition should be boolean")818 },819 )?;820 if assertion_result {821 evaluate(context, returned)?822 } else if let Some(msg) = msg {823 create_error_result(crate::Error::AssertionFailed(evaluate(context, msg)?))?824 } else {825 create_error_result(crate::Error::AssertionFailed(Val::Null))?826 }827 }828 Error(e) => push(829 &loc,830 || "error statement".to_owned(),831 || {832 create_error_result(crate::Error::RuntimeError(833 evaluate(context, e)?.try_cast_str("error text should be string")?,834 ))?835 },836 )?,837 IfElse {838 cond,839 cond_then,840 cond_else,841 } => {842 if evaluate(context.clone(), &cond.0)?843 .try_cast_bool("if condition should be boolean")?844 {845 evaluate(context, cond_then)?846 } else {847 match cond_else {848 Some(v) => evaluate(context, v)?,849 None => Val::Null,850 }851 }852 }853 Import(path) => {854 let mut tmp = loc855 .clone()856 .expect("imports can't be used without loc_data")857 .0;858 let import_location = Rc::make_mut(&mut tmp);859 import_location.pop();860 with_state(|s| s.import_file(&import_location, path))?861 }862 ImportStr(path) => {863 let mut tmp = loc864 .clone()865 .expect("imports can't be used without loc_data")866 .0;867 let import_location = Rc::make_mut(&mut tmp);868 import_location.pop();869 Val::Str(with_state(|s| s.import_file_str(&import_location, path))?)870 }871 Literal(LiteralType::Super) => return create_error_result(crate::Error::StandaloneSuper),872 })873}crates/jrsonnet-evaluator/src/val.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/val.rs
+++ b/crates/jrsonnet-evaluator/src/val.rs
@@ -58,6 +58,7 @@
#[derive(Debug, PartialEq)]
pub struct FuncDesc {
+ pub name: Rc<str>,
pub ctx: Context,
pub params: ParamsDesc,
pub body: LocExpr,