difftreelog
fix lazy array initialization
in: master
2 files changed
crates/jrsonnet-evaluator/src/evaluate/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/evaluate/mod.rs
+++ b/crates/jrsonnet-evaluator/src/evaluate/mod.rs
@@ -18,8 +18,8 @@
throw,
typed::Typed,
val::{CachedUnbound, IndexableVal, StrValue, Thunk, ThunkValue},
- Context, GcHashMap, ObjValue, ObjValueBuilder, ObjectAssertion, Pending, Result, State,
- Unbound, Val,
+ Context, GcHashMap, ObjValue, ObjValueBuilder, ObjectAssertion, Pending, Result, ResultExt,
+ State, Unbound, Val,
};
pub mod destructure;
pub mod operator;
@@ -134,13 +134,13 @@
let fctx = Pending::new();
let mut new_bindings = GcHashMap::with_capacity(var.capacity_hint());
- let value = Thunk::evaluated(Val::Arr(ArrValue::lazy(Cc::new(vec![
+ let value = Thunk::evaluated(Val::Arr(ArrValue::lazy(vec![
Thunk::evaluated(Val::Str(StrValue::Flat(field.clone()))),
Thunk::new(ObjectFieldThunk {
field: field.clone(),
obj: obj.clone(),
}),
- ]))));
+ ])));
destruct(var, value, fctx.clone(), &mut new_bindings)?;
let ctx = ctx
.clone()
crates/jrsonnet-parser/src/expr.rsdiffbeforeafterboth1use std::{2 fmt::{self, Debug, Display},3 ops::Deref,4 rc::Rc,5};67use jrsonnet_gcmodule::Trace;8use jrsonnet_interner::IStr;9#[cfg(feature = "serde")]10use serde::{Deserialize, Serialize};11#[cfg(feature = "structdump")]12use structdump::Codegen;1314use crate::source::Source;1516#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]17#[cfg_attr(feature = "structdump", derive(Codegen))]18#[derive(Debug, PartialEq, Trace)]19pub enum FieldName {20 /// {fixed: 2}21 Fixed(IStr),22 /// {["dyn"+"amic"]: 3}23 Dyn(LocExpr),24}2526#[cfg_attr(feature = "structdump", derive(Codegen))]27#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]28#[derive(Debug, Clone, Copy, PartialEq, Eq, Trace)]29pub enum Visibility {30 /// :31 Normal,32 /// ::33 Hidden,34 /// :::35 Unhide,36}3738impl Visibility {39 pub fn is_visible(&self) -> bool {40 matches!(self, Self::Normal | Self::Unhide)41 }42}4344#[cfg_attr(feature = "structdump", derive(Codegen))]45#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]46#[derive(Clone, Debug, PartialEq, Trace)]47pub struct AssertStmt(pub LocExpr, pub Option<LocExpr>);4849#[cfg_attr(feature = "structdump", derive(Codegen))]50#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]51#[derive(Debug, PartialEq, Trace)]52pub struct FieldMember {53 pub name: FieldName,54 pub plus: bool,55 pub params: Option<ParamsDesc>,56 pub visibility: Visibility,57 pub value: LocExpr,58}5960#[cfg_attr(feature = "structdump", derive(Codegen))]61#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]62#[derive(Debug, PartialEq, Trace)]63pub enum Member {64 Field(FieldMember),65 BindStmt(BindSpec),66 AssertStmt(AssertStmt),67}6869#[cfg_attr(feature = "structdump", derive(Codegen))]70#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]71#[derive(Debug, Clone, Copy, PartialEq, Eq, Trace)]72pub enum UnaryOpType {73 Plus,74 Minus,75 BitNot,76 Not,77}7879impl Display for UnaryOpType {80 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {81 use UnaryOpType::*;82 write!(83 f,84 "{}",85 match self {86 Plus => "+",87 Minus => "-",88 BitNot => "~",89 Not => "!",90 }91 )92 }93}9495#[cfg_attr(feature = "structdump", derive(Codegen))]96#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]97#[derive(Debug, Clone, Copy, PartialEq, Eq, Trace)]98pub enum BinaryOpType {99 Mul,100 Div,101102 /// Implemented as intrinsic, put here for completeness103 Mod,104105 Add,106 Sub,107108 Lhs,109 Rhs,110111 Lt,112 Gt,113 Lte,114 Gte,115116 BitAnd,117 BitOr,118 BitXor,119120 Eq,121 Neq,122123 And,124 Or,125 #[cfg(feature = "exp-null-coaelse")]126 NullCoaelse,127128 // Equialent to std.objectHasEx(a, b, true)129 In,130}131132impl Display for BinaryOpType {133 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {134 use BinaryOpType::*;135 write!(136 f,137 "{}",138 match self {139 Mul => "*",140 Div => "/",141 Mod => "%",142 Add => "+",143 Sub => "-",144 Lhs => "<<",145 Rhs => ">>",146 Lt => "<",147 Gt => ">",148 Lte => "<=",149 Gte => ">=",150 BitAnd => "&",151 BitOr => "|",152 BitXor => "^",153 Eq => "==",154 Neq => "!=",155 And => "&&",156 Or => "||",157 In => "in",158 #[cfg(feature = "exp-null-coaelse")]159 NullCoaelse => "??",160 }161 )162 }163}164165/// name, default value166#[cfg_attr(feature = "structdump", derive(Codegen))]167#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]168#[derive(Debug, PartialEq, Trace)]169pub struct Param(pub Destruct, pub Option<LocExpr>);170171/// Defined function parameters172#[cfg_attr(feature = "structdump", derive(Codegen))]173#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]174#[derive(Debug, Clone, PartialEq, Trace)]175pub struct ParamsDesc(pub Rc<Vec<Param>>);176177impl Deref for ParamsDesc {178 type Target = Vec<Param>;179 fn deref(&self) -> &Self::Target {180 &self.0181 }182}183184#[cfg_attr(feature = "structdump", derive(Codegen))]185#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]186#[derive(Debug, PartialEq, Trace)]187pub struct ArgsDesc {188 pub unnamed: Vec<LocExpr>,189 pub named: Vec<(IStr, LocExpr)>,190}191impl ArgsDesc {192 pub fn new(unnamed: Vec<LocExpr>, named: Vec<(IStr, LocExpr)>) -> Self {193 Self { unnamed, named }194 }195}196197#[cfg_attr(feature = "structdump", derive(Codegen))]198#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]199#[derive(Debug, Clone, PartialEq, Eq, Trace)]200pub enum DestructRest {201 /// ...rest202 Keep(IStr),203 /// ...204 Drop,205}206207#[cfg_attr(feature = "structdump", derive(Codegen))]208#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]209#[derive(Debug, Clone, PartialEq, Trace)]210pub enum Destruct {211 Full(IStr),212 #[cfg(feature = "exp-destruct")]213 Skip,214 #[cfg(feature = "exp-destruct")]215 Array {216 start: Vec<Destruct>,217 rest: Option<DestructRest>,218 end: Vec<Destruct>,219 },220 #[cfg(feature = "exp-destruct")]221 Object {222 fields: Vec<(IStr, Option<Destruct>, Option<LocExpr>)>,223 rest: Option<DestructRest>,224 },225}226impl Destruct {227 /// Name of destructure, used for function parameter names228 pub fn name(&self) -> Option<IStr> {229 match self {230 Self::Full(name) => Some(name.clone()),231 #[cfg(feature = "exp-destruct")]232 _ => None,233 }234 }235 pub fn capacity_hint(&self) -> usize {236 #[cfg(feature = "exp-destruct")]237 fn cap_rest(rest: &Option<DestructRest>) -> usize {238 match rest {239 Some(DestructRest::Keep(_)) => 1,240 Some(DestructRest::Drop) => 0,241 None => 0,242 }243 }244 match self {245 Self::Full(_) => 1,246 #[cfg(feature = "exp-destruct")]247 Self::Skip => 0,248 #[cfg(feature = "exp-destruct")]249 Self::Array { start, rest, end } => {250 start.iter().map(Destruct::capacity_hint).sum::<usize>()251 + end.iter().map(Destruct::capacity_hint).sum::<usize>()252 + cap_rest(rest)253 }254 #[cfg(feature = "exp-destruct")]255 Self::Object { fields, rest } => {256 let mut out = 0;257 for (_, into, _) in fields {258 match into {259 Some(v) => out += v.capacity_hint(),260 // Field is destructured to default name261 None => out += 1,262 }263 }264 out + cap_rest(rest)265 }266 }267 }268}269270#[cfg_attr(feature = "structdump", derive(Codegen))]271#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]272#[derive(Debug, Clone, PartialEq, Trace)]273pub enum BindSpec {274 Field {275 into: Destruct,276 value: LocExpr,277 },278 Function {279 name: IStr,280 params: ParamsDesc,281 value: LocExpr,282 },283}284impl BindSpec {285 pub fn capacity_hint(&self) -> usize {286 match self {287 BindSpec::Field { into, .. } => into.capacity_hint(),288 BindSpec::Function { .. } => 1,289 }290 }291}292293#[cfg_attr(feature = "structdump", derive(Codegen))]294#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]295#[derive(Debug, PartialEq, Trace)]296pub struct IfSpecData(pub LocExpr);297298#[cfg_attr(feature = "structdump", derive(Codegen))]299#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]300#[derive(Debug, PartialEq, Trace)]301pub struct ForSpecData(pub Destruct, pub LocExpr);302303#[cfg_attr(feature = "structdump", derive(Codegen))]304#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]305#[derive(Debug, PartialEq, Trace)]306pub enum CompSpec {307 IfSpec(IfSpecData),308 ForSpec(ForSpecData),309}310311#[cfg_attr(feature = "structdump", derive(Codegen))]312#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]313#[derive(Debug, PartialEq, Trace)]314pub struct ObjComp {315 pub pre_locals: Vec<BindSpec>,316 pub field: FieldMember,317 pub post_locals: Vec<BindSpec>,318 pub compspecs: Vec<CompSpec>,319}320321#[cfg_attr(feature = "structdump", derive(Codegen))]322#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]323#[derive(Debug, PartialEq, Trace)]324pub enum ObjBody {325 MemberList(Vec<Member>),326 ObjComp(ObjComp),327}328329#[cfg_attr(feature = "structdump", derive(Codegen))]330#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]331#[derive(Debug, PartialEq, Eq, Clone, Copy, Trace)]332pub enum LiteralType {333 This,334 Super,335 Dollar,336 Null,337 True,338 False,339}340341#[cfg_attr(feature = "structdump", derive(Codegen))]342#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]343#[derive(Debug, PartialEq, Trace)]344pub struct SliceDesc {345 pub start: Option<LocExpr>,346 pub end: Option<LocExpr>,347 pub step: Option<LocExpr>,348}349350/// Syntax base351#[cfg_attr(feature = "structdump", derive(Codegen))]352#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]353#[derive(Debug, PartialEq, Trace)]354pub enum Expr {355 Literal(LiteralType),356357 /// String value: "hello"358 Str(IStr),359 /// Number: 1, 2.0, 2e+20360 Num(f64),361 /// Variable name: test362 Var(IStr),363364 /// Array of expressions: [1, 2, "Hello"]365 Arr(Vec<LocExpr>),366 /// Array comprehension:367 /// ```jsonnet368 /// ingredients: [369 /// { kind: kind, qty: 4 / 3 }370 /// for kind in [371 /// 'Honey Syrup',372 /// 'Lemon Juice',373 /// 'Farmers Gin',374 /// ]375 /// ],376 /// ```377 ArrComp(LocExpr, Vec<CompSpec>),378379 /// Object: {a: 2}380 Obj(ObjBody),381 /// Object extension: var1 {b: 2}382 ObjExtend(LocExpr, ObjBody),383384 /// (obj)385 Parened(LocExpr),386387 /// -2388 UnaryOp(UnaryOpType, LocExpr),389 /// 2 - 2390 BinaryOp(LocExpr, BinaryOpType, LocExpr),391 /// assert 2 == 2 : "Math is broken"392 AssertExpr(AssertStmt, LocExpr),393 /// local a = 2; { b: a }394 LocalExpr(Vec<BindSpec>, LocExpr),395396 /// import "hello"397 Import(LocExpr),398 /// importStr "file.txt"399 ImportStr(LocExpr),400 /// importBin "file.txt"401 ImportBin(LocExpr),402 /// error "I'm broken"403 ErrorStmt(LocExpr),404 /// a(b, c)405 Apply(LocExpr, ArgsDesc, bool),406 /// a[b], a.b, a?.b407 Index {408 indexable: LocExpr,409 parts: Vec<IndexPart>,410 },411 /// function(x) x412 Function(ParamsDesc, LocExpr),413 /// if true == false then 1 else 2414 IfElse {415 cond: IfSpecData,416 cond_then: LocExpr,417 cond_else: Option<LocExpr>,418 },419 Slice(LocExpr, SliceDesc),420}421422#[cfg_attr(feature = "structdump", derive(Codegen))]423#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]424#[derive(Debug, PartialEq, Trace)]425pub struct IndexPart {426 pub value: LocExpr,427 #[cfg(feature = "exp-null-coaelse")]428 pub null_coaelse: bool,429}430431/// file, begin offset, end offset432#[cfg_attr(feature = "structdump", derive(Codegen))]433#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]434#[derive(Clone, PartialEq, Eq, Trace)]435#[trace(skip)]436#[repr(C)]437pub struct ExprLocation(pub Source, pub u32, pub u32);438impl ExprLocation {439 pub fn belongs_to(&self, other: &ExprLocation) -> bool {440 other.0 == self.0 && other.1 <= self.1 && other.2 >= self.2441 }442}443444#[cfg(target_pointer_width = "64")]445static_assertions::assert_eq_size!(ExprLocation, [u8; 16]);446447impl Debug for ExprLocation {448 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {449 write!(f, "{:?}:{:?}-{:?}", self.0, self.1, self.2)450 }451}452453/// Holds AST expression and its location in source file454#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]455#[cfg_attr(feature = "structdump", derive(Codegen))]456#[derive(Clone, PartialEq, Trace)]457pub struct LocExpr(pub Rc<Expr>, pub ExprLocation);458459#[cfg(target_pointer_width = "64")]460static_assertions::assert_eq_size!(LocExpr, [u8; 24]);461462impl Debug for LocExpr {463 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {464 if f.alternate() {465 write!(f, "{:#?}", self.0)?;466 } else {467 write!(f, "{:?}", self.0)?;468 }469 write!(f, " from {:?}", self.1)?;470 Ok(())471 }472}1use std::{2 fmt::{self, Debug, Display},3 ops::Deref,4 rc::Rc,5};67use jrsonnet_gcmodule::Trace;8use jrsonnet_interner::IStr;9#[cfg(feature = "serde")]10use serde::{Deserialize, Serialize};11#[cfg(feature = "structdump")]12use structdump::Codegen;1314use crate::source::Source;1516#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]17#[cfg_attr(feature = "structdump", derive(Codegen))]18#[derive(Debug, PartialEq, Trace)]19pub enum FieldName {20 /// {fixed: 2}21 Fixed(IStr),22 /// {["dyn"+"amic"]: 3}23 Dyn(LocExpr),24}2526#[cfg_attr(feature = "structdump", derive(Codegen))]27#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]28#[derive(Debug, Clone, Copy, PartialEq, Eq, Trace)]29#[repr(u8)]30pub enum Visibility {31 /// :32 Normal,33 /// ::34 Hidden,35 /// :::36 Unhide,37}3839impl Visibility {40 pub fn is_visible(&self) -> bool {41 matches!(self, Self::Normal | Self::Unhide)42 }43}4445#[cfg_attr(feature = "structdump", derive(Codegen))]46#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]47#[derive(Clone, Debug, PartialEq, Trace)]48pub struct AssertStmt(pub LocExpr, pub Option<LocExpr>);4950#[cfg_attr(feature = "structdump", derive(Codegen))]51#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]52#[derive(Debug, PartialEq, Trace)]53pub struct FieldMember {54 pub name: FieldName,55 pub plus: bool,56 pub params: Option<ParamsDesc>,57 pub visibility: Visibility,58 pub value: LocExpr,59}6061#[cfg_attr(feature = "structdump", derive(Codegen))]62#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]63#[derive(Debug, PartialEq, Trace)]64pub enum Member {65 Field(FieldMember),66 BindStmt(BindSpec),67 AssertStmt(AssertStmt),68}6970#[cfg_attr(feature = "structdump", derive(Codegen))]71#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]72#[derive(Debug, Clone, Copy, PartialEq, Eq, Trace)]73pub enum UnaryOpType {74 Plus,75 Minus,76 BitNot,77 Not,78}7980impl Display for UnaryOpType {81 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {82 use UnaryOpType::*;83 write!(84 f,85 "{}",86 match self {87 Plus => "+",88 Minus => "-",89 BitNot => "~",90 Not => "!",91 }92 )93 }94}9596#[cfg_attr(feature = "structdump", derive(Codegen))]97#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]98#[derive(Debug, Clone, Copy, PartialEq, Eq, Trace)]99pub enum BinaryOpType {100 Mul,101 Div,102103 /// Implemented as intrinsic, put here for completeness104 Mod,105106 Add,107 Sub,108109 Lhs,110 Rhs,111112 Lt,113 Gt,114 Lte,115 Gte,116117 BitAnd,118 BitOr,119 BitXor,120121 Eq,122 Neq,123124 And,125 Or,126 #[cfg(feature = "exp-null-coaelse")]127 NullCoaelse,128129 // Equialent to std.objectHasEx(a, b, true)130 In,131}132133impl Display for BinaryOpType {134 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {135 use BinaryOpType::*;136 write!(137 f,138 "{}",139 match self {140 Mul => "*",141 Div => "/",142 Mod => "%",143 Add => "+",144 Sub => "-",145 Lhs => "<<",146 Rhs => ">>",147 Lt => "<",148 Gt => ">",149 Lte => "<=",150 Gte => ">=",151 BitAnd => "&",152 BitOr => "|",153 BitXor => "^",154 Eq => "==",155 Neq => "!=",156 And => "&&",157 Or => "||",158 In => "in",159 #[cfg(feature = "exp-null-coaelse")]160 NullCoaelse => "??",161 }162 )163 }164}165166/// name, default value167#[cfg_attr(feature = "structdump", derive(Codegen))]168#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]169#[derive(Debug, PartialEq, Trace)]170pub struct Param(pub Destruct, pub Option<LocExpr>);171172/// Defined function parameters173#[cfg_attr(feature = "structdump", derive(Codegen))]174#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]175#[derive(Debug, Clone, PartialEq, Trace)]176pub struct ParamsDesc(pub Rc<Vec<Param>>);177178impl Deref for ParamsDesc {179 type Target = Vec<Param>;180 fn deref(&self) -> &Self::Target {181 &self.0182 }183}184185#[cfg_attr(feature = "structdump", derive(Codegen))]186#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]187#[derive(Debug, PartialEq, Trace)]188pub struct ArgsDesc {189 pub unnamed: Vec<LocExpr>,190 pub named: Vec<(IStr, LocExpr)>,191}192impl ArgsDesc {193 pub fn new(unnamed: Vec<LocExpr>, named: Vec<(IStr, LocExpr)>) -> Self {194 Self { unnamed, named }195 }196}197198#[cfg_attr(feature = "structdump", derive(Codegen))]199#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]200#[derive(Debug, Clone, PartialEq, Eq, Trace)]201pub enum DestructRest {202 /// ...rest203 Keep(IStr),204 /// ...205 Drop,206}207208#[cfg_attr(feature = "structdump", derive(Codegen))]209#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]210#[derive(Debug, Clone, PartialEq, Trace)]211pub enum Destruct {212 Full(IStr),213 #[cfg(feature = "exp-destruct")]214 Skip,215 #[cfg(feature = "exp-destruct")]216 Array {217 start: Vec<Destruct>,218 rest: Option<DestructRest>,219 end: Vec<Destruct>,220 },221 #[cfg(feature = "exp-destruct")]222 Object {223 fields: Vec<(IStr, Option<Destruct>, Option<LocExpr>)>,224 rest: Option<DestructRest>,225 },226}227impl Destruct {228 /// Name of destructure, used for function parameter names229 pub fn name(&self) -> Option<IStr> {230 match self {231 Self::Full(name) => Some(name.clone()),232 #[cfg(feature = "exp-destruct")]233 _ => None,234 }235 }236 pub fn capacity_hint(&self) -> usize {237 #[cfg(feature = "exp-destruct")]238 fn cap_rest(rest: &Option<DestructRest>) -> usize {239 match rest {240 Some(DestructRest::Keep(_)) => 1,241 Some(DestructRest::Drop) => 0,242 None => 0,243 }244 }245 match self {246 Self::Full(_) => 1,247 #[cfg(feature = "exp-destruct")]248 Self::Skip => 0,249 #[cfg(feature = "exp-destruct")]250 Self::Array { start, rest, end } => {251 start.iter().map(Destruct::capacity_hint).sum::<usize>()252 + end.iter().map(Destruct::capacity_hint).sum::<usize>()253 + cap_rest(rest)254 }255 #[cfg(feature = "exp-destruct")]256 Self::Object { fields, rest } => {257 let mut out = 0;258 for (_, into, _) in fields {259 match into {260 Some(v) => out += v.capacity_hint(),261 // Field is destructured to default name262 None => out += 1,263 }264 }265 out + cap_rest(rest)266 }267 }268 }269}270271#[cfg_attr(feature = "structdump", derive(Codegen))]272#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]273#[derive(Debug, Clone, PartialEq, Trace)]274pub enum BindSpec {275 Field {276 into: Destruct,277 value: LocExpr,278 },279 Function {280 name: IStr,281 params: ParamsDesc,282 value: LocExpr,283 },284}285impl BindSpec {286 pub fn capacity_hint(&self) -> usize {287 match self {288 BindSpec::Field { into, .. } => into.capacity_hint(),289 BindSpec::Function { .. } => 1,290 }291 }292}293294#[cfg_attr(feature = "structdump", derive(Codegen))]295#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]296#[derive(Debug, PartialEq, Trace)]297pub struct IfSpecData(pub LocExpr);298299#[cfg_attr(feature = "structdump", derive(Codegen))]300#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]301#[derive(Debug, PartialEq, Trace)]302pub struct ForSpecData(pub Destruct, pub LocExpr);303304#[cfg_attr(feature = "structdump", derive(Codegen))]305#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]306#[derive(Debug, PartialEq, Trace)]307pub enum CompSpec {308 IfSpec(IfSpecData),309 ForSpec(ForSpecData),310}311312#[cfg_attr(feature = "structdump", derive(Codegen))]313#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]314#[derive(Debug, PartialEq, Trace)]315pub struct ObjComp {316 pub pre_locals: Vec<BindSpec>,317 pub field: FieldMember,318 pub post_locals: Vec<BindSpec>,319 pub compspecs: Vec<CompSpec>,320}321322#[cfg_attr(feature = "structdump", derive(Codegen))]323#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]324#[derive(Debug, PartialEq, Trace)]325pub enum ObjBody {326 MemberList(Vec<Member>),327 ObjComp(ObjComp),328}329330#[cfg_attr(feature = "structdump", derive(Codegen))]331#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]332#[derive(Debug, PartialEq, Eq, Clone, Copy, Trace)]333pub enum LiteralType {334 This,335 Super,336 Dollar,337 Null,338 True,339 False,340}341342#[cfg_attr(feature = "structdump", derive(Codegen))]343#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]344#[derive(Debug, PartialEq, Trace)]345pub struct SliceDesc {346 pub start: Option<LocExpr>,347 pub end: Option<LocExpr>,348 pub step: Option<LocExpr>,349}350351/// Syntax base352#[cfg_attr(feature = "structdump", derive(Codegen))]353#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]354#[derive(Debug, PartialEq, Trace)]355pub enum Expr {356 Literal(LiteralType),357358 /// String value: "hello"359 Str(IStr),360 /// Number: 1, 2.0, 2e+20361 Num(f64),362 /// Variable name: test363 Var(IStr),364365 /// Array of expressions: [1, 2, "Hello"]366 Arr(Vec<LocExpr>),367 /// Array comprehension:368 /// ```jsonnet369 /// ingredients: [370 /// { kind: kind, qty: 4 / 3 }371 /// for kind in [372 /// 'Honey Syrup',373 /// 'Lemon Juice',374 /// 'Farmers Gin',375 /// ]376 /// ],377 /// ```378 ArrComp(LocExpr, Vec<CompSpec>),379380 /// Object: {a: 2}381 Obj(ObjBody),382 /// Object extension: var1 {b: 2}383 ObjExtend(LocExpr, ObjBody),384385 /// (obj)386 Parened(LocExpr),387388 /// -2389 UnaryOp(UnaryOpType, LocExpr),390 /// 2 - 2391 BinaryOp(LocExpr, BinaryOpType, LocExpr),392 /// assert 2 == 2 : "Math is broken"393 AssertExpr(AssertStmt, LocExpr),394 /// local a = 2; { b: a }395 LocalExpr(Vec<BindSpec>, LocExpr),396397 /// import "hello"398 Import(LocExpr),399 /// importStr "file.txt"400 ImportStr(LocExpr),401 /// importBin "file.txt"402 ImportBin(LocExpr),403 /// error "I'm broken"404 ErrorStmt(LocExpr),405 /// a(b, c)406 Apply(LocExpr, ArgsDesc, bool),407 /// a[b], a.b, a?.b408 Index {409 indexable: LocExpr,410 parts: Vec<IndexPart>,411 },412 /// function(x) x413 Function(ParamsDesc, LocExpr),414 /// if true == false then 1 else 2415 IfElse {416 cond: IfSpecData,417 cond_then: LocExpr,418 cond_else: Option<LocExpr>,419 },420 Slice(LocExpr, SliceDesc),421}422423#[cfg_attr(feature = "structdump", derive(Codegen))]424#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]425#[derive(Debug, PartialEq, Trace)]426pub struct IndexPart {427 pub value: LocExpr,428 #[cfg(feature = "exp-null-coaelse")]429 pub null_coaelse: bool,430}431432/// file, begin offset, end offset433#[cfg_attr(feature = "structdump", derive(Codegen))]434#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]435#[derive(Clone, PartialEq, Eq, Trace)]436#[trace(skip)]437#[repr(C)]438pub struct ExprLocation(pub Source, pub u32, pub u32);439impl ExprLocation {440 pub fn belongs_to(&self, other: &ExprLocation) -> bool {441 other.0 == self.0 && other.1 <= self.1 && other.2 >= self.2442 }443}444445#[cfg(target_pointer_width = "64")]446static_assertions::assert_eq_size!(ExprLocation, [u8; 16]);447448impl Debug for ExprLocation {449 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {450 write!(f, "{:?}:{:?}-{:?}", self.0, self.1, self.2)451 }452}453454/// Holds AST expression and its location in source file455#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]456#[cfg_attr(feature = "structdump", derive(Codegen))]457#[derive(Clone, PartialEq, Trace)]458pub struct LocExpr(pub Rc<Expr>, pub ExprLocation);459460#[cfg(target_pointer_width = "64")]461static_assertions::assert_eq_size!(LocExpr, [u8; 24]);462463impl Debug for LocExpr {464 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {465 if f.alternate() {466 write!(f, "{:#?}", self.0)?;467 } else {468 write!(f, "{:?}", self.0)?;469 }470 write!(f, " from {:?}", self.1)?;471 Ok(())472 }473}