1use std::cmp::Ordering;23use jrsonnet_parser::{BinaryOpType, LocExpr, UnaryOpType};45use crate::{6 arr::ArrValue, error::ErrorKind::*, evaluate, stdlib::std_format, throw, typed::Typed,7 val::equals, Context, Result, 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}20212223pub fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {24 use Val::*;25 Ok(match (a, b) {26 (Str(a), Str(b)) if a.is_empty() => Val::Str(b.clone()),27 (Str(a), Str(b)) if b.is_empty() => Val::Str(a.clone()),28 (Str(v1), Str(v2)) => Str(((**v1).to_owned() + v2).into()),2930 31 (Num(a), Str(b)) => Str(format!("{a}{b}").into()),32 (Str(a), Num(b)) => Str(format!("{a}{b}").into()),3334 (Str(a), o) | (o, Str(a)) if a.is_empty() => Val::Str(o.clone().to_string()?),35 (Str(a), o) => Str(format!("{a}{}", o.clone().to_string()?).into()),36 (o, Str(a)) => Str(format!("{}{a}", o.clone().to_string()?).into()),3738 (Obj(v1), Obj(v2)) => Obj(v2.extend_from(v1.clone())),39 (Arr(a), Arr(b)) => Val::Arr(ArrValue::extended(a.clone(), b.clone())),4041 (Num(v1), Num(v2)) => Val::new_checked_num(v1 + v2)?,42 _ => throw!(BinaryOperatorDoesNotOperateOnValues(43 BinaryOpType::Add,44 a.value_type(),45 b.value_type(),46 )),47 })48}4950pub fn evaluate_mod_op(a: &Val, b: &Val) -> Result<Val> {51 use Val::*;52 match (a, b) {53 (Num(a), Num(b)) => {54 if *b == 0.0 {55 throw!(DivisionByZero)56 }57 Ok(Num(a % b))58 }59 (Str(str), vals) => String::into_untyped(std_format(str.clone(), vals.clone())?),60 (a, b) => throw!(BinaryOperatorDoesNotOperateOnValues(61 BinaryOpType::Mod,62 a.value_type(),63 b.value_type()64 )),65 }66}6768pub fn evaluate_binary_op_special(69 ctx: Context,70 a: &LocExpr,71 op: BinaryOpType,72 b: &LocExpr,73) -> Result<Val> {74 use BinaryOpType::*;75 use Val::*;76 Ok(match (evaluate(ctx.clone(), a)?, op, b) {77 (Bool(true), Or, _o) => Val::Bool(true),78 (Bool(false), And, _o) => Val::Bool(false),79 (a, op, eb) => evaluate_binary_op_normal(&a, op, &evaluate(ctx, eb)?)?,80 })81}8283pub fn evaluate_compare_op(a: &Val, b: &Val, op: BinaryOpType) -> Result<Ordering> {84 use Val::*;85 Ok(match (a, b) {86 (Str(a), Str(b)) => a.cmp(b),87 (Num(a), Num(b)) => a.partial_cmp(b).expect("jsonnet numbers are non NaN"),88 (Arr(a), Arr(b)) => {89 let ai = a.iter();90 let bi = b.iter();9192 for (a, b) in ai.zip(bi) {93 let ord = evaluate_compare_op(&a?, &b?, op)?;94 if !ord.is_eq() {95 return Ok(ord);96 }97 }9899 a.len().cmp(&b.len())100 }101 (_, _) => throw!(BinaryOperatorDoesNotOperateOnValues(102 op,103 a.value_type(),104 b.value_type()105 )),106 })107}108109pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {110 use BinaryOpType::*;111 use Val::*;112 Ok(match (a, op, b) {113 (a, Add, b) => evaluate_add_op(a, b)?,114115 (a, Eq, b) => Bool(equals(a, b)?),116 (a, Neq, b) => Bool(!equals(a, b)?),117118 (a, Lt, b) => Bool(evaluate_compare_op(a, b, Lt)?.is_lt()),119 (a, Gt, b) => Bool(evaluate_compare_op(a, b, Gt)?.is_gt()),120 (a, Lte, b) => Bool(evaluate_compare_op(a, b, Lte)?.is_le()),121 (a, Gte, b) => Bool(evaluate_compare_op(a, b, Gte)?.is_ge()),122123 (Str(a), In, Obj(obj)) => Bool(obj.has_field_ex(a.clone(), true)),124 (a, Mod, b) => evaluate_mod_op(a, b)?,125126 (Str(v1), Mul, Num(v2)) => Str(v1.repeat(*v2 as usize).into()),127128 129 (Bool(a), And, Bool(b)) => Bool(*a && *b),130 (Bool(a), Or, Bool(b)) => Bool(*a || *b),131132 133 (Num(v1), Mul, Num(v2)) => Val::new_checked_num(v1 * v2)?,134 (Num(v1), Div, Num(v2)) => {135 if *v2 == 0.0 {136 throw!(DivisionByZero)137 }138 Val::new_checked_num(v1 / v2)?139 }140141 (Num(v1), Sub, Num(v2)) => Val::new_checked_num(v1 - v2)?,142143 (Num(v1), BitAnd, Num(v2)) => Num(f64::from((*v1 as i32) & (*v2 as i32))),144 (Num(v1), BitOr, Num(v2)) => Num(f64::from((*v1 as i32) | (*v2 as i32))),145 (Num(v1), BitXor, Num(v2)) => Num(f64::from((*v1 as i32) ^ (*v2 as i32))),146 (Num(v1), Lhs, Num(v2)) => {147 if *v2 < 0.0 {148 throw!("shift by negative exponent")149 }150 Num(f64::from((*v1 as i32) << (*v2 as i32)))151 }152 (Num(v1), Rhs, Num(v2)) => {153 if *v2 < 0.0 {154 throw!("shift by negative exponent")155 }156 Num(f64::from((*v1 as i32) >> (*v2 as i32)))157 }158159 _ => throw!(BinaryOperatorDoesNotOperateOnValues(160 op,161 a.value_type(),162 b.value_type(),163 )),164 })165}