1use crate::{2 binding, context_creator, create_error, function_default, function_rhs, future_wrapper,3 lazy_binding, lazy_val, push, Context, ContextCreator, FuncDesc, LazyBinding, ObjMember,4 ObjValue, Result, Val,5};6use closure::closure;7use jsonnet_parser::{8 ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember, ForSpecData,9 IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType, Visibility,10};11use std::{12 collections::{BTreeMap, HashMap},13 rc::Rc,14};1516pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, LazyBinding) {17 let b = b.clone();18 if let Some(args) = &b.params {19 let args = args.clone();20 (21 b.name.clone(),22 lazy_binding!(move |this, super_obj| Ok(lazy_val!(23 closure!(clone b, clone args, clone context_creator, || Ok(evaluate_method(24 context_creator.0(this.clone(), super_obj.clone())?,25 &b.value,26 args.clone()27 )))28 ))),29 )30 } else {31 (32 b.name.clone(),33 lazy_binding!(move |this, super_obj| {34 Ok(lazy_val!(35 closure!(clone context_creator, clone b, || evaluate(36 context_creator.0(this.clone(), super_obj.clone())?,37 &b.value38 ))39 ))40 }),41 )42 }43}4445pub fn evaluate_method(ctx: Context, expr: &LocExpr, arg_spec: ParamsDesc) -> Val {46 Val::Func(FuncDesc {47 ctx,48 params: arg_spec,49 eval_rhs: function_rhs!(closure!(clone expr, |ctx| evaluate(ctx, &expr))),50 eval_default: function_default!(closure!(|ctx, default| evaluate(ctx, &default))),51 })52}5354pub fn evaluate_field_name(55 context: Context,56 field_name: &jsonnet_parser::FieldName,57) -> Result<Option<String>> {58 Ok(match field_name {59 jsonnet_parser::FieldName::Fixed(n) => Some(n.clone()),60 jsonnet_parser::FieldName::Dyn(expr) => {61 let value = evaluate(context, expr)?.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),8485 (Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),86 (Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),87 (Val::Str(v1), Val::Bool(v2)) => Val::Str(format!("{}{}", v1, v2)),88 (Val::Bool(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),89 (Val::Str(v1), Val::Null) => Val::Str(format!("{}null", v1)),90 (Val::Null, Val::Str(v2)) => Val::Str(format!("null{}", v2)),9192 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),93 (Val::Arr(a), Val::Arr(b)) => Val::Arr([&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)),121122 123 (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 127 (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 133 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),134 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => Val::Num(v1 / v2),135136 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),137138 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),139 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),140 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),141 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),142143 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {144 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)145 }146 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {147 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)148 }149 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {150 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)151 }152 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {153 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)154 }155 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {156 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)157 }158159 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),160 })161}162163future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);164future_wrapper!(ObjValue, FutureObjValue);165166pub fn evaluate_comp(167 context: Context,168 value: &LocExpr,169 specs: &[CompSpec],170) -> Result<Option<Vec<Val>>> {171 Ok(match specs.get(0) {172 None => Some(vec![evaluate(context, &value)?]),173 Some(CompSpec::IfSpec(IfSpecData(cond))) => {174 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {175 evaluate_comp(context, value, &specs[1..])?176 } else {177 None178 }179 }180 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {181 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {182 Val::Arr(list) => {183 let mut out = Vec::new();184 for item in list {185 let item = item.clone().unwrap_if_lazy()?;186 out.push(evaluate_comp(187 context.with_var(var.clone(), item)?,188 value,189 &specs[1..],190 )?);191 }192 Some(out.iter().flatten().flatten().cloned().collect())193 }194 _ => panic!("for expression evaluated to non-iterable value"),195 }196 }197 })198}199200201pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {202 Ok(match object {203 ObjBody::MemberList(members) => {204 let new_bindings = FutureNewBindings::new();205 let future_this = FutureObjValue::new();206 let context_creator = context_creator!(207 closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {208 Ok(context.clone().extend(209 new_bindings.clone().unwrap(),210 context.clone().dollar().clone().or_else(||this.clone()),211 Some(this.unwrap()),212 super_obj213 )?)214 })215 );216 {217 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();218 for (n, b) in members219 .iter()220 .filter_map(|m| match m {221 Member::BindStmt(b) => Some(b.clone()),222 _ => None,223 })224 .map(|b| evaluate_binding(&b, context_creator.clone()))225 {226 bindings.insert(n, b);227 }228 new_bindings.fill(bindings);229 }230231 let mut new_members = BTreeMap::new();232 for member in members.into_iter() {233 match member {234 Member::Field(FieldMember {235 name,236 plus,237 params: None,238 visibility,239 value,240 }) => {241 let name = evaluate_field_name(context.clone(), &name)?;242 if name.is_none() {243 continue;244 }245 let name = name.unwrap();246 new_members.insert(247 name.clone(),248 ObjMember {249 add: plus,250 visibility: visibility.clone(),251 invoke: binding!(252 closure!(clone name, clone value, clone context_creator, |this, super_obj| {253 push(value.clone(), "object ".to_owned()+&name+" field", ||{254 let context = context_creator.0(this, super_obj)?;255 evaluate(256 context,257 &value,258 )?.unwrap_if_lazy()259 })260 })261 ),262 },263 );264 }265 Member::Field(FieldMember {266 name,267 params: Some(params),268 value,269 ..270 }) => {271 let name = evaluate_field_name(context.clone(), &name)?;272 if name.is_none() {273 continue;274 }275 let name = name.unwrap();276 new_members.insert(277 name,278 ObjMember {279 add: false,280 visibility: Visibility::Hidden,281 invoke: binding!(282 closure!(clone value, clone context_creator, |this, super_obj| {283 284 Ok(evaluate_method(285 context_creator.0(this, super_obj)?,286 &value.clone(),287 params.clone(),288 ))289 })290 ),291 },292 );293 }294 Member::BindStmt(_) => {}295 Member::AssertStmt(_) => {}296 }297 }298 future_this.fill(ObjValue::new(None, Rc::new(new_members)))299 }300 _ => todo!(),301 })302}303304pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {305 use Expr::*;306 let locexpr = expr.clone();307 let LocExpr(expr, loc) = expr;308 Ok(match &**expr {309 Literal(LiteralType::This) => Val::Obj(310 context311 .this()312 .clone()313 .unwrap_or_else(|| panic!("this not found")),314 ),315 Literal(LiteralType::Dollar) => Val::Obj(316 context317 .dollar()318 .clone()319 .unwrap_or_else(|| panic!("dollar not found")),320 ),321 Literal(LiteralType::True) => Val::Bool(true),322 Literal(LiteralType::False) => Val::Bool(false),323 Literal(LiteralType::Null) => Val::Null,324 Parened(e) => evaluate(context, e)?,325 Str(v) => Val::Str(v.clone()),326 Num(v) => Val::Num(*v),327 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,328 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,329 Var(name) => Val::Lazy(context.binding(&name)).unwrap_if_lazy()?,330 Index(LocExpr(v, _), index) if matches!(&**v, Expr::Literal(LiteralType::Super)) => {331 let name = evaluate(context.clone(), index)?.try_cast_str("object index")?;332 context333 .super_obj()334 .clone()335 .expect("no super found")336 .get_raw(&name, &context.this().clone().expect("no this found"))?337 .expect("value not found")338 }339 Index(value, index) => {340 match (341 evaluate(context.clone(), value)?.unwrap_if_lazy()?,342 evaluate(context, index)?,343 ) {344 (Val::Obj(v), Val::Str(s)) => {345 if let Some(v) = v.get(&s)? {346 v.unwrap_if_lazy()?347 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {348 Val::Intristic(n, s)349 } else {350 create_error(crate::Error::NoSuchField(s))?351 }352 }353 (Val::Arr(v), Val::Num(n)) => v354 .get(n as usize)355 .unwrap_or_else(|| panic!("out of bounds"))356 .clone()357 .unwrap_if_lazy()?,358 (Val::Str(s), Val::Num(n)) => {359 Val::Str(s.chars().skip(n as usize).take(1).collect())360 }361 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),362 }363 }364 LocalExpr(bindings, returned) => {365 let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();366 let future_context = Context::new_future();367368 let context_creator = context_creator!(369 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))370 );371372 for (k, v) in bindings373 .iter()374 .map(|b| evaluate_binding(b, context_creator.clone()))375 {376 new_bindings.insert(k, v);377 }378379 let context = context380 .extend(new_bindings, None, None, None)?381 .into_future(future_context);382 evaluate(context, &returned.clone())?383 }384 Arr(items) => {385 let mut out = Vec::with_capacity(items.len());386 for item in items {387 out.push(Val::Lazy(lazy_val!(388 closure!(clone context, clone item, || {389 evaluate(context.clone(), &item)390 })391 )));392 }393 Val::Arr(out)394 }395 ArrComp(expr, compspecs) => Val::Arr(396 397 evaluate_comp(context, expr, compspecs)?.unwrap(),398 ),399 Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),400 ObjExtend(s, t) => evaluate_add_op(401 &evaluate(context.clone(), s)?,402 &Val::Obj(evaluate_object(context, t.clone())?),403 )?,404 Apply(value, ArgsDesc(args)) => {405 let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;406 match value {407 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {408 409 ("std", "length") => {410 assert_eq!(args.len(), 1);411 let expr = &args.get(0).unwrap().1;412 match evaluate(context, expr)? {413 Val::Str(n) => Val::Num(n.chars().count() as f64),414 Val::Arr(i) => Val::Num(i.len() as f64),415 v => panic!("can't get length of {:?}", v),416 }417 }418 419 ("std", "type") => {420 assert_eq!(args.len(), 1);421 let expr = &args.get(0).unwrap().1;422 Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())423 }424 425 ("std", "makeArray") => {426 assert_eq!(args.len(), 2);427 if let (Val::Num(v), Val::Func(d)) = (428 evaluate(context.clone(), &args[0].1)?,429 evaluate(context, &args[1].1)?,430 ) {431 assert!(v >= 0.0);432 let mut out = Vec::with_capacity(v as usize);433 for i in 0..v as usize {434 out.push(d.evaluate(vec![(None, Val::Num(i as f64))])?)435 }436 Val::Arr(out)437 } else {438 panic!("bad makeArray call");439 }440 }441 442 ("std", "codepoint") => {443 assert_eq!(args.len(), 1);444 if let Val::Str(s) = evaluate(context, &args[0].1)? {445 assert!(446 s.chars().count() == 1,447 "std.codepoint should receive single char string"448 );449 Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)450 } else {451 panic!("bad codepoint call");452 }453 }454 455 ("std", "objectFieldsEx") => {456 assert_eq!(args.len(), 2);457 if let (Val::Obj(body), Val::Bool(_include_hidden)) = (458 evaluate(context.clone(), &args[0].1)?,459 evaluate(context, &args[1].1)?,460 ) {461 462 Val::Arr(body.fields().into_iter().map(Val::Str).collect())463 } else {464 panic!("bad objectFieldsEx call");465 }466 }467 ("std", "primitiveEquals") => {468 assert_eq!(args.len(), 2);469 let (a, b) = (470 evaluate(context.clone(), &args[0].1)?,471 evaluate(context, &args[1].1)?,472 );473 Val::Bool(a == b)474 }475 ("std", "modulo") => {476 assert_eq!(args.len(), 2);477 if let (Val::Num(a), Val::Num(b)) = (478 evaluate(context.clone(), &args[0].1)?,479 evaluate(context, &args[1].1)?,480 ) {481 Val::Num(a % b)482 } else {483 panic!("bad modulo call");484 }485 }486 ("std", "floor") => {487 assert_eq!(args.len(), 1);488 if let Val::Num(a) = evaluate(context, &args[0].1)? {489 Val::Num(a.floor())490 } else {491 panic!("bad floor call");492 }493 }494 (ns, name) => panic!("Intristic not found: {}.{}", ns, name),495 },496 Val::Func(f) => push(locexpr, "function call".to_owned(), || {497 f.evaluate(498 args.clone()499 .into_iter()500 .map(move |a| {501 (502 a.clone().0,503 Val::Lazy(lazy_val!(504 closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))505 )),506 )507 })508 .collect(),509 )510 })?,511 _ => panic!("{:?} is not a function", value),512 }513 }514 Function(params, body) => evaluate_method(context, body, params.clone()),515 AssertExpr(AssertStmt(value, msg), returned) => {516 let assertion_result = push(value.clone(), "assertion condition".to_owned(), || {517 evaluate(context.clone(), &value)?518 .try_cast_bool("assertion condition should be boolean")519 })?;520 if assertion_result {521 push(522 returned.clone(),523 "assert 'return' branch".to_owned(),524 || evaluate(context, returned),525 )?526 } else if let Some(msg) = msg {527 panic!(528 "assertion failed ({:?}): {}",529 value,530 evaluate(context, msg)?.try_cast_str("assertion message should be string")?531 );532 } else {533 panic!("assertion failed ({:?}): no message", value);534 }535 }536 Error(e) => create_error(crate::Error::RuntimeError(537 evaluate(context, e)?.try_cast_str("error text should be string")?,538 ))?,539 IfElse {540 cond,541 cond_then,542 cond_else,543 } => {544 let condition_result = push(cond.0.clone(), "if condition".to_owned(), || {545 evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")546 })?;547 if condition_result {548 push(549 cond_then.clone(),550 "if condition 'then' branch".to_owned(),551 || evaluate(context, cond_then),552 )?553 } else {554 match cond_else {555 Some(v) => push(v.clone(), "if condition 'else' branch".to_owned(), || {556 evaluate(context, v)557 })?,558 None => Val::Null,559 }560 }561 }562 _ => panic!(563 "evaluation not implemented: {:?}",564 LocExpr(expr.clone(), loc.clone())565 ),566 })567}