difftreelog
fix(evaluator) correct handling of true || expr
in: master
1 file changed
crates/jsonnet-evaluator/src/evaluate.rsdiffbeforeafterboth1use 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::Str(v1), v2) => Val::Str(format!("{}{:?}", v1, v2)),81 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),82 (Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),83 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),84 _ => panic!("can't add: {:?} and {:?}", a, b),85 })86}8788pub fn evaluate_binary_op(context: Context, a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {89 Ok(match (a, op, b) {90 (Val::Lazy(a), o, b) => evaluate_binary_op(context, &a.evaluate()?, o, b)?,91 (a, o, Val::Lazy(b)) => evaluate_binary_op(context, a, o, &b.evaluate()?)?,9293 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,9495 (Val::Str(v1), BinaryOpType::Ne, Val::Str(v2)) => Val::Bool(v1 != v2),9697 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),98 (Val::Str(format), BinaryOpType::Mod, args) => evaluate(99 context100 .with_var("__tmp__format__".to_owned(), Val::Str(format.to_owned()))?101 .with_var(102 "__tmp__args__".to_owned(),103 match args {104 Val::Arr(v) => Val::Arr(v.clone()),105 v => Val::Arr(vec![v.clone()]),106 },107 )?,108 &el!(Expr::Apply(109 el!(Expr::Index(110 el!(Expr::Var("std".to_owned())),111 el!(Expr::Str("format".to_owned()))112 )),113 ArgsDesc(vec![114 Arg(None, el!(Expr::Var("__tmp__format__".to_owned()))),115 Arg(None, el!(Expr::Var("__tmp__args__".to_owned())))116 ])117 )),118 )?,119120 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),121 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),122123 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),124 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => Val::Num(v1 / v2),125 (Val::Num(v1), BinaryOpType::Mod, Val::Num(v2)) => Val::Num(v1 % v2),126127 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),128129 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {130 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)131 }132 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {133 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)134 }135136 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),137 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),138 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),139 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),140141 (Val::Num(v1), BinaryOpType::Eq, Val::Num(v2)) => Val::Bool((v1 - v2).abs() < f64::EPSILON),142 (Val::Num(v1), BinaryOpType::Ne, Val::Num(v2)) => Val::Bool((v1 - v2).abs() > f64::EPSILON),143144 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {145 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)146 }147 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {148 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)149 }150 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {151 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)152 }153 (a, BinaryOpType::Eq, b) => Val::Bool(a == b),154 (a, BinaryOpType::Ne, b) => Val::Bool(a != b),155 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),156 })157}158159future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);160future_wrapper!(ObjValue, FutureObjValue);161162pub fn evaluate_comp(163 context: Context,164 value: &LocExpr,165 specs: &[CompSpec],166) -> Result<Option<Vec<Val>>> {167 Ok(match specs.get(0) {168 None => Some(vec![evaluate(context, &value)?]),169 Some(CompSpec::IfSpec(IfSpecData(cond))) => {170 if evaluate(context.clone(), &cond)?.try_cast_bool("if spec")? {171 evaluate_comp(context, value, &specs[1..])?172 } else {173 None174 }175 }176 Some(CompSpec::ForSpec(ForSpecData(var, expr))) => {177 match evaluate(context.clone(), &expr)?.unwrap_if_lazy()? {178 Val::Arr(list) => {179 let mut out = Vec::new();180 for item in list {181 let item = item.clone();182 out.push(evaluate_comp(183 context.with_var(var.clone(), item)?,184 value,185 &specs[1..],186 )?);187 }188 Some(out.iter().flatten().flatten().cloned().collect())189 }190 _ => panic!("for expression evaluated to non-iterable value"),191 }192 }193 })194}195196// TODO: Asserts197pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {198 Ok(match object {199 ObjBody::MemberList(members) => {200 let new_bindings = FutureNewBindings::new();201 let future_this = FutureObjValue::new();202 let context_creator = context_creator!(203 closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {204 Ok(context.clone().extend(205 new_bindings.clone().unwrap(),206 context.clone().dollar().clone().or_else(||this.clone()),207 Some(this.unwrap()),208 super_obj209 )?)210 })211 );212 {213 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();214 for (n, b) in members215 .iter()216 .filter_map(|m| match m {217 Member::BindStmt(b) => Some(b.clone()),218 _ => None,219 })220 .map(|b| evaluate_binding(&b, context_creator.clone()))221 {222 bindings.insert(n, b);223 }224 new_bindings.fill(bindings);225 }226227 let mut new_members = BTreeMap::new();228 for member in members.into_iter() {229 match member {230 Member::Field(FieldMember {231 name,232 plus,233 params: None,234 visibility,235 value,236 }) => {237 let name = evaluate_field_name(context.clone(), &name)?;238 new_members.insert(239 name.clone(),240 ObjMember {241 add: plus,242 visibility: visibility.clone(),243 invoke: binding!(244 closure!(clone name, clone value, clone context_creator, |this, super_obj| {245 push(value.clone(), "object ".to_owned()+&name+" field", ||{246 let context = context_creator.0(this, super_obj)?;247 evaluate(248 context,249 &value,250 )?.unwrap_if_lazy()251 })252 })253 ),254 },255 );256 }257 Member::Field(FieldMember {258 name,259 params: Some(params),260 value,261 ..262 }) => {263 let name = evaluate_field_name(context.clone(), &name)?;264 new_members.insert(265 name,266 ObjMember {267 add: false,268 visibility: Visibility::Hidden,269 invoke: binding!(270 closure!(clone value, clone context_creator, |this, super_obj| {271 // TODO: Assert272 Ok(evaluate_method(273 context_creator.0(this, super_obj)?,274 &value.clone(),275 params.clone(),276 ))277 })278 ),279 },280 );281 }282 Member::BindStmt(_) => {}283 Member::AssertStmt(_) => {}284 }285 }286 future_this.fill(ObjValue::new(None, Rc::new(new_members)))287 }288 _ => todo!(),289 })290}291292pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {293 use Expr::*;294 let locexpr = expr.clone();295 let LocExpr(expr, loc) = expr;296 Ok(match &**expr {297 Literal(LiteralType::This) => Val::Obj(298 context299 .this()300 .clone()301 .unwrap_or_else(|| panic!("this not found")),302 ),303 Literal(LiteralType::Super) => Val::Obj(304 context305 .super_obj()306 .clone()307 .unwrap_or_else(|| panic!("super not found")),308 ),309 Literal(LiteralType::True) => Val::Bool(true),310 Literal(LiteralType::False) => Val::Bool(false),311 Literal(LiteralType::Null) => Val::Null,312 Parened(e) => evaluate(context, e)?,313 Str(v) => Val::Str(v.clone()),314 Num(v) => Val::Num(*v),315 BinaryOp(v1, o, v2) => {316 let a = evaluate(context.clone(), v1)?.unwrap_if_lazy()?;317 let op = *o;318 let b = evaluate(context.clone(), v2)?.unwrap_if_lazy()?;319 evaluate_binary_op(context, &a, op, &b)?320 }321 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,322 Var(name) => Val::Lazy(context.binding(&name)).unwrap_if_lazy()?,323 Index(value, index) => {324 match (325 evaluate(context.clone(), value)?.unwrap_if_lazy()?,326 evaluate(context.clone(), index)?,327 ) {328 (Val::Obj(v), Val::Str(s)) => {329 if let Some(v) = v.get(&s)? {330 v.unwrap_if_lazy()?331 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {332 Val::Intristic(n, s)333 } else {334 create_error(crate::Error::NoSuchField(s))?335 }336 }337 (Val::Arr(v), Val::Num(n)) => v338 .get(n as usize)339 .unwrap_or_else(|| panic!("out of bounds"))340 .clone(),341 (Val::Str(s), Val::Num(n)) => {342 Val::Str(s.chars().skip(n as usize).take(1).collect())343 }344 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),345 }346 }347 LocalExpr(bindings, returned) => {348 let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();349 let future_context = Context::new_future();350351 let context_creator = context_creator!(352 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))353 );354355 for (k, v) in bindings356 .iter()357 .map(|b| evaluate_binding(b, context_creator.clone()))358 {359 new_bindings.insert(k, v);360 }361362 let context = context363 .extend(new_bindings, None, None, None)?364 .into_future(future_context);365 evaluate(context, &returned.clone())?366 }367 Arr(items) => {368 let mut out = Vec::with_capacity(items.len());369 for item in items {370 out.push(evaluate(context.clone(), item)?);371 }372 Val::Arr(out)373 }374 ArrComp(expr, compspecs) => Val::Arr(375 // First compspec should be forspec, so no "None" possible here376 evaluate_comp(context, expr, compspecs)?.unwrap(),377 ),378 Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),379 Apply(value, ArgsDesc(args)) => {380 let value = evaluate(context.clone(), value)?.unwrap_if_lazy()?;381 match value {382 Val::Intristic(ns, name) => match (&ns as &str, &name as &str) {383 // arr/string/function384 ("std", "length") => {385 assert_eq!(args.len(), 1);386 let expr = &args.get(0).unwrap().1;387 match evaluate(context, expr)? {388 Val::Str(n) => Val::Num(n.chars().count() as f64),389 Val::Arr(i) => Val::Num(i.len() as f64),390 v => panic!("can't get length of {:?}", v),391 }392 }393 // any394 ("std", "type") => {395 assert_eq!(args.len(), 1);396 let expr = &args.get(0).unwrap().1;397 Val::Str(evaluate(context, expr)?.value_type()?.name().to_owned())398 }399 // length, idx=>any400 ("std", "makeArray") => {401 assert_eq!(args.len(), 2);402 if let (Val::Num(v), Val::Func(d)) = (403 evaluate(context.clone(), &args[0].1)?,404 evaluate(context, &args[1].1)?,405 ) {406 assert!(v > 0.0);407 let mut out = Vec::with_capacity(v as usize);408 for i in 0..v as usize {409 out.push(d.evaluate(vec![(None, Val::Num(i as f64))])?)410 }411 Val::Arr(out)412 } else {413 panic!("bad makeArray call");414 }415 }416 // string417 ("std", "codepoint") => {418 assert_eq!(args.len(), 1);419 if let Val::Str(s) = evaluate(context, &args[0].1)? {420 assert!(421 s.chars().count() == 1,422 "std.codepoint should receive single char string"423 );424 Val::Num(s.chars().take(1).next().unwrap() as u32 as f64)425 } else {426 panic!("bad codepoint call");427 }428 }429 // object, includeHidden430 ("std", "objectFieldsEx") => {431 assert_eq!(args.len(), 2);432 if let (Val::Obj(body), Val::Bool(_include_hidden)) = (433 evaluate(context.clone(), &args[0].1)?,434 evaluate(context, &args[1].1)?,435 ) {436 // TODO: handle visibility (_include_hidden)437 Val::Arr(body.fields().into_iter().map(Val::Str).collect())438 } else {439 panic!("bad objectFieldsEx call");440 }441 }442 (ns, name) => panic!("Intristic not found: {}.{}", ns, name),443 },444 Val::Func(f) => push(locexpr.clone(), "function call".to_owned(), || {445 f.evaluate(446 args.clone()447 .into_iter()448 .map(move |a| {449 (450 a.clone().0,451 Val::Lazy(lazy_val!(452 closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))453 )),454 )455 })456 .collect(),457 )458 })?,459 _ => panic!("{:?} is not a function", value),460 }461 }462 Function(params, body) => evaluate_method(context, body, params.clone()),463 AssertExpr(AssertStmt(value, msg), returned) => {464 if push(value.clone(), "assertion condition".to_owned(), || {465 evaluate(context.clone(), &value)?466 .try_cast_bool("assertion condition should be boolean")467 })? {468 push(469 returned.clone(),470 "assert 'return' branch".to_owned(),471 || evaluate(context, returned),472 )?473 } else if let Some(msg) = msg {474 panic!(475 "assertion failed ({:?}): {}",476 value,477 evaluate(context, msg)?.try_cast_str("assertion message should be string")?478 );479 } else {480 panic!("assertion failed ({:?}): no message", value);481 }482 }483 Error(e) => create_error(crate::Error::RuntimeError(484 evaluate(context, e)?.try_cast_str("error text should be string")?,485 ))?,486 IfElse {487 cond,488 cond_then,489 cond_else,490 } => {491 if push(cond.0.clone(), "if condition".to_owned(), || {492 evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")493 })? {494 push(495 cond_then.clone(),496 "if condition 'then' branch".to_owned(),497 || evaluate(context, cond_then),498 )?499 } else {500 match cond_else {501 Some(v) => push(v.clone(), "if condition 'else' branch".to_owned(), || {502 evaluate(context, v)503 })?,504 None => Val::Bool(false),505 }506 }507 }508 _ => panic!(509 "evaluation not implemented: {:?}",510 LocExpr(expr.clone(), loc.clone())511 ),512 })513}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 (UnaryOpType::Minus, Val::Num(n)) => Val::Num(-*n),72 (UnaryOpType::BitNot, Val::Num(n)) => Val::Num(!(*n as i32) as f64),73 (op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),74 })75}7677pub(crate) fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {78 Ok(match (a, b) {79 (Val::Str(v1), Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),80 (Val::Str(v1), Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),81 (Val::Num(v1), Val::Str(v2)) => Val::Str(format!("{}{}", v1, v2)),82 (Val::Str(v1), v2) => Val::Str(format!("{}{:?}", v1, v2)),83 (Val::Obj(v1), Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),84 (Val::Arr(a), Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),85 (Val::Num(v1), Val::Num(v2)) => Val::Num(v1 + v2),86 _ => panic!("can't add: {:?} and {:?}", a, b),87 })88}8990pub fn evaluate_binary_op_special(91 context: Context,92 a: &LocExpr,93 op: BinaryOpType,94 b: &LocExpr,95) -> Result<Val> {96 Ok(97 match (evaluate(context.clone(), &a)?.unwrap_if_lazy()?, op, b) {98 (Val::Bool(true), BinaryOpType::Or, _o) => Val::Bool(true),99 (Val::Bool(false), BinaryOpType::And, _o) => Val::Bool(false),100 (a, op, eb) => evaluate_binary_op_normal(101 context.clone(),102 &a,103 op,104 &evaluate(context, eb)?.unwrap_if_lazy()?,105 )?,106 },107 )108}109110pub fn evaluate_binary_op_normal(111 context: Context,112 a: &Val,113 op: BinaryOpType,114 b: &Val,115) -> Result<Val> {116 Ok(match (a, op, b) {117 (a, BinaryOpType::Add, b) => evaluate_add_op(a, b)?,118119 (Val::Str(v1), BinaryOpType::Ne, Val::Str(v2)) => Val::Bool(v1 != v2),120121 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),122 (Val::Str(format), BinaryOpType::Mod, args) => evaluate(123 context124 .with_var("__tmp__format__".to_owned(), Val::Str(format.to_owned()))?125 .with_var(126 "__tmp__args__".to_owned(),127 match args {128 Val::Arr(v) => Val::Arr(v.clone()),129 v => Val::Arr(vec![v.clone()]),130 },131 )?,132 &el!(Expr::Apply(133 el!(Expr::Index(134 el!(Expr::Var("std".to_owned())),135 el!(Expr::Str("format".to_owned()))136 )),137 ArgsDesc(vec![138 Arg(None, el!(Expr::Var("__tmp__format__".to_owned()))),139 Arg(None, el!(Expr::Var("__tmp__args__".to_owned())))140 ])141 )),142 )?,143144 (Val::Bool(a), BinaryOpType::And, Val::Bool(b)) => Val::Bool(*a && *b),145 (Val::Bool(a), BinaryOpType::Or, Val::Bool(b)) => Val::Bool(*a || *b),146147 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),148 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => Val::Num(v1 / v2),149 (Val::Num(v1), BinaryOpType::Mod, Val::Num(v2)) => Val::Num(v1 % v2),150151 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),152153 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {154 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)155 }156 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {157 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)158 }159160 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => Val::Bool(v1 < v2),161 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => Val::Bool(v1 > v2),162 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => Val::Bool(v1 <= v2),163 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => Val::Bool(v1 >= v2),164165 (Val::Num(v1), BinaryOpType::Eq, Val::Num(v2)) => Val::Bool((v1 - v2).abs() < f64::EPSILON),166 (Val::Num(v1), BinaryOpType::Ne, Val::Num(v2)) => Val::Bool((v1 - v2).abs() > f64::EPSILON),167168 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {169 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)170 }171 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {172 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)173 }174 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {175 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)176 }177 (a, BinaryOpType::Eq, b) => Val::Bool(a == b),178 (a, BinaryOpType::Ne, b) => Val::Bool(a != b),179 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),180 })181}182183future_wrapper!(HashMap<String, LazyBinding>, FutureNewBindings);184future_wrapper!(ObjValue, FutureObjValue);185186pub fn evaluate_comp(187 context: Context,188 value: &LocExpr,189 specs: &[CompSpec],190) -> Result<Option<Vec<Val>>> {191 Ok(match specs.get(0) {192 None => Some(vec![evaluate(context, &value)?]),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 {205 let item = item.clone();206 out.push(evaluate_comp(207 context.with_var(var.clone(), item)?,208 value,209 &specs[1..],210 )?);211 }212 Some(out.iter().flatten().flatten().cloned().collect())213 }214 _ => panic!("for expression evaluated to non-iterable value"),215 }216 }217 })218}219220// TODO: Asserts221pub fn evaluate_object(context: Context, object: ObjBody) -> Result<ObjValue> {222 Ok(match object {223 ObjBody::MemberList(members) => {224 let new_bindings = FutureNewBindings::new();225 let future_this = FutureObjValue::new();226 let context_creator = context_creator!(227 closure!(clone context, clone new_bindings, clone future_this, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {228 Ok(context.clone().extend(229 new_bindings.clone().unwrap(),230 context.clone().dollar().clone().or_else(||this.clone()),231 Some(this.unwrap()),232 super_obj233 )?)234 })235 );236 {237 let mut bindings: HashMap<String, LazyBinding> = HashMap::new();238 for (n, b) in members239 .iter()240 .filter_map(|m| match m {241 Member::BindStmt(b) => Some(b.clone()),242 _ => None,243 })244 .map(|b| evaluate_binding(&b, context_creator.clone()))245 {246 bindings.insert(n, b);247 }248 new_bindings.fill(bindings);249 }250251 let mut new_members = BTreeMap::new();252 for member in members.into_iter() {253 match member {254 Member::Field(FieldMember {255 name,256 plus,257 params: None,258 visibility,259 value,260 }) => {261 let name = evaluate_field_name(context.clone(), &name)?;262 new_members.insert(263 name.clone(),264 ObjMember {265 add: plus,266 visibility: visibility.clone(),267 invoke: binding!(268 closure!(clone name, clone value, clone context_creator, |this, super_obj| {269 push(value.clone(), "object ".to_owned()+&name+" field", ||{270 let context = context_creator.0(this, super_obj)?;271 evaluate(272 context,273 &value,274 )?.unwrap_if_lazy()275 })276 })277 ),278 },279 );280 }281 Member::Field(FieldMember {282 name,283 params: Some(params),284 value,285 ..286 }) => {287 let name = evaluate_field_name(context.clone(), &name)?;288 new_members.insert(289 name,290 ObjMember {291 add: false,292 visibility: Visibility::Hidden,293 invoke: binding!(294 closure!(clone value, clone context_creator, |this, super_obj| {295 // TODO: Assert296 Ok(evaluate_method(297 context_creator.0(this, super_obj)?,298 &value.clone(),299 params.clone(),300 ))301 })302 ),303 },304 );305 }306 Member::BindStmt(_) => {}307 Member::AssertStmt(_) => {}308 }309 }310 future_this.fill(ObjValue::new(None, Rc::new(new_members)))311 }312 _ => todo!(),313 })314}315316pub fn evaluate(context: Context, expr: &LocExpr) -> Result<Val> {317 use Expr::*;318 let locexpr = expr.clone();319 let LocExpr(expr, loc) = expr;320 Ok(match &**expr {321 Literal(LiteralType::This) => Val::Obj(322 context323 .this()324 .clone()325 .unwrap_or_else(|| panic!("this not found")),326 ),327 Literal(LiteralType::Super) => Val::Obj(328 context329 .super_obj()330 .clone()331 .unwrap_or_else(|| panic!("super not found")),332 ),333 Literal(LiteralType::Dollar) => Val::Obj(334 context335 .dollar()336 .clone()337 .unwrap_or_else(|| panic!("dollar not found")),338 ),339 Literal(LiteralType::True) => Val::Bool(true),340 Literal(LiteralType::False) => Val::Bool(false),341 Literal(LiteralType::Null) => Val::Null,342 Parened(e) => evaluate(context, e)?,343 Str(v) => Val::Str(v.clone()),344 Num(v) => Val::Num(*v),345 BinaryOp(v1, o, v2) => evaluate_binary_op_special(context, &v1, *o, &v2)?,346 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)?)?,347 Var(name) => Val::Lazy(context.binding(&name)).unwrap_if_lazy()?,348 Index(value, index) => {349 match (350 evaluate(context.clone(), value)?.unwrap_if_lazy()?,351 evaluate(context.clone(), index)?,352 ) {353 (Val::Obj(v), Val::Str(s)) => {354 if let Some(v) = v.get(&s)? {355 v.unwrap_if_lazy()?356 } else if let Some(Val::Str(n)) = v.get("__intristic_namespace__")? {357 Val::Intristic(n, s)358 } else {359 create_error(crate::Error::NoSuchField(s))?360 }361 }362 (Val::Arr(v), Val::Num(n)) => v363 .get(n as usize)364 .unwrap_or_else(|| panic!("out of bounds"))365 .clone(),366 (Val::Str(s), Val::Num(n)) => {367 Val::Str(s.chars().skip(n as usize).take(1).collect())368 }369 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),370 }371 }372 LocalExpr(bindings, returned) => {373 let mut new_bindings: HashMap<String, LazyBinding> = HashMap::new();374 let future_context = Context::new_future();375376 let context_creator = context_creator!(377 closure!(clone future_context, |_, _| Ok(future_context.clone().unwrap()))378 );379380 for (k, v) in bindings381 .iter()382 .map(|b| evaluate_binding(b, context_creator.clone()))383 {384 new_bindings.insert(k, v);385 }386387 let context = context388 .extend(new_bindings, None, None, None)?389 .into_future(future_context);390 evaluate(context, &returned.clone())?391 }392 Arr(items) => {393 let mut out = Vec::with_capacity(items.len());394 for item in items {395 out.push(evaluate(context.clone(), item)?);396 }397 Val::Arr(out)398 }399 ArrComp(expr, compspecs) => Val::Arr(400 // First compspec should be forspec, so no "None" possible here401 evaluate_comp(context, expr, compspecs)?.unwrap(),402 ),403 Obj(body) => Val::Obj(evaluate_object(context, body.clone())?),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 // arr/string/function409 ("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 // any419 ("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 // length, idx=>any425 ("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 // string442 ("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 // object, includeHidden455 ("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 // TODO: handle visibility (_include_hidden)462 Val::Arr(body.fields().into_iter().map(Val::Str).collect())463 } else {464 panic!("bad objectFieldsEx call");465 }466 }467 (ns, name) => panic!("Intristic not found: {}.{}", ns, name),468 },469 Val::Func(f) => push(locexpr.clone(), "function call".to_owned(), || {470 f.evaluate(471 args.clone()472 .into_iter()473 .map(move |a| {474 (475 a.clone().0,476 Val::Lazy(lazy_val!(477 closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))478 )),479 )480 })481 .collect(),482 )483 })?,484 _ => panic!("{:?} is not a function", value),485 }486 }487 Function(params, body) => evaluate_method(context, body, params.clone()),488 AssertExpr(AssertStmt(value, msg), returned) => {489 if push(value.clone(), "assertion condition".to_owned(), || {490 evaluate(context.clone(), &value)?491 .try_cast_bool("assertion condition should be boolean")492 })? {493 push(494 returned.clone(),495 "assert 'return' branch".to_owned(),496 || evaluate(context, returned),497 )?498 } else if let Some(msg) = msg {499 panic!(500 "assertion failed ({:?}): {}",501 value,502 evaluate(context, msg)?.try_cast_str("assertion message should be string")?503 );504 } else {505 panic!("assertion failed ({:?}): no message", value);506 }507 }508 Error(e) => create_error(crate::Error::RuntimeError(509 evaluate(context, e)?.try_cast_str("error text should be string")?,510 ))?,511 IfElse {512 cond,513 cond_then,514 cond_else,515 } => {516 if push(cond.0.clone(), "if condition".to_owned(), || {517 evaluate(context.clone(), &cond.0)?.try_cast_bool("if condition should be boolean")518 })? {519 push(520 cond_then.clone(),521 "if condition 'then' branch".to_owned(),522 || evaluate(context, cond_then),523 )?524 } else {525 match cond_else {526 Some(v) => push(v.clone(), "if condition 'else' branch".to_owned(), || {527 evaluate(context, v)528 })?,529 None => Val::Bool(false),530 }531 }532 }533 _ => panic!(534 "evaluation not implemented: {:?}",535 LocExpr(expr.clone(), loc.clone())536 ),537 })538}