difftreelog
feat(parser) impl Display for op types
in: master
1 file changed
crates/jrsonnet-parser/src/expr.rsdiffbeforeafterboth1#[cfg(feature = "deserialize")]2use serde::Deserialize;3#[cfg(feature = "serialize")]4use serde::Serialize;5use std::{6 fmt::{Debug, Display},7 ops::Deref,8 path::PathBuf,9 rc::Rc,10};11#[cfg(feature = "dump")]12use structdump_derive::Codegen;1314#[cfg_attr(feature = "dump", derive(Codegen))]15#[cfg_attr(feature = "serialize", derive(Serialize))]16#[cfg_attr(feature = "deserialize", derive(Deserialize))]17#[derive(Debug, PartialEq)]18pub enum FieldName {19 /// {fixed: 2}20 Fixed(Rc<str>),21 /// {["dyn"+"amic"]: 3}22 Dyn(LocExpr),23}2425#[cfg_attr(feature = "dump", derive(Codegen))]26#[cfg_attr(feature = "serialize", derive(Serialize))]27#[cfg_attr(feature = "deserialize", derive(Deserialize))]28#[derive(Debug, Clone, Copy, PartialEq)]29pub enum Visibility {30 /// :31 Normal,32 /// ::33 Hidden,34 /// :::35 Unhide,36}3738#[cfg_attr(feature = "dump", derive(Codegen))]39#[cfg_attr(feature = "serialize", derive(Serialize))]40#[cfg_attr(feature = "deserialize", derive(Deserialize))]41#[derive(Debug, PartialEq)]42pub struct AssertStmt(pub LocExpr, pub Option<LocExpr>);4344#[cfg_attr(feature = "dump", derive(Codegen))]45#[cfg_attr(feature = "serialize", derive(Serialize))]46#[cfg_attr(feature = "deserialize", derive(Deserialize))]47#[derive(Debug, PartialEq)]48pub struct FieldMember {49 pub name: FieldName,50 pub plus: bool,51 pub params: Option<ParamsDesc>,52 pub visibility: Visibility,53 pub value: LocExpr,54}5556#[cfg_attr(feature = "dump", derive(Codegen))]57#[cfg_attr(feature = "serialize", derive(Serialize))]58#[cfg_attr(feature = "deserialize", derive(Deserialize))]59#[derive(Debug, PartialEq)]60pub enum Member {61 Field(FieldMember),62 BindStmt(BindSpec),63 AssertStmt(AssertStmt),64}6566#[cfg_attr(feature = "dump", derive(Codegen))]67#[cfg_attr(feature = "serialize", derive(Serialize))]68#[cfg_attr(feature = "deserialize", derive(Deserialize))]69#[derive(Debug, Clone, Copy, PartialEq)]70pub enum UnaryOpType {71 Plus,72 Minus,73 BitNot,74 Not,75}76impl Display for UnaryOpType {77 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {78 use UnaryOpType::*;79 write!(80 f,81 "{}",82 match self {83 Plus => "+",84 Minus => "-",85 BitNot => "~",86 Not => "!",87 }88 )89 }90}9192#[cfg_attr(feature = "dump", derive(Codegen))]93#[cfg_attr(feature = "serialize", derive(Serialize))]94#[cfg_attr(feature = "deserialize", derive(Deserialize))]95#[derive(Debug, Clone, Copy, PartialEq)]96pub enum BinaryOpType {97 Mul,98 Div,99100 Add,101 Sub,102103 Lhs,104 Rhs,105106 Lt,107 Gt,108 Lte,109 Gte,110111 BitAnd,112 BitOr,113 BitXor,114115 And,116 Or,117}118impl Display for BinaryOpType {119 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {120 use BinaryOpType::*;121 write!(122 f,123 "{}",124 match self {125 Mul => "*",126 Div => "/",127 Add => "+",128 Sub => "-",129 Lhs => "<<",130 Rhs => ">>",131 Lt => "<",132 Gt => ">",133 Lte => "<=",134 Gte => ">=",135 BitAnd => "&",136 BitOr => "|",137 BitXor => "^",138 And => "&&",139 Or => "||",140 }141 )142 }143}144145/// name, default value146#[cfg_attr(feature = "dump", derive(Codegen))]147#[cfg_attr(feature = "serialize", derive(Serialize))]148#[cfg_attr(feature = "deserialize", derive(Deserialize))]149#[derive(Debug, PartialEq)]150pub struct Param(pub Rc<str>, pub Option<LocExpr>);151152/// Defined function parameters153#[cfg_attr(feature = "dump", derive(Codegen))]154#[cfg_attr(feature = "serialize", derive(Serialize))]155#[cfg_attr(feature = "deserialize", derive(Deserialize))]156#[derive(Debug, Clone, PartialEq)]157pub struct ParamsDesc(pub Rc<Vec<Param>>);158impl Deref for ParamsDesc {159 type Target = Vec<Param>;160 fn deref(&self) -> &Self::Target {161 &self.0162 }163}164165#[cfg_attr(feature = "dump", derive(Codegen))]166#[cfg_attr(feature = "serialize", derive(Serialize))]167#[cfg_attr(feature = "deserialize", derive(Deserialize))]168#[derive(Debug, PartialEq)]169pub struct Arg(pub Option<String>, pub LocExpr);170171#[cfg_attr(feature = "dump", derive(Codegen))]172#[cfg_attr(feature = "serialize", derive(Serialize))]173#[cfg_attr(feature = "deserialize", derive(Deserialize))]174#[derive(Debug, PartialEq)]175pub struct ArgsDesc(pub Vec<Arg>);176impl Deref for ArgsDesc {177 type Target = Vec<Arg>;178 fn deref(&self) -> &Self::Target {179 &self.0180 }181}182183#[cfg_attr(feature = "dump", derive(Codegen))]184#[cfg_attr(feature = "serialize", derive(Serialize))]185#[cfg_attr(feature = "deserialize", derive(Deserialize))]186#[derive(Debug, Clone, PartialEq)]187pub struct BindSpec {188 pub name: Rc<str>,189 pub params: Option<ParamsDesc>,190 pub value: LocExpr,191}192193#[cfg_attr(feature = "dump", derive(Codegen))]194#[cfg_attr(feature = "serialize", derive(Serialize))]195#[cfg_attr(feature = "deserialize", derive(Deserialize))]196#[derive(Debug, PartialEq)]197pub struct IfSpecData(pub LocExpr);198199#[cfg_attr(feature = "dump", derive(Codegen))]200#[cfg_attr(feature = "serialize", derive(Serialize))]201#[cfg_attr(feature = "deserialize", derive(Deserialize))]202#[derive(Debug, PartialEq)]203pub struct ForSpecData(pub Rc<str>, pub LocExpr);204205#[cfg_attr(feature = "dump", derive(Codegen))]206#[cfg_attr(feature = "serialize", derive(Serialize))]207#[cfg_attr(feature = "deserialize", derive(Deserialize))]208#[derive(Debug, PartialEq)]209pub enum CompSpec {210 IfSpec(IfSpecData),211 ForSpec(ForSpecData),212}213214#[cfg_attr(feature = "dump", derive(Codegen))]215#[cfg_attr(feature = "serialize", derive(Serialize))]216#[cfg_attr(feature = "deserialize", derive(Deserialize))]217#[derive(Debug, PartialEq)]218pub struct ObjComp {219 pub pre_locals: Vec<BindSpec>,220 pub key: LocExpr,221 pub value: LocExpr,222 pub post_locals: Vec<BindSpec>,223 pub compspecs: Vec<CompSpec>,224}225226#[cfg_attr(feature = "dump", derive(Codegen))]227#[cfg_attr(feature = "serialize", derive(Serialize))]228#[cfg_attr(feature = "deserialize", derive(Deserialize))]229#[derive(Debug, PartialEq)]230pub enum ObjBody {231 MemberList(Vec<Member>),232 ObjComp(ObjComp),233}234235#[cfg_attr(feature = "dump", derive(Codegen))]236#[cfg_attr(feature = "serialize", derive(Serialize))]237#[cfg_attr(feature = "deserialize", derive(Deserialize))]238#[derive(Debug, PartialEq, Clone, Copy)]239pub enum LiteralType {240 This,241 Super,242 Dollar,243 Null,244 True,245 False,246}247248#[derive(Debug, PartialEq)]249pub struct SliceDesc {250 pub start: Option<LocExpr>,251 pub end: Option<LocExpr>,252 pub step: Option<LocExpr>,253}254255/// Syntax base256#[cfg_attr(feature = "dump", derive(Codegen))]257#[cfg_attr(feature = "serialize", derive(Serialize))]258#[cfg_attr(feature = "deserialize", derive(Deserialize))]259#[derive(Debug, PartialEq)]260pub enum Expr {261 Literal(LiteralType),262263 /// String value: "hello"264 Str(Rc<str>),265 /// Number: 1, 2.0, 2e+20266 Num(f64),267 /// Variable name: test268 Var(Rc<str>),269270 /// Array of expressions: [1, 2, "Hello"]271 Arr(Vec<LocExpr>),272 /// Array comprehension:273 /// ```jsonnet274 /// ingredients: [275 /// { kind: kind, qty: 4 / 3 }276 /// for kind in [277 /// 'Honey Syrup',278 /// 'Lemon Juice',279 /// 'Farmers Gin',280 /// ]281 /// ],282 /// ```283 ArrComp(LocExpr, Vec<CompSpec>),284285 /// Object: {a: 2}286 Obj(ObjBody),287 /// Object extension: var1 {b: 2}288 ObjExtend(LocExpr, ObjBody),289290 /// (obj)291 Parened(LocExpr),292293 /// -2294 UnaryOp(UnaryOpType, LocExpr),295 /// 2 - 2296 BinaryOp(LocExpr, BinaryOpType, LocExpr),297 /// assert 2 == 2 : "Math is broken"298 AssertExpr(AssertStmt, LocExpr),299 /// local a = 2; { b: a }300 LocalExpr(Vec<BindSpec>, LocExpr),301302 /// import "hello"303 Import(PathBuf),304 /// importStr "file.txt"305 ImportStr(PathBuf),306 /// error "I'm broken"307 ErrorStmt(LocExpr),308 /// a(b, c)309 Apply(LocExpr, ArgsDesc, bool),310 /// a[b]311 Index(LocExpr, LocExpr),312 /// function(x) x313 Function(ParamsDesc, LocExpr),314 /// if true == false then 1 else 2315 IfElse {316 cond: IfSpecData,317 cond_then: LocExpr,318 cond_else: Option<LocExpr>,319 },320}321322/// file, begin offset, end offset323#[cfg_attr(feature = "dump", derive(Codegen))]324#[cfg_attr(feature = "serialize", derive(Serialize))]325#[cfg_attr(feature = "deserialize", derive(Deserialize))]326#[derive(Clone, PartialEq)]327pub struct ExprLocation(pub Rc<PathBuf>, pub usize, pub usize);328impl Debug for ExprLocation {329 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {330 write!(f, "{:?}:{:?}-{:?}", self.0, self.1, self.2)331 }332}333334/// Holds AST expression and its location in source file335#[cfg_attr(feature = "dump", derive(Codegen))]336#[cfg_attr(feature = "serialize", derive(Serialize))]337#[cfg_attr(feature = "deserialize", derive(Deserialize))]338#[derive(Clone, PartialEq)]339pub struct LocExpr(pub Rc<Expr>, pub Option<ExprLocation>);340impl Debug for LocExpr {341 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {342 write!(f, "{:?} from {:?}", self.0, self.1)343 }344}345346/// Creates LocExpr from Expr and ExprLocation components347#[macro_export]348macro_rules! loc_expr {349 ($expr:expr, $need_loc:expr,($name:expr, $start:expr, $end:expr)) => {350 LocExpr(351 std::rc::Rc::new($expr),352 if $need_loc {353 Some(ExprLocation($name, $start, $end))354 } else {355 None356 },357 )358 };359}360361/// Creates LocExpr without location info362#[macro_export]363macro_rules! loc_expr_todo {364 ($expr:expr) => {365 LocExpr(Rc::new($expr), None)366 };367}