1use jrsonnet_parser::{BinaryOpType, LocExpr, UnaryOpType};23use crate::{4 builtin::std_format, error::Error::*, evaluate, throw, typed::Typed, val::equals, Context,5 Result, State, Val,6};78pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {9 use UnaryOpType::*;10 use Val::*;11 Ok(match (op, b) {12 (Not, Bool(v)) => Bool(!v),13 (Minus, Num(n)) => Num(-*n),14 (BitNot, Num(n)) => Num(!(*n as i32) as f64),15 (op, o) => throw!(UnaryOperatorDoesNotOperateOnType(op, o.value_type())),16 })17}1819pub fn evaluate_add_op(s: State, a: &Val, b: &Val) -> Result<Val> {20 use Val::*;21 Ok(match (a, b) {22 (Str(v1), Str(v2)) => Str(((**v1).to_owned() + v2).into()),2324 25 (Num(n), Str(o)) => Str(format!("{}{}", n, o).into()),26 (Str(o), Num(n)) => Str(format!("{}{}", o, n).into()),2728 (Str(a), o) => Str(format!("{}{}", a, o.clone().to_string(s)?).into()),29 (o, Str(a)) => Str(format!("{}{}", o.clone().to_string(s)?, a).into()),3031 (Obj(v1), Obj(v2)) => Obj(v2.extend_from(v1.clone())),32 (Arr(a), Arr(b)) => {33 let mut out = Vec::with_capacity(a.len() + b.len());34 out.extend(a.iter_lazy());35 out.extend(b.iter_lazy());36 Arr(out.into())37 }38 (Num(v1), Num(v2)) => Val::new_checked_num(v1 + v2)?,39 _ => throw!(BinaryOperatorDoesNotOperateOnValues(40 BinaryOpType::Add,41 a.value_type(),42 b.value_type(),43 )),44 })45}4647pub fn evaluate_mod_op(s: State, a: &Val, b: &Val) -> Result<Val> {48 use Val::*;49 match (a, b) {50 (Num(a), Num(b)) => Ok(Num(a % b)),51 (Str(str), vals) => {52 String::into_untyped(std_format(s.clone(), str.clone(), vals.clone())?, s)53 }54 (a, b) => throw!(BinaryOperatorDoesNotOperateOnValues(55 BinaryOpType::Mod,56 a.value_type(),57 b.value_type()58 )),59 }60}6162pub fn evaluate_binary_op_special(63 s: State,64 ctx: Context,65 a: &LocExpr,66 op: BinaryOpType,67 b: &LocExpr,68) -> Result<Val> {69 use BinaryOpType::*;70 use Val::*;71 Ok(match (evaluate(s.clone(), ctx.clone(), a)?, op, b) {72 (Bool(true), Or, _o) => Val::Bool(true),73 (Bool(false), And, _o) => Val::Bool(false),74 (a, op, eb) => evaluate_binary_op_normal(s.clone(), &a, op, &evaluate(s, ctx, eb)?)?,75 })76}7778pub fn evaluate_binary_op_normal(s: State, a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {79 use BinaryOpType::*;80 use Val::*;81 Ok(match (a, op, b) {82 (a, Add, b) => evaluate_add_op(s, a, b)?,8384 (a, Eq, b) => Bool(equals(s, a, b)?),85 (a, Neq, b) => Bool(!equals(s, a, b)?),8687 (Str(a), In, Obj(obj)) => Bool(obj.has_field_ex(a.clone(), true)),88 (a, Mod, b) => evaluate_mod_op(s, a, b)?,8990 (Str(v1), Mul, Num(v2)) => Str(v1.repeat(*v2 as usize).into()),9192 93 (Bool(a), And, Bool(b)) => Bool(*a && *b),94 (Bool(a), Or, Bool(b)) => Bool(*a || *b),9596 97 (Str(v1), Lt, Str(v2)) => Bool(v1 < v2),98 (Str(v1), Gt, Str(v2)) => Bool(v1 > v2),99 (Str(v1), Lte, Str(v2)) => Bool(v1 <= v2),100 (Str(v1), Gte, Str(v2)) => Bool(v1 >= v2),101102 103 (Num(v1), Mul, Num(v2)) => Val::new_checked_num(v1 * v2)?,104 (Num(v1), Div, Num(v2)) => {105 if *v2 <= f64::EPSILON {106 throw!(DivisionByZero)107 }108 Val::new_checked_num(v1 / v2)?109 }110111 (Num(v1), Sub, Num(v2)) => Val::new_checked_num(v1 - v2)?,112113 (Num(v1), Lt, Num(v2)) => Bool(v1 < v2),114 (Num(v1), Gt, Num(v2)) => Bool(v1 > v2),115 (Num(v1), Lte, Num(v2)) => Bool(v1 <= v2),116 (Num(v1), Gte, Num(v2)) => Bool(v1 >= v2),117118 (Num(v1), BitAnd, Num(v2)) => Num(((*v1 as i32) & (*v2 as i32)) as f64),119 (Num(v1), BitOr, Num(v2)) => Num(((*v1 as i32) | (*v2 as i32)) as f64),120 (Num(v1), BitXor, Num(v2)) => Num(((*v1 as i32) ^ (*v2 as i32)) as f64),121 (Num(v1), Lhs, Num(v2)) => {122 if *v2 < 0.0 {123 throw!(RuntimeError("shift by negative exponent".into()))124 }125 Num(((*v1 as i32) << (*v2 as i32)) as f64)126 }127 (Num(v1), Rhs, Num(v2)) => {128 if *v2 < 0.0 {129 throw!(RuntimeError("shift by negative exponent".into()))130 }131 Num(((*v1 as i32) >> (*v2 as i32)) as f64)132 }133134 _ => throw!(BinaryOperatorDoesNotOperateOnValues(135 op,136 a.value_type(),137 b.value_type(),138 )),139 })140}