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.rsdiffbeforeafterboth1use std::mem::replace;23use jrsonnet_interner::IStr;4use jrsonnet_parser::{function::FunctionSignature, ExprParams};5use rustc_hash::FxHashMap;67use super::arglike::ArgsLike;8use crate::{9 bail,10 destructure::destruct,11 error::{ErrorKind::*, Result},12 evaluate_named, evaluate_named_param,13 gc::WithCapacityExt as _,14 Context, Pending, Thunk, Val,15};1617/// Creates correct [context](Context) for function body evaluation returning error on invalid call.18///19/// ## Parameters20/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)21/// * `body_ctx`: used for default parameter values' execution and for body execution (if set)22/// * `params`: function parameters' definition23/// * `args`: passed function arguments24/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily25pub fn parse_function_call(26 ctx: Context,27 body_ctx: Context,28 params: &ExprParams,29 args: &dyn ArgsLike,30 tailstrict: bool,31) -> Result<Context> {32 let mut passed_args = FxHashMap::with_capacity(params.binds_len());33 if args.unnamed_len() > params.signature.len() {34 bail!(TooManyArgsFunctionHas(35 params.signature.len(),36 params.signature.clone(),37 ))38 }3940 let mut filled_named = 0;41 let mut filled_positionals = 0;4243 args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {44 destruct(45 ¶ms.exprs[id].destruct,46 arg,47 Pending::new_filled(ctx.clone()),48 &mut passed_args,49 )?;50 filled_positionals += 1;51 Ok(())52 })?;5354 args.named_iter(ctx, tailstrict, &mut |name, value| {55 // FIXME: O(n) for arg existence check56 if !params.exprs.iter().any(|p| &p.destruct.name() == name) {57 bail!(UnknownFunctionParameter(name.clone()));58 }59 if passed_args.insert(name.clone(), value).is_some() {60 bail!(BindingParameterASecondTime(name.clone()));61 }62 filled_named += 1;63 Ok(())64 })?;6566 if filled_named + filled_positionals < params.len() {67 // Some args are unset, but maybe we have defaults for them68 // Default values should be created in newly created context69 let fctx = Context::new_future();70 let mut defaults =71 FxHashMap::with_capacity(params.binds_len() - filled_named - filled_positionals);7273 for (idx, into, default) in params74 .exprs75 .iter()76 .enumerate()77 .filter_map(|(i, p)| Some((i, &p.destruct, p.default.as_ref()?)))78 {79 if let Some(name) = into.name().0 {80 if passed_args.contains_key(&name) {81 continue;82 }83 } else if idx < filled_positionals {84 continue;85 }8687 destruct(88 &into,89 {90 let ctx = fctx.clone();91 let name = into.name();92 let value = default.clone();93 Thunk!(move || evaluate_named_param(ctx.unwrap(), &value, name))94 },95 fctx.clone(),96 &mut defaults,97 )?;98 if !into.name().is_anonymous() {99 filled_named += 1;100 } else {101 filled_positionals += 1;102 }103 }104105 // Some args still weren't filled106 if filled_named + filled_positionals != params.len() {107 for param in params.exprs.iter().skip(args.unnamed_len()) {108 let mut found = false;109 args.named_names(&mut |name| {110 if ¶m.destruct.name() == name {111 found = true;112 }113 });114 if !found {115 bail!(FunctionParameterNotBoundInCall(116 param.destruct.name(),117 params.signature.clone()118 ));119 }120 }121 unreachable!();122 }123124 Ok(body_ctx125 .extend_bindings(passed_args)126 .extend_bindings(defaults)127 .into_future(fctx))128 } else {129 let body_ctx = body_ctx.extend_bindings(passed_args);130 Ok(body_ctx)131 }132}133134/// You shouldn't probally use this function, use `jrsonnet_macros::builtin` instead135///136/// ## Parameters137/// * `ctx`: used for passed argument expressions' execution and for body execution (if `body_ctx` is not set)138/// * `params`: function parameters' definition139/// * `args`: passed function arguments140/// * `tailstrict`: if set to `true` function arguments are eagerly executed, otherwise - lazily141pub fn parse_builtin_call(142 ctx: Context,143 params: FunctionSignature,144 args: &dyn ArgsLike,145 tailstrict: bool,146) -> Result<Vec<Option<Thunk<Val>>>> {147 let mut passed_args: Vec<Option<Thunk<Val>>> = vec![None; params.len()];148 if args.unnamed_len() > params.len() {149 bail!(TooManyArgsFunctionHas(params.len(), params,))150 }151152 let mut filled_args = 0;153154 args.unnamed_iter(ctx.clone(), tailstrict, &mut |id, arg| {155 passed_args[id] = Some(arg);156 filled_args += 1;157 Ok(())158 })?;159160 args.named_iter(ctx, tailstrict, &mut |name, arg| {161 // FIXME: O(n) for arg existence check162 let id = params163 .iter()164 .position(|p| p.name() == name)165 .ok_or_else(|| UnknownFunctionParameter(name.clone()))?;166 if replace(&mut passed_args[id], Some(arg)).is_some() {167 bail!(BindingParameterASecondTime(name.clone()));168 }169 filled_args += 1;170 Ok(())171 })?;172173 if filled_args < params.len() {174 for (id, _) in params.iter().enumerate().filter(|(_, p)| p.has_default()) {175 if passed_args[id].is_some() {176 continue;177 }178 filled_args += 1;179 }180181 // Some args still wasn't filled182 if filled_args != params.len() {183 for param in params.iter().skip(args.unnamed_len()) {184 let mut found = false;185 args.named_names(&mut |name| {186 if param.name() == name {187 found = true;188 }189 });190 if !found {191 bail!(FunctionParameterNotBoundInCall(192 param.name().clone(),193 params,194 ));195 }196 }197 unreachable!();198 }199 }200 Ok(passed_args)201}202203/// Creates Context, which has all argument default values applied204/// and with unbound values causing error to be returned205pub fn parse_default_function_call(body_ctx: Context, params: &ExprParams) -> Result<Context> {206 let fctx = Context::new_future();207208 let mut bindings = FxHashMap::with_capacity(params.binds_len());209210 for param in params.exprs.iter() {211 if let Some(v) = ¶m.default {212 destruct(213 ¶m.destruct.clone(),214 {215 let ctx = fctx.clone();216 let name = param.destruct.name();217 let value = v.clone();218 Thunk!(move || evaluate_named_param(ctx.unwrap(), &value, name))219 },220 fctx.clone(),221 &mut bindings,222 )?;223 } else {224 destruct(225 ¶m.destruct,226 {227 let param_name = param.destruct.name();228 let params = params.clone();229 Thunk!(move || Err(FunctionParameterNotBoundInCall(230 param_name,231 params.signature.clone()232 )233 .into()))234 },235 fctx.clone(),236 &mut bindings,237 )?;238 }239 }240241 Ok(body_ctx.extend_bindings(bindings).into_future(fctx))242}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,
},