1use std::cmp::Ordering;23use jrsonnet_parser::{BinaryOpType, LocExpr, UnaryOpType};45use crate::{6 arr::ArrValue,7 error::ErrorKind::*,8 evaluate,9 stdlib::std_format,10 throw,11 typed::Typed,12 val::{equals, StrValue},13 Context, Result, Val,14};1516pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {17 use UnaryOpType::*;18 use Val::*;19 Ok(match (op, b) {20 (Not, Bool(v)) => Bool(!v),21 (Minus, Num(n)) => Num(-*n),22 (BitNot, Num(n)) => Num(f64::from(!(*n as i32))),23 (op, o) => throw!(UnaryOperatorDoesNotOperateOnType(op, o.value_type())),24 })25}2627pub fn evaluate_add_op(a: &Val, b: &Val) -> Result<Val> {28 use Val::*;29 Ok(match (a, b) {30 (Str(v1), Str(v2)) => Str(StrValue::concat(v1.clone(), v2.clone())),3132 33 (Num(a), Str(b)) => Str(StrValue::Flat(format!("{a}{b}").into())),34 (Str(a), Num(b)) => Str(StrValue::Flat(format!("{a}{b}").into())),3536 (Str(a), o) | (o, Str(a)) if a.is_empty() => {37 Val::Str(StrValue::Flat(o.clone().to_string()?))38 }39 (Str(a), o) => Str(StrValue::Flat(40 format!("{a}{}", o.clone().to_string()?).into(),41 )),42 (o, Str(a)) => Str(StrValue::Flat(43 format!("{}{a}", o.clone().to_string()?).into(),44 )),4546 (Obj(v1), Obj(v2)) => Obj(v2.extend_from(v1.clone())),47 (Arr(a), Arr(b)) => Val::Arr(ArrValue::extended(a.clone(), b.clone())),4849 (Num(v1), Num(v2)) => Val::new_checked_num(v1 + v2)?,50 #[cfg(feature = "exp-bigint")]51 (BigInt(a), BigInt(b)) => BigInt(Box::new((&**a).clone() + (&**b).clone())),52 _ => throw!(BinaryOperatorDoesNotOperateOnValues(53 BinaryOpType::Add,54 a.value_type(),55 b.value_type(),56 )),57 })58}5960pub fn evaluate_mod_op(a: &Val, b: &Val) -> Result<Val> {61 use Val::*;62 match (a, b) {63 (Num(a), Num(b)) => {64 if *b == 0.0 {65 throw!(DivisionByZero)66 }67 Ok(Num(a % b))68 }69 (Str(str), vals) => {70 String::into_untyped(std_format(&str.clone().into_flat(), vals.clone())?)71 }72 (a, b) => throw!(BinaryOperatorDoesNotOperateOnValues(73 BinaryOpType::Mod,74 a.value_type(),75 b.value_type()76 )),77 }78}7980pub fn evaluate_binary_op_special(81 ctx: Context,82 a: &LocExpr,83 op: BinaryOpType,84 b: &LocExpr,85) -> Result<Val> {86 use BinaryOpType::*;87 use Val::*;88 Ok(match (evaluate(ctx.clone(), a)?, op, b) {89 (Bool(true), Or, _o) => Val::Bool(true),90 (Bool(false), And, _o) => Val::Bool(false),91 #[cfg(feature = "exp-null-coaelse")]92 (Null, NullCoaelse, eb) => evaluate(ctx, eb)?,93 #[cfg(feature = "exp-null-coaelse")]94 (a, NullCoaelse, _o) => a,95 (a, op, eb) => evaluate_binary_op_normal(&a, op, &evaluate(ctx, eb)?)?,96 })97}9899pub fn evaluate_compare_op(a: &Val, b: &Val, op: BinaryOpType) -> Result<Ordering> {100 use Val::*;101 Ok(match (a, b) {102 (Str(a), Str(b)) => a.cmp(b),103 (Num(a), Num(b)) => a.partial_cmp(b).expect("jsonnet numbers are non NaN"),104 #[cfg(feature = "exp-bigint")]105 (BigInt(a), BigInt(b)) => a.cmp(b),106 (Arr(a), Arr(b)) => {107 if let (Some(ai), Some(bi)) = (a.iter_cheap(), b.iter_cheap()) {108 for (a, b) in ai.zip(bi) {109 let ord = evaluate_compare_op(&a, &b, op)?;110 if !ord.is_eq() {111 return Ok(ord);112 }113 }114 } else {115 let ai = a.iter();116 let bi = b.iter();117118 for (a, b) in ai.zip(bi) {119 let ord = evaluate_compare_op(&a?, &b?, op)?;120 if !ord.is_eq() {121 return Ok(ord);122 }123 }124 }125 a.len().cmp(&b.len())126 }127 (_, _) => throw!(BinaryOperatorDoesNotOperateOnValues(128 op,129 a.value_type(),130 b.value_type()131 )),132 })133}134135pub fn evaluate_binary_op_normal(a: &Val, op: BinaryOpType, b: &Val) -> Result<Val> {136 use BinaryOpType::*;137 use Val::*;138 Ok(match (a, op, b) {139 (a, Add, b) => evaluate_add_op(a, b)?,140141 (a, Eq, b) => Bool(equals(a, b)?),142 (a, Neq, b) => Bool(!equals(a, b)?),143144 (a, Lt, b) => Bool(evaluate_compare_op(a, b, Lt)?.is_lt()),145 (a, Gt, b) => Bool(evaluate_compare_op(a, b, Gt)?.is_gt()),146 (a, Lte, b) => Bool(evaluate_compare_op(a, b, Lte)?.is_le()),147 (a, Gte, b) => Bool(evaluate_compare_op(a, b, Gte)?.is_ge()),148149 (Str(a), In, Obj(obj)) => Bool(obj.has_field_ex(a.clone().into_flat(), true)),150 (a, Mod, b) => evaluate_mod_op(a, b)?,151152 (Str(v1), Mul, Num(v2)) => Str(StrValue::Flat(v1.to_string().repeat(*v2 as usize).into())),153154 155 (Bool(a), And, Bool(b)) => Bool(*a && *b),156 (Bool(a), Or, Bool(b)) => Bool(*a || *b),157158 159 (Num(v1), Mul, Num(v2)) => Val::new_checked_num(v1 * v2)?,160 (Num(v1), Div, Num(v2)) => {161 if *v2 == 0.0 {162 throw!(DivisionByZero)163 }164 Val::new_checked_num(v1 / v2)?165 }166167 (Num(v1), Sub, Num(v2)) => Val::new_checked_num(v1 - v2)?,168169 (Num(v1), BitAnd, Num(v2)) => Num(f64::from((*v1 as i32) & (*v2 as i32))),170 (Num(v1), BitOr, Num(v2)) => Num(f64::from((*v1 as i32) | (*v2 as i32))),171 (Num(v1), BitXor, Num(v2)) => Num(f64::from((*v1 as i32) ^ (*v2 as i32))),172 (Num(v1), Lhs, Num(v2)) => {173 if *v2 < 0.0 {174 throw!("shift by negative exponent")175 }176 Num(f64::from((*v1 as i32) << (*v2 as i32)))177 }178 (Num(v1), Rhs, Num(v2)) => {179 if *v2 < 0.0 {180 throw!("shift by negative exponent")181 }182 Num(f64::from((*v1 as i32) >> (*v2 as i32)))183 }184185 186 #[cfg(feature = "exp-bigint")]187 (BigInt(a), Mul, BigInt(b)) => BigInt(Box::new((&**a).clone() * (&**b).clone())),188 #[cfg(feature = "exp-bigint")]189 (BigInt(a), Sub, BigInt(b)) => BigInt(Box::new((&**a).clone() - (&**b).clone())),190191 _ => throw!(BinaryOperatorDoesNotOperateOnValues(192 op,193 a.value_type(),194 b.value_type(),195 )),196 })197}