difftreelog
feat convert ParamName into enum
in: master
10 files changed
crates/jrsonnet-evaluator/src/evaluate/destructure.rsdiffbeforeafterboth1use 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}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.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,
},