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
--- 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)?;
 		}
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
after · crates/jrsonnet-evaluator/src/function/prepared.rs
1use jrsonnet_parser::function::FunctionSignature;2use jrsonnet_parser::{ExprParams, IStr};3use rustc_hash::{FxHashMap, FxHashSet};45use crate::destructure::destruct;6use crate::gc::WithCapacityExt;7use crate::{bail, error::ErrorKind::*, Result};8use crate::{evaluate_named_param, Context, ContextBuilder, Pending, Thunk, Val};910pub struct PreparedCall {11	// Param, named input.12	named: Vec<(usize, usize)>,13	defaults: Vec<usize>,14}1516pub fn prepare_call(17	params: FunctionSignature,18	unnamed: usize,19	named: &[IStr],20) -> Result<PreparedCall> {21	if unnamed > params.len() {22		bail!(TooManyArgsFunctionHas(params.len(), params))23	}2425	let expected_defaults = params.len() - unnamed - named.len();26	let mut ops = PreparedCall {27		named: Vec::with_capacity(named.len()),28		defaults: Vec::with_capacity(expected_defaults),29	};3031	// FIXME: bitmask32	let mut passed: FxHashSet<usize> = (0..unnamed).collect();3334	for (input_id, name) in named.iter().enumerate() {35		// FIXME: O(n) for arg existence check36		let Some(param_idx) = params.iter().position(|p| p.name() == name) else {37			bail!(UnknownFunctionParameter(name.clone()));38		};39		if !passed.insert(param_idx) {40			bail!(BindingParameterASecondTime(name.clone()));41		}42		ops.named.push((param_idx, input_id));43	}4445	if named.len() + unnamed < params.len() {46		let mut defaults = 0;4748		for (param_id, param) in params49			.iter()50			.enumerate()51			.skip(unnamed)52			.filter(|p| p.1.has_default())53		{54			// Skip already passed parameters55			if !param.name().is_anonymous() && passed.contains(&param_id) {56				continue;57			}58			defaults += 1;5960			ops.defaults.push(param_id);61		}6263		// Some args still weren't filled64		if defaults != expected_defaults {65			for param in params.iter().skip(unnamed) {66				let mut found = false;67				for name in named {68					if param.name() == name {69						found = true;70					}71				}72				if !found {73					bail!(FunctionParameterNotBoundInCall(74						param.name().clone(),75						params76					));77				}78			}79			unreachable!();80		}81	}8283	Ok(ops)84}85pub fn parse_prepared_function_call(86	body_ctx: Context,87	prepared: &PreparedCall,88	params: &ExprParams,89	unnamed: &[Thunk<Val>],90	named: &[Thunk<Val>],91) -> Result<Context> {92	let mut passed_args = FxHashMap::with_capacity(params.binds_len());9394	let destruct_ctx = Pending::new();9596	for (param_idx, unnamed) in unnamed.iter().enumerate() {97		destruct(98			&params.exprs[param_idx].destruct,99			unnamed.clone(),100			destruct_ctx.clone(),101			&mut passed_args,102		)?;103	}104105	for (param_idx, arg_idx) in prepared.named.iter().copied() {106		destruct(107			&params.exprs[param_idx].destruct,108			named[arg_idx].clone(),109			destruct_ctx.clone(),110			&mut passed_args,111		)?;112	}113114	if prepared.defaults.is_empty() {115		let body_ctx = body_ctx116			.extend_bindings(passed_args)117			.into_future(destruct_ctx);118		Ok(body_ctx)119	} else {120		let fctx = Context::new_future();121		let mut defaults = FxHashMap::with_capacity(params.binds_len() - passed_args.len());122		for param_idx in prepared.defaults.iter().copied() {123			// let param = params.0.rc_idx(param_idx);124			destruct(125				&params.exprs[param_idx].destruct,126				{127					let ctx = fctx.clone();128					let params = params.clone();129					Thunk!(move || {130						let param = &params.exprs[param_idx];131						let name = param.destruct.name();132						let value = param.default.as_ref().expect("default exists");133						evaluate_named_param(ctx.unwrap(), value, name)134					})135				},136				fctx.clone(),137				&mut defaults,138			)?;139		}140141		let mut ctx = ContextBuilder::extend(body_ctx);142		ctx.binds(passed_args);143		ctx.binds(defaults);144		Ok(ctx.build().into_future(fctx).into_future(destruct_ctx))145	}146}147pub fn parse_prepared_builtin_call(148	prepared: &PreparedCall,149	params: FunctionSignature,150	unnamed: &[Thunk<Val>],151	named: &[Thunk<Val>],152) -> Result<Vec<Option<Thunk<Val>>>> {153	let mut passed_args = vec![None; params.len()];154155	for (param_idx, unnamed) in unnamed.iter().enumerate() {156		passed_args[param_idx] = Some(unnamed.clone());157	}158159	for (param_idx, arg_idx) in prepared.named.iter().copied() {160		passed_args[param_idx] = Some(named[arg_idx].clone());161	}162163	Ok(passed_args)164}
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,
                         },