1use std::convert::TryInto;23use jrsonnet_parser::{BinaryOpType, LocExpr, UnaryOpType};45use crate::{6 builtin::std_format, error::Error::*, evaluate, throw, val::equals, Context, Result, Val,7};89pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {10 use UnaryOpType::*;11 use Val::*;12 Ok(match (op, b) {13 (Not, Bool(v)) => Bool(!v),14 (Minus, Num(n)) => Num(-*n),15 (BitNot, Num(n)) => Num(!(*n as i32) as f64),16 (op, o) => throw!(UnaryOperatorDoesNotOperateOnType(op, o.value_type())),17 })18}1920pub fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {21 use Val::*;22 Ok(match (a, b) {23 (Str(v1), Str(v2)) => Str(((**v1).to_owned() + v2).into()),2425 26 (Num(n), Str(o)) => Str(format!("{}{}", n, o).into()),27 (Str(o), Num(n)) => Str(format!("{}{}", o, n).into()),2829 (Str(s), o) => Str(format!("{}{}", s, o.clone().to_string()?).into()),30 (o, Str(s)) => Str(format!("{}{}", o.clone().to_string()?, s).into()),3132 (Obj(v1), Obj(v2)) => Obj(v2.extend_from(v1.clone())),33 (Arr(a), Arr(b)) => {34 let mut out = Vec::with_capacity(a.len() + b.len());35 out.extend(a.iter_lazy());36 out.extend(b.iter_lazy());37 Arr(out.into())38 }39 (Num(v1), Num(v2)) => Val::new_checked_num(v1 + v2)?,40 _ => throw!(BinaryOperatorDoesNotOperateOnValues(41 BinaryOpType::Add,42 a.value_type(),43 b.value_type(),44 )),45 })46}4748pub fn evaluate_mod_op(a: &Val, b: &Val) -> Result<Val> {49 use Val::*;50 match (a, b) {51 (Num(a), Num(b)) => Ok(Num(a % b)),52 (Str(str), vals) => std_format(str.clone(), vals.clone())?.try_into(),53 (a, b) => throw!(BinaryOperatorDoesNotOperateOnValues(54 BinaryOpType::Mod,55 a.value_type(),56 b.value_type()57 )),58 }59}6061pub fn evaluate_binary_op_special(62 context: Context,63 a: &LocExpr,64 op: BinaryOpType,65 b: &LocExpr,66) -> Result<Val> {67 use BinaryOpType::*;68 use Val::*;69 Ok(match (evaluate(context.clone(), a)?, op, b) {70 (Bool(true), Or, _o) => Val::Bool(true),71 (Bool(false), And, _o) => Val::Bool(false),72 (a, op, eb) => evaluate_binary_op_normal(&a, op, &evaluate(context, eb)?)?,73 })74}7576pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {77 use BinaryOpType::*;78 use Val::*;79 Ok(match (a, op, b) {80 (a, Add, b) => evaluate_add_op(a, b)?,8182 (a, Eq, b) => Bool(equals(a, b)?),83 (a, Neq, b) => Bool(!equals(a, b)?),8485 (Str(a), In, Obj(obj)) => Bool(obj.has_field_ex(a.clone(), true)),86 (a, Mod, b) => evaluate_mod_op(a, b)?,8788 (Str(v1), Mul, Num(v2)) => Str(v1.repeat(*v2 as usize).into()),8990 91 (Bool(a), And, Bool(b)) => Bool(*a && *b),92 (Bool(a), Or, Bool(b)) => Bool(*a || *b),9394 95 (Str(v1), Lt, Str(v2)) => Bool(v1 < v2),96 (Str(v1), Gt, Str(v2)) => Bool(v1 > v2),97 (Str(v1), Lte, Str(v2)) => Bool(v1 <= v2),98 (Str(v1), Gte, Str(v2)) => Bool(v1 >= v2),99100 101 (Num(v1), Mul, Num(v2)) => Val::new_checked_num(v1 * v2)?,102 (Num(v1), Div, Num(v2)) => {103 if *v2 <= f64::EPSILON {104 throw!(DivisionByZero)105 }106 Val::new_checked_num(v1 / v2)?107 }108109 (Num(v1), Sub, Num(v2)) => Val::new_checked_num(v1 - v2)?,110111 (Num(v1), Lt, Num(v2)) => Bool(v1 < v2),112 (Num(v1), Gt, Num(v2)) => Bool(v1 > v2),113 (Num(v1), Lte, Num(v2)) => Bool(v1 <= v2),114 (Num(v1), Gte, Num(v2)) => Bool(v1 >= v2),115116 (Num(v1), BitAnd, Num(v2)) => Num(((*v1 as i32) & (*v2 as i32)) as f64),117 (Num(v1), BitOr, Num(v2)) => Num(((*v1 as i32) | (*v2 as i32)) as f64),118 (Num(v1), BitXor, Num(v2)) => Num(((*v1 as i32) ^ (*v2 as i32)) as f64),119 (Num(v1), Lhs, Num(v2)) => {120 if *v2 < 0.0 {121 throw!(RuntimeError("shift by negative exponent".into()))122 }123 Num(((*v1 as i32) << (*v2 as i32)) as f64)124 }125 (Num(v1), Rhs, Num(v2)) => {126 if *v2 < 0.0 {127 throw!(RuntimeError("shift by negative exponent".into()))128 }129 Num(((*v1 as i32) >> (*v2 as i32)) as f64)130 }131132 _ => throw!(BinaryOperatorDoesNotOperateOnValues(133 op,134 a.value_type(),135 b.value_type(),136 )),137 })138}