difftreelog
style fix clippy warnings
in: master
30 files changed
bindings/jrsonnet-web/src/lib.rsdiffbeforeafterboth--- a/bindings/jrsonnet-web/src/lib.rs
+++ b/bindings/jrsonnet-web/src/lib.rs
@@ -65,6 +65,11 @@
}
fn unwrap_val_ref(value: &JsValue) -> Result<<WasmVal as RefFromWasmAbi>::Anchor, JsValue> {
+ #[allow(
+ clippy::cast_sign_loss,
+ clippy::cast_possible_truncation,
+ reason = "defined to be u32"
+ )]
let ptr = get(value, &JsValue::from_str("__wbg_ptr"))
.ok()
.and_then(|v| v.as_f64())
@@ -371,14 +376,14 @@
impl WasmArrValue {
#[wasm_bindgen(getter)]
pub fn length(&self) -> u32 {
- self.arr.len()
+ self.arr.len32()
}
pub fn at(&self, index: u32) -> Result<Option<WasmVal>, JsValue> {
let result = self.state.as_ref().map_or_else(
- || self.arr.get(index),
+ || self.arr.get32(index),
|state| {
let _guard = state.try_enter();
- self.arr.get(index)
+ self.arr.get32(index)
},
);
result
bindings/jsonnet/src/lib.rsdiffbeforeafterboth--- a/bindings/jsonnet/src/lib.rs
+++ b/bindings/jsonnet/src/lib.rs
@@ -66,14 +66,12 @@
#[cfg(target_family = "unix")]
{
use std::os::unix::ffi::OsStrExt;
- let str = CString::new(input.as_os_str().as_bytes()).expect("input has zero byte in it");
- str
+ CString::new(input.as_os_str().as_bytes()).expect("input has zero byte in it")
}
#[cfg(not(target_family = "unix"))]
{
let str = input.as_os_str().to_str().expect("bad utf-8");
- let cstr = CString::new(str).expect("input has NUL inside");
- cstr
+ CString::new(str).expect("input has NUL inside")
}
}
cmds/jrb/src/main.rsdiffbeforeafterboth--- a/cmds/jrb/src/main.rs
+++ b/cmds/jrb/src/main.rs
@@ -104,6 +104,7 @@
}
}
+#[allow(clippy::too_many_lines)]
fn main() {
tracing_subscriber::fmt().init();
crates/jrsonnet-evaluator/src/analyze.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/analyze.rs
+++ b/crates/jrsonnet-evaluator/src/analyze.rs
@@ -28,7 +28,10 @@
use rustc_hash::FxHashMap;
use smallvec::SmallVec;
-use crate::error::{format_found, suggest_names};
+use crate::{
+ arr::arridx,
+ error::{format_found, suggest_names},
+};
#[derive(Debug, Clone, Copy)]
#[must_use]
@@ -658,7 +661,7 @@
stack: &'s mut AnalysisStack,
bomb: DropBomb,
}
-impl<'s> PendingBody<'s> {
+impl PendingBody<'_> {
/// After the body is processed, drop the frame's locals and emit any
/// "unused local" warnings.
fn finish(self) {
@@ -704,7 +707,7 @@
.drain(closures.first_in_frame.idx()..)
.collect();
for (i, def) in drained.iter().enumerate().rev() {
- let id = LocalId(closures.first_in_frame.0 + i as u32);
+ let id = LocalId(closures.first_in_frame.0 + arridx(i));
let stack_locals = stack
.local_by_name
.get_mut(&def.name)
@@ -819,7 +822,7 @@
let (this_refs, rest) = refs.split_at(*refs_len);
refs = rest;
let start = next_id;
- next_id += *dest_count as u32;
+ next_id += arridx(*dest_count);
Closure {
references: this_refs,
ids: start..next_id,
@@ -1043,7 +1046,7 @@
}
fn next_local_id(&self) -> LocalId {
- LocalId(self.local_defs.len() as u32)
+ LocalId(arridx(self.local_defs.len()))
}
fn report_error(&mut self, msg: impl Into<String>, span: Option<Span>) {
@@ -1565,7 +1568,7 @@
let mut pending = alloc.finish();
let mut l_binds: Vec<LBind> = Vec::with_capacity(binds.len());
- for (bind, destruct) in binds.iter().zip(destructs.into_iter()) {
+ for (bind, destruct) in binds.iter().zip(destructs) {
let mut value_taint = AnalysisResult::default();
let (value_shape, value) = pending
.stack
@@ -1608,7 +1611,7 @@
let mut pending = alloc.finish();
let mut l_params: Vec<LParam> = Vec::with_capacity(params.exprs.len());
- for (p, destruct) in params.exprs.iter().zip(param_destructs.into_iter()) {
+ for (p, destruct) in params.exprs.iter().zip(param_destructs) {
let mut value_taint = AnalysisResult::default();
let default = p.default.as_ref().map_or_else(
|| None,
crates/jrsonnet-evaluator/src/arr/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/arr/mod.rs
+++ b/crates/jrsonnet-evaluator/src/arr/mod.rs
@@ -2,6 +2,7 @@
any::Any,
fmt::{self},
num::NonZeroU32,
+ ops::{Bound, RangeBounds},
rc::Rc,
};
@@ -104,20 +105,37 @@
Self::new(RangeArray::new_inclusive(a, b))
}
+ #[inline]
#[must_use]
- pub fn slice(self, index: Option<i32>, end: Option<i32>, step: Option<NonZeroU32>) -> Self {
+ pub fn slice(self, range: impl RangeBounds<usize>) -> Self {
+ fn map_bound(start: bool, bound: Bound<&usize>) -> Option<i32> {
+ match bound {
+ Bound::Included(&v) => Some(i32::try_from(v).unwrap_or(i32::MAX)),
+ Bound::Excluded(&v) => Some(
+ i32::try_from(v)
+ .unwrap_or(i32::MAX)
+ .saturating_add(if start { 1 } else { -1 }),
+ ),
+ Bound::Unbounded => None,
+ }
+ }
+ self.slice32(
+ map_bound(true, range.start_bound()),
+ map_bound(false, range.end_bound()),
+ None,
+ )
+ }
+
+ #[must_use]
+ pub fn slice32(self, index: Option<i32>, end: Option<i32>, step: Option<NonZeroU32>) -> Self {
let get_idx = |pos: Option<i32>, len: u32, default| match pos {
- #[expect(
- clippy::cast_sign_loss,
- reason = "abs value is used, len is limited to u31"
- )]
Some(v) if v < 0 => len.saturating_add_signed(v),
#[expect(clippy::cast_sign_loss, reason = "abs value is used")]
Some(v) => (v as u32).min(len),
None => default,
};
- let index = get_idx(index, self.len(), 0);
- let end = get_idx(end, self.len(), self.len());
+ let index = get_idx(index, self.len32(), 0);
+ let end = get_idx(end, self.len32(), self.len32());
let step = step.unwrap_or_else(|| NonZeroU32::new(1).expect("1 != 0"));
if index >= end {
@@ -126,24 +144,29 @@
Self::new(SliceArray {
inner: self,
- #[expect(clippy::cast_possible_truncation, reason = "len is limited to u31")]
- from: index as u32,
- #[expect(clippy::cast_possible_truncation, reason = "len is limited to u31")]
- to: end as u32,
+ from: index,
+ to: end,
step: step.get(),
})
}
/// Array length.
- pub fn len(&self) -> u32 {
- self.0.len()
+ #[inline]
+ pub fn len32(&self) -> u32 {
+ self.0.len32()
}
+ pub fn len(&self) -> usize {
+ self.len32() as usize
+ }
+
/// Is array contains no elements?
+ #[inline]
pub fn is_empty(&self) -> bool {
self.0.is_empty()
}
+ #[inline]
pub fn is_cheap(&self) -> bool {
self.0.is_cheap()
}
@@ -151,24 +174,37 @@
/// Get array element by index, evaluating it, if it is lazy.
///
/// Returns `None` on out-of-bounds condition.
- pub fn get(&self, index: u32) -> Result<Option<Val>> {
- self.0.get(index)
+ #[inline]
+ pub fn get32(&self, index: u32) -> Result<Option<Val>> {
+ self.0.get32(index)
+ }
+
+ pub fn get(&self, index: usize) -> Result<Option<Val>> {
+ let Ok(i) = u32::try_from(index) else {
+ return Ok(None);
+ };
+ self.get32(i)
}
/// Get array element by index, without evaluation.
///
/// Returns `None` on out-of-bounds condition.
- pub fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
- self.0.get_lazy(index)
+ #[inline]
+ pub fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
+ self.0.get_lazy32(index)
+ }
+
+ pub fn get_lazy(&self, index: usize) -> Option<Thunk<Val>> {
+ u32::try_from(index).ok().and_then(|i| self.get_lazy32(i))
}
pub fn iter(&self) -> impl ArrayLikeIter<Result<Val>> + '_ {
- (0..self.len()).map(|i| self.get(i).transpose().expect("length checked"))
+ (0..self.len32()).map(|i| self.get32(i).transpose().expect("length checked"))
}
/// Iterate over elements, returning lazy values.
pub fn iter_lazy(&self) -> impl ArrayLikeIter<Thunk<Val>> + '_ {
- (0..self.len()).map(|i| self.get_lazy(i).expect("length checked"))
+ (0..self.len32()).map(|i| self.get_lazy32(i).expect("length checked"))
}
/// Return a reversed view on current array.
@@ -201,3 +237,18 @@
Self::new(iter.into_iter().collect::<Vec<_>>())
}
}
+
+/// Checks that the usize does not exceed 4g with debug assertions enabled
+/// Should only be used on values that can't reasonably exceed this value
+#[inline]
+pub(crate) fn arridx(i: usize) -> u32 {
+ #[allow(
+ clippy::cast_possible_truncation,
+ reason = "array indexes never exceed 4g"
+ )]
+ if cfg!(debug_assertions) {
+ u32::try_from(i).expect("4g hard limit")
+ } else {
+ i as u32
+ }
+}
crates/jrsonnet-evaluator/src/arr/spec.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/arr/spec.rs
+++ b/crates/jrsonnet-evaluator/src/arr/spec.rs
@@ -9,7 +9,7 @@
use jrsonnet_gcmodule::{Cc, Trace};
use jrsonnet_interner::{IBytes, IStr};
-use super::ArrValue;
+use super::{ArrValue, arridx};
use crate::{
Context, Error, ObjValue, Result, Thunk, Val,
analyze::{ClosureShape, LExpr},
@@ -21,12 +21,12 @@
};
pub trait ArrayLike: Any + Trace + Debug {
- fn len(&self) -> u32;
+ fn len32(&self) -> u32;
fn is_empty(&self) -> bool {
- self.len() == 0
+ self.len32() == 0
}
- fn get(&self, index: u32) -> Result<Option<Val>>;
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>>;
+ fn get32(&self, index: u32) -> Result<Option<Val>>;
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>>;
fn is_cheap(&self) -> bool {
false
@@ -40,15 +40,15 @@
where
T: Any + Trace + Debug + ArrayCheap,
{
- fn len(&self) -> u32 {
+ fn len32(&self) -> u32 {
<T as ArrayCheap>::len(self)
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
Ok(<T as ArrayCheap>::get(self, index))
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
<T as ArrayCheap>::get(self, index).map(Thunk::evaluated)
}
@@ -80,16 +80,16 @@
}
}
impl ArrayLike for SliceArray {
- fn len(&self) -> u32 {
+ fn len32(&self) -> u32 {
(self.to - self.from).div_ceil(self.step)
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
- self.inner.get(self.map_idx(index))
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
+ self.inner.get32(self.map_idx(index))
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
- self.inner.get_lazy(self.map_idx(index))
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
+ self.inner.get_lazy32(self.map_idx(index))
}
fn is_cheap(&self) -> bool {
@@ -99,7 +99,7 @@
impl ArrayCheap for IBytes {
fn len(&self) -> u32 {
- self.as_slice().len() as u32
+ arridx(self.as_slice().len())
}
fn get(&self, index: u32) -> Option<Val> {
self.as_slice()
@@ -132,11 +132,11 @@
}
}
impl ArrayLike for ExprArray {
- fn len(&self) -> u32 {
- self.cached.borrow().len() as u32
+ fn len32(&self) -> u32 {
+ arridx(self.cached.borrow().len())
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
- if index >= self.len() {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
+ if index >= self.len32() {
return Ok(None);
}
match &self.cached.borrow()[index as usize] {
@@ -157,7 +157,7 @@
self.cached.borrow_mut()[index as usize] = ArrayThunk::Computed(new_value.clone());
Ok(Some(new_value))
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
#[derive(Trace)]
struct ExprArrThunk {
expr: ExprArray,
@@ -168,13 +168,13 @@
fn get(&self) -> Result<Self::Output> {
self.expr
- .get(self.index)
+ .get32(self.index)
.transpose()
.expect("index checked")
}
}
- if index >= self.len() {
+ if index >= self.len32() {
return None;
}
match &self.cached.borrow()[index as usize] {
@@ -202,8 +202,8 @@
}
impl ExtendedArray {
pub fn new(a: ArrValue, b: ArrValue) -> Option<Self> {
- let a_len = a.len();
- let b_len = b.len();
+ let a_len = a.len32();
+ let b_len = b.len32();
let len = a_len.checked_add(b_len)?;
Some(Self {
a,
@@ -251,22 +251,22 @@
}
}
impl ArrayLike for ExtendedArray {
- fn get(&self, index: u32) -> Result<Option<Val>> {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
if self.split > index {
- self.a.get(index)
+ self.a.get32(index)
} else {
- self.b.get(index - self.split)
+ self.b.get32(index - self.split)
}
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
if self.split > index {
- self.a.get_lazy(index)
+ self.a.get_lazy32(index)
} else {
- self.b.get_lazy(index - self.split)
+ self.b.get_lazy32(index - self.split)
}
}
- fn len(&self) -> u32 {
+ fn len32(&self) -> u32 {
self.len
}
@@ -280,18 +280,18 @@
T: IntoUntyped + Trace + fmt::Debug,
for<'a> &'a T: IntoUntyped,
{
- fn len(&self) -> u32 {
+ fn len32(&self) -> u32 {
self.as_slice().len().try_into().unwrap_or(u32::MAX)
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
let Some(elem) = self.as_slice().get(index as usize) else {
return Ok(None);
};
IntoUntyped::into_untyped(elem).map(Some)
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
let elem = self.as_slice().get(index as usize)?;
Some(IntoUntyped::into_lazy_untyped(elem))
}
@@ -343,16 +343,16 @@
#[derive(Debug, Trace)]
pub struct ReverseArray(pub ArrValue);
impl ArrayLike for ReverseArray {
- fn len(&self) -> u32 {
- self.0.len()
+ fn len32(&self) -> u32 {
+ self.0.len32()
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
- self.0.get(self.0.len() - index - 1)
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
+ self.0.get32(self.0.len32() - index - 1)
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
- self.0.get_lazy(self.0.len() - index - 1)
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
+ self.0.get_lazy32(self.0.len32() - index - 1)
}
fn is_cheap(&self) -> bool {
@@ -374,7 +374,7 @@
}
impl MappedArray {
pub fn new(inner: ArrValue, mapper: ArrayMapper) -> Self {
- let len = inner.len();
+ let len = inner.len32();
Self {
inner,
cached: Cc::new(RefCell::new(vec![ArrayThunk::Waiting; len as usize])),
@@ -389,12 +389,12 @@
}
}
impl ArrayLike for MappedArray {
- fn len(&self) -> u32 {
- self.cached.borrow().len() as u32
+ fn len32(&self) -> u32 {
+ arridx(self.cached.borrow().len())
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
- if index >= self.len() {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
+ if index >= self.len32() {
return Ok(None);
}
match &self.cached.borrow()[index as usize] {
@@ -413,7 +413,7 @@
let val = self
.inner
- .get(index)
+ .get32(index)
.transpose()
.expect("index checked")
.and_then(|r| self.evaluate(index, r));
@@ -428,7 +428,7 @@
self.cached.borrow_mut()[index as usize] = ArrayThunk::Computed(new_value.clone());
Ok(Some(new_value))
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
#[derive(Trace)]
struct MappedArrayThunk {
arr: MappedArray,
@@ -438,11 +438,14 @@
type Output = Val;
fn get(&self) -> Result<Self::Output> {
- self.arr.get(self.index).transpose().expect("index checked")
+ self.arr
+ .get32(self.index)
+ .transpose()
+ .expect("index checked")
}
}
- if index >= self.len() {
+ if index >= self.len32() {
return None;
}
match &self.cached.borrow()[index as usize] {
@@ -471,12 +474,12 @@
}
}
impl ArrayLike for MakeArray {
- fn len(&self) -> u32 {
- self.cached.borrow().len() as u32
+ fn len32(&self) -> u32 {
+ arridx(self.cached.borrow().len())
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
- if index >= self.len() {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
+ if index >= self.len32() {
return Ok(None);
}
match &self.cached.borrow()[index as usize] {
@@ -493,7 +496,7 @@
unreachable!()
};
- let val = self.mapper.call(index as u32);
+ let val = self.mapper.call(index);
let new_value = match val {
Ok(v) => v,
@@ -505,7 +508,7 @@
self.cached.borrow_mut()[index as usize] = ArrayThunk::Computed(new_value.clone());
Ok(Some(new_value))
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
#[derive(Trace)]
struct MakeArrayThunk {
arr: MakeArray,
@@ -515,11 +518,14 @@
type Output = Val;
fn get(&self) -> Result<Self::Output> {
- self.arr.get(self.index).transpose().expect("index checked")
+ self.arr
+ .get32(self.index)
+ .transpose()
+ .expect("index checked")
}
}
- if index >= self.len() {
+ if index >= self.len32() {
return None;
}
match &self.cached.borrow()[index as usize] {
@@ -543,7 +549,7 @@
}
impl RepeatedArray {
pub fn new(data: ArrValue, repeats: u32) -> Option<Self> {
- let total_len = data.len().checked_mul(repeats)?;
+ let total_len = data.len32().checked_mul(repeats)?;
Some(Self {
data,
repeats,
@@ -554,25 +560,25 @@
if index > self.total_len {
return None;
}
- Some(index % self.data.len())
+ Some(index % self.data.len32())
}
}
impl ArrayLike for RepeatedArray {
- fn len(&self) -> u32 {
+ fn len32(&self) -> u32 {
self.total_len
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
let Some(idx) = self.map_idx(index) else {
return Ok(None);
};
- self.data.get(idx)
+ self.data.get32(idx)
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
let idx = self.map_idx(index)?;
- self.data.get_lazy(idx)
+ self.data.get_lazy32(idx)
}
fn is_cheap(&self) -> bool {
@@ -593,18 +599,18 @@
}
impl ArrayLike for PickObjectValues {
- fn len(&self) -> u32 {
- self.keys.len() as u32
+ fn len32(&self) -> u32 {
+ arridx(self.keys.len())
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
let Some(key) = self.keys.as_slice().get(index as usize) else {
return Ok(None);
};
Ok(Some(self.obj.get_or_bail(key.clone())?))
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
let key = self.keys.as_slice().get(index as usize)?;
Some(self.obj.get_lazy_or_bail(key.clone()))
}
@@ -633,11 +639,11 @@
}
impl ArrayLike for PickObjectKeyValues {
- fn len(&self) -> u32 {
- self.keys.len() as u32
+ fn len32(&self) -> u32 {
+ arridx(self.keys.len())
}
- fn get(&self, index: u32) -> Result<Option<Val>> {
+ fn get32(&self, index: u32) -> Result<Option<Val>> {
let Some(key) = self.keys.as_slice().get(index as usize) else {
return Ok(None);
};
@@ -650,7 +656,7 @@
))
}
- fn get_lazy(&self, index: u32) -> Option<Thunk<Val>> {
+ fn get_lazy32(&self, index: u32) -> Option<Thunk<Val>> {
let key = self.keys.as_slice().get(index as usize)?;
// Nothing can fail in the key part, yet value is still
// lazy-evaluated
crates/jrsonnet-evaluator/src/ctx.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/ctx.rs
+++ b/crates/jrsonnet-evaluator/src/ctx.rs
@@ -122,7 +122,7 @@
pub fn enter(self, sup_this: SupThis, build: impl FnOnce(&LocalsFrame, &Context)) -> Context {
let locals = LocalsFrame::new_once(self.n_locals);
let val = Context(Cc::new(ContextInternal {
- captures: self.captures.clone(),
+ captures: self.captures,
locals,
sup_this: Some(sup_this),
}));
crates/jrsonnet-evaluator/src/evaluate/compspec.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/evaluate/compspec.rs
+++ b/crates/jrsonnet-evaluator/src/evaluate/compspec.rs
@@ -97,7 +97,7 @@
let value_ctx = inner_ctx
.pack_captures_sup_this(self.frame_shape)
.enter(|fill, ctx| {
- fill_letrec_binds(fill, &ctx, self.locals);
+ fill_letrec_binds(fill, ctx, self.locals);
});
evaluate_field_member_static(self.builder, inner_ctx, value_ctx, self.field)
}
@@ -336,6 +336,7 @@
Ok(())
}
+#[allow(clippy::too_many_lines)]
fn evaluate_compspecs(
ctx: Context,
specs: &[LCompSpec],
@@ -381,7 +382,7 @@
for (i, item) in arr.iter().enumerate() {
let item = item?;
let inner_ctx = ctx.pack_captures_sup_this(frame_shape).enter(|fill, ctx| {
- destruct(dst, fill, Thunk::evaluated(item), &ctx);
+ destruct(dst, fill, Thunk::evaluated(item), ctx);
});
evaluate_compspecs(
inner_ctx,
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::{
Context, LocalsFrame, PackedContext, Result, SupThis, Thunk, Unbound, Val,
analyze::{
- ClosureShape, LBind, LDestruct, LDestructField, LDestructRest, LExpr, LLocalExpr, LocalSlot,
+ ClosureShape, LBind, LDestruct, LDestructField, LDestructRest, LLocalExpr, LocalSlot,
},
bail,
evaluate::evaluate,
@@ -19,7 +19,7 @@
fill: &LocalsFrame,
value: Thunk<Val>,
- a_ctx: &Context,
+ ctx: &Context,
) {
let min_len = start.len() + end.len();
let has_rest = rest.is_some();
@@ -29,14 +29,14 @@
bail!("expected array");
};
if !has_rest {
- if arr.len() as usize != min_len {
- bail!("expected {} elements, got {}", min_len, arr.len())
+ if arr.len() != min_len {
+ bail!("expected {} elements, got {}", min_len, arr.len32())
}
- } else if (arr.len() as usize) < min_len {
+ } else if arr.len() < min_len {
bail!(
"expected at least {} elements, but array was only {}",
min_len,
- arr.len()
+ arr.len32()
)
}
Ok(arr)
@@ -47,13 +47,13 @@
destruct(
d,
fill,
- Thunk!(move || Ok(full.evaluate()?.get(i as u32)?.expect("length is checked"))),
- a_ctx,
+ Thunk!(move || Ok(full.evaluate()?.get(i)?.expect("length is checked"))),
+ ctx,
);
}
- let start_len = start.len() as u32;
- let end_len = end.len() as u32;
+ let start_len = start.len();
+ let end_len = end.len();
if let Some(LDestructRest::Keep(slot)) = rest {
let full = full.clone();
@@ -62,11 +62,7 @@
Thunk!(move || {
let full = full.evaluate()?;
let to = full.len() - end_len;
- Ok(Val::Arr(full.slice(
- Some(start_len as i32),
- Some(to as i32),
- None,
- )))
+ Ok(Val::Arr(full.slice(start_len..to)))
}),
);
}
@@ -79,10 +75,10 @@
Thunk!(move || {
let full = full.evaluate()?;
Ok(full
- .get(full.len() - end_len + i as u32)?
+ .get(full.len() - end_len + i)?
.expect("length is checked"))
}),
- a_ctx,
+ ctx,
);
}
}
@@ -94,7 +90,7 @@
fill: &LocalsFrame,
value: Thunk<Val>,
- a_ctx: &Context,
+ ctx: &Context,
) {
use jrsonnet_interner::IStr;
use rustc_hash::FxHashSet;
@@ -118,7 +114,7 @@
}
}
if !has_rest {
- let len = obj.len();
+ let len = obj.len32();
if len as usize > field_names.len() {
bail!("too many fields, and rest not found");
}
@@ -142,10 +138,11 @@
for field in fields {
let field_name = field.name.clone();
- let default_thunk: Option<Thunk<Val>> = field
- .default
- .as_ref()
- .map(|(shape, expr)| build_b_thunk(a_ctx, shape, expr.clone()));
+ let default_thunk: Option<Thunk<Val>> = field.default.as_ref().map(|(shape, expr)| {
+ let expr = expr.clone();
+ let env = Context::enter_using(ctx, shape);
+ Thunk!(move || evaluate(env, &expr))
+ });
let field_full = full.clone();
let value_thunk = Thunk!(move || {
@@ -157,7 +154,7 @@
});
if let Some(into) = &field.into {
- destruct(into, fill, value_thunk, a_ctx);
+ destruct(into, fill, value_thunk, ctx);
} else {
unreachable!("analyzer lowers object-destruct shorthands into `into`");
}
@@ -177,21 +174,18 @@
#[cfg(feature = "exp-destruct")]
LDestruct::Object { fields, rest } => destruct_object(fields, rest.as_ref(), fill, value, a_ctx),
}
-}
-
-pub fn build_b_thunk(a_ctx: &Context, shape: &ClosureShape, expr: Rc<LExpr>) -> Thunk<Val> {
- let env = Context::enter_using(a_ctx, shape);
- Thunk!(move || evaluate(env, &expr))
-}
-pub fn build_b_thunk_uno(a_ctx: &Context, shape: Rc<(ClosureShape, LExpr)>) -> Thunk<Val> {
- let env = Context::enter_using(a_ctx, &shape.0);
- Thunk!(move || evaluate(env, &shape.1))
}
pub fn fill_letrec_binds(fill: &LocalsFrame, ctx: &Context, binds: &[LBind]) {
for bind in binds {
- let value_thunk = build_b_thunk(ctx, &bind.value_shape, bind.value.clone());
- destruct(&bind.destruct, fill, value_thunk, ctx);
+ let expr = bind.value.clone();
+ let env = Context::enter_using(ctx, &bind.value_shape);
+ destruct(
+ &bind.destruct,
+ fill,
+ Thunk!(move || evaluate(env, &expr)),
+ ctx,
+ );
}
}
crates/jrsonnet-evaluator/src/evaluate/mod.rsdiffbeforeafterboth1use std::rc::Rc;23use jrsonnet_gcmodule::{Cc, Trace};4use jrsonnet_interner::IStr;5use jrsonnet_ir::ImportKind;6use jrsonnet_types::ValType;78use self::{9 compspec::{evaluate_arr_comp, evaluate_obj_comp},10 destructure::{build_b_thunk_uno, evaluate_local_expr, evaluate_locals_unbound},11 operator::evaluate_binary_op_special,12};13use crate::{14 Context, Error, ObjValue, ObjValueBuilder, ObjectAssertion, Result, ResultExt as _, SupThis,15 Unbound, Val,16 analyze::{17 ClosureShape, LArgsDesc, LAssertStmt, LExpr, LFieldMember, LFieldName, LFunction,18 LIndexPart, LObjAsserts, LObjBody, LObjMembers, LSlot,19 },20 arr::ArrValue,21 bail,22 error::{ErrorKind::*, suggest_object_fields},23 evaluate::{destructure::fill_letrec_binds, operator::evaluate_unary_op},24 function::{CallLocation, FuncDesc, FuncVal, prepared::PreparedFuncVal},25 in_frame,26 typed::{BoundedUsize, FromUntyped as _},27 val::{CachedUnbound, Thunk},28 with_state,29};3031pub mod compspec;32pub mod destructure;33pub mod operator;3435// This is the amount of bytes that need to be left on the stack before increasing the size.36// It must be at least as large as the stack required by any code that does not call37// `ensure_sufficient_stack`.38const RED_ZONE: usize = 100 * 1024;3940// Only the first stack that is pushed, grows exponentially (2^n * STACK_PER_RECURSION) from then41// on. This flag has performance relevant characteristics. Don't set it too high.42const STACK_PER_RECURSION: usize = 1024 * 1024;4344/// Grows the stack on demand to prevent stack overflow. Call this in strategic locations45/// to "break up" recursive calls. E.g. almost any call to `visit_expr` or equivalent can benefit46/// from this.47///48/// Should not be sprinkled around carelessly, as it causes a little bit of overhead.49#[inline]50pub fn ensure_sufficient_stack<R>(f: impl FnOnce() -> R) -> R {51 stacker::maybe_grow(RED_ZONE, STACK_PER_RECURSION, f)52}5354pub fn evaluate_trivial(expr: &LExpr) -> Option<Val> {55 // TODO: Eager trivial array56 Some(match expr {57 LExpr::Str(s) => Val::string(s.clone()),58 LExpr::Num(n) => Val::Num(*n),59 LExpr::Bool(false) => Val::Bool(false),60 LExpr::Bool(true) => Val::Bool(true),61 LExpr::Null => Val::Null,62 _ => return None,63 })64}6566pub fn evaluate_method(ctx: Context, name: IStr, func: &Rc<LFunction>) -> Val {67 Val::Func(FuncVal::Normal(Cc::new(FuncDesc {68 name,69 body_captures: ctx.pack_captures_sup_this(&func.body_shape),70 func: func.clone(),71 })))72}7374pub fn evaluate_field_name(ctx: Context, field_name: &LFieldName) -> Result<Option<IStr>> {75 Ok(match field_name {76 LFieldName::Fixed(n) => Some(n.clone()),77 LFieldName::Dyn(expr) => in_frame(78 // TODO: Spanned<LFieldName>79 CallLocation::native(),80 || "evaluating field name".to_string(),81 || {82 let v = evaluate(ctx.clone(), expr)?;83 Ok(if matches!(v, Val::Null) {84 None85 } else {86 Some(IStr::from_untyped(v)?)87 })88 },89 )?,90 })91}9293pub fn evaluate_thunk(ctx: Context, expr: Rc<LExpr>, tailstrict: bool) -> Result<Thunk<Val>> {94 match &*expr {95 LExpr::Slot(LSlot::Local(i)) => return Ok(ctx.local(*i)),96 LExpr::Slot(LSlot::Capture(i)) => return Ok(ctx.capture(*i)),97 _ => {98 if let Some(v) = evaluate_trivial(&expr) {99 return Ok(Thunk::evaluated(v));100 }101 }102 }103 Ok(if tailstrict {104 Thunk::evaluated(evaluate(ctx, &expr)?)105 } else {106 Thunk!(move || { evaluate(ctx, &expr) })107 })108}109110mod names {111 use crate::names;112113 names! {114 anonymous: "anonymous",115 }116}117118pub fn evaluate(ctx: Context, expr: &LExpr) -> Result<Val> {119 Ok(match expr {120 LExpr::Null => Val::Null,121 LExpr::Bool(b) => Val::Bool(*b),122 LExpr::Str(s) => Val::string(s.clone()),123 LExpr::Num(n) => Val::Num(*n),124 LExpr::Slot(slot) => ctx.slot(*slot).evaluate()?,125 LExpr::BadLocal(name) => panic!("unresolvable reference: {name}"),126 LExpr::Arr { shape, items } => Val::Arr(ArrValue::expr(ctx, shape, items.clone())),127 LExpr::UnaryOp(op, value) => {128 let value = evaluate(ctx, value)?;129 evaluate_unary_op(*op, &value)?130 }131 LExpr::BinaryOp { lhs, op, rhs } => evaluate_binary_op_special(ctx, lhs, *op, rhs)?,132 LExpr::LocalExpr(local_expr) => evaluate_local_expr(ctx, local_expr)?,133 LExpr::IfElse {134 cond,135 cond_then,136 cond_else,137 } => {138 let cond_val = evaluate(ctx.clone(), cond)?;139 let Val::Bool(b) = cond_val else {140 bail!(TypeMismatch(141 "if condition",142 vec![ValType::Bool],143 cond_val.value_type()144 ))145 };146 if b {147 evaluate(ctx, cond_then)?148 } else if let Some(e) = cond_else {149 evaluate(ctx, e)?150 } else {151 Val::Null152 }153 }154 LExpr::Error(s, e) => in_frame(155 CallLocation::new(s),156 || "error statement".to_owned(),157 || bail!(RuntimeError(evaluate(ctx, e)?.to_string()?,)),158 )?,159 LExpr::AssertExpr { assert, rest } => {160 evaluate_assert(ctx.clone(), assert)?;161 evaluate(ctx, rest)?162 }163164 LExpr::Function(func) => evaluate_method(165 ctx,166 func.name.clone().unwrap_or_else(names::anonymous),167 func,168 ),169 LExpr::IdentityFunction => Val::Func(FuncVal::identity()),170 LExpr::Apply {171 applicable,172 args,173 tailstrict,174 } => evaluate_apply(175 ctx,176 applicable,177 args,178 CallLocation::new(&args.span),179 *tailstrict,180 )?,181 LExpr::Index { indexable, parts } => evaluate_index(ctx, indexable, parts)?,182 LExpr::Obj(body) => evaluate_obj_body(None, ctx, body)?,183 LExpr::ObjExtend(lhs, body) => {184 let lhs_val = evaluate(ctx.clone(), lhs)?;185 let Val::Obj(lhs_obj) = lhs_val else {186 bail!(TypeMismatch(187 "object extend lhs",188 vec![ValType::Obj],189 lhs_val.value_type(),190 ))191 };192 evaluate_obj_body(Some(lhs_obj), ctx, body)?193 }194 LExpr::ArrComp(comp) => evaluate_arr_comp(ctx, comp)?,195 LExpr::Slice(slice) => {196 let val = evaluate(ctx.clone(), &slice.value)?;197 let indexable = val.into_indexable()?;198 let start = slice199 .start200 .as_ref()201 .map(|e| evaluate(ctx.clone(), e))202 .transpose()?203 .map(|v| -> Result<i32> { i32::from_untyped(v).description("slice start value") })204 .transpose()?;205 let end = slice206 .end207 .as_ref()208 .map(|e| evaluate(ctx.clone(), e))209 .transpose()?210 .map(|v| -> Result<i32> { i32::from_untyped(v).description("slice end value") })211 .transpose()?;212 let step = slice213 .step214 .as_ref()215 .map(|e| evaluate(ctx, e))216 .transpose()?217 .map(|v| -> Result<BoundedUsize<1, { i32::MAX as usize }>> {218 BoundedUsize::from_untyped(v).description("slice step value")219 })220 .transpose()?;221 Val::from(indexable.slice(start, end, step)?)222 }223 LExpr::Super => Val::Obj(ctx.try_sup_this()?.standalone_super().ok_or(NoSuperFound)?),224 LExpr::Import {225 kind,226 kind_span,227 path,228 } => with_state(|state| {229 let resolved = state.resolve_from(kind_span.0.source_path(), &path.clone())?;230 Ok::<_, Error>(match kind.value {231 ImportKind::Normal => in_frame(232 CallLocation::new(&kind.span),233 || "import".to_string(),234 || state.import_resolved(resolved),235 )?,236 ImportKind::Str => Val::string(state.import_resolved_str(resolved)?),237 ImportKind::Bin => Val::arr(state.import_resolved_bin(resolved)?),238 })239 })?,240 })241}242243fn evaluate_apply(244 ctx: Context,245 applicable: &LExpr,246 args: &LArgsDesc,247 loc: CallLocation<'_>,248 tailstrict: bool,249) -> Result<Val> {250 let func_val = evaluate(ctx.clone(), applicable)?;251 let Val::Func(func) = func_val else {252 bail!(OnlyFunctionsCanBeCalledGot(func_val.value_type()))253 };254255 if func.is_identity() && args.names.is_empty() && args.unnamed.len() == 1 {256 return evaluate_thunk(ctx, args.unnamed[0].clone(), tailstrict)?.evaluate();257 }258259 let name = func.name();260261 if args.names.is_empty() && args.unnamed.len() == 1 && func.params().len() == 1 {262 use crate::function::prepared::PreparedCall;263 let prepared_inline = PreparedCall::empty();264 let arg = evaluate_thunk(ctx, args.unnamed[0].clone(), tailstrict)?;265 let arg_slice = std::slice::from_ref(&arg);266 return in_frame(267 loc,268 || format!("function <{name}> call"),269 || {270 func.evaluate_prepared(271 &prepared_inline,272 CallLocation::native(),273 arg_slice,274 &[],275 tailstrict,276 )277 },278 );279 }280281 let unnamed = args282 .unnamed283 .iter()284 .cloned()285 .map(|e| evaluate_thunk(ctx.clone(), e, tailstrict))286 .collect::<Result<Vec<_>>>()?;287288 // Fast path: positional-only multi-arg call fully covering the289 // params, no defaults.290 if args.names.is_empty() && unnamed.len() == func.params().len() {291 use crate::function::prepared::PreparedCall;292 let prepared_inline = PreparedCall::empty();293 return in_frame(294 loc,295 || format!("function <{name}> call"),296 || {297 func.evaluate_prepared(298 &prepared_inline,299 CallLocation::native(),300 &unnamed,301 &[],302 tailstrict,303 )304 },305 );306 }307308 let named = args309 .values310 .iter()311 .cloned()312 .map(|e| evaluate_thunk(ctx.clone(), e, tailstrict))313 .collect::<Result<Vec<_>>>()?;314 let prepare = PreparedFuncVal::new(func, unnamed.len(), &args.names)315 .with_description_src(loc, || format!("function <{name}> preparation"))?;316 in_frame(317 loc,318 || format!("function <{name}> call"),319 || prepare.call(CallLocation::native(), &unnamed, &named),320 )321}322323fn evaluate_index(ctx: Context, indexable: &LExpr, parts: &[LIndexPart]) -> Result<Val> {324 let mut parts = parts.iter();325 let mut indexable = if matches!(indexable, LExpr::Super) {326 let part = parts.next().expect("at least part should exist");327 // sup_this existence check might also be skipped here for null-coalesce...328 // But I believe this might cause errors.329 let sup_this = ctx.try_sup_this()?;330331 if !sup_this.has_super() {332 #[cfg(feature = "exp-null-coaelse")]333 if part.null_coaelse {334 return Ok(Val::Null);335 }336 bail!(NoSuperFound);337 }338 let name = evaluate(ctx.clone(), &part.value)?;339340 let Val::Str(name) = name else {341 bail!(ValueIndexMustBeTypeGot(342 ValType::Obj,343 ValType::Str,344 name.value_type(),345 ))346 };347348 let name = name.into_flat();349 match sup_this350 .get_super(name.clone())351 .with_description_src(&part.span, || format!("super field <{name}> access"))?352 {353 Some(v) => v,354 #[cfg(feature = "exp-null-coaelse")]355 None if part.null_coaelse => return Ok(Val::Null),356 None => {357 let suggestions = suggest_object_fields(358 &sup_this.standalone_super().expect("super exists"),359 name.clone(),360 );361 bail!(NoSuchField(name, suggestions))362 }363 }364 } else {365 evaluate(ctx.clone(), indexable)?366 };367368 for part in parts {369 let ctx = ctx.clone();370 let loc = CallLocation::new(&part.span);371 let value = indexable;372 let key_val = evaluate(ctx, &part.value)?;373 indexable = match (&value, &key_val) {374 (Val::Obj(obj), Val::Str(key)) => {375 let key = key.clone().into_flat();376 match obj377 .get(key.clone())378 .with_description_src(loc, || format!("field <{key}> access"))?379 {380 Some(v) => v,381 #[cfg(feature = "exp-null-coaelse")]382 None if part.null_coaelse => return Ok(Val::Null),383 None => {384 return Err(Error::from(NoSuchField(385 key.clone(),386 suggest_object_fields(obj, key.clone()),387 )))388 .with_description_src(loc, || format!("field <{key}> access"));389 }390 }391 }392 (Val::Arr(arr), Val::Num(idx)) => {393 let n = idx.get();394 if n.fract() > f64::EPSILON {395 bail!(FractionalIndex)396 }397 let len = arr.len();398 if n < 0.0 || n > f64::from(len) {399 bail!(ArrayBoundsError(n, len));400 }401 #[expect(402 clippy::cast_possible_truncation,403 clippy::cast_sign_loss,404 reason = "n is checked positive"405 )]406 let i = n as u32;407 arr.get(i)408 .with_description_src(loc, || format!("element <{i}> access"))?409 .ok_or_else(|| ArrayBoundsError(n, len))?410 }411 (Val::Str(s), Val::Num(idx)) => {412 let n = idx.get();413 if n.fract() > f64::EPSILON {414 bail!(FractionalIndex)415 }416 #[expect(417 clippy::cast_possible_truncation,418 clippy::cast_sign_loss,419 reason = "n is checked positive, overflow will truncate as expected"420 )]421 let i = n as usize;422 let flat = s.clone().into_flat();423 #[allow(clippy::cast_possible_truncation, reason = "string is max 4g")]424 if n >= 0.0425 && n <= f64::from(u32::MAX)426 && let Some(char) = flat.chars().nth(i)427 {428 Val::string(char)429 } else {430 let len = flat.chars().count();431 bail!(StringBoundsError(n, len as u32))432 }433 }434 #[cfg(feature = "exp-null-coaelse")]435 (Val::Null, _) if part.null_coaelse => return Ok(Val::Null),436 _ => bail!(ValueIndexMustBeTypeGot(437 value.value_type(),438 ValType::Str,439 key_val.value_type()440 )),441 };442 }443 Ok(indexable)444}445446fn evaluate_obj_body(super_obj: Option<ObjValue>, ctx: Context, body: &LObjBody) -> Result<Val> {447 match body {448 LObjBody::MemberList(members) => evaluate_obj_members(super_obj, ctx, members),449 LObjBody::ObjComp(comp) => evaluate_obj_comp(super_obj, ctx, comp),450 }451}452453pub fn evaluate_field_member_unbound<B: Unbound<Bound = Context> + Clone>(454 builder: &mut ObjValueBuilder,455 ctx: Context,456 uctx: B,457 field: &LFieldMember,458) -> Result<()> {459 #[derive(Trace)]460 struct UnboundValue<B: Trace> {461 uctx: B,462 value: Rc<(ClosureShape, LExpr)>,463 name: IStr,464 }465 impl<B: Unbound<Bound = Context>> Unbound for UnboundValue<B> {466 type Bound = Val;467 fn bind(&self, sup_this: SupThis) -> Result<Val> {468 let a_ctx = self.uctx.bind(sup_this)?;469 let b_ctx = Context::enter_using(&a_ctx, &self.value.0);470 evaluate(b_ctx, &self.value.1)471 }472 }473474 let LFieldMember {475 name,476 plus,477 visibility,478 value,479 } = field;480 let Some(name) = evaluate_field_name(ctx, name)? else {481 return Ok(());482 };483484 builder485 .field(name.clone())486 .with_add(*plus)487 .with_visibility(*visibility)488 .bindable(UnboundValue {489 uctx,490 value: value.clone(),491 name,492 })493}494pub fn evaluate_field_member_static(495 builder: &mut ObjValueBuilder,496 field_ctx: Context,497 value_ctx: Context,498 field: &LFieldMember,499) -> Result<()> {500 let LFieldMember {501 name,502 plus,503 visibility,504 value,505 } = field;506 let Some(name) = evaluate_field_name(field_ctx, name)? else {507 return Ok(());508 };509510 let thunk = build_b_thunk_uno(&value_ctx, value.clone());511 builder512 .field(name)513 .with_add(*plus)514 .with_visibility(*visibility)515 .try_thunk(thunk)?;516 Ok(())517}518519fn evaluate_obj_members(520 super_obj: Option<ObjValue>,521 ctx: Context,522 members: &LObjMembers,523) -> Result<Val> {524 let mut builder = ObjValueBuilder::with_capacity(members.fields.len());525 if let Some(sup) = super_obj {526 builder.with_super(sup);527 }528529 let needs_unbound = members.this.is_some() || members.uses_super;530531 if needs_unbound {532 let uctx = CachedUnbound::new(evaluate_locals_unbound(533 &ctx,534 &members.frame_shape,535 members.this,536 members.locals.clone(),537 ));538 for field in &members.fields {539 evaluate_field_member_unbound(&mut builder, ctx.clone(), uctx.clone(), field)?;540 }541 if let Some(asserts_block) = &members.asserts {542 builder.assert(evaluate_object_assertions_unbound(543 uctx,544 asserts_block.clone(),545 ));546 }547 } else {548 let a_ctx = ctx549 .pack_captures_sup_this(&members.frame_shape)550 .enter(|fill, ctx| {551 fill_letrec_binds(fill, ctx, &members.locals);552 });553 for field in &members.fields {554 evaluate_field_member_static(&mut builder, ctx.clone(), a_ctx.clone(), field)?;555 }556 if let Some(asserts_block) = &members.asserts {557 builder.assert(evaluate_object_assertions_static(558 a_ctx,559 asserts_block.clone(),560 ));561 }562 }563564 Ok(Val::Obj(builder.build()))565}566567pub fn evaluate_assert(ctx: Context, assertion: &LAssertStmt) -> Result<()> {568 let LAssertStmt { cond, message } = assertion;569 let assertion_result = in_frame(570 CallLocation::new(&cond.span),571 || "assertion condition".to_owned(),572 || bool::from_untyped(evaluate(ctx.clone(), cond)?),573 )?;574 if !assertion_result {575 in_frame(576 CallLocation::new(&cond.span),577 || "assertion failure".to_owned(),578 || {579 if let Some(msg) = message {580 bail!(AssertionFailed(evaluate(ctx, msg)?.to_string()?));581 }582 bail!(AssertionFailed(Val::Null.to_string()?));583 },584 )?;585 }586 Ok(())587}588589fn evaluate_object_assertions_unbound<B: Unbound<Bound = Context>>(590 uctx: B,591 asserts: Rc<LObjAsserts>,592) -> impl ObjectAssertion {593 #[derive(Trace)]594 struct ObjectAssert<B: Trace> {595 uctx: B,596 asserts: Rc<LObjAsserts>,597 }598 impl<B: Unbound<Bound = Context>> ObjectAssertion for ObjectAssert<B> {599 fn run(&self, sup_this: SupThis) -> Result<()> {600 let a_ctx = self.uctx.bind(sup_this)?;601 let assert_env = Context::enter_using(&a_ctx, &self.asserts.shape);602 for assert in &self.asserts.asserts {603 evaluate_assert(assert_env.clone(), assert)?;604 }605 Ok(())606 }607 }608 ObjectAssert { uctx, asserts }609}610fn evaluate_object_assertions_static(611 a_ctx: Context,612 asserts: Rc<LObjAsserts>,613) -> impl ObjectAssertion {614 #[derive(Trace)]615 struct ObjectAssert {616 assert_env: Context,617 asserts: Rc<LObjAsserts>,618 }619 impl ObjectAssertion for ObjectAssert {620 fn run(&self, _sup_this: SupThis) -> Result<()> {621 for assert in &self.asserts.asserts {622 evaluate_assert(self.assert_env.clone(), assert)?;623 }624 Ok(())625 }626 }627 let assert_env = Context::enter_using(&a_ctx, &asserts.shape);628 ObjectAssert {629 assert_env,630 asserts,631 }632}1use std::rc::Rc;23use jrsonnet_gcmodule::{Cc, Trace};4use jrsonnet_interner::IStr;5use jrsonnet_ir::ImportKind;6use jrsonnet_types::ValType;78use self::{9 compspec::{evaluate_arr_comp, evaluate_obj_comp},10 destructure::{evaluate_local_expr, evaluate_locals_unbound},11 operator::evaluate_binary_op_special,12};13use crate::{14 Context, Error, ObjValue, ObjValueBuilder, ObjectAssertion, Result, ResultExt as _, SupThis,15 Unbound, Val,16 analyze::{17 ClosureShape, LArgsDesc, LAssertStmt, LExpr, LFieldMember, LFieldName, LFunction,18 LIndexPart, LObjAsserts, LObjBody, LObjMembers, LSlot,19 },20 arr::ArrValue,21 bail,22 error::{ErrorKind::*, suggest_object_fields},23 evaluate::{destructure::fill_letrec_binds, operator::evaluate_unary_op},24 function::{CallLocation, FuncDesc, FuncVal, prepared::PreparedFuncVal},25 in_frame,26 typed::{BoundedUsize, FromUntyped as _},27 val::{CachedUnbound, Thunk},28 with_state,29};3031pub mod compspec;32pub mod destructure;33pub mod operator;3435// This is the amount of bytes that need to be left on the stack before increasing the size.36// It must be at least as large as the stack required by any code that does not call37// `ensure_sufficient_stack`.38const RED_ZONE: usize = 100 * 1024;3940// Only the first stack that is pushed, grows exponentially (2^n * STACK_PER_RECURSION) from then41// on. This flag has performance relevant characteristics. Don't set it too high.42const STACK_PER_RECURSION: usize = 1024 * 1024;4344/// Grows the stack on demand to prevent stack overflow. Call this in strategic locations45/// to "break up" recursive calls. E.g. almost any call to `visit_expr` or equivalent can benefit46/// from this.47///48/// Should not be sprinkled around carelessly, as it causes a little bit of overhead.49#[inline]50pub fn ensure_sufficient_stack<R>(f: impl FnOnce() -> R) -> R {51 stacker::maybe_grow(RED_ZONE, STACK_PER_RECURSION, f)52}5354pub fn evaluate_trivial(expr: &LExpr) -> Option<Val> {55 // TODO: Eager trivial array56 Some(match expr {57 LExpr::Str(s) => Val::string(s.clone()),58 LExpr::Num(n) => Val::Num(*n),59 LExpr::Bool(false) => Val::Bool(false),60 LExpr::Bool(true) => Val::Bool(true),61 LExpr::Null => Val::Null,62 _ => return None,63 })64}6566pub fn evaluate_method(ctx: Context, name: IStr, func: &Rc<LFunction>) -> Val {67 Val::Func(FuncVal::Normal(Cc::new(FuncDesc {68 name,69 body_captures: ctx.pack_captures_sup_this(&func.body_shape),70 func: func.clone(),71 })))72}7374pub fn evaluate_field_name(ctx: Context, field_name: &LFieldName) -> Result<Option<IStr>> {75 Ok(match field_name {76 LFieldName::Fixed(n) => Some(n.clone()),77 LFieldName::Dyn(expr) => in_frame(78 // TODO: Spanned<LFieldName>79 CallLocation::native(),80 || "evaluating field name".to_string(),81 || {82 let v = evaluate(ctx.clone(), expr)?;83 Ok(if matches!(v, Val::Null) {84 None85 } else {86 Some(IStr::from_untyped(v)?)87 })88 },89 )?,90 })91}9293pub fn evaluate_thunk(ctx: Context, expr: Rc<LExpr>, tailstrict: bool) -> Result<Thunk<Val>> {94 match &*expr {95 LExpr::Slot(LSlot::Local(i)) => return Ok(ctx.local(*i)),96 LExpr::Slot(LSlot::Capture(i)) => return Ok(ctx.capture(*i)),97 _ => {98 if let Some(v) = evaluate_trivial(&expr) {99 return Ok(Thunk::evaluated(v));100 }101 }102 }103 Ok(if tailstrict {104 Thunk::evaluated(evaluate(ctx, &expr)?)105 } else {106 Thunk!(move || { evaluate(ctx, &expr) })107 })108}109110mod names {111 use crate::names;112113 names! {114 anonymous: "anonymous",115 }116}117118#[allow(clippy::too_many_lines)]119pub fn evaluate(ctx: Context, expr: &LExpr) -> Result<Val> {120 Ok(match expr {121 LExpr::Null => Val::Null,122 LExpr::Bool(b) => Val::Bool(*b),123 LExpr::Str(s) => Val::string(s.clone()),124 LExpr::Num(n) => Val::Num(*n),125 LExpr::Slot(slot) => ctx.slot(*slot).evaluate()?,126 LExpr::BadLocal(name) => panic!("unresolvable reference: {name}"),127 LExpr::Arr { shape, items } => Val::Arr(ArrValue::expr(ctx, shape, items.clone())),128 LExpr::UnaryOp(op, value) => {129 let value = evaluate(ctx, value)?;130 evaluate_unary_op(*op, &value)?131 }132 LExpr::BinaryOp { lhs, op, rhs } => evaluate_binary_op_special(ctx, lhs, *op, rhs)?,133 LExpr::LocalExpr(local_expr) => evaluate_local_expr(ctx, local_expr)?,134 LExpr::IfElse {135 cond,136 cond_then,137 cond_else,138 } => {139 let cond_val = evaluate(ctx.clone(), cond)?;140 let Val::Bool(b) = cond_val else {141 bail!(TypeMismatch(142 "if condition",143 vec![ValType::Bool],144 cond_val.value_type()145 ))146 };147 if b {148 evaluate(ctx, cond_then)?149 } else if let Some(e) = cond_else {150 evaluate(ctx, e)?151 } else {152 Val::Null153 }154 }155 LExpr::Error(s, e) => in_frame(156 CallLocation::new(s),157 || "error statement".to_owned(),158 || bail!(RuntimeError(evaluate(ctx, e)?.to_string()?,)),159 )?,160 LExpr::AssertExpr { assert, rest } => {161 evaluate_assert(ctx.clone(), assert)?;162 evaluate(ctx, rest)?163 }164165 LExpr::Function(func) => evaluate_method(166 ctx,167 func.name.clone().unwrap_or_else(names::anonymous),168 func,169 ),170 LExpr::IdentityFunction => Val::Func(FuncVal::identity()),171 LExpr::Apply {172 applicable,173 args,174 tailstrict,175 } => evaluate_apply(176 ctx,177 applicable,178 args,179 CallLocation::new(&args.span),180 *tailstrict,181 )?,182 LExpr::Index { indexable, parts } => evaluate_index(ctx, indexable, parts)?,183 LExpr::Obj(body) => evaluate_obj_body(None, ctx, body)?,184 LExpr::ObjExtend(lhs, body) => {185 let lhs_val = evaluate(ctx.clone(), lhs)?;186 let Val::Obj(lhs_obj) = lhs_val else {187 bail!(TypeMismatch(188 "object extend lhs",189 vec![ValType::Obj],190 lhs_val.value_type(),191 ))192 };193 evaluate_obj_body(Some(lhs_obj), ctx, body)?194 }195 LExpr::ArrComp(comp) => evaluate_arr_comp(ctx, comp)?,196 LExpr::Slice(slice) => {197 let val = evaluate(ctx.clone(), &slice.value)?;198 let indexable = val.into_indexable()?;199 let start = slice200 .start201 .as_ref()202 .map(|e| evaluate(ctx.clone(), e))203 .transpose()?204 .map(|v| -> Result<i32> { i32::from_untyped(v).description("slice start value") })205 .transpose()?;206 let end = slice207 .end208 .as_ref()209 .map(|e| evaluate(ctx.clone(), e))210 .transpose()?211 .map(|v| -> Result<i32> { i32::from_untyped(v).description("slice end value") })212 .transpose()?;213 let step = slice214 .step215 .as_ref()216 .map(|e| evaluate(ctx, e))217 .transpose()?218 .map(|v| -> Result<BoundedUsize<1, { i32::MAX as usize }>> {219 BoundedUsize::from_untyped(v).description("slice step value")220 })221 .transpose()?;222 Val::from(indexable.slice32(start, end, step)?)223 }224 LExpr::Super => Val::Obj(ctx.try_sup_this()?.standalone_super().ok_or(NoSuperFound)?),225 LExpr::Import {226 kind,227 kind_span,228 path,229 } => with_state(|state| {230 let resolved = state.resolve_from(kind_span.0.source_path(), &path.clone())?;231 Ok::<_, Error>(match kind.value {232 ImportKind::Normal => in_frame(233 CallLocation::new(&kind.span),234 || "import".to_string(),235 || state.import_resolved(resolved),236 )?,237 ImportKind::Str => Val::string(state.import_resolved_str(resolved)?),238 ImportKind::Bin => Val::arr(state.import_resolved_bin(resolved)?),239 })240 })?,241 })242}243244fn evaluate_apply(245 ctx: Context,246 applicable: &LExpr,247 args: &LArgsDesc,248 loc: CallLocation<'_>,249 tailstrict: bool,250) -> Result<Val> {251 let func_val = evaluate(ctx.clone(), applicable)?;252 let Val::Func(func) = func_val else {253 bail!(OnlyFunctionsCanBeCalledGot(func_val.value_type()))254 };255256 if func.is_identity() && args.names.is_empty() && args.unnamed.len() == 1 {257 return evaluate_thunk(ctx, args.unnamed[0].clone(), tailstrict)?.evaluate();258 }259260 let name = func.name();261262 if args.names.is_empty() && args.unnamed.len() == 1 && func.params().len() == 1 {263 use crate::function::prepared::PreparedCall;264 let prepared_inline = PreparedCall::empty();265 let arg = evaluate_thunk(ctx, args.unnamed[0].clone(), tailstrict)?;266 let arg_slice = std::slice::from_ref(&arg);267 return in_frame(268 loc,269 || format!("function <{name}> call"),270 || {271 func.evaluate_prepared(272 &prepared_inline,273 CallLocation::native(),274 arg_slice,275 &[],276 tailstrict,277 )278 },279 );280 }281282 let unnamed = args283 .unnamed284 .iter()285 .cloned()286 .map(|e| evaluate_thunk(ctx.clone(), e, tailstrict))287 .collect::<Result<Vec<_>>>()?;288289 // Fast path: positional-only multi-arg call fully covering the290 // params, no defaults.291 if args.names.is_empty() && unnamed.len() == func.params().len() {292 use crate::function::prepared::PreparedCall;293 let prepared_inline = PreparedCall::empty();294 return in_frame(295 loc,296 || format!("function <{name}> call"),297 || {298 func.evaluate_prepared(299 &prepared_inline,300 CallLocation::native(),301 &unnamed,302 &[],303 tailstrict,304 )305 },306 );307 }308309 let named = args310 .values311 .iter()312 .cloned()313 .map(|e| evaluate_thunk(ctx.clone(), e, tailstrict))314 .collect::<Result<Vec<_>>>()?;315 let prepare = PreparedFuncVal::new(func, unnamed.len(), &args.names)316 .with_description_src(loc, || format!("function <{name}> preparation"))?;317 in_frame(318 loc,319 || format!("function <{name}> call"),320 || prepare.call(CallLocation::native(), &unnamed, &named),321 )322}323324#[allow(clippy::too_many_lines)]325fn evaluate_index(ctx: Context, indexable: &LExpr, parts: &[LIndexPart]) -> Result<Val> {326 let mut parts = parts.iter();327 let mut indexable = if matches!(indexable, LExpr::Super) {328 let part = parts.next().expect("at least part should exist");329 // sup_this existence check might also be skipped here for null-coalesce...330 // But I believe this might cause errors.331 let sup_this = ctx.try_sup_this()?;332333 if !sup_this.has_super() {334 #[cfg(feature = "exp-null-coaelse")]335 if part.null_coaelse {336 return Ok(Val::Null);337 }338 bail!(NoSuperFound);339 }340 let name = evaluate(ctx.clone(), &part.value)?;341342 let Val::Str(name) = name else {343 bail!(ValueIndexMustBeTypeGot(344 ValType::Obj,345 ValType::Str,346 name.value_type(),347 ))348 };349350 let name = name.into_flat();351 match sup_this352 .get_super(name.clone())353 .with_description_src(&part.span, || format!("super field <{name}> access"))?354 {355 Some(v) => v,356 #[cfg(feature = "exp-null-coaelse")]357 None if part.null_coaelse => return Ok(Val::Null),358 None => {359 let suggestions = suggest_object_fields(360 &sup_this.standalone_super().expect("super exists"),361 name.clone(),362 );363 bail!(NoSuchField(name, suggestions))364 }365 }366 } else {367 evaluate(ctx.clone(), indexable)?368 };369370 for part in parts {371 let ctx = ctx.clone();372 let loc = CallLocation::new(&part.span);373 let value = indexable;374 let key_val = evaluate(ctx, &part.value)?;375 indexable = match (&value, &key_val) {376 (Val::Obj(obj), Val::Str(key)) => {377 let key = key.clone().into_flat();378 match obj379 .get(key.clone())380 .with_description_src(loc, || format!("field <{key}> access"))?381 {382 Some(v) => v,383 #[cfg(feature = "exp-null-coaelse")]384 None if part.null_coaelse => return Ok(Val::Null),385 None => {386 return Err(Error::from(NoSuchField(387 key.clone(),388 suggest_object_fields(obj, key.clone()),389 )))390 .with_description_src(loc, || format!("field <{key}> access"));391 }392 }393 }394 (Val::Arr(arr), Val::Num(idx)) => {395 let n = idx.get();396 if n.fract() > f64::EPSILON {397 bail!(FractionalIndex)398 }399 let len = arr.len32();400 if n < 0.0 || n > f64::from(len) {401 bail!(ArrayBoundsError(n, len));402 }403 #[expect(404 clippy::cast_possible_truncation,405 clippy::cast_sign_loss,406 reason = "n is checked range"407 )]408 let i = n as u32;409 arr.get32(i)410 .with_description_src(loc, || format!("element <{i}> access"))?411 .ok_or_else(|| ArrayBoundsError(n, len))?412 }413 (Val::Str(s), Val::Num(idx)) => {414 let n = idx.get();415 if n.fract() > f64::EPSILON {416 bail!(FractionalIndex)417 }418 #[expect(419 clippy::cast_possible_truncation,420 clippy::cast_sign_loss,421 reason = "n is checked positive, overflow will truncate as expected"422 )]423 let i = n as usize;424 let flat = s.clone().into_flat();425 #[allow(clippy::cast_possible_truncation, reason = "string is max 4g")]426 if n >= 0.0427 && n <= f64::from(u32::MAX)428 && let Some(char) = flat.chars().nth(i)429 {430 Val::string(char)431 } else {432 let len = flat.chars().count();433 bail!(StringBoundsError(n, len as u32))434 }435 }436 #[cfg(feature = "exp-null-coaelse")]437 (Val::Null, _) if part.null_coaelse => return Ok(Val::Null),438 _ => bail!(ValueIndexMustBeTypeGot(439 value.value_type(),440 ValType::Str,441 key_val.value_type()442 )),443 };444 }445 Ok(indexable)446}447448fn evaluate_obj_body(super_obj: Option<ObjValue>, ctx: Context, body: &LObjBody) -> Result<Val> {449 match body {450 LObjBody::MemberList(members) => evaluate_obj_members(super_obj, ctx, members),451 LObjBody::ObjComp(comp) => evaluate_obj_comp(super_obj, ctx, comp),452 }453}454455pub fn evaluate_field_member_unbound<B: Unbound<Bound = Context> + Clone>(456 builder: &mut ObjValueBuilder,457 ctx: Context,458 uctx: B,459 field: &LFieldMember,460) -> Result<()> {461 #[derive(Trace)]462 struct UnboundValue<B: Trace> {463 uctx: B,464 value: Rc<(ClosureShape, LExpr)>,465 name: IStr,466 }467 impl<B: Unbound<Bound = Context>> Unbound for UnboundValue<B> {468 type Bound = Val;469 fn bind(&self, sup_this: SupThis) -> Result<Val> {470 let a_ctx = self.uctx.bind(sup_this)?;471 let b_ctx = Context::enter_using(&a_ctx, &self.value.0);472 evaluate(b_ctx, &self.value.1)473 }474 }475476 let LFieldMember {477 name,478 plus,479 visibility,480 value,481 } = field;482 let Some(name) = evaluate_field_name(ctx, name)? else {483 return Ok(());484 };485486 builder487 .field(name.clone())488 .with_add(*plus)489 .with_visibility(*visibility)490 .bindable(UnboundValue {491 uctx,492 value: value.clone(),493 name,494 })495}496pub fn evaluate_field_member_static(497 builder: &mut ObjValueBuilder,498 field_ctx: Context,499 value_ctx: Context,500 field: &LFieldMember,501) -> Result<()> {502 let LFieldMember {503 name,504 plus,505 visibility,506 value,507 } = field;508 let Some(name) = evaluate_field_name(field_ctx, name)? else {509 return Ok(());510 };511512 let env = Context::enter_using(&value_ctx, &value.0);513 let value = value.clone();514 builder515 .field(name)516 .with_add(*plus)517 .with_visibility(*visibility)518 .try_thunk(Thunk!(move || evaluate(env, &value.1)))?;519 Ok(())520}521522fn evaluate_obj_members(523 super_obj: Option<ObjValue>,524 ctx: Context,525 members: &LObjMembers,526) -> Result<Val> {527 let mut builder = ObjValueBuilder::with_capacity(members.fields.len());528 if let Some(sup) = super_obj {529 builder.with_super(sup);530 }531532 let needs_unbound = members.this.is_some() || members.uses_super;533534 if needs_unbound {535 let uctx = CachedUnbound::new(evaluate_locals_unbound(536 &ctx,537 &members.frame_shape,538 members.this,539 members.locals.clone(),540 ));541 for field in &members.fields {542 evaluate_field_member_unbound(&mut builder, ctx.clone(), uctx.clone(), field)?;543 }544 if let Some(asserts_block) = &members.asserts {545 builder.assert(evaluate_object_assertions_unbound(546 uctx,547 asserts_block.clone(),548 ));549 }550 } else {551 let a_ctx = ctx552 .pack_captures_sup_this(&members.frame_shape)553 .enter(|fill, ctx| {554 fill_letrec_binds(fill, ctx, &members.locals);555 });556 for field in &members.fields {557 evaluate_field_member_static(&mut builder, ctx.clone(), a_ctx.clone(), field)?;558 }559 if let Some(asserts_block) = &members.asserts {560 builder.assert(evaluate_object_assertions_static(561 a_ctx,562 asserts_block.clone(),563 ));564 }565 }566567 Ok(Val::Obj(builder.build()))568}569570pub fn evaluate_assert(ctx: Context, assertion: &LAssertStmt) -> Result<()> {571 let LAssertStmt { cond, message } = assertion;572 let assertion_result = in_frame(573 CallLocation::new(&cond.span),574 || "assertion condition".to_owned(),575 || bool::from_untyped(evaluate(ctx.clone(), cond)?),576 )?;577 if !assertion_result {578 in_frame(579 CallLocation::new(&cond.span),580 || "assertion failure".to_owned(),581 || {582 if let Some(msg) = message {583 bail!(AssertionFailed(evaluate(ctx, msg)?.to_string()?));584 }585 bail!(AssertionFailed(Val::Null.to_string()?));586 },587 )?;588 }589 Ok(())590}591592fn evaluate_object_assertions_unbound<B: Unbound<Bound = Context>>(593 uctx: B,594 asserts: Rc<LObjAsserts>,595) -> impl ObjectAssertion {596 #[derive(Trace)]597 struct ObjectAssert<B: Trace> {598 uctx: B,599 asserts: Rc<LObjAsserts>,600 }601 impl<B: Unbound<Bound = Context>> ObjectAssertion for ObjectAssert<B> {602 fn run(&self, sup_this: SupThis) -> Result<()> {603 let a_ctx = self.uctx.bind(sup_this)?;604 let assert_env = Context::enter_using(&a_ctx, &self.asserts.shape);605 for assert in &self.asserts.asserts {606 evaluate_assert(assert_env.clone(), assert)?;607 }608 Ok(())609 }610 }611 ObjectAssert { uctx, asserts }612}613fn evaluate_object_assertions_static(614 a_ctx: Context,615 asserts: Rc<LObjAsserts>,616) -> impl ObjectAssertion {617 #[derive(Trace)]618 struct ObjectAssert {619 assert_env: Context,620 asserts: Rc<LObjAsserts>,621 }622 impl ObjectAssertion for ObjectAssert {623 fn run(&self, _sup_this: SupThis) -> Result<()> {624 for assert in &self.asserts.asserts {625 evaluate_assert(self.assert_env.clone(), assert)?;626 }627 Ok(())628 }629 }630 let assert_env = Context::enter_using(&a_ctx, &asserts.shape);631 ObjectAssert {632 assert_env,633 asserts,634 }635}crates/jrsonnet-evaluator/src/function/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/function/mod.rs
+++ b/crates/jrsonnet-evaluator/src/function/mod.rs
@@ -11,12 +11,10 @@
prepared::{PreparedCall, parse_prepared_builtin_call},
};
use crate::{
- PackedContextSupThis, Result, Thunk, Val,
+ Context, PackedContextSupThis, Result, Thunk, Val,
analyze::LFunction,
- evaluate::{
- destructure::{build_b_thunk, destruct},
- ensure_sufficient_stack, evaluate, evaluate_trivial,
- },
+ arr::arridx,
+ evaluate::{destructure::destruct, ensure_sufficient_stack, evaluate, evaluate_trivial},
function::builtin::BuiltinFunc,
};
@@ -83,7 +81,7 @@
&self.func.params[param_idx].destruct,
fill,
thunk.clone(),
- &ctx,
+ ctx,
);
}
for &(param_idx, arg_idx) in prepared.named() {
@@ -91,15 +89,22 @@
&self.func.params[param_idx].destruct,
fill,
named[arg_idx].clone(),
- &ctx,
+ ctx,
);
}
for ¶m_idx in prepared.defaults() {
let param = &self.func.params[param_idx];
let (shape, expr) = param.default.as_ref().expect("default exists");
- let thunk = build_b_thunk(&ctx, shape, expr.clone());
- destruct(¶m.destruct, fill, thunk, &ctx);
+ let expr = expr.clone();
+ let env = Context::enter_using(ctx, shape);
+
+ destruct(
+ ¶m.destruct,
+ fill,
+ Thunk!(move || evaluate(env, &expr)),
+ ctx,
+ );
}
});
@@ -152,8 +157,8 @@
}
}
/// Amount of non-default required arguments
- pub fn params_len(&self) -> u32 {
- self.params().iter().filter(|p| !p.has_default()).count() as u32
+ pub fn params_len32(&self) -> u32 {
+ arridx(self.params().iter().filter(|p| !p.has_default()).count())
}
/// Function name, as defined in code.
pub fn name(&self) -> IStr {
crates/jrsonnet-evaluator/src/integrations/serde.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/integrations/serde.rs
+++ b/crates/jrsonnet-evaluator/src/integrations/serde.rs
@@ -182,7 +182,7 @@
#[cfg(feature = "exp-bigint")]
Self::BigInt(b) => b.serialize(serializer),
Self::Arr(arr) => {
- let mut seq = serializer.serialize_seq(Some(arr.len() as usize))?;
+ let mut seq = serializer.serialize_seq(Some(arr.len()))?;
for (i, element) in arr.iter().enumerate() {
let mut serde_error = None;
in_description_frame(
@@ -203,7 +203,7 @@
seq.end()
}
Self::Obj(obj) => {
- let mut map = serializer.serialize_map(Some(obj.len() as usize))?;
+ let mut map = serializer.serialize_map(Some(obj.len32() as usize))?;
for (field, value) in obj.iter(
#[cfg(feature = "exp-preserve-order")]
true,
crates/jrsonnet-evaluator/src/obj/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/obj/mod.rs
+++ b/crates/jrsonnet-evaluator/src/obj/mod.rs
@@ -23,7 +23,7 @@
use crate::{
CcUnbound, MaybeUnbound, Result, Thunk, Unbound, Val,
- arr::{PickObjectKeyValues, PickObjectValues},
+ arr::{PickObjectKeyValues, PickObjectValues, arridx},
bail,
error::{ErrorKind::*, suggest_object_fields},
evaluate::operator::evaluate_add_op,
@@ -510,11 +510,14 @@
// }
/// Returns amount of visible object fields
/// If object only contains hidden fields - may return zero.
- pub fn len(&self) -> u32 {
+ pub fn len(&self) -> usize {
self.fields_visibility()
.values()
.filter(|d| d.visible())
- .count() as u32
+ .count()
+ }
+ pub fn len32(&self) -> u32 {
+ arridx(self.len())
}
/// For each field, calls callback.
/// If callback returns false - ends iteration prematurely.
@@ -625,7 +628,7 @@
Entry::Vacant(v) => {
v.insert(CacheValue::Pending);
}
- };
+ }
}
let result = self.get_idx_uncached(key, core);
{
crates/jrsonnet-evaluator/src/stack.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/stack.rs
+++ b/crates/jrsonnet-evaluator/src/stack.rs
@@ -11,7 +11,7 @@
struct NightlyLocalKey<T>(pub T);
#[cfg(nightly)]
impl<T> NightlyLocalKey<T> {
- #[inline(always)]
+ #[inline]
fn with<U>(&self, v: impl FnOnce(&T) -> U) -> U {
v(&self.0)
}
crates/jrsonnet-evaluator/src/trace/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/trace/mod.rs
+++ b/crates/jrsonnet-evaluator/src/trace/mod.rs
@@ -197,7 +197,7 @@
w = align
)?;
} else {
- write!(out, "{:<p$}{}", "", el.desc, p = self.padding,)?;
+ write!(out, "{:<p$}{}", "", el.desc, p = self.padding)?;
}
}
Ok(())
@@ -258,6 +258,7 @@
}
#[cfg(feature = "explaining-traces")]
impl TraceFormat for HiDocFormat {
+ #[allow(clippy::too_many_lines)]
fn write_trace(&self, out: &mut dyn fmt::Write, error: &Error) -> Result<(), fmt::Error> {
struct ResetData {
loc: Span,
crates/jrsonnet-evaluator/src/typed/conversions.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/typed/conversions.rs
+++ b/crates/jrsonnet-evaluator/src/typed/conversions.rs
@@ -637,7 +637,7 @@
}
<Self as Typed>::TYPE.check(&value)?;
// Any::downcast_ref::<ByteArray>(&a);
- let mut out = Vec::with_capacity(a.len() as usize);
+ let mut out = Vec::with_capacity(a.len());
for e in a.iter() {
let r = e?;
out.push(u8::from_untyped(r)?);
crates/jrsonnet-evaluator/src/val.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/val.rs
+++ b/crates/jrsonnet-evaluator/src/val.rs
@@ -277,7 +277,7 @@
/// For strings, will create a copy of specified interval.
///
/// For arrays, nothing will be copied on this call, instead [`ArrValue::Slice`] view will be returned.
- pub fn slice(
+ pub fn slice32(
self,
index: Option<i32>,
end: Option<i32>,
@@ -321,7 +321,7 @@
.into(),
))
}
- Self::Arr(arr) => Ok(Self::Arr(arr.clone().slice(
+ Self::Arr(arr) => Ok(Self::Arr(arr.clone().slice32(
index,
end,
#[expect(
@@ -658,7 +658,7 @@
if ArrValue::ptr_eq(a, b) {
return Ok(true);
}
- if a.len() != b.len() {
+ if a.len32() != b.len32() {
return Ok(false);
}
for (a, b) in a.iter().zip(b.iter()) {
crates/jrsonnet-formatter/src/lib.rsdiffbeforeafterboth--- a/crates/jrsonnet-formatter/src/lib.rs
+++ b/crates/jrsonnet-formatter/src/lib.rs
@@ -477,8 +477,7 @@
&mut out,
);
- let mut compspecs = compspecs.into_iter().peekable();
- while let Some(mem) = compspecs.next() {
+ for mem in compspecs {
if mem.should_start_with_newline {
p!(out, nl);
}
crates/jrsonnet-peg-parser/src/lib.rsdiffbeforeafterboth--- a/crates/jrsonnet-peg-parser/src/lib.rs
+++ b/crates/jrsonnet-peg-parser/src/lib.rs
@@ -273,19 +273,15 @@
Expr::ArrComp(Box::new(expr), specs)
}
pub rule number_expr(s: &ParserSettings) -> Expr
- = n:number() {? if let Some(n) = NumValue::new(n) {
- Ok(Expr::Num(n))
- } else {
- Err("!!!numbers are finite")
- }}
+ = n:number() {? NumValue::new(n).map_or_else(|| Err("!!!numbers are finite"), |n| Ok(Expr::Num(n)))}
rule spanned<T: Acyclic>(x: rule<T>, s: &ParserSettings) -> Spanned<T>
- = a:position!() n:x() b:position!() { Spanned::new(n, Span(s.source.clone(), a as u32, b as u32)) }
+ = a:position!() n:x() b:position!() { Spanned::new(n, Span(s.source.clone(), codeidx(a), codeidx(b))) }
pub rule var_expr(s: &ParserSettings) -> Expr
= n:spanned(<id()>, s) { Expr::Var(n) }
pub rule id_loc(s: &ParserSettings) -> Spanned<Expr>
- = a:position!() n:id() b:position!() { Spanned::new(Expr::Str(n), Span(s.source.clone(), a as u32,b as u32)) }
+ = a:position!() n:id() b:position!() { Spanned::new(Expr::Str(n), Span(s.source.clone(), codeidx(a), codeidx(b))) }
pub rule if_then_else_expr(s: &ParserSettings) -> Expr
= cond:ifspec(s) _ keyword("then") _ cond_then:expr(s) cond_else:(_ keyword("else") _ e:expr(s) {e})? {Expr::IfElse(Box::new(IfElse{
cond,
@@ -421,6 +417,10 @@
}
}
+fn codeidx(i: usize) -> u32 {
+ u32::try_from(i).expect("code has 4g hard limit")
+}
+
pub type ParseError = peg::error::ParseError<peg::str::LineCol>;
pub fn parse(str: &str, settings: &ParserSettings) -> Result<Expr, ParseError> {
jsonnet_parser::jsonnet(str, settings)
@@ -428,7 +428,10 @@
/// Used for importstr values
pub fn string_to_expr(str: IStr, settings: &ParserSettings) -> Spanned<Expr> {
let len = str.len();
- Spanned::new(Expr::Str(str), Span(settings.source.clone(), 0, len as u32))
+ Spanned::new(
+ Expr::Str(str),
+ Span(settings.source.clone(), 0, codeidx(len)),
+ )
}
#[cfg(test)]
crates/jrsonnet-pkg/src/install/accessor.rsdiffbeforeafterboth--- a/crates/jrsonnet-pkg/src/install/accessor.rs
+++ b/crates/jrsonnet-pkg/src/install/accessor.rs
@@ -66,6 +66,10 @@
Ok(Some(out))
}
#[allow(clippy::significant_drop_tightening, reason = "false-positive")]
+ #[allow(
+ clippy::iter_not_returning_iterator,
+ reason = "idk for a better name, it is still inner iteration"
+ )]
pub fn iter<E>(
&self,
subdir: &SubDir,
crates/jrsonnet-rowan-parser/src/parser.rsdiffbeforeafterboth--- a/crates/jrsonnet-rowan-parser/src/parser.rs
+++ b/crates/jrsonnet-rowan-parser/src/parser.rs
@@ -226,12 +226,12 @@
self.nth_at(0, kind)
}
pub fn nth_at(&self, n: usize, kind: SyntaxKind) -> bool {
- if n == 0 {
- if let ExpectedSyntax::Unnamed(kinds) = self.expected_syntax_tracking_state.get() {
- let kinds = kinds.with(kind);
- self.expected_syntax_tracking_state
- .set(ExpectedSyntax::Unnamed(kinds));
- }
+ if n == 0
+ && let ExpectedSyntax::Unnamed(kinds) = self.expected_syntax_tracking_state.get()
+ {
+ let kinds = kinds.with(kind);
+ self.expected_syntax_tracking_state
+ .set(ExpectedSyntax::Unnamed(kinds));
}
self.nth(n) == kind
}
crates/jrsonnet-stdlib/src/arrays.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/arrays.rs
+++ b/crates/jrsonnet-stdlib/src/arrays.rs
@@ -52,7 +52,7 @@
step: Option<Option<BoundedUsize<1, { i32::MAX as usize }>>>,
) -> Result<Val> {
indexable
- .slice(index.flatten(), end.flatten(), step.flatten())
+ .slice32(index.flatten(), end.flatten(), step.flatten())
.map(Val::from)
}
@@ -204,14 +204,14 @@
let item = item?.clone();
if let Val::Arr(items) = item {
if !first {
- out.reserve(joiner_items.len() as usize);
+ out.reserve(joiner_items.len());
// TODO: extend
for item in joiner_items.iter() {
out.push(item?);
}
}
first = false;
- out.reserve(items.len() as usize);
+ out.reserve(items.len());
for item in items.iter() {
out.push(item?);
}
@@ -372,10 +372,10 @@
#[builtin]
pub fn builtin_remove_at(arr: ArrValue, at: i32) -> Result<ArrValue> {
- let newArrLeft = arr.clone().slice(None, Some(at), None);
- let newArrRight = arr.slice(Some(at + 1), None, None);
+ let newArrLeft = arr.clone().slice32(None, Some(at), None);
+ let newArrRight = arr.slice32(Some(at + 1), None, None);
- Ok(ArrValue::extended(newArrLeft, newArrRight).ok_or_else(|| error!("array is too large"))?)
+ ArrValue::extended(newArrLeft, newArrRight).ok_or_else(|| error!("array is too large"))
}
#[builtin]
crates/jrsonnet-stdlib/src/manifest/xml.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/manifest/xml.rs
+++ b/crates/jrsonnet-stdlib/src/manifest/xml.rs
@@ -44,15 +44,15 @@
bail!("JSONML value should have tag (array length should be >=1)");
}
let tag = String::from_untyped(
- arr.get(0)
+ arr.get32(0)
.description("getting JSONML tag")?
.expect("length checked"),
)
.description("parsing JSONML tag")?;
- let (has_attrs, attrs) = if arr.len() >= 2 {
+ let (has_attrs, attrs) = if arr.len32() >= 2 {
let maybe_attrs = arr
- .get(1)
+ .get32(1)
.with_description(|| "getting JSONML attrs")?
.expect("length checked");
if let Val::Obj(attrs) = maybe_attrs {
@@ -68,13 +68,7 @@
attrs,
children: in_description_frame(
|| "parsing children".to_owned(),
- || {
- FromUntyped::from_untyped(Val::Arr(arr.slice(
- Some(if has_attrs { 2 } else { 1 }),
- None,
- None,
- )))
- },
+ || FromUntyped::from_untyped(Val::Arr(arr.slice(if has_attrs { 2 } else { 1 }..))),
)?,
})
}
crates/jrsonnet-stdlib/src/misc.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/misc.rs
+++ b/crates/jrsonnet-stdlib/src/misc.rs
@@ -16,10 +16,10 @@
pub fn builtin_length(x: Either![IStr, ArrValue, ObjValue, FuncVal]) -> u32 {
use Either4::*;
match x {
- A(x) => x.chars().count() as u32,
- B(x) => x.len(),
- C(x) => x.len(),
- D(f) => f.params_len(),
+ A(x) => u32::try_from(x.chars().count()).expect("4g limit"),
+ B(x) => x.len32(),
+ C(x) => x.len32(),
+ D(f) => f.params_len32(),
}
}
@@ -102,7 +102,7 @@
} else if b.len() == a.len() {
return equals(&Val::Arr(a), &Val::Arr(b));
}
- for (a, b) in a.iter().take(b.len() as usize).zip(b.iter()) {
+ for (a, b) in a.iter().take(b.len()).zip(b.iter()) {
let a = a?;
let b = b?;
if !equals(&a, &b)? {
@@ -127,7 +127,7 @@
return equals(&Val::Arr(a), &Val::Arr(b));
}
let a_len = a.len();
- for (a, b) in a.iter().skip((a_len - b.len()) as usize).zip(b.iter()) {
+ for (a, b) in a.iter().skip(a_len - b.len()).zip(b.iter()) {
let a = a?;
let b = b?;
if !equals(&a, &b)? {
crates/jrsonnet-stdlib/src/sets.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/sets.rs
+++ b/crates/jrsonnet-stdlib/src/sets.rs
@@ -8,13 +8,13 @@
#[allow(non_snake_case)]
pub fn builtin_set_member(x: Thunk<Val>, arr: ArrValue, #[default] keyF: KeyF) -> Result<bool> {
let mut low = 0;
- let mut high = arr.len();
+ let mut high = arr.len32();
let x = keyF.eval(x)?;
while low < high {
let middle = u32::midpoint(high, low);
- let comp = keyF.eval(arr.get_lazy(middle).expect("in bounds"))?;
+ let comp = keyF.eval(arr.get_lazy32(middle).expect("in bounds"))?;
match Val::try_cmp(&comp, &x)? {
Ordering::Less => low = middle + 1,
Ordering::Equal => return Ok(true),
crates/jrsonnet-stdlib/src/sort.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/sort.rs
+++ b/crates/jrsonnet-stdlib/src/sort.rs
@@ -69,7 +69,7 @@
fn sort_keyf(values: ArrValue, keyf: KeyF) -> Result<Vec<Thunk<Val>>> {
// Slow path, user provided key getter
- let mut vk = Vec::with_capacity(values.len() as usize);
+ let mut vk = Vec::with_capacity(values.len());
for value in values.iter_lazy() {
vk.push((value.clone(), keyf.eval(value)?));
}
@@ -137,7 +137,7 @@
fn uniq_keyf(arr: ArrValue, keyf: KeyF) -> Result<Vec<Thunk<Val>>> {
let mut out = Vec::new();
- let last_value = arr.get_lazy(0).unwrap();
+ let last_value = arr.get_lazy32(0).unwrap();
let mut last_key = keyf.eval(last_value.clone())?;
out.push(last_value);
crates/jrsonnet-types/src/lib.rsdiffbeforeafterboth--- a/crates/jrsonnet-types/src/lib.rs
+++ b/crates/jrsonnet-types/src/lib.rs
@@ -103,10 +103,8 @@
Self::BoundedNumber(a, b) => write!(
f,
"BoundedNumber<{}, {}>",
- a.map(|e| e.to_string())
- .unwrap_or_else(|| "open".to_owned()),
- b.map(|e| e.to_string())
- .unwrap_or_else(|| "open".to_owned())
+ a.map_or_else(|| "open".to_owned(), |e| e.to_string()),
+ b.map_or_else(|| "open".to_owned(), |e| e.to_string())
)?,
Self::ArrayRef(a) => print_array(a, f)?,
Self::Array(a) => print_array(a, f)?,
tests/tests/cpp_test_suite.rsdiffbeforeafterboth--- a/tests/tests/cpp_test_suite.rs
+++ b/tests/tests/cpp_test_suite.rs
@@ -60,7 +60,7 @@
let _entered = s.enter();
let trace_format = CompactFormat {
- resolver: resolver.clone(),
+ resolver,
max_trace: 20,
padding: 4,
};
xtask/src/bench.rsdiffbeforeafterboth--- a/xtask/src/bench.rs
+++ b/xtask/src/bench.rs
@@ -91,6 +91,10 @@
let start = Instant::now();
let child = cmd.spawn()?;
+ #[allow(
+ clippy::cast_possible_wrap,
+ reason = "it is signed, but libc didn't set unsigned for it"
+ )]
let pid = child.id() as libc::pid_t;
// We'll reap via wait4 ourselves; don't let std touch this handle again.
mem::forget(child);
@@ -133,10 +137,10 @@
);
eprintln!(
" max_rss: {} ± {} KiB [{}..{}]",
- r.max_rss_kib.mean as i64,
- r.max_rss_kib.stddev as i64,
- r.max_rss_kib.min as i64,
- r.max_rss_kib.max as i64,
+ r.max_rss_kib.mean.trunc(),
+ r.max_rss_kib.stddev.trunc(),
+ r.max_rss_kib.min.trunc(),
+ r.max_rss_kib.max.trunc(),
);
Ok(())
}
xtask/src/sourcegen/mod.rsdiffbeforeafterboth--- a/xtask/src/sourcegen/mod.rs
+++ b/xtask/src/sourcegen/mod.rs
@@ -113,6 +113,7 @@
Ok(())
}
+#[allow(clippy::too_many_lines)]
fn generate_syntax_kinds(kinds: &KindsSrc, grammar: &AstSrc, lexer: bool) -> Result<String> {
let t_macros = kinds.tokens().filter_map(TokenKind::expand_t_macros);
let token_kinds = kinds.tokens().map(|t| t.expand_kind(lexer));