1use crate::{2 binding, bool_val, context_creator, function_default, function_rhs, future_wrapper, lazy_val,3 Binding, Context, ContextCreator, FuncDesc, ObjMember, ObjValue, Val,4};5use closure::closure;6use jsonnet_parser::{7 ArgsDesc, BinaryOpType, BindSpec, Expr, FieldMember, LiteralType, Member, ObjBody, ParamsDesc,8 UnaryOpType, Visibility,9};10use std::{11 collections::{BTreeMap, HashMap},12 rc::Rc,13};1415pub fn evaluate_binding(b: &BindSpec, context_creator: ContextCreator) -> (String, Binding) {16 let b = b.clone();17 if let Some(args) = &b.params {18 let args = args.clone();19 (20 b.name.clone(),21 binding!(move |this, super_obj| Val::Lazy(lazy_val!(22 closure!(clone b, clone args, clone context_creator, || evaluate_method(23 context_creator.0(this.clone(), super_obj.clone()),24 &b.value,25 args.clone()26 ))27 ))),28 )29 } else {30 (31 b.name.clone(),32 binding!(move |this, super_obj| {33 println!("Evaluating binding");34 Val::Lazy(lazy_val!(35 closure!(clone context_creator, clone b, || evaluate(36 context_creator.0(this.clone(), super_obj.clone()),37 &b.value38 ))39 ))40 }),41 )42 }43}4445pub fn evaluate_method(ctx: Context, expr: &Expr, arg_spec: ParamsDesc) -> Val {46 Val::Func(FuncDesc {47 ctx,48 params: arg_spec,49 eval_rhs: function_rhs!(closure!(clone expr, |ctx| evaluate(ctx, &expr))),50 eval_default: function_default!(|ctx, default| evaluate(ctx, &default)),51 })52}5354pub fn evaluate_field_name(context: Context, field_name: &jsonnet_parser::FieldName) -> String {55 match field_name {56 jsonnet_parser::FieldName::Fixed(n) => n.clone(),57 jsonnet_parser::FieldName::Dyn(expr) => {58 let name = evaluate(context, expr).unwrap_if_lazy();59 match name {60 Val::Str(n) => n,61 _ => panic!(62 "dynamic field name can be only evaluated to 'string', got: {:?}",63 name64 ),65 }66 }67 }68}6970pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Val {71 match (op, b) {72 (o, Val::Lazy(l)) => evaluate_unary_op(o, &l.0()),73 (UnaryOpType::Not, Val::Literal(LiteralType::True)) => Val::Literal(LiteralType::False),74 (UnaryOpType::Not, Val::Literal(LiteralType::False)) => Val::Literal(LiteralType::True),75 (op, o) => panic!("unary op not implemented: {:?} {:?}", op, o),76 }77}7879pub fn evaluate_binary_op(a: &Val, op: BinaryOpType, b: &Val) -> Val {80 match (a, op, b) {81 (Val::Lazy(a), o, b) => evaluate_binary_op(&a.0(), o, b),82 (a, o, Val::Lazy(b)) => evaluate_binary_op(a, o, &b.0()),8384 (Val::Str(v1), BinaryOpType::Add, Val::Str(v2)) => Val::Str(v1.to_owned() + &v2),85 (Val::Str(v1), BinaryOpType::Eq, Val::Str(v2)) => bool_val(v1 == v2),86 (Val::Str(v1), BinaryOpType::Ne, Val::Str(v2)) => bool_val(v1 != v2),8788 (Val::Str(v1), BinaryOpType::Add, Val::Num(v2)) => Val::Str(format!("{}{}", v1, v2)),89 (Val::Str(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Str(v1.repeat(*v2 as usize)),9091 (Val::Obj(v1), BinaryOpType::Add, Val::Obj(v2)) => Val::Obj(v2.with_super(v1.clone())),9293 (Val::Arr(a), BinaryOpType::Add, Val::Arr(b)) => Val::Arr([&a[..], &b[..]].concat()),9495 (Val::Num(v1), BinaryOpType::Mul, Val::Num(v2)) => Val::Num(v1 * v2),96 (Val::Num(v1), BinaryOpType::Div, Val::Num(v2)) => Val::Num(v1 / v2),97 (Val::Num(v1), BinaryOpType::Mod, Val::Num(v2)) => Val::Num(v1 % v2),9899 (Val::Num(v1), BinaryOpType::Add, Val::Num(v2)) => Val::Num(v1 + v2),100 (Val::Num(v1), BinaryOpType::Sub, Val::Num(v2)) => Val::Num(v1 - v2),101102 (Val::Num(v1), BinaryOpType::Lhs, Val::Num(v2)) => {103 Val::Num(((*v1 as i32) << (*v2 as i32)) as f64)104 }105 (Val::Num(v1), BinaryOpType::Rhs, Val::Num(v2)) => {106 Val::Num(((*v1 as i32) >> (*v2 as i32)) as f64)107 }108109 (Val::Num(v1), BinaryOpType::Lt, Val::Num(v2)) => bool_val(v1 < v2),110 (Val::Num(v1), BinaryOpType::Gt, Val::Num(v2)) => bool_val(v1 > v2),111 (Val::Num(v1), BinaryOpType::Lte, Val::Num(v2)) => bool_val(v1 <= v2),112 (Val::Num(v1), BinaryOpType::Gte, Val::Num(v2)) => bool_val(v1 >= v2),113114 (Val::Num(v1), BinaryOpType::Eq, Val::Num(v2)) => bool_val((v1 - v2).abs() < f64::EPSILON),115 (Val::Num(v1), BinaryOpType::Ne, Val::Num(v2)) => bool_val((v1 - v2).abs() > f64::EPSILON),116117 (Val::Num(v1), BinaryOpType::BitAnd, Val::Num(v2)) => {118 Val::Num(((*v1 as i32) & (*v2 as i32)) as f64)119 }120 (Val::Num(v1), BinaryOpType::BitOr, Val::Num(v2)) => {121 Val::Num(((*v1 as i32) | (*v2 as i32)) as f64)122 }123 (Val::Num(v1), BinaryOpType::BitXor, Val::Num(v2)) => {124 Val::Num(((*v1 as i32) ^ (*v2 as i32)) as f64)125 }126 _ => panic!("no rules for binary operation: {:?} {:?} {:?}", a, op, b),127 }128}129130future_wrapper!(HashMap<String, Binding>, FutureNewBindings);131future_wrapper!(ObjValue, FutureObjValue);132133134pub fn evaluate_object(context: Context, object: ObjBody) -> ObjValue {135 match object {136 ObjBody::MemberList(members) => {137 let future_bindings = FutureNewBindings::new();138 let future_this = FutureObjValue::new();139 let context_creator = context_creator!(140 closure!(clone context, clone future_bindings, |this: Option<ObjValue>, super_obj: Option<ObjValue>| {141 println!("Context created");142 context.clone().extend(143 future_bindings.clone().unwrap(),144 context.clone().dollar().clone().or_else(||this.clone()),145 this,146 super_obj147 )148 })149 );150 let mut bindings: HashMap<String, Binding> = HashMap::new();151 for (n, b) in members152 .iter()153 .filter_map(|m| match m {154 Member::BindStmt(b) => Some(b.clone()),155 _ => None,156 })157 .map(|b| {158 evaluate_binding(&b, context_creator.clone())159 })160 {161 bindings.insert(n, b);162 }163 future_bindings.fill(bindings);164165 println!("Bindings filled");166 let mut new_members = BTreeMap::new();167 for member in members.into_iter() {168 match member {169 Member::Field(FieldMember {170 name,171 plus,172 params: None,173 visibility,174 value,175 }) => {176 let name = evaluate_field_name(context.clone(), &name);177 new_members.insert(178 name,179 ObjMember {180 add: plus,181 visibility: visibility.clone(),182 invoke: binding!(183 closure!(clone value, clone context_creator, clone future_this, |this, super_obj| {184 185 186 evaluate(187 context_creator.0(this, super_obj),188 &value,189 )190 })191 ),192 },193 );194 }195 Member::Field(FieldMember {196 name,197 params: Some(params),198 value,199 ..200 }) => {201 let name = evaluate_field_name(context.clone(), &name);202 new_members.insert(203 name,204 ObjMember {205 add: false,206 visibility: Visibility::Hidden,207 invoke: binding!(208 closure!(clone value, clone context_creator, clone future_this, |this, super_obj| {209 210 211 evaluate_method(212 context_creator.0(this, super_obj),213 &value.clone(),214 params.clone(),215 )216 })217 ),218 },219 );220 }221 Member::BindStmt(_) => {}222 Member::AssertStmt(_) => {}223 }224 }225 future_this.fill(ObjValue::new(None, Rc::new(new_members)))226 }227 _ => todo!(),228 }229}230231pub fn evaluate(context: Context, expr: &Expr) -> Val {232 use Expr::*;233 match &*expr {234 Literal(LiteralType::This) => {235 println!("{:?}", context.this());236 Val::Obj(237 context238 .this()239 .clone()240 .unwrap_or_else(|| panic!("this not found")),241 )242 }243 Literal(LiteralType::Super) => Val::Obj(244 context245 .super_obj()246 .clone()247 .unwrap_or_else(|| panic!("super not found")),248 ),249 Literal(t) => Val::Literal(t.clone()),250 Parened(e) => evaluate(context, e),251 Str(v) => Val::Str(v.clone()),252 Num(v) => Val::Num(*v),253 BinaryOp(v1, o, v2) => {254 evaluate_binary_op(&evaluate(context.clone(), v1), *o, &evaluate(context, v2))255 }256 UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(context, v)),257 Var(name) => {258 let variable = context.binding(&name);259 variable.0(None, None).unwrap_if_lazy()260 }261 Index(box value, box index) => {262 match (263 evaluate(context.clone(), value).unwrap_if_lazy(),264 evaluate(context, index),265 ) {266 (Val::Obj(v), Val::Str(s)) => v267 .get(&s)268 .unwrap_or_else(|| panic!("{} not found in {:?}", s, v)),269 (Val::Arr(v), Val::Num(n)) => v270 .get(n as usize)271 .unwrap_or_else(|| panic!("out of bounds"))272 .clone(),273 (v, i) => todo!("not implemented: {:?}[{:?}]", v, i.unwrap_if_lazy()),274 }275 }276 LocalExpr(bindings, returned) => {277 let mut new_bindings: HashMap<String, Binding> = HashMap::new();278 let future_context = Context::new_future();279280 let context_creator = context_creator!(281 closure!(clone future_context, |_, _| future_context.clone().unwrap())282 );283284 for (k, v) in bindings285 .iter()286 .map(move |b| evaluate_binding(b, context_creator.clone()))287 {288 new_bindings.insert(k, v);289 }290291 let context = context292 .extend(new_bindings, None, None, None)293 .into_future(future_context);294 evaluate(context, &*returned.clone())295 }296 Obj(body) => Val::Obj(evaluate_object(context, body.clone())),297 Apply(box value, ArgsDesc(args)) => {298 let value = evaluate(context.clone(), value).unwrap_if_lazy();299 match value {300 Val::Func(f) => f.evaluate(301 args.clone()302 .into_iter()303 .map(|a| {304 (305 a.clone().0,306 Val::Lazy(lazy_val!(307 closure!(clone context, clone a, || evaluate(context.clone(), &a.clone().1))308 )),309 )310 })311 .collect(),312 ),313 _ => panic!("{:?} is not a function", value),314 }315 }316 Function(params, body) => evaluate_method(context, body, params.clone()),317 Error(e) => panic!("error: {}", evaluate(context, e)),318 IfElse {319 cond,320 cond_then,321 cond_else,322 } => match evaluate(context.clone(), &cond.0).unwrap_if_lazy() {323 Val::Literal(LiteralType::True) => evaluate(context, cond_then),324 Val::Literal(LiteralType::False) => match cond_else {325 Some(v) => evaluate(context, v),326 None => Val::Literal(LiteralType::False),327 },328 v => panic!("if condition evaluated to {:?} (boolean needed instead)", v),329 },330 _ => panic!("evaluation not implemented: {:?}", expr),331 }332}