difftreelog
feat convert ParamName into enum
in: master
10 files changed
crates/jrsonnet-evaluator/src/evaluate/destructure.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/evaluate/destructure.rs
+++ b/crates/jrsonnet-evaluator/src/evaluate/destructure.rs
@@ -5,7 +5,7 @@
use crate::{
bail,
error::{ErrorKind::*, Result},
- evaluate, evaluate_method, evaluate_named, Context, Pending, Thunk, Val,
+ evaluate_method, evaluate_named_param, Context, Pending, Thunk, Val,
};
#[allow(clippy::too_many_lines)]
@@ -170,10 +170,7 @@
let value = value.clone();
let data = {
let fctx = fctx.clone();
- Thunk!(move || name.0.map_or_else(
- || evaluate(fctx.unwrap(), &value),
- |name| evaluate_named(fctx.unwrap(), &value, name),
- ))
+ Thunk!(move || evaluate_named_param(fctx.unwrap(), &value, name))
};
destruct(into, data, fctx, new_bindings)?;
}
crates/jrsonnet-evaluator/src/evaluate/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/evaluate/mod.rs
+++ b/crates/jrsonnet-evaluator/src/evaluate/mod.rs
@@ -380,9 +380,9 @@
}
pub fn evaluate_named_param(ctx: Context, expr: &Spanned<Expr>, name: ParamName) -> Result<Val> {
- match name.0 {
- Some(name) => evaluate_named(ctx, expr, name),
- None => evaluate(ctx, expr),
+ match name {
+ ParamName::Named(name) => evaluate_named(ctx, expr, name),
+ ParamName::Unnamed => evaluate(ctx, expr),
}
}
crates/jrsonnet-evaluator/src/function/builtin.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/function/builtin.rs
+++ b/crates/jrsonnet-evaluator/src/function/builtin.rs
@@ -1,8 +1,6 @@
use std::any::Any;
-use std::fmt;
-use jrsonnet_gcmodule::{cc_dyn, Acyclic, Trace, TraceBox};
-use jrsonnet_interner::IStr;
+use jrsonnet_gcmodule::{cc_dyn, Trace, TraceBox};
use jrsonnet_parser::function::{FunctionSignature, ParamDefault, ParamName, ParamParse};
use super::{arglike::ArgsLike, parse::parse_builtin_call, CallLocation};
@@ -10,8 +8,8 @@
#[macro_export]
macro_rules! params {
- (@name unnamed) => { ParamName::ANONYMOUS };
- (@name named $name:literal) => { ParamName::new($crate::IStr::from($name)) };
+ (@name unnamed) => { ParamName::Unnamed };
+ (@name named $name:literal) => { ParamName::Named($crate::IStr::from($name)) };
($($(#[$meta:meta])* [$kind:ident $(($lit:literal))? => $default:expr]),* $(,)?) => {
thread_local! {
static PARAMS: FunctionSignature = FunctionSignature::new([
@@ -79,7 +77,7 @@
params: FunctionSignature::new(
params
.into_iter()
- .map(|n| ParamParse::new(ParamName::new(n.into()), ParamDefault::None))
+ .map(|n| ParamParse::new(ParamName::Named(n.into()), ParamDefault::None))
.collect(),
),
handler: TraceBox(Box::new(handler)),
@@ -88,7 +86,7 @@
}
impl Builtin for NativeCallback {
- fn name(&self) -> &str {
+ fn name(&self) -> &'static str {
// TODO: standard natives gets their names from definition
// But builitins should already have them
"<native>"
crates/jrsonnet-evaluator/src/function/mod.rsdiffbeforeafterboth1use std::{fmt::Debug, rc::Rc};23pub use arglike::{ArgLike, ArgsLike, TlaArg};4use educe::Educe;5use jrsonnet_gcmodule::{Cc, Trace};6use jrsonnet_interner::IStr;7pub use jrsonnet_macros::builtin;8use jrsonnet_parser::{Destruct, Expr, ExprParams, Span, Spanned};910use self::{11 arglike::OptionalContext,12 builtin::{Builtin, StaticBuiltin},13 native::NativeDesc,14 parse::{parse_default_function_call, parse_function_call},15};16use crate::{17 bail, error::ErrorKind::*, evaluate, evaluate_trivial, function::builtin::BuiltinFunc, params,18 Context, ContextBuilder, Result, Thunk, Val,19};2021pub mod arglike;22pub mod builtin;23pub mod native;24pub mod parse;25pub mod prepared;2627pub use jrsonnet_parser::function::*;2829/// Function callsite location.30/// Either from other jsonnet code, specified by expression location, or from native (without location).31#[derive(Clone, Copy)]32pub struct CallLocation<'l>(pub Option<&'l Span>);33impl<'l> CallLocation<'l> {34 /// Construct new location for calls coming from specified jsonnet expression location.35 pub const fn new(loc: &'l Span) -> Self {36 Self(Some(loc))37 }38}39impl CallLocation<'static> {40 /// Construct new location for calls coming from native code.41 pub const fn native() -> Self {42 Self(None)43 }44}4546/// Represents Jsonnet function defined in code.47#[derive(Trace, Educe)]48#[educe(Debug, PartialEq)]49pub struct FuncDesc {50 /// # Example51 ///52 /// In expressions like this, deducted to `a`, unspecified otherwise.53 /// ```jsonnet54 /// local a = function() ...55 /// local a() ...56 /// { a: function() ... }57 /// { a() = ... }58 /// ```59 pub name: IStr,60 /// Context, in which this function was evaluated.61 ///62 /// # Example63 /// In64 /// ```jsonnet65 /// local a = 2;66 /// function() ...67 /// ```68 /// context will contain `a`.69 pub ctx: Context,7071 /// Function parameter definition72 pub params: ExprParams,73 /// Function body74 pub body: Rc<Spanned<Expr>>,75}76impl FuncDesc {77 /// Create body context, but fill arguments without defaults with lazy error78 pub fn default_body_context(&self) -> Result<Context> {79 parse_default_function_call(self.ctx.clone(), &self.params)80 }8182 /// Create context, with which body code will run83 pub fn call_body_context(84 &self,85 call_ctx: Context,86 args: &dyn ArgsLike,87 tailstrict: bool,88 ) -> Result<Context> {89 parse_function_call(call_ctx, self.ctx.clone(), &self.params, args, tailstrict)90 }9192 pub fn evaluate_trivial(&self) -> Option<Val> {93 evaluate_trivial(&self.body)94 }95}9697/// Represents a Jsonnet function value, including plain functions and user-provided builtins.98#[allow(clippy::module_name_repetitions)]99#[derive(Trace, Clone)]100pub enum FuncVal {101 /// Identity function, kept this way for comparsions.102 Id,103 /// Plain function implemented in jsonnet.104 Normal(Cc<FuncDesc>),105 /// Function without arguments works just as a fancy thunk value.106 Thunk(Thunk<Val>),107 /// Standard library function.108 StaticBuiltin(#[trace(skip)] &'static dyn StaticBuiltin),109 /// User-provided function.110 Builtin(BuiltinFunc),111}112113impl Debug for FuncVal {114 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {115 match self {116 Self::Id => f.debug_tuple("Id").finish(),117 Self::Thunk(arg0) => f.debug_tuple("Thunk").field(arg0).finish(),118 Self::Normal(arg0) => f.debug_tuple("Normal").field(arg0).finish(),119 Self::StaticBuiltin(arg0) => {120 f.debug_tuple("StaticBuiltin").field(&arg0.name()).finish()121 }122 Self::Builtin(arg0) => f.debug_tuple("Builtin").field(&arg0.name()).finish(),123 }124 }125}126127#[allow(clippy::unnecessary_wraps)]128#[builtin]129const fn builtin_id(x: Val) -> Val {130 x131}132static ID: &builtin_id = &builtin_id {};133134impl FuncVal {135 pub fn builtin(builtin: impl Builtin) -> Self {136 Self::Builtin(BuiltinFunc::new(builtin))137 }138 pub fn static_builtin(static_builtin: &'static dyn StaticBuiltin) -> Self {139 Self::StaticBuiltin(static_builtin)140 }141142 pub fn params(&self) -> FunctionSignature {143 match self {144 Self::Id => ID.params(),145 Self::StaticBuiltin(i) => i.params(),146 Self::Builtin(i) => i.params(),147 Self::Normal(p) => p.params.signature.clone(),148 Self::Thunk(_) => FunctionSignature::empty(),149 }150 }151 /// Amount of non-default required arguments152 pub fn params_len(&self) -> usize {153 self.params().iter().filter(|p| !p.has_default()).count()154 }155 /// Function name, as defined in code.156 pub fn name(&self) -> IStr {157 match self {158 Self::Id => "id".into(),159 Self::Normal(normal) => normal.name.clone(),160 Self::StaticBuiltin(builtin) => builtin.name().into(),161 Self::Builtin(builtin) => builtin.name().into(),162 Self::Thunk(_) => "thunk".into(),163 }164 }165 /// Call function using arguments evaluated in specified `call_ctx` [`Context`].166 ///167 /// If `tailstrict` is specified - then arguments will be evaluated before being passed to function body.168 pub fn evaluate(169 &self,170 call_ctx: Context,171 loc: CallLocation<'_>,172 args: &dyn ArgsLike,173 tailstrict: bool,174 ) -> Result<Val> {175 match self {176 Self::Id => ID.call(call_ctx, loc, args),177 Self::Normal(func) => {178 let body_ctx = func.call_body_context(call_ctx, args, tailstrict)?;179 evaluate(body_ctx, &func.body)180 }181 Self::Thunk(thunk) => {182 if !args.is_empty() {183 bail!(TooManyArgsFunctionHas(0, FunctionSignature::empty()))184 }185 thunk.evaluate()186 }187 Self::StaticBuiltin(b) => b.call(call_ctx, loc, args),188 Self::Builtin(b) => b.call(call_ctx, loc, args),189 }190 }191 pub fn evaluate_simple<A: ArgsLike + OptionalContext>(192 &self,193 args: &A,194 tailstrict: bool,195 ) -> Result<Val> {196 self.evaluate(197 ContextBuilder::new().build(),198 CallLocation::native(),199 args,200 tailstrict,201 )202 }203 /// Convert jsonnet function to plain `Fn` value.204 pub fn into_native<D: NativeDesc>(self) -> D::Value {205 D::into_native(self)206 }207208 /// Is this function an indentity function.209 ///210 /// Currently only works for builtin `std.id`, aka `Self::Id` value, and `function(x) x`.211 ///212 /// This function should only be used for optimization, not for the conditional logic, i.e code should work with syntetic identity function too213 pub fn is_identity(&self) -> bool {214 match self {215 Self::Id => true,216 Self::Normal(desc) => {217 if desc.params.len() != 1 {218 return false;219 }220 let param = &desc.params.exprs[0];221 if param.default.is_some() {222 return false;223 }224225 #[allow(clippy::infallible_destructuring_match)]226 let id = match ¶m.destruct {227 Destruct::Full(id) => id,228 #[cfg(feature = "exp-destruct")]229 _ => return false,230 };231 **desc.body == Expr::Var(id.clone())232 }233 _ => false,234 }235 }236 /// Identity function value.237 pub const fn identity() -> Self {238 Self::Id239 }240241 pub fn evaluate_trivial(&self) -> Option<Val> {242 match self {243 Self::Normal(n) => n.evaluate_trivial(),244 _ => None,245 }246 }247}248249impl<T> From<T> for FuncVal250where251 T: Builtin,252{253 fn from(value: T) -> Self {254 Self::builtin(value)255 }256}257impl From<&'static dyn StaticBuiltin> for FuncVal {258 fn from(value: &'static dyn StaticBuiltin) -> Self {259 Self::static_builtin(value)260 }261}crates/jrsonnet-evaluator/src/function/parse.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/function/parse.rs
+++ b/crates/jrsonnet-evaluator/src/function/parse.rs
@@ -1,7 +1,9 @@
use std::mem::replace;
-use jrsonnet_interner::IStr;
-use jrsonnet_parser::{function::FunctionSignature, ExprParams};
+use jrsonnet_parser::{
+ function::{FunctionSignature, ParamName},
+ ExprParams,
+};
use rustc_hash::FxHashMap;
use super::arglike::ArgsLike;
@@ -9,7 +11,7 @@
bail,
destructure::destruct,
error::{ErrorKind::*, Result},
- evaluate_named, evaluate_named_param,
+ evaluate_named_param,
gc::WithCapacityExt as _,
Context, Pending, Thunk, Val,
};
@@ -76,7 +78,7 @@
.enumerate()
.filter_map(|(i, p)| Some((i, &p.destruct, p.default.as_ref()?)))
{
- if let Some(name) = into.name().0 {
+ if let ParamName::Named(name) = into.name() {
if passed_args.contains_key(&name) {
continue;
}
crates/jrsonnet-evaluator/src/function/prepared.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/function/prepared.rs
+++ b/crates/jrsonnet-evaluator/src/function/prepared.rs
@@ -4,9 +4,8 @@
use crate::destructure::destruct;
use crate::gc::WithCapacityExt;
-use crate::val::ThunkValue as _;
use crate::{bail, error::ErrorKind::*, Result};
-use crate::{evaluate_named, evaluate_named_param, Context, ContextBuilder, Pending, Thunk, Val};
+use crate::{evaluate_named_param, Context, ContextBuilder, Pending, Thunk, Val};
pub struct PreparedCall {
// Param, named input.
crates/jrsonnet-interner/src/lib.rsdiffbeforeafterboth--- a/crates/jrsonnet-interner/src/lib.rs
+++ b/crates/jrsonnet-interner/src/lib.rs
@@ -31,6 +31,10 @@
false
}
}
+
+/// SAFETY:
+///
+/// `IStr` is acyclic
unsafe impl Acyclic for IStr {}
impl IStr {
crates/jrsonnet-parser/src/expr.rsdiffbeforeafterboth--- a/crates/jrsonnet-parser/src/expr.rs
+++ b/crates/jrsonnet-parser/src/expr.rs
@@ -227,11 +227,11 @@
impl Destruct {
/// Name of destructure, used for function parameter names
pub fn name(&self) -> ParamName {
- ParamName(match self {
- Self::Full(name) => Some(name.clone()),
+ match self {
+ Self::Full(name) => ParamName::Named(name.clone()),
#[cfg(feature = "exp-destruct")]
- _ => None,
- })
+ _ => ParamName::Unnamed,
+ }
}
pub fn binds_len(&self) -> usize {
#[cfg(feature = "exp-destruct")]
crates/jrsonnet-parser/src/function.rsdiffbeforeafterboth--- a/crates/jrsonnet-parser/src/function.rs
+++ b/crates/jrsonnet-parser/src/function.rs
@@ -6,32 +6,38 @@
use jrsonnet_interner::IStr;
#[derive(Clone, Acyclic, Debug, PartialEq, Eq)]
-pub struct ParamName(pub Option<IStr>);
+pub enum ParamName {
+ Unnamed,
+ Named(IStr),
+}
impl ParamName {
- pub const ANONYMOUS: Self = Self(None);
- pub fn new(name: IStr) -> Self {
- Self(Some(name))
- }
pub fn as_str(&self) -> Option<&str> {
- self.0.as_deref()
+ match self {
+ ParamName::Unnamed => None,
+ ParamName::Named(istr) => Some(istr),
+ }
}
pub fn is_anonymous(&self) -> bool {
- self.0.is_none()
+ matches!(self, Self::Unnamed)
+ }
+ pub fn is_named(&self) -> bool {
+ matches!(self, Self::Named(_))
}
}
impl PartialEq<IStr> for ParamName {
fn eq(&self, other: &IStr) -> bool {
- self.0
- .as_ref()
- .map_or(false, |s| s.as_bytes() == other.as_bytes())
+ match self {
+ ParamName::Unnamed => false,
+ ParamName::Named(istr) => istr == other,
+ }
}
}
impl fmt::Display for ParamName {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
- match &self.0 {
- Some(v) => write!(f, "{v}"),
- None => write!(f, "<unnamed>"),
+ match &self {
+ Self::Named(v) => write!(f, "{v}"),
+ Self::Unnamed => write!(f, "<unnamed>"),
}
}
}
crates/jrsonnet-parser/src/snapshots/jrsonnet_parser__tests__default_param_before_nondefault.snapdiffbeforeafterboth--- a/crates/jrsonnet-parser/src/snapshots/jrsonnet_parser__tests__default_param_before_nondefault.snap
+++ b/crates/jrsonnet-parser/src/snapshots/jrsonnet_parser__tests__default_param_before_nondefault.snap
@@ -28,18 +28,14 @@
signature: FunctionSignature(
[
ParamParse {
- name: ParamName(
- Some(
- "foo",
- ),
+ name: Named(
+ "foo",
),
default: Exists,
},
ParamParse {
- name: ParamName(
- Some(
- "bar",
- ),
+ name: Named(
+ "bar",
),
default: None,
},