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.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;
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.rsdiffbeforeafterboth1use jrsonnet_parser::function::FunctionSignature;2use jrsonnet_parser::{ExprParams, IStr};3use rustc_hash::{FxHashMap, FxHashSet};45use crate::destructure::destruct;6use crate::gc::WithCapacityExt;7use crate::val::ThunkValue as _;8use crate::{bail, error::ErrorKind::*, Result};9use crate::{evaluate_named, evaluate_named_param, Context, ContextBuilder, Pending, Thunk, Val};1011pub struct PreparedCall {12 // Param, named input.13 named: Vec<(usize, usize)>,14 defaults: Vec<usize>,15}1617pub fn prepare_call(18 params: FunctionSignature,19 unnamed: usize,20 named: &[IStr],21) -> Result<PreparedCall> {22 if unnamed > params.len() {23 bail!(TooManyArgsFunctionHas(params.len(), params))24 }2526 let expected_defaults = params.len() - unnamed - named.len();27 let mut ops = PreparedCall {28 named: Vec::with_capacity(named.len()),29 defaults: Vec::with_capacity(expected_defaults),30 };3132 // FIXME: bitmask33 let mut passed: FxHashSet<usize> = (0..unnamed).collect();3435 for (input_id, name) in named.iter().enumerate() {36 // FIXME: O(n) for arg existence check37 let Some(param_idx) = params.iter().position(|p| p.name() == name) else {38 bail!(UnknownFunctionParameter(name.clone()));39 };40 if !passed.insert(param_idx) {41 bail!(BindingParameterASecondTime(name.clone()));42 }43 ops.named.push((param_idx, input_id));44 }4546 if named.len() + unnamed < params.len() {47 let mut defaults = 0;4849 for (param_id, param) in params50 .iter()51 .enumerate()52 .skip(unnamed)53 .filter(|p| p.1.has_default())54 {55 // Skip already passed parameters56 if !param.name().is_anonymous() && passed.contains(¶m_id) {57 continue;58 }59 defaults += 1;6061 ops.defaults.push(param_id);62 }6364 // Some args still weren't filled65 if defaults != expected_defaults {66 for param in params.iter().skip(unnamed) {67 let mut found = false;68 for name in named {69 if param.name() == name {70 found = true;71 }72 }73 if !found {74 bail!(FunctionParameterNotBoundInCall(75 param.name().clone(),76 params77 ));78 }79 }80 unreachable!();81 }82 }8384 Ok(ops)85}86pub fn parse_prepared_function_call(87 body_ctx: Context,88 prepared: &PreparedCall,89 params: &ExprParams,90 unnamed: &[Thunk<Val>],91 named: &[Thunk<Val>],92) -> Result<Context> {93 let mut passed_args = FxHashMap::with_capacity(params.binds_len());9495 let destruct_ctx = Pending::new();9697 for (param_idx, unnamed) in unnamed.iter().enumerate() {98 destruct(99 ¶ms.exprs[param_idx].destruct,100 unnamed.clone(),101 destruct_ctx.clone(),102 &mut passed_args,103 )?;104 }105106 for (param_idx, arg_idx) in prepared.named.iter().copied() {107 destruct(108 ¶ms.exprs[param_idx].destruct,109 named[arg_idx].clone(),110 destruct_ctx.clone(),111 &mut passed_args,112 )?;113 }114115 if prepared.defaults.is_empty() {116 let body_ctx = body_ctx117 .extend_bindings(passed_args)118 .into_future(destruct_ctx);119 Ok(body_ctx)120 } else {121 let fctx = Context::new_future();122 let mut defaults = FxHashMap::with_capacity(params.binds_len() - passed_args.len());123 for param_idx in prepared.defaults.iter().copied() {124 // let param = params.0.rc_idx(param_idx);125 destruct(126 ¶ms.exprs[param_idx].destruct,127 {128 let ctx = fctx.clone();129 let params = params.clone();130 Thunk!(move || {131 let param = ¶ms.exprs[param_idx];132 let name = param.destruct.name();133 let value = param.default.as_ref().expect("default exists");134 evaluate_named_param(ctx.unwrap(), value, name)135 })136 },137 fctx.clone(),138 &mut defaults,139 )?;140 }141142 let mut ctx = ContextBuilder::extend(body_ctx);143 ctx.binds(passed_args);144 ctx.binds(defaults);145 Ok(ctx.build().into_future(fctx).into_future(destruct_ctx))146 }147}148pub fn parse_prepared_builtin_call(149 prepared: &PreparedCall,150 params: FunctionSignature,151 unnamed: &[Thunk<Val>],152 named: &[Thunk<Val>],153) -> Result<Vec<Option<Thunk<Val>>>> {154 let mut passed_args = vec![None; params.len()];155156 for (param_idx, unnamed) in unnamed.iter().enumerate() {157 passed_args[param_idx] = Some(unnamed.clone());158 }159160 for (param_idx, arg_idx) in prepared.named.iter().copied() {161 passed_args[param_idx] = Some(named[arg_idx].clone());162 }163164 Ok(passed_args)165}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(¶m_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 ¶ms.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 ¶ms.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 ¶ms.exprs[param_idx].destruct,126 {127 let ctx = fctx.clone();128 let params = params.clone();129 Thunk!(move || {130 let param = ¶ms.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}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,
},