1use std::cmp::Ordering;23use jrsonnet_parser::{BinaryOpType, LocExpr, UnaryOpType};45use crate::{6 error::Error::*, evaluate, stdlib::std_format, throw, typed::Typed, val::equals, Context,7 Result, State, Val,8};910pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {11 use UnaryOpType::*;12 use Val::*;13 Ok(match (op, b) {14 (Not, Bool(v)) => Bool(!v),15 (Minus, Num(n)) => Num(-*n),16 (BitNot, Num(n)) => Num(f64::from(!(*n as i32))),17 (op, o) => throw!(UnaryOperatorDoesNotOperateOnType(op, o.value_type())),18 })19}2021pub fn evaluate_add_op(s: State, a: &Val, b: &Val) -> Result<Val> {22 use Val::*;23 Ok(match (a, b) {24 (Str(a), Str(b)) if a.is_empty() => Val::Str(b.clone()),25 (Str(a), Str(b)) if b.is_empty() => Val::Str(a.clone()),26 (Str(v1), Str(v2)) => Str(((**v1).to_owned() + v2).into()),2728 29 (Num(a), Str(b)) => Str(format!("{a}{b}").into()),30 (Str(a), Num(b)) => Str(format!("{a}{b}").into()),3132 (Str(a), o) | (o, Str(a)) if a.is_empty() => Val::Str(o.clone().to_string(s)?),33 (Str(a), o) => Str(format!("{a}{}", o.clone().to_string(s)?).into()),34 (o, Str(a)) => Str(format!("{}{a}", o.clone().to_string(s)?).into()),3536 (Obj(v1), Obj(v2)) => Obj(v2.extend_from(v1.clone())),37 (Arr(a), Arr(b)) => {38 let mut out = Vec::with_capacity(a.len() + b.len());39 out.extend(a.iter_lazy());40 out.extend(b.iter_lazy());41 Arr(out.into())42 }43 (Num(v1), Num(v2)) => Val::new_checked_num(v1 + v2)?,44 _ => throw!(BinaryOperatorDoesNotOperateOnValues(45 BinaryOpType::Add,46 a.value_type(),47 b.value_type(),48 )),49 })50}5152pub fn evaluate_mod_op(s: State, a: &Val, b: &Val) -> Result<Val> {53 use Val::*;54 match (a, b) {55 (Num(a), Num(b)) => {56 if *b == 0.0 {57 throw!(DivisionByZero)58 }59 Ok(Num(a % b))60 }61 (Str(str), vals) => {62 String::into_untyped(std_format(s.clone(), str.clone(), vals.clone())?, s)63 }64 (a, b) => throw!(BinaryOperatorDoesNotOperateOnValues(65 BinaryOpType::Mod,66 a.value_type(),67 b.value_type()68 )),69 }70}7172pub fn evaluate_binary_op_special(73 s: State,74 ctx: Context,75 a: &LocExpr,76 op: BinaryOpType,77 b: &LocExpr,78) -> Result<Val> {79 use BinaryOpType::*;80 use Val::*;81 Ok(match (evaluate(s.clone(), ctx.clone(), a)?, op, b) {82 (Bool(true), Or, _o) => Val::Bool(true),83 (Bool(false), And, _o) => Val::Bool(false),84 (a, op, eb) => evaluate_binary_op_normal(s.clone(), &a, op, &evaluate(s, ctx, eb)?)?,85 })86}8788pub fn evaluate_compare_op(s: State, a: &Val, op: BinaryOpType, b: &Val) -> Result<Ordering> {89 use Val::*;90 Ok(match (a, b) {91 (Str(a), Str(b)) => a.cmp(b),92 (Num(a), Num(b)) => a.partial_cmp(b).expect("jsonnet numbers are non NaN"),93 (Arr(a), Arr(b)) => {94 let ai = a.iter(s.clone());95 let bi = b.iter(s.clone());9697 for (a, b) in ai.zip(bi) {98 let ord = evaluate_compare_op(s.clone(), &a?, op, &b?)?;99 if !ord.is_eq() {100 return Ok(ord);101 }102 }103104 a.len().cmp(&b.len())105 }106 (_, _) => throw!(BinaryOperatorDoesNotOperateOnValues(107 op,108 a.value_type(),109 b.value_type()110 )),111 })112}113114pub fn evaluate_binary_op_normal(s: State, a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {115 use BinaryOpType::*;116 use Val::*;117 Ok(match (a, op, b) {118 (a, Add, b) => evaluate_add_op(s, a, b)?,119120 (a, Eq, b) => Bool(equals(s, a, b)?),121 (a, Neq, b) => Bool(!equals(s, a, b)?),122123 (a, Lt, b) => Bool(evaluate_compare_op(s, a, Lt, b)?.is_lt()),124 (a, Gt, b) => Bool(evaluate_compare_op(s, a, Gt, b)?.is_gt()),125 (a, Lte, b) => Bool(evaluate_compare_op(s, a, Lte, b)?.is_le()),126 (a, Gte, b) => Bool(evaluate_compare_op(s, a, Gte, b)?.is_ge()),127128 (Str(a), In, Obj(obj)) => Bool(obj.has_field_ex(a.clone(), true)),129 (a, Mod, b) => evaluate_mod_op(s, a, b)?,130131 (Str(v1), Mul, Num(v2)) => Str(v1.repeat(*v2 as usize).into()),132133 134 (Bool(a), And, Bool(b)) => Bool(*a && *b),135 (Bool(a), Or, Bool(b)) => Bool(*a || *b),136137 138 (Num(v1), Mul, Num(v2)) => Val::new_checked_num(v1 * v2)?,139 (Num(v1), Div, Num(v2)) => {140 if *v2 == 0.0 {141 throw!(DivisionByZero)142 }143 Val::new_checked_num(v1 / v2)?144 }145146 (Num(v1), Sub, Num(v2)) => Val::new_checked_num(v1 - v2)?,147148 (Num(v1), BitAnd, Num(v2)) => Num(f64::from((*v1 as i32) & (*v2 as i32))),149 (Num(v1), BitOr, Num(v2)) => Num(f64::from((*v1 as i32) | (*v2 as i32))),150 (Num(v1), BitXor, Num(v2)) => Num(f64::from((*v1 as i32) ^ (*v2 as i32))),151 (Num(v1), Lhs, Num(v2)) => {152 if *v2 < 0.0 {153 throw!(RuntimeError("shift by negative exponent".into()))154 }155 Num(f64::from((*v1 as i32) << (*v2 as i32)))156 }157 (Num(v1), Rhs, Num(v2)) => {158 if *v2 < 0.0 {159 throw!(RuntimeError("shift by negative exponent".into()))160 }161 Num(f64::from((*v1 as i32) >> (*v2 as i32)))162 }163164 _ => throw!(BinaryOperatorDoesNotOperateOnValues(165 op,166 a.value_type(),167 b.value_type(),168 )),169 })170}