difftreelog
refactor extended strings
in: master
16 files changed
crates/jrsonnet-evaluator/src/evaluate/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/evaluate/mod.rs
+++ b/crates/jrsonnet-evaluator/src/evaluate/mod.rs
@@ -17,7 +17,7 @@
function::{CallLocation, FuncDesc, FuncVal},
tb, throw,
typed::Typed,
- val::{CachedUnbound, IndexableVal, Thunk, ThunkValue},
+ val::{CachedUnbound, IndexableVal, StrValue, Thunk, ThunkValue},
Context, GcHashMap, ObjValue, ObjValueBuilder, ObjectAssertion, Pending, Result, State,
Unbound, Val,
};
@@ -36,7 +36,7 @@
}
}
Some(match &*expr.0 {
- Expr::Str(s) => Val::Str(s.clone()),
+ Expr::Str(s) => Val::Str(StrValue::Flat(s.clone())),
Expr::Num(n) => Val::Num(*n),
Expr::Literal(LiteralType::False) => Val::Bool(false),
Expr::Literal(LiteralType::True) => Val::Bool(true),
@@ -135,7 +135,7 @@
let fctx = Pending::new();
let mut new_bindings = GcHashMap::with_capacity(var.capacity_hint());
let value = Thunk::evaluated(Val::Arr(ArrValue::lazy(Cc::new(vec![
- Thunk::evaluated(Val::Str(field.clone())),
+ Thunk::evaluated(Val::Str(StrValue::Flat(field.clone()))),
Thunk::new(tb!(ObjectFieldThunk {
field: field.clone(),
obj: obj.clone(),
@@ -436,7 +436,7 @@
Literal(LiteralType::False) => Val::Bool(false),
Literal(LiteralType::Null) => Val::Null,
Parened(e) => evaluate(ctx, e)?,
- Str(v) => Val::Str(v.clone()),
+ Str(v) => Val::Str(StrValue::Flat(v.clone())),
Num(v) => Val::new_checked_num(*v)?,
BinaryOp(v1, o, v2) => evaluate_binary_op_special(ctx, v1, *o, v2)?,
UnaryOp(o, v) => evaluate_unary_op(*o, &evaluate(ctx, v)?)?,
@@ -457,14 +457,14 @@
ctx.super_obj()
.clone()
.expect("no super found")
- .get_for(name, ctx.this().clone().expect("no this found"))?
+ .get_for(name.into_flat(), ctx.this().clone().expect("no this found"))?
.expect("value not found")
}
Index(value, index) => match (evaluate(ctx.clone(), value)?, evaluate(ctx, index)?) {
(Val::Obj(v), Val::Str(key)) => State::push(
CallLocation::new(loc),
|| format!("field <{key}> access"),
- || match v.get(key.clone()) {
+ || match v.get(key.clone().into_flat()) {
Ok(Some(v)) => Ok(v),
#[cfg(not(feature = "friendly-errors"))]
Ok(None) => throw!(NoSuchField(key.clone(), vec![])),
@@ -476,7 +476,10 @@
#[cfg(feature = "exp-preserve-order")]
false,
) {
- let conf = strsim::jaro_winkler(&field as &str, &key as &str);
+ let conf = strsim::jaro_winkler(
+ &field as &str,
+ &key.clone().into_flat() as &str,
+ );
if conf < 0.8 {
continue;
}
@@ -485,7 +488,7 @@
heap.sort_by(|a, b| b.0.partial_cmp(&a.0).unwrap_or(Ordering::Equal));
throw!(NoSuchField(
- key.clone(),
+ key.clone().into_flat(),
heap.into_iter().map(|(_, v)| v).collect()
))
}
@@ -505,7 +508,7 @@
v.get(n as usize)?
.ok_or_else(|| ArrayBoundsError(n as usize, v.len()))?
}
- (Val::Arr(_), Val::Str(n)) => throw!(AttemptedIndexAnArrayWithString(n)),
+ (Val::Arr(_), Val::Str(n)) => throw!(AttemptedIndexAnArrayWithString(n.into_flat())),
(Val::Arr(_), n) => throw!(ValueIndexMustBeTypeGot(
ValType::Arr,
ValType::Num,
@@ -514,16 +517,18 @@
(Val::Str(s), Val::Num(n)) => Val::Str({
let v: IStr = s
+ .clone()
+ .into_flat()
.chars()
.skip(n as usize)
.take(1)
.collect::<String>()
.into();
if v.is_empty() {
- let size = s.chars().count();
+ let size = s.into_flat().chars().count();
throw!(StringBoundsError(n as usize, size))
}
- v
+ StrValue::Flat(v)
}),
(Val::Str(_), n) => throw!(ValueIndexMustBeTypeGot(
ValType::Str,
@@ -654,7 +659,7 @@
|| format!("import {:?}", path.clone()),
|| s.import_resolved(resolved_path),
)?,
- ImportStr(_) => Val::Str(s.import_resolved_str(resolved_path)?),
+ ImportStr(_) => Val::Str(StrValue::Flat(s.import_resolved_str(resolved_path)?)),
ImportBin(_) => Val::Arr(ArrValue::bytes(s.import_resolved_bin(resolved_path)?)),
_ => unreachable!(),
}
crates/jrsonnet-evaluator/src/evaluate/operator.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/evaluate/operator.rs
+++ b/crates/jrsonnet-evaluator/src/evaluate/operator.rs
@@ -3,8 +3,14 @@
use jrsonnet_parser::{BinaryOpType, LocExpr, UnaryOpType};
use crate::{
- arr::ArrValue, error::ErrorKind::*, evaluate, stdlib::std_format, throw, typed::Typed,
- val::equals, Context, Result, Val,
+ arr::ArrValue,
+ error::ErrorKind::*,
+ evaluate,
+ stdlib::std_format,
+ throw,
+ typed::Typed,
+ val::{equals, StrValue},
+ Context, Result, Val,
};
pub fn evaluate_unary_op(op: UnaryOpType, b: &Val) -> Result<Val> {
@@ -25,15 +31,21 @@
Ok(match (a, b) {
(Str(a), Str(b)) if a.is_empty() => Val::Str(b.clone()),
(Str(a), Str(b)) if b.is_empty() => Val::Str(a.clone()),
- (Str(v1), Str(v2)) => Str(((**v1).to_owned() + v2).into()),
+ (Str(v1), Str(v2)) => Str(StrValue::concat(v1.clone(), v2.clone())),
// Can't use generic json serialization way, because it depends on number to string concatenation (std.jsonnet:890)
- (Num(a), Str(b)) => Str(format!("{a}{b}").into()),
- (Str(a), Num(b)) => Str(format!("{a}{b}").into()),
+ (Num(a), Str(b)) => Str(StrValue::Flat(format!("{a}{b}").into())),
+ (Str(a), Num(b)) => Str(StrValue::Flat(format!("{a}{b}").into())),
- (Str(a), o) | (o, Str(a)) if a.is_empty() => Val::Str(o.clone().to_string()?),
- (Str(a), o) => Str(format!("{a}{}", o.clone().to_string()?).into()),
- (o, Str(a)) => Str(format!("{}{a}", o.clone().to_string()?).into()),
+ (Str(a), o) | (o, Str(a)) if a.is_empty() => {
+ Val::Str(StrValue::Flat(o.clone().to_string()?))
+ }
+ (Str(a), o) => Str(StrValue::Flat(
+ format!("{a}{}", o.clone().to_string()?).into(),
+ )),
+ (o, Str(a)) => Str(StrValue::Flat(
+ format!("{}{a}", o.clone().to_string()?).into(),
+ )),
(Obj(v1), Obj(v2)) => Obj(v2.extend_from(v1.clone())),
(Arr(a), Arr(b)) => Val::Arr(ArrValue::extended(a.clone(), b.clone())),
@@ -56,7 +68,9 @@
}
Ok(Num(a % b))
}
- (Str(str), vals) => String::into_untyped(std_format(str.clone(), vals.clone())?),
+ (Str(str), vals) => {
+ String::into_untyped(std_format(&str.clone().into_flat(), vals.clone())?)
+ }
(a, b) => throw!(BinaryOperatorDoesNotOperateOnValues(
BinaryOpType::Mod,
a.value_type(),
@@ -120,10 +134,10 @@
(a, Lte, b) => Bool(evaluate_compare_op(a, b, Lte)?.is_le()),
(a, Gte, b) => Bool(evaluate_compare_op(a, b, Gte)?.is_ge()),
- (Str(a), In, Obj(obj)) => Bool(obj.has_field_ex(a.clone(), true)),
+ (Str(a), In, Obj(obj)) => Bool(obj.has_field_ex(a.clone().into_flat(), true)),
(a, Mod, b) => evaluate_mod_op(a, b)?,
- (Str(v1), Mul, Num(v2)) => Str(v1.repeat(*v2 as usize).into()),
+ (Str(v1), Mul, Num(v2)) => Str(StrValue::Flat(v1.to_string().repeat(*v2 as usize).into())),
// Bool X Bool
(Bool(a), And, Bool(b)) => Bool(*a && *b),
crates/jrsonnet-evaluator/src/function/arglike.rsdiffbeforeafterboth1use hashbrown::HashMap;2use jrsonnet_gcmodule::Trace;3use jrsonnet_interner::IStr;4use jrsonnet_parser::{ArgsDesc, LocExpr};56use crate::{7 error::Result, evaluate, gc::GcHashMap, tb, typed::Typed, val::ThunkValue, Context, Thunk, Val,8};910/// Marker for arguments, which can be evaluated with context set to None11pub trait OptionalContext {}1213#[derive(Trace)]14struct EvaluateThunk {15 ctx: Context,16 expr: LocExpr,17}18impl ThunkValue for EvaluateThunk {19 type Output = Val;20 fn get(self: Box<Self>) -> Result<Val> {21 evaluate(self.ctx, &self.expr)22 }23}2425pub trait ArgLike {26 fn evaluate_arg(&self, ctx: Context, tailstrict: bool) -> Result<Thunk<Val>>;27}2829impl ArgLike for &LocExpr {30 fn evaluate_arg(&self, ctx: Context, tailstrict: bool) -> Result<Thunk<Val>> {31 Ok(if tailstrict {32 Thunk::evaluated(evaluate(ctx, self)?)33 } else {34 Thunk::new(tb!(EvaluateThunk {35 ctx,36 expr: (*self).clone(),37 }))38 })39 }40}4142impl<T> ArgLike for T43where44 T: Typed + Clone,45{46 fn evaluate_arg(&self, _ctx: Context, _tailstrict: bool) -> Result<Thunk<Val>> {47 let val = T::into_untyped(self.clone())?;48 Ok(Thunk::evaluated(val))49 }50}51impl<T> OptionalContext for T where T: Typed + Clone {}5253#[derive(Clone, Trace)]54pub enum TlaArg {55 String(IStr),56 Code(LocExpr),57 Val(Val),58}59impl ArgLike for TlaArg {60 fn evaluate_arg(&self, ctx: Context, tailstrict: bool) -> Result<Thunk<Val>> {61 match self {62 TlaArg::String(s) => Ok(Thunk::evaluated(Val::Str(s.clone()))),63 TlaArg::Code(code) => Ok(if tailstrict {64 Thunk::evaluated(evaluate(ctx, code)?)65 } else {66 Thunk::new(tb!(EvaluateThunk {67 ctx,68 expr: code.clone(),69 }))70 }),71 TlaArg::Val(val) => Ok(Thunk::evaluated(val.clone())),72 }73 }74}7576mod sealed {77 /// Implemented for `ArgsLike`, where only unnamed arguments present78 pub trait Unnamed {}79 /// Implemented for `ArgsLike`, where only named arguments present80 pub trait Named {}81}8283pub trait ArgsLike {84 fn unnamed_len(&self) -> usize;85 fn unnamed_iter(86 &self,87 ctx: Context,88 tailstrict: bool,89 handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,90 ) -> Result<()>;91 fn named_iter(92 &self,93 ctx: Context,94 tailstrict: bool,95 handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,96 ) -> Result<()>;97 fn named_names(&self, handler: &mut dyn FnMut(&IStr));98}99100impl ArgsLike for Vec<Val> {101 fn unnamed_len(&self) -> usize {102 self.len()103 }104 fn unnamed_iter(105 &self,106 _ctx: Context,107 _tailstrict: bool,108 handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,109 ) -> Result<()> {110 for (idx, el) in self.iter().enumerate() {111 handler(idx, Thunk::evaluated(el.clone()))?;112 }113 Ok(())114 }115 fn named_iter(116 &self,117 _ctx: Context,118 _tailstrict: bool,119 _handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,120 ) -> Result<()> {121 Ok(())122 }123 fn named_names(&self, _handler: &mut dyn FnMut(&IStr)) {}124}125impl OptionalContext for Vec<Val> {}126127impl ArgsLike for ArgsDesc {128 fn unnamed_len(&self) -> usize {129 self.unnamed.len()130 }131132 fn unnamed_iter(133 &self,134 ctx: Context,135 tailstrict: bool,136 handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,137 ) -> Result<()> {138 for (id, arg) in self.unnamed.iter().enumerate() {139 handler(140 id,141 if tailstrict {142 Thunk::evaluated(evaluate(ctx.clone(), arg)?)143 } else {144 Thunk::new(tb!(EvaluateThunk {145 ctx: ctx.clone(),146 expr: arg.clone(),147 }))148 },149 )?;150 }151 Ok(())152 }153154 fn named_iter(155 &self,156 ctx: Context,157 tailstrict: bool,158 handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,159 ) -> Result<()> {160 for (name, arg) in &self.named {161 handler(162 name,163 if tailstrict {164 Thunk::evaluated(evaluate(ctx.clone(), arg)?)165 } else {166 Thunk::new(tb!(EvaluateThunk {167 ctx: ctx.clone(),168 expr: arg.clone(),169 }))170 },171 )?;172 }173 Ok(())174 }175176 fn named_names(&self, handler: &mut dyn FnMut(&IStr)) {177 for (name, _) in &self.named {178 handler(name);179 }180 }181}182183impl<V: ArgLike, S> sealed::Named for HashMap<IStr, V, S> {}184impl<V: ArgLike, S> ArgsLike for HashMap<IStr, V, S> {185 fn unnamed_len(&self) -> usize {186 0187 }188189 fn unnamed_iter(190 &self,191 _ctx: Context,192 _tailstrict: bool,193 _handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,194 ) -> Result<()> {195 Ok(())196 }197198 fn named_iter(199 &self,200 ctx: Context,201 tailstrict: bool,202 handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,203 ) -> Result<()> {204 for (name, value) in self.iter() {205 handler(name, value.evaluate_arg(ctx.clone(), tailstrict)?)?;206 }207 Ok(())208 }209210 fn named_names(&self, handler: &mut dyn FnMut(&IStr)) {211 for (name, _) in self.iter() {212 handler(name);213 }214 }215}216impl<V, S> OptionalContext for HashMap<IStr, V, S> where V: ArgLike + OptionalContext {}217218impl<A: ArgLike> ArgsLike for GcHashMap<IStr, A> {219 fn unnamed_len(&self) -> usize {220 self.0.unnamed_len()221 }222223 fn unnamed_iter(224 &self,225 ctx: Context,226 tailstrict: bool,227 handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,228 ) -> Result<()> {229 self.0.unnamed_iter(ctx, tailstrict, handler)230 }231232 fn named_iter(233 &self,234 ctx: Context,235 tailstrict: bool,236 handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,237 ) -> Result<()> {238 self.0.named_iter(ctx, tailstrict, handler)239 }240241 fn named_names(&self, handler: &mut dyn FnMut(&IStr)) {242 self.0.named_names(handler);243 }244}245246macro_rules! impl_args_like {247 ($count:expr; $($gen:ident)*) => {248 impl<$($gen: ArgLike,)*> sealed::Unnamed for ($($gen,)*) {}249 impl<$($gen: ArgLike,)*> ArgsLike for ($($gen,)*) {250 fn unnamed_len(&self) -> usize {251 $count252 }253 #[allow(non_snake_case, unused_assignments)]254 fn unnamed_iter(255 &self,256 ctx: Context,257 tailstrict: bool,258 handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,259 ) -> Result<()> {260 let mut i = 0usize;261 let ($($gen,)*) = self;262 $(263 handler(i, $gen.evaluate_arg(ctx.clone(), tailstrict)?)?;264 i+=1;265 )*266 Ok(())267 }268 fn named_iter(269 &self,270 _ctx: Context,271 _tailstrict: bool,272 _handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,273 ) -> Result<()> {274 Ok(())275 }276 fn named_names(&self, _handler: &mut dyn FnMut(&IStr)) {}277 }278 impl<$($gen: ArgLike,)*> OptionalContext for ($($gen,)*) where $($gen: OptionalContext),* {}279280 impl<$($gen: ArgLike,)*> sealed::Named for ($((IStr, $gen),)*) {}281 impl<$($gen: ArgLike,)*> ArgsLike for ($((IStr, $gen),)*) {282 fn unnamed_len(&self) -> usize {283 0284 }285 fn unnamed_iter(286 &self,287 _ctx: Context,288 _tailstrict: bool,289 _handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,290 ) -> Result<()> {291 Ok(())292 }293 #[allow(non_snake_case)]294 fn named_iter(295 &self,296 ctx: Context,297 tailstrict: bool,298 handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,299 ) -> Result<()> {300 let ($($gen,)*) = self;301 $(302 handler(&$gen.0, $gen.1.evaluate_arg(ctx.clone(), tailstrict)?)?;303 )*304 Ok(())305 }306 #[allow(non_snake_case)]307 fn named_names(&self, handler: &mut dyn FnMut(&IStr)) {308 let ($($gen,)*) = self;309 $(310 handler(&$gen.0);311 )*312 }313 }314 impl<$($gen: ArgLike,)*> OptionalContext for ($((IStr, $gen),)*) where $($gen: OptionalContext),* {}315 };316 ($count:expr; $($cur:ident)* @ $c:ident $($rest:ident)*) => {317 impl_args_like!($count; $($cur)*);318 impl_args_like!($count + 1usize; $($cur)* $c @ $($rest)*);319 };320 ($count:expr; $($cur:ident)* @) => {321 impl_args_like!($count; $($cur)*);322 }323}324impl_args_like! {325 // First argument is already in position, so count starts from 1326 1usize; A @ B C D E F G H I J K L327}328329impl ArgsLike for () {330 fn unnamed_len(&self) -> usize {331 0332 }333334 fn unnamed_iter(335 &self,336 _ctx: Context,337 _tailstrict: bool,338 _handler: &mut dyn FnMut(usize, Thunk<Val>) -> Result<()>,339 ) -> Result<()> {340 Ok(())341 }342343 fn named_iter(344 &self,345 _ctx: Context,346 _tailstrict: bool,347 _handler: &mut dyn FnMut(&IStr, Thunk<Val>) -> Result<()>,348 ) -> Result<()> {349 Ok(())350 }351352 fn named_names(&self, _handler: &mut dyn FnMut(&IStr)) {}353}354impl OptionalContext for () {}crates/jrsonnet-evaluator/src/integrations/serde.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/integrations/serde.rs
+++ b/crates/jrsonnet-evaluator/src/integrations/serde.rs
@@ -7,7 +7,7 @@
Deserialize, Serialize,
};
-use crate::{arr::ArrValue, error::Result, ObjValueBuilder, State, Val};
+use crate::{arr::ArrValue, error::Result, val::StrValue, ObjValueBuilder, State, Val};
impl<'de> Deserialize<'de> for Val {
fn deserialize<D>(deserializer: D) -> Result<Val, D::Error>
@@ -49,7 +49,7 @@
where
E: serde::de::Error,
{
- Ok(Val::Str(v.into()))
+ Ok(Val::Str(StrValue::Flat(v.into())))
}
// visit_num! {
@@ -152,7 +152,7 @@
match self {
Val::Bool(v) => serializer.serialize_bool(*v),
Val::Null => serializer.serialize_none(),
- Val::Str(s) => serializer.serialize_str(s),
+ Val::Str(s) => serializer.serialize_str(&s.clone().into_flat()),
Val::Num(n) => serializer.serialize_f64(*n),
Val::Arr(arr) => {
let mut seq = serializer.serialize_seq(Some(arr.len()))?;
crates/jrsonnet-evaluator/src/manifest.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/manifest.rs
+++ b/crates/jrsonnet-evaluator/src/manifest.rs
@@ -147,7 +147,7 @@
}
}
Val::Null => buf.push_str("null"),
- Val::Str(s) => escape_string_json_buf(s, buf),
+ Val::Str(s) => escape_string_json_buf(&s.clone().into_flat(), buf),
Val::Num(n) => write!(buf, "{n}").unwrap(),
Val::Arr(items) => {
buf.push('[');
@@ -256,7 +256,7 @@
let Val::Str(s) = val else {
throw!("output should be string for string manifest format, got {}", val.value_type())
};
- out.write_str(&s).unwrap();
+ write!(out, "{s}").unwrap();
Ok(())
}
}
crates/jrsonnet-evaluator/src/stdlib/format.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/stdlib/format.rs
+++ b/crates/jrsonnet-evaluator/src/stdlib/format.rs
@@ -589,6 +589,7 @@
.ok_or_else(|| InvalidUnicodeCodepointGot(n as u32))?,
),
Val::Str(s) => {
+ let s = s.into_flat();
if s.chars().count() != 1 {
throw!("%c expected 1 char string, got {}", s.chars().count(),);
}
crates/jrsonnet-evaluator/src/stdlib/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/stdlib/mod.rs
+++ b/crates/jrsonnet-evaluator/src/stdlib/mod.rs
@@ -2,13 +2,12 @@
#![allow(clippy::unnecessary_wraps)]
use format::{format_arr, format_obj};
-use jrsonnet_interner::IStr;
use crate::{error::Result, function::CallLocation, State, Val};
pub mod format;
-pub fn std_format(str: IStr, vals: Val) -> Result<String> {
+pub fn std_format(str: &str, vals: Val) -> Result<String> {
State::push(
CallLocation::native(),
|| format!("std.format of {str}"),
crates/jrsonnet-evaluator/src/typed/conversions.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/typed/conversions.rs
+++ b/crates/jrsonnet-evaluator/src/typed/conversions.rs
@@ -11,7 +11,7 @@
function::{native::NativeDesc, FuncDesc, FuncVal},
throw,
typed::CheckType,
- val::IndexableVal,
+ val::{IndexableVal, StrValue},
ObjValue, ObjValueBuilder, Val,
};
@@ -187,13 +187,13 @@
const TYPE: &'static ComplexValType = &ComplexValType::Simple(ValType::Str);
fn into_untyped(value: Self) -> Result<Val> {
- Ok(Val::Str(value))
+ Ok(Val::Str(StrValue::Flat(value)))
}
fn from_untyped(value: Val) -> Result<Self> {
<Self as Typed>::TYPE.check(&value)?;
match value {
- Val::Str(s) => Ok(s),
+ Val::Str(s) => Ok(s.into_flat()),
_ => unreachable!(),
}
}
@@ -203,7 +203,7 @@
const TYPE: &'static ComplexValType = &ComplexValType::Simple(ValType::Str);
fn into_untyped(value: Self) -> Result<Val> {
- Ok(Val::Str(value.into()))
+ Ok(Val::Str(StrValue::Flat(value.into())))
}
fn from_untyped(value: Val) -> Result<Self> {
@@ -219,13 +219,13 @@
const TYPE: &'static ComplexValType = &ComplexValType::Char;
fn into_untyped(value: Self) -> Result<Val> {
- Ok(Val::Str(value.to_string().into()))
+ Ok(Val::Str(StrValue::Flat(value.to_string().into())))
}
fn from_untyped(value: Val) -> Result<Self> {
<Self as Typed>::TYPE.check(&value)?;
match value {
- Val::Str(s) => Ok(s.chars().next().unwrap()),
+ Val::Str(s) => Ok(s.into_flat().chars().next().unwrap()),
_ => unreachable!(),
}
}
@@ -480,7 +480,7 @@
fn into_untyped(value: Self) -> Result<Val> {
match value {
- IndexableVal::Str(s) => Ok(Val::Str(s)),
+ IndexableVal::Str(s) => Ok(Val::Str(StrValue::Flat(s))),
IndexableVal::Arr(a) => Ok(Val::Arr(a)),
}
}
crates/jrsonnet-evaluator/src/typed/mod.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/typed/mod.rs
+++ b/crates/jrsonnet-evaluator/src/typed/mod.rs
@@ -150,7 +150,7 @@
Self::Any => Ok(()),
Self::Simple(t) => t.check(value),
Self::Char => match value {
- Val::Str(s) if s.len() == 1 || s.chars().count() == 1 => Ok(()),
+ Val::Str(s) if s.len() == 1 || s.clone().into_flat().chars().count() == 1 => Ok(()),
v => Err(TypeError::ExpectedGot(self.clone(), v.value_type()).into()),
},
Self::BoundedNumber(from, to) => {
crates/jrsonnet-evaluator/src/val.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/val.rs
+++ b/crates/jrsonnet-evaluator/src/val.rs
@@ -1,4 +1,9 @@
-use std::{cell::RefCell, fmt::Debug, mem::replace};
+use std::{
+ cell::RefCell,
+ fmt::{self, Debug, Display},
+ mem::replace,
+ rc::Rc,
+};
use jrsonnet_gcmodule::{Cc, Trace};
use jrsonnet_interner::IStr;
@@ -117,7 +122,7 @@
}
impl<T: Debug + Trace> Debug for Thunk<T> {
- fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+ fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "Lazy")
}
}
@@ -187,6 +192,87 @@
}
}
+#[derive(Debug, Clone, Trace)]
+pub enum StrValue {
+ Flat(IStr),
+ Tree(Rc<(StrValue, StrValue, usize)>),
+}
+impl StrValue {
+ pub fn concat(a: StrValue, b: StrValue) -> Self {
+ if a.is_empty() {
+ b
+ } else if b.is_empty() {
+ a
+ } else {
+ let len = a.len() + b.len();
+ Self::Tree(Rc::new((a, b, len)))
+ }
+ }
+ pub fn into_flat(self) -> IStr {
+ match self {
+ StrValue::Flat(f) => f,
+ StrValue::Tree(_) => {
+ let mut buf = String::new();
+ self.into_flat_buf(&mut buf);
+ buf.into()
+ }
+ }
+ }
+ fn into_flat_buf(&self, out: &mut String) {
+ match self {
+ StrValue::Flat(f) => out.push_str(f),
+ StrValue::Tree(t) => {
+ t.0.into_flat_buf(out);
+ t.1.into_flat_buf(out);
+ }
+ }
+ }
+ pub fn len(&self) -> usize {
+ match self {
+ StrValue::Flat(v) => v.len(),
+ StrValue::Tree(t) => t.2,
+ }
+ }
+ pub fn is_empty(&self) -> bool {
+ match self {
+ Self::Flat(v) => v.is_empty(),
+ _ => false,
+ }
+ }
+}
+impl Display for StrValue {
+ fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+ match self {
+ StrValue::Flat(v) => write!(f, "{v}"),
+ StrValue::Tree(t) => {
+ write!(f, "{}", t.0)?;
+ write!(f, "{}", t.1)
+ }
+ }
+ }
+}
+impl PartialEq for StrValue {
+ fn eq(&self, other: &Self) -> bool {
+ let a = self.clone().into_flat();
+ let b = other.clone().into_flat();
+ a == b
+ }
+}
+impl Eq for StrValue {}
+impl PartialOrd for StrValue {
+ fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
+ let a = self.clone().into_flat();
+ let b = other.clone().into_flat();
+ Some(a.cmp(&b))
+ }
+}
+impl Ord for StrValue {
+ fn cmp(&self, other: &Self) -> std::cmp::Ordering {
+ self.partial_cmp(other)
+ .expect("partial_cmp always returns Some")
+ }
+}
+
/// Represents any valid Jsonnet value.
#[derive(Debug, Clone, Trace)]
pub enum Val {
@@ -195,7 +281,7 @@
/// Represents a Jsonnet null value.
Null,
/// Represents a Jsonnet string.
- Str(IStr),
+ Str(StrValue),
/// Represents a Jsonnet number.
/// Should be finite, and not NaN
/// This restriction isn't enforced by enum, as enum field can't be marked as private
@@ -208,10 +294,12 @@
Func(FuncVal),
}
+static_assertions::assert_eq_size!(Val, [u8; 24]);
+
impl From<IndexableVal> for Val {
fn from(v: IndexableVal) -> Self {
match v {
- IndexableVal::Str(s) => Self::Str(s),
+ IndexableVal::Str(s) => Self::Str(StrValue::Flat(s)),
IndexableVal::Arr(a) => Self::Arr(a),
}
}
@@ -232,7 +320,7 @@
}
pub fn as_str(&self) -> Option<IStr> {
match self {
- Self::Str(s) => Some(s.clone()),
+ Self::Str(s) => Some(s.clone().into_flat()),
_ => None,
}
}
@@ -295,14 +383,14 @@
Self::Bool(true) => "true".into(),
Self::Bool(false) => "false".into(),
Self::Null => "null".into(),
- Self::Str(s) => s.clone(),
+ Self::Str(s) => s.clone().into_flat(),
_ => self.manifest(ToStringFormat).map(IStr::from)?,
})
}
pub fn into_indexable(self) -> Result<IndexableVal> {
Ok(match self {
- Val::Str(s) => IndexableVal::Str(s),
+ Val::Str(s) => IndexableVal::Str(s.into_flat()),
Val::Arr(arr) => IndexableVal::Arr(arr),
_ => throw!(ValueIsNotIndexable(self.value_type())),
})
crates/jrsonnet-stdlib/src/arrays.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/arrays.rs
+++ b/crates/jrsonnet-stdlib/src/arrays.rs
@@ -37,12 +37,13 @@
func: NativeFn<((Either![String, Any],), Any)>,
arr: IndexableVal,
) -> Result<IndexableVal> {
+ use std::fmt::Write;
match arr {
IndexableVal::Str(str) => {
let mut out = String::new();
for c in str.chars() {
match func(Either2::A(c.to_string()))?.0 {
- Val::Str(o) => out.push_str(&o),
+ Val::Str(o) => write!(out, "{o}").unwrap(),
Val::Null => continue,
_ => throw!("in std.join all items should be strings"),
};
@@ -101,6 +102,7 @@
#[builtin]
pub fn builtin_join(sep: IndexableVal, arr: ArrValue) -> Result<IndexableVal> {
+ use std::fmt::Write;
Ok(match sep {
IndexableVal::Arr(joiner_items) => {
let mut out = Vec::new();
@@ -141,7 +143,7 @@
out += &sep;
}
first = false;
- out += &item;
+ write!(out, "{item}").unwrap()
} else if matches!(item, Val::Null) {
continue;
} else {
crates/jrsonnet-stdlib/src/lib.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/lib.rs
+++ b/crates/jrsonnet-stdlib/src/lib.rs
@@ -320,15 +320,19 @@
}
#[cfg(feature = "legacy-this-file")]
fn initialize(&self, s: State, source: Source) -> Context {
+ use jrsonnet_evaluator::val::StrValue;
+
let mut builder = ObjValueBuilder::new();
builder.with_super(self.stdlib_obj.clone());
builder
.member("thisFile".into())
.hide()
- .value(Val::Str(match source.source_path().path() {
- Some(p) => self.settings().path_resolver.resolve(p).into(),
- None => source.source_path().to_string().into(),
- }))
+ .value(Val::Str(StrValue::Flat(
+ match source.source_path().path() {
+ Some(p) => self.settings().path_resolver.resolve(p).into(),
+ None => source.source_path().to_string().into(),
+ },
+ )))
.expect("this object builder is empty");
let stdlib_with_this_file = builder.build();
crates/jrsonnet-stdlib/src/manifest/yaml.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/manifest/yaml.rs
+++ b/crates/jrsonnet-stdlib/src/manifest/yaml.rs
@@ -118,6 +118,7 @@
}
Val::Null => buf.push_str("null"),
Val::Str(s) => {
+ let s = s.clone().into_flat();
if s.is_empty() {
buf.push_str("\"\"");
} else if let Some(s) = s.strip_suffix('\n') {
@@ -128,10 +129,10 @@
buf.push_str(&options.padding);
buf.push_str(line);
}
- } else if !options.quote_keys && !yaml_needs_quotes(s) {
- buf.push_str(s);
+ } else if !options.quote_keys && !yaml_needs_quotes(&s) {
+ buf.push_str(&s);
} else {
- escape_string_json_buf(s, buf);
+ escape_string_json_buf(&s, buf);
}
}
Val::Num(n) => write!(buf, "{}", *n).unwrap(),
crates/jrsonnet-stdlib/src/objects.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/objects.rs
+++ b/crates/jrsonnet-stdlib/src/objects.rs
@@ -1,5 +1,9 @@
use jrsonnet_evaluator::{
- error::Result, function::builtin, typed::VecVal, val::Val, IStr, ObjValue,
+ error::Result,
+ function::builtin,
+ typed::VecVal,
+ val::{StrValue, Val},
+ IStr, ObjValue,
};
use jrsonnet_gcmodule::Cc;
@@ -17,7 +21,10 @@
preserve_order,
);
Ok(VecVal(Cc::new(
- out.into_iter().map(Val::Str).collect::<Vec<_>>(),
+ out.into_iter()
+ .map(StrValue::Flat)
+ .map(Val::Str)
+ .collect::<Vec<_>>(),
)))
}
crates/jrsonnet-stdlib/src/operator.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/operator.rs
+++ b/crates/jrsonnet-stdlib/src/operator.rs
@@ -7,7 +7,7 @@
operator::evaluate_mod_op,
stdlib::std_format,
typed::{Any, Either, Either2},
- val::{equals, primitive_equals},
+ val::{equals, primitive_equals, StrValue},
IStr, Val,
};
@@ -17,7 +17,7 @@
Ok(Any(evaluate_mod_op(
&match a {
A(v) => Val::Num(v),
- B(s) => Val::Str(s),
+ B(s) => Val::Str(StrValue::Flat(s)),
},
&b.0,
)?))
@@ -35,5 +35,5 @@
#[builtin]
pub fn builtin_format(str: IStr, vals: Any) -> Result<String> {
- std_format(str, vals.0)
+ std_format(&str, vals.0)
}
crates/jrsonnet-stdlib/src/strings.rsdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/strings.rs
+++ b/crates/jrsonnet-stdlib/src/strings.rs
@@ -3,7 +3,7 @@
function::builtin,
throw,
typed::{Either2, VecVal, M1},
- val::ArrValue,
+ val::{ArrValue, StrValue},
Either, IStr, Val,
};
use jrsonnet_gcmodule::Cc;
@@ -34,9 +34,12 @@
Ok(VecVal(Cc::new(match maxsplits {
A(n) => str
.splitn(n + 1, &c as &str)
- .map(|s| Val::Str(s.into()))
+ .map(|s| Val::Str(StrValue::Flat(s.into())))
+ .collect(),
+ B(_) => str
+ .split(&c as &str)
+ .map(|s| Val::Str(StrValue::Flat(s.into())))
.collect(),
- B(_) => str.split(&c as &str).map(|s| Val::Str(s.into())).collect(),
})))
}