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 el, Arg, ArgsDesc, AssertStmt, BinaryOpType, BindSpec, CompSpec, Expr, FieldMember,9 ForSpecData, IfSpecData, LiteralType, LocExpr, Member, ObjBody, ParamsDesc, UnaryOpType,10 Visibility,11};12use std::{13 collections::{BTreeMap, HashMap},14 rc::Rc,15};1617pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, LazyBinding) {18 let b = b.clone();19 if let Some(args) = &b.params {20 let args = args.clone();21 (22 b.name.clone(),23 lazy_binding!(move |this, super_obj| Ok(lazy_val!(24 closure!(clone b, clone args, clone context_creator, || Ok(evaluate_method(25 context_creator.0(this.clone(), super_obj.clone())?,26 &b.value,27 args.clone()28 )))29 ))),30 )31 } else {32 (33 b.name.clone(),34 lazy_binding!(move |this, super_obj| {35 Ok(lazy_val!(36 closure!(clone context_creator, clone b, || evaluate(37 context_creator.0(this.clone(), super_obj.clone())?,38 &b.value39 ))40 ))41 }),42 )43 }44}4546pub fn evaluate_method(ctx: Context, expr: &LocExpr, arg_spec: ParamsDesc) -> Val {47 Val::Func(FuncDesc {48 ctx,49 params: arg_spec,50 eval_rhs: function_rhs!(closure!(clone expr, |ctx| evaluate(ctx, &expr))),51 eval_default: function_default!(closure!(|ctx, default| evaluate(ctx, &default))),52 })53}5455pub fn evaluate_field_name(56 context: Context,57 field_name: &jsonnet_parser::FieldName,58) -> Result<String> {59 Ok(match field_name {60 jsonnet_parser::FieldName::Fixed(n) => n.clone(),61 jsonnet_parser::FieldName::Dyn(expr) => {62 evaluate(context, expr)?.try_cast_str("dynamic field name")?63 }64 })65}6667pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {68 Ok(match (op, b) {69 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.evaluate()?)?,70 (UnaryOpType::Not, Val::Bool(v)) => Val::Bool(!v),71 (op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),72 })73}7475pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {76 Ok(match (a, b) {77 (Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),78 (Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),79 (Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),80 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),81 (Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),82 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),83 _ => panic!("can't add: {:?} and {:?}", a, b),84 })85}8687pub fn evaluate_binary_op(context: Context, a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {88 Ok(match (a, op, b) {89 (Val::Lazy(a), o, b) => evaluate_binary_op(context, &a.evaluate()?, o, b)?,90 (a, o, Val::Lazy(b)) => evaluate_binary_op(context, a, o, &b.evaluate()?)?,9192 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,9394 (Val::Str(v1), BinaryOpType::Ne, Val::Str(v2)) => Val::Bool(v1 != v2),9596 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),97 (Val::Str(format), BinaryOpType::Mod, args) => evaluate(98 context99 .with_var("__tmp__format__".to_owned(), Val::Str(format.to_owned()))?100 .with_var(101 "__tmp__args__".to_owned(),102 match args {103 Val::Arr(v) => Val::Arr(v.clone()),104 v => Val::Arr(vec![v.clone()]),105 },106 )?,107 &el!(Expr::Apply(108 el!(Expr::Index(109 el!(Expr::Var("std".to_owned())),110 el!(Expr::Str("format".to_owned()))111 )),112 ArgsDesc(vec![113 Arg(None, el!(Expr::Var("__tmp__format__".to_owned()))),114 Arg(None, el!(Expr::Var("__tmp__args__".to_owned())))115 ])116 )),117 )?,118119 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),120 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),121122 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),123 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => Val::Num(v1 / v2),124 (Val::Num(v1), BinaryOpType::Mod, Val::Num(v2)) => Val::Num(v1 % v2),125126 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),127128 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {129 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)130 }131 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {132 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)133 }134135 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),136 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),137 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),138 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),139140 (Val::Num(v1), BinaryOpType::Eq, Val::Num(v2)) => Val::Bool((v1 - v2).abs() < f64::EPSILON),141 (Val::Num(v1), BinaryOpType::Ne, Val::Num(v2)) => Val::Bool((v1 - v2).abs() > f64::EPSILON),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 (a, BinaryOpType::Eq, b) => Val::Bool(a == b),153 (a, BinaryOpType::Ne, b) => Val::Bool(a != b),154 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),155 })156}157158future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);159future_wrapper!(ObjValue, FutureObjValue);160161pub fn evaluate_comp(162 context: Context,163 value: &LocExpr,164 specs: &[CompSpec],165) -> Result<Option<Vec<Val>>> {166 Ok(match specs.get(0) {167 None => Some(vec![evaluate(context, &value)?]),168 Some(CompSpec::IfSpec(IfSpecData(cond))) => {169 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {170 evaluate_comp(context, value, &specs[1..])?171 } else {172 None173 }174 }175 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {176 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {177 Val::Arr(list) => {178 let mut out = Vec::new();179 for item in list {180 let item = item.clone();181 out.push(evaluate_comp(182 context.with_var(var.clone(), item)?,183 value,184 &specs[1..],185 )?);186 }187 Some(out.iter().flatten().flatten().cloned().collect())188 }189 _ => panic!("for expression evaluated to non-iterable value"),190 }191 }192 })193}194195196pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {197 Ok(match object {198 ObjBody::MemberList(members) => {199 let new_bindings = FutureNewBindings::new();200 let future_this = FutureObjValue::new();201 let context_creator = context_creator!(202 closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {203 Ok(context.clone().extend(204 new_bindings.clone().unwrap(),205 context.clone().dollar().clone().or_else(||this.clone()),206 Some(this.unwrap()),207 super_obj208 )?)209 })210 );211 {212 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();213 for (n, b) in members214 .iter()215 .filter_map(|m| match m {216 Member::BindStmt(b) => Some(b.clone()),217 _ => None,218 })219 .map(|b| evaluate_binding(&b, context_creator.clone()))220 {221 bindings.insert(n, b);222 }223 new_bindings.fill(bindings);224 }225226 let mut new_members = BTreeMap::new();227 for member in members.into_iter() {228 match member {229 Member::Field(FieldMember {230 name,231 plus,232 params: None,233 visibility,234 value,235 }) => {236 let name = evaluate_field_name(context.clone(), &name)?;237 new_members.insert(238 name,239 ObjMember {240 add: plus,241 visibility: visibility.clone(),242 invoke: binding!(243 closure!(clone value, clone context_creator, |this, super_obj| {244 let context = context_creator.0(this, super_obj)?;245 246 evaluate(247 context,248 &value,249 )?.unwrap_if_lazy()250 })251 ),252 },253 );254 }255 Member::Field(FieldMember {256 name,257 params: Some(params),258 value,259 ..260 }) => {261 let name = evaluate_field_name(context.clone(), &name)?;262 new_members.insert(263 name,264 ObjMember {265 add: false,266 visibility: Visibility::Hidden,267 invoke: binding!(268 closure!(clone value, clone context_creator, |this, super_obj| {269 270 Ok(evaluate_method(271 context_creator.0(this, super_obj)?,272 &value.clone(),273 params.clone(),274 ))275 })276 ),277 },278 );279 }280 Member::BindStmt(_) => {}281 Member::AssertStmt(_) => {}282 }283 }284 future_this.fill(ObjValue::new(None, Rc::new(new_members)))285 }286 _ => todo!(),287 })288}289290pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {291 use Expr::*;292 push(expr.clone(), "expr".to_owned(), || {293 let LocExpr(expr, loc) = expr;294 Ok(match &**expr {295 Literal(LiteralType::This) => Val::Obj(296 context297 .this()298 .clone()299 .unwrap_or_else(|| panic!("this not found")),300 ),301 Literal(LiteralType::Super) => Val::Obj(302 context303 .super_obj()304 .clone()305 .unwrap_or_else(|| panic!("super not found")),306 ),307 Literal(LiteralType::True) => Val::Bool(true),308 Literal(LiteralType::False) => Val::Bool(false),309 Literal(LiteralType::Null) => Val::Null,310 Parened(e) => evaluate(context, e)?,311 Str(v) => Val::Str(v.clone()),312 Num(v) => Val::Num(*v),313 BinaryOp(v1, o, v2) => {314 let a = evaluate(context.clone(), v1)?.unwrap_if_lazy()?;315 let op = *o;316 let b = evaluate(context.clone(), v2)?.unwrap_if_lazy()?;317 evaluate_binary_op(context, &a, op, &b)?318 }319 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,320 Var(name) => Val::Lazy(context.binding(&name)).unwrap_if_lazy()?,321 Index(value, index) => {322 match (323 evaluate(context.clone(), value)?.unwrap_if_lazy()?,324 evaluate(context.clone(), index)?,325 ) {326 (Val::Obj(v), Val::Str(s)) => {327 if let Some(v) = v.get(&s)? {328 v.unwrap_if_lazy()?329 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {330 Val::Intristic(n, s)331 } else {332 create_error(crate::Error::NoSuchField(s))?333 }334 }335 (Val::Arr(v), Val::Num(n)) => v336 .get(n as usize)337 .unwrap_or_else(|| panic!("out of bounds"))338 .clone(),339 (Val::Str(s), Val::Num(n)) => {340 Val::Str(s.chars().skip(n as usize).take(1).collect())341 }342 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),343 }344 }345 LocalExpr(bindings, returned) => {346 let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();347 let future_context = Context::new_future();348349 let context_creator = context_creator!(350 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))351 );352353 for (k, v) in bindings354 .iter()355 .map(|b| evaluate_binding(b, context_creator.clone()))356 {357 new_bindings.insert(k, v);358 }359360 let context = context361 .extend(new_bindings, None, None, None)?362 .into_future(future_context);363 evaluate(context, &returned.clone())?364 }365 Arr(items) => {366 let mut out = Vec::with_capacity(items.len());367 for item in items {368 out.push(evaluate(context.clone(), item)?);369 }370 Val::Arr(out)371 }372 ArrComp(expr, compspecs) => Val::Arr(373 374 evaluate_comp(context, expr, compspecs)?.unwrap(),375 ),376 Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),377 Apply(value, ArgsDesc(args)) => {378 let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;379 match value {380 381 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {382 383 ("std", "length") => {384 assert_eq!(args.len(), 1);385 let expr = &args.get(0).unwrap().1;386 match evaluate(context, expr)? {387 Val::Str(n) => Val::Num(n.chars().count() as f64),388 Val::Arr(i) => Val::Num(i.len() as f64),389 v => panic!("can't get length of {:?}", v),390 }391 }392 393 ("std", "type") => {394 assert_eq!(args.len(), 1);395 let expr = &args.get(0).unwrap().1;396 Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())397 }398 399 ("std", "makeArray") => {400 assert_eq!(args.len(), 2);401 if let (Val::Num(v), Val::Func(d)) = (402 evaluate(context.clone(), &args[0].1)?,403 evaluate(context, &args[1].1)?,404 ) {405 assert!(v > 0.0);406 let mut out = Vec::with_capacity(v as usize);407 for i in 0..v as usize {408 out.push(d.evaluate(vec![(None, Val::Num(i as f64))])?)409 }410 Val::Arr(out)411 } else {412 panic!("bad makeArray call");413 }414 }415 416 ("std", "codepoint") => {417 assert_eq!(args.len(), 1);418 if let Val::Str(s) = evaluate(context, &args[0].1)? {419 assert!(420 s.chars().count() == 1,421 "std.codepoint should receive single char string"422 );423 Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)424 } else {425 panic!("bad codepoint call");426 }427 }428 429 ("std", "objectFieldsEx") => {430 assert_eq!(args.len(), 2);431 if let (Val::Obj(body), Val::Bool(_include_hidden)) = (432 evaluate(context.clone(), &args[0].1)?,433 evaluate(context, &args[1].1)?,434 ) {435 436 Val::Arr(body.fields().into_iter().map(Val::Str).collect())437 } else {438 panic!("bad objectFieldsEx call");439 }440 }441 (ns, name) => panic!("Intristic not found: {}.{}", ns, name),442 },443 Val::Func(f) => f.evaluate(444 args.clone()445 .into_iter()446 .map(move |a| {447 (448 a.clone().0,449 Val::Lazy(lazy_val!(450 closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))451 )),452 )453 })454 .collect(),455 )?,456 _ => panic!("{:?} is not a function", value),457 }458 }459 Function(params, body) => evaluate_method(context, body, params.clone()),460 AssertExpr(AssertStmt(value, msg), returned) => {461 if evaluate(context.clone(), &value)?462 .try_cast_bool("assertion condition should be boolean")?463 {464 evaluate(context, returned)?465 } else if let Some(msg) = msg {466 panic!(467 "assertion failed ({:?}): {}",468 value,469 evaluate(context, msg)?470 .try_cast_str("assertion message should be string")?471 );472 } else {473 panic!("assertion failed ({:?}): no message", value);474 }475 }476 Error(e) => create_error(crate::Error::RuntimeError(477 evaluate(context, e)?.try_cast_str("error text should be string")?,478 ))?,479 IfElse {480 cond,481 cond_then,482 cond_else,483 } => {484 if evaluate(context.clone(), &cond.0)?485 .try_cast_bool("if condition should be boolean")?486 {487 evaluate(context, cond_then)?488 } else {489 match cond_else {490 Some(v) => evaluate(context, v)?,491 None => Val::Bool(false),492 }493 }494 }495 _ => panic!(496 "evaluation not implemented: {:?}",497 LocExpr(expr.clone(), loc.clone())498 ),499 })500 })501}