git.delta.rocks / jrsonnet / refs/commits / b3f009bb7b2e

difftreelog

feat convert ParamName into enum

sytvpywqYaroslav Bolyukin2026-03-21parent: #b915f23.patch.diff
in: master

10 files changed

modifiedcrates/jrsonnet-evaluator/src/evaluate/destructure.rsdiffbeforeafterboth
after · crates/jrsonnet-evaluator/src/evaluate/destructure.rs
1use jrsonnet_interner::IStr;2use jrsonnet_parser::{BindSpec, Destruct};3use rustc_hash::FxHashMap;45use crate::{6	bail,7	error::{ErrorKind::*, Result},8	evaluate_method, evaluate_named_param, Context, Pending, Thunk, Val,9};1011#[allow(clippy::too_many_lines)]12#[allow(unused_variables)]13pub fn destruct(14	d: &Destruct,15	parent: Thunk<Val>,16	fctx: Pending<Context>,17	new_bindings: &mut FxHashMap<IStr, Thunk<Val>>,18) -> Result<()> {19	match d {20		Destruct::Full(v) => {21			let old = new_bindings.insert(v.clone(), parent);22			if old.is_some() {23				bail!(DuplicateLocalVar(v.clone()))24			}25		}26		#[cfg(feature = "exp-destruct")]27		Destruct::Skip => {}28		#[cfg(feature = "exp-destruct")]29		Destruct::Array { start, rest, end } => {30			use jrsonnet_parser::DestructRest;3132			let min_len = start.len() + end.len();33			let has_rest = rest.is_some();34			let full = Thunk!(move || {35				let v = parent.evaluate()?;36				let Val::Arr(arr) = v else {37					bail!("expected array");38				};39				if !has_rest {40					if arr.len() != min_len {41						bail!("expected {} elements, got {}", min_len, arr.len())42					}43				} else if arr.len() < min_len {44					bail!(45						"expected at least {} elements, but array was only {}",46						min_len,47						arr.len()48					)49				}50				Ok(arr)51			});5253			{54				for (i, d) in start.iter().enumerate() {55					let full = full.clone();56					destruct(57						d,58						Thunk!(move || Ok(full.evaluate()?.get(i)?.expect("length is checked"))),59						fctx.clone(),60						new_bindings,61					)?;62				}63			}6465			match rest {66				Some(DestructRest::Keep(v)) => {67					let start = start.len();68					let end = end.len();69					let full = full.clone();70					destruct(71						&Destruct::Full(v.clone()),72						Thunk!(move || {73							let full = full.evaluate()?;74							let to = full.len() - end;75							Ok(Val::Arr(full.slice(76								Some(start as i32),77								Some(to as i32),78								None,79							)))80						}),81						fctx.clone(),82						new_bindings,83					)?;84				}85				Some(DestructRest::Drop) | None => {}86			}8788			{89				for (i, d) in end.iter().enumerate() {90					let full = full.clone();91					let end = end.len();92					destruct(93						d,94						Thunk!(move || {95							let full = full.evaluate()?;96							Ok(full.get(full.len() - end + i)?.expect("length is checked"))97						}),98						fctx.clone(),99						new_bindings,100					)?;101				}102			}103		}104		#[cfg(feature = "exp-destruct")]105		Destruct::Object { fields, rest } => {106			let field_names: Vec<_> = fields107				.iter()108				.map(|f| (f.0.clone(), f.2.is_some()))109				.collect();110			let has_rest = rest.is_some();111			let full = Thunk!(move || {112				let v = parent.evaluate()?;113				let Val::Obj(obj) = v else {114					bail!("expected object");115				};116				for (field, has_default) in &field_names {117					if !has_default && !obj.has_field_ex(field.clone(), true) {118						bail!("missing field: {field}");119					}120				}121				if !has_rest {122					let len = obj.len();123					if len > field_names.len() {124						bail!("too many fields, and rest not found");125					}126				}127				Ok(obj)128			});129130			for (field, d, default) in fields {131				let default = default.clone().map(|e| (fctx.clone(), e));132				let value = {133					let field = field.clone();134					let full = full.clone();135					Thunk!(move || {136						let full = full.evaluate()?;137						if let Some(field) = full.get(field)? {138							Ok(field)139						} else {140							let (fctx, expr) = default.as_ref().expect("shape is checked");141							Ok(evaluate(fctx.clone().unwrap(), expr)?)142						}143					})144				};145146				if let Some(d) = d {147					destruct(d, value, fctx.clone(), new_bindings)?;148				} else {149					destruct(150						&Destruct::Full(field.clone()),151						value,152						fctx.clone(),153						new_bindings,154					)?;155				}156			}157		}158	}159	Ok(())160}161162pub fn evaluate_dest(163	d: &BindSpec,164	fctx: Pending<Context>,165	new_bindings: &mut FxHashMap<IStr, Thunk<Val>>,166) -> Result<()> {167	match d {168		BindSpec::Field { into, value } => {169			let name = into.name();170			let value = value.clone();171			let data = {172				let fctx = fctx.clone();173				Thunk!(move || evaluate_named_param(fctx.unwrap(), &value, name))174			};175			destruct(into, data, fctx, new_bindings)?;176		}177		BindSpec::Function {178			name,179			params,180			value,181		} => {182			let params = params.clone();183			let name = name.clone();184			let value = value.clone();185			let old = new_bindings.insert(name.clone(), {186				let name = name.clone();187				Thunk!(move || Ok(evaluate_method(fctx.unwrap(), name, params, value)))188			});189			if old.is_some() {190				bail!(DuplicateLocalVar(name))191			}192		}193	}194	Ok(())195}
modifiedcrates/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),
 	}
 }
 
modifiedcrates/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>"
modifiedcrates/jrsonnet-evaluator/src/function/mod.rsdiffbeforeafterboth
--- a/crates/jrsonnet-evaluator/src/function/mod.rs
+++ b/crates/jrsonnet-evaluator/src/function/mod.rs
@@ -14,8 +14,8 @@
 	parse::{parse_default_function_call, parse_function_call},
 };
 use crate::{
-	bail, error::ErrorKind::*, evaluate, evaluate_trivial, function::builtin::BuiltinFunc, params,
-	Context, ContextBuilder, Result, Thunk, Val,
+	bail, error::ErrorKind::*, evaluate, evaluate_trivial, function::builtin::BuiltinFunc, Context,
+	ContextBuilder, Result, Thunk, Val,
 };
 
 pub mod arglike;
modifiedcrates/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;
 				}
modifiedcrates/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.
modifiedcrates/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 {
modifiedcrates/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")]
modifiedcrates/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>"),
 		}
 	}
 }
modifiedcrates/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,
                         },