difftreelog
Merge pull request #56 from messense/member-count-builtin
in: master
3 files changed
crates/jrsonnet-evaluator/src/builtin/mod.rsdiffbeforeafterboth1use crate::{2 equals,3 error::{Error::*, Result},4 operator::evaluate_mod_op,5 parse_args, primitive_equals, push, throw, with_state, ArrValue, Context, EvaluationState,6 FuncVal, IndexableVal, LazyVal, Val,7};8use format::{format_arr, format_obj};9use jrsonnet_gc::Gc;10use jrsonnet_interner::IStr;11use jrsonnet_parser::{ArgsDesc, ExprLocation};12use jrsonnet_types::ty;13use std::{collections::HashMap, path::PathBuf, rc::Rc};1415pub mod stdlib;16pub use stdlib::*;1718use self::manifest::{escape_string_json, manifest_json_ex, ManifestJsonOptions, ManifestType};1920pub mod format;21pub mod manifest;22pub mod sort;2324pub fn std_format(str: IStr, vals: Val) -> Result<Val> {25 push(26 Some(&ExprLocation(Rc::from(PathBuf::from("std.jsonnet")), 0, 0)),27 || format!("std.format of {}", str),28 || {29 Ok(match vals {30 Val::Arr(vals) => Val::Str(format_arr(&str, &vals.evaluated()?)?.into()),31 Val::Obj(obj) => Val::Str(format_obj(&str, &obj)?.into()),32 o => Val::Str(format_arr(&str, &[o])?.into()),33 })34 },35 )36}3738pub fn std_slice(39 indexable: IndexableVal,40 index: Option<usize>,41 end: Option<usize>,42 step: Option<usize>,43) -> Result<Val> {44 let index = index.unwrap_or(0);45 let end = end.unwrap_or_else(|| match &indexable {46 IndexableVal::Str(_) => usize::MAX,47 IndexableVal::Arr(v) => v.len(),48 });49 let step = step.unwrap_or(1);50 match &indexable {51 IndexableVal::Str(s) => Ok(Val::Str(52 (s.chars()53 .skip(index)54 .take(end - index)55 .step_by(step)56 .collect::<String>())57 .into(),58 )),59 IndexableVal::Arr(arr) => Ok(Val::Arr(60 (arr.iter()61 .skip(index)62 .take(end - index)63 .step_by(step)64 .collect::<Result<Vec<Val>>>()?)65 .into(),66 )),67 }68}6970type Builtin = fn(context: Context, loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val>;7172type BuiltinsType = HashMap<Box<str>, Builtin>;7374thread_local! {75 static BUILTINS: BuiltinsType = {76 [77 ("length".into(), builtin_length as Builtin),78 ("type".into(), builtin_type),79 ("makeArray".into(), builtin_make_array),80 ("codepoint".into(), builtin_codepoint),81 ("objectFieldsEx".into(), builtin_object_fields_ex),82 ("objectHasEx".into(), builtin_object_has_ex),83 ("slice".into(), builtin_slice),84 ("substr".into(), builtin_substr),85 ("primitiveEquals".into(), builtin_primitive_equals),86 ("equals".into(), builtin_equals),87 ("modulo".into(), builtin_modulo),88 ("mod".into(), builtin_mod),89 ("floor".into(), builtin_floor),90 ("ceil".into(), builtin_ceil),91 ("log".into(), builtin_log),92 ("pow".into(), builtin_pow),93 ("sqrt".into(), builtin_sqrt),94 ("sin".into(), builtin_sin),95 ("cos".into(), builtin_cos),96 ("tan".into(), builtin_tan),97 ("asin".into(), builtin_asin),98 ("acos".into(), builtin_acos),99 ("atan".into(), builtin_atan),100 ("exp".into(), builtin_exp),101 ("mantissa".into(), builtin_mantissa),102 ("exponent".into(), builtin_exponent),103 ("extVar".into(), builtin_ext_var),104 ("native".into(), builtin_native),105 ("filter".into(), builtin_filter),106 ("map".into(), builtin_map),107 ("flatMap".into(), builtin_flatmap),108 ("foldl".into(), builtin_foldl),109 ("foldr".into(), builtin_foldr),110 ("sortImpl".into(), builtin_sort_impl),111 ("format".into(), builtin_format),112 ("range".into(), builtin_range),113 ("char".into(), builtin_char),114 ("encodeUTF8".into(), builtin_encode_utf8),115 ("decodeUTF8".into(), builtin_decode_utf8),116 ("md5".into(), builtin_md5),117 ("base64".into(), builtin_base64),118 ("base64DecodeBytes".into(), builtin_base64_decode_bytes),119 ("base64Decode".into(), builtin_base64_decode),120 ("trace".into(), builtin_trace),121 ("join".into(), builtin_join),122 ("escapeStringJson".into(), builtin_escape_string_json),123 ("manifestJsonEx".into(), builtin_manifest_json_ex),124 ("reverse".into(), builtin_reverse),125 ("id".into(), builtin_id),126 ("strReplace".into(), builtin_str_replace),127 ("splitLimit".into(), builtin_splitlimit),128 ("parseJson".into(), builtin_parse_json),129 ("asciiUpper".into(), builtin_ascii_upper),130 ("asciiLower".into(), builtin_ascii_lower),131 ].iter().cloned().collect()132 };133}134135fn builtin_length(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {136 parse_args!(context, "length", args, 1, [137 0, x: ty!((string | object | array));138 ], {139 Ok(match x {140 Val::Str(n) => Val::Num(n.chars().count() as f64),141 Val::Arr(a) => Val::Num(a.len() as f64),142 Val::Obj(o) => Val::Num(143 o.fields_visibility()144 .into_iter()145 .filter(|(_k, v)| *v)146 .count() as f64,147 ),148 _ => unreachable!(),149 })150 })151}152153fn builtin_type(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {154 parse_args!(context, "type", args, 1, [155 0, x: ty!(any);156 ], {157 Ok(Val::Str(x.value_type().name().into()))158 })159}160161fn builtin_make_array(162 context: Context,163 _loc: Option<&ExprLocation>,164 args: &ArgsDesc,165) -> Result<Val> {166 parse_args!(context, "makeArray", args, 2, [167 0, sz: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;168 1, func: ty!(function) => Val::Func;169 ], {170 let mut out = Vec::with_capacity(sz as usize);171 for i in 0..sz as usize {172 out.push(LazyVal::new_resolved(func.evaluate_values(173 context.clone(),174 &[Val::Num(i as f64)]175 )?))176 }177 Ok(Val::Arr(out.into()))178 })179}180181fn builtin_codepoint(182 context: Context,183 _loc: Option<&ExprLocation>,184 args: &ArgsDesc,185) -> Result<Val> {186 parse_args!(context, "codepoint", args, 1, [187 0, str: ty!(char) => Val::Str;188 ], {189 Ok(Val::Num(str.chars().next().unwrap() as u32 as f64))190 })191}192193fn builtin_object_fields_ex(194 context: Context,195 _loc: Option<&ExprLocation>,196 args: &ArgsDesc,197) -> Result<Val> {198 parse_args!(context, "objectFieldsEx", args, 2, [199 0, obj: ty!(object) => Val::Obj;200 1, inc_hidden: ty!(boolean) => Val::Bool;201 ], {202 let out = obj.fields_ex(inc_hidden);203 Ok(Val::Arr(out.into_iter().map(Val::Str).collect::<Vec<_>>().into()))204 })205}206207fn builtin_object_has_ex(208 context: Context,209 _loc: Option<&ExprLocation>,210 args: &ArgsDesc,211) -> Result<Val> {212 parse_args!(context, "objectHasEx", args, 3, [213 0, obj: ty!(object) => Val::Obj;214 1, f: ty!(string) => Val::Str;215 2, inc_hidden: ty!(boolean) => Val::Bool;216 ], {217 Ok(Val::Bool(obj.has_field_ex(f, inc_hidden)))218 })219}220221fn builtin_parse_json(222 context: Context,223 _loc: Option<&ExprLocation>,224 args: &ArgsDesc,225) -> Result<Val> {226 parse_args!(context, "parseJson", args, 1, [227 0, s: ty!(string) => Val::Str;228 ], {229 let state = EvaluationState::default();230 let path = PathBuf::from("std.parseJson").into();231 state.evaluate_snippet_raw(path ,s)232 })233}234235fn builtin_slice(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {236 parse_args!(context, "slice", args, 4, [237 0, indexable: ty!((string | array));238 1, index: ty!((number | null));239 2, end: ty!((number | null));240 3, step: ty!((number | null));241 ], {242 std_slice(243 indexable.into_indexable()?,244 index.try_cast_nullable_num("index")?.map(|v| v as usize),245 end.try_cast_nullable_num("end")?.map(|v| v as usize),246 step.try_cast_nullable_num("step")?.map(|v| v as usize),247 )248 })249}250251fn builtin_substr(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {252 parse_args!(context, "substr", args, 3, [253 0, str: ty!(string) => Val::Str;254 1, from: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;255 2, len: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;256 ], {257 let out: String = str.chars().skip(from as usize).take(len as usize).collect();258 Ok(Val::Str(out.into()))259 })260}261262fn builtin_primitive_equals(263 context: Context,264 _loc: Option<&ExprLocation>,265 args: &ArgsDesc,266) -> Result<Val> {267 parse_args!(context, "primitiveEquals", args, 2, [268 0, a: ty!(any);269 1, b: ty!(any);270 ], {271 Ok(Val::Bool(primitive_equals(&a, &b)?))272 })273}274275fn builtin_equals(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {276 parse_args!(context, "equals", args, 2, [277 0, a: ty!(any);278 1, b: ty!(any);279 ], {280 Ok(Val::Bool(equals(&a, &b)?))281 })282}283284fn builtin_modulo(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {285 parse_args!(context, "modulo", args, 2, [286 0, a: ty!(number) => Val::Num;287 1, b: ty!(number) => Val::Num;288 ], {289 Ok(Val::Num(a % b))290 })291}292293fn builtin_mod(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {294 parse_args!(context, "mod", args, 2, [295 0, a: ty!((number | string));296 1, b: ty!(any);297 ], {298 evaluate_mod_op(&a, &b)299 })300}301302fn builtin_floor(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {303 parse_args!(context, "floor", args, 1, [304 0, x: ty!(number) => Val::Num;305 ], {306 Ok(Val::Num(x.floor()))307 })308}309310fn builtin_ceil(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {311 parse_args!(context, "ceil", args, 1, [312 0, x: ty!(number) => Val::Num;313 ], {314 Ok(Val::Num(x.ceil()))315 })316}317318fn builtin_log(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {319 parse_args!(context, "log", args, 1, [320 0, n: ty!(number) => Val::Num;321 ], {322 Ok(Val::Num(n.ln()))323 })324}325326fn builtin_pow(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {327 parse_args!(context, "pow", args, 2, [328 0, x: ty!(number) => Val::Num;329 1, n: ty!(number) => Val::Num;330 ], {331 Ok(Val::Num(x.powf(n)))332 })333}334335fn builtin_sqrt(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {336 parse_args!(context, "sqrt", args, 1, [337 0, x: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;338 ], {339 Ok(Val::Num(x.sqrt()))340 })341}342343fn builtin_sin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {344 parse_args!(context, "sin", args, 1, [345 0, x: ty!(number) => Val::Num;346 ], {347 Ok(Val::Num(x.sin()))348 })349}350351fn builtin_cos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {352 parse_args!(context, "cos", args, 1, [353 0, x: ty!(number) => Val::Num;354 ], {355 Ok(Val::Num(x.cos()))356 })357}358359fn builtin_tan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {360 parse_args!(context, "tan", args, 1, [361 0, x: ty!(number) => Val::Num;362 ], {363 Ok(Val::Num(x.tan()))364 })365}366367fn builtin_asin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {368 parse_args!(context, "asin", args, 1, [369 0, x: ty!(number) => Val::Num;370 ], {371 Ok(Val::Num(x.asin()))372 })373}374375fn builtin_acos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {376 parse_args!(context, "acos", args, 1, [377 0, x: ty!(number) => Val::Num;378 ], {379 Ok(Val::Num(x.acos()))380 })381}382383fn builtin_atan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {384 parse_args!(context, "atan", args, 1, [385 0, x: ty!(number) => Val::Num;386 ], {387 Ok(Val::Num(x.atan()))388 })389}390391fn builtin_exp(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {392 parse_args!(context, "exp", args, 1, [393 0, x: ty!(number) => Val::Num;394 ], {395 Ok(Val::Num(x.exp()))396 })397}398399fn frexp(s: f64) -> (f64, i16) {400 if 0.0 == s {401 (s, 0)402 } else {403 let lg = s.abs().log2();404 let x = (lg - lg.floor() - 1.0).exp2();405 let exp = lg.floor() + 1.0;406 (s.signum() * x, exp as i16)407 }408}409410fn builtin_mantissa(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {411 parse_args!(context, "mantissa", args, 1, [412 0, x: ty!(number) => Val::Num;413 ], {414 Ok(Val::Num(frexp(x).0))415 })416}417418fn builtin_exponent(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {419 parse_args!(context, "exponent", args, 1, [420 0, x: ty!(number) => Val::Num;421 ], {422 Ok(Val::Num(frexp(x).1.into()))423 })424}425426fn builtin_ext_var(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {427 parse_args!(context, "extVar", args, 1, [428 0, x: ty!(string) => Val::Str;429 ], {430 Ok(with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or(UndefinedExternalVariable(x))?)431 })432}433434fn builtin_native(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {435 parse_args!(context, "native", args, 1, [436 0, x: ty!(string) => Val::Str;437 ], {438 Ok(with_state(|s| s.settings().ext_natives.get(&x).cloned()).map(|v| Val::Func(Gc::new(FuncVal::NativeExt(x.clone(), v)))).ok_or(UndefinedExternalFunction(x))?)439 })440}441442fn builtin_filter(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {443 parse_args!(context, "filter", args, 2, [444 0, func: ty!(function) => Val::Func;445 1, arr: ty!(array) => Val::Arr;446 ], {447 Ok(Val::Arr(arr.filter(|val| func448 .evaluate_values(context.clone(), &[val.clone()])?449 .try_cast_bool("filter predicate"))?))450 })451}452453fn builtin_map(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {454 parse_args!(context, "map", args, 2, [455 0, func: ty!(function) => Val::Func;456 1, arr: ty!(array) => Val::Arr;457 ], {458 Ok(Val::Arr(arr.map(|val| func459 .evaluate_values(context.clone(), &[val]))?))460 })461}462463fn builtin_flatmap(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {464 parse_args!(context, "flatMap", args, 2, [465 0, func: ty!(function) => Val::Func;466 1, arr: ty!((array | string));467 ], {468 match arr {469 Val::Str(s) => {470 let mut out = String::new();471 for c in s.chars() {472 match func.evaluate_values(context.clone(), &[Val::Str(c.to_string().into())])? {473 Val::Str(o) => out.push_str(&o),474 _ => throw!(RuntimeError("in std.join all items should be strings".into())),475 };476 }477 Ok(Val::Str(out.into()))478 },479 Val::Arr(a) => {480 let mut out = Vec::new();481 for el in a.iter() {482 let el = el?;483 match func.evaluate_values(context.clone(), &[el])? {484 Val::Arr(o) => for oe in o.iter() {485 out.push(oe?)486 },487 _ => throw!(RuntimeError("in std.join all items should be arrays".into())),488 };489 }490 Ok(Val::Arr(out.into()))491 },492 _ => unreachable!(),493 }494 })495}496497fn builtin_foldl(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {498 parse_args!(context, "foldl", args, 3, [499 0, func: ty!(function) => Val::Func;500 1, arr: ty!(array) => Val::Arr;501 2, init: ty!(any);502 ], {503 let mut acc = init;504 for i in arr.iter() {505 acc = func.evaluate_values(context.clone(), &[acc, i?])?;506 }507 Ok(acc)508 })509}510511fn builtin_foldr(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {512 parse_args!(context, "foldr", args, 3, [513 0, func: ty!(function) => Val::Func;514 1, arr: ty!(array) => Val::Arr;515 2, init: ty!(any);516 ], {517 let mut acc = init;518 for i in arr.iter().rev() {519 acc = func.evaluate_values(context.clone(), &[acc, i?])?;520 }521 Ok(acc)522 })523}524525#[allow(non_snake_case)]526fn builtin_sort_impl(527 context: Context,528 _loc: Option<&ExprLocation>,529 args: &ArgsDesc,530) -> Result<Val> {531 parse_args!(context, "sort", args, 2, [532 0, arr: ty!(array) => Val::Arr;533 1, keyF: ty!(function) => Val::Func;534 ], {535 if arr.len() <= 1 {536 return Ok(Val::Arr(arr))537 }538 Ok(Val::Arr(ArrValue::Eager(sort::sort(context, arr.evaluated()?, &keyF)?)))539 })540}541542fn builtin_format(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {543 parse_args!(context, "format", args, 2, [544 0, str: ty!(string) => Val::Str;545 1, vals: ty!(any)546 ], {547 std_format(str, vals)548 })549}550551fn builtin_range(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {552 parse_args!(context, "range", args, 2, [553 0, from: ty!(number) => Val::Num;554 1, to: ty!(number) => Val::Num;555 ], {556 if to < from {557 return Ok(Val::Arr(ArrValue::new_eager()))558 }559 let mut out = Vec::with_capacity((1+to as usize-from as usize).max(0));560 for i in from as usize..=to as usize {561 out.push(Val::Num(i as f64));562 }563 Ok(Val::Arr(out.into()))564 })565}566567fn builtin_char(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {568 parse_args!(context, "char", args, 1, [569 0, n: ty!(number) => Val::Num;570 ], {571 let mut out = String::new();572 out.push(std::char::from_u32(n as u32).ok_or_else(||573 InvalidUnicodeCodepointGot(n as u32)574 )?);575 Ok(Val::Str(out.into()))576 })577}578579fn builtin_encode_utf8(580 context: Context,581 _loc: Option<&ExprLocation>,582 args: &ArgsDesc,583) -> Result<Val> {584 parse_args!(context, "encodeUTF8", args, 1, [585 0, str: ty!(string) => Val::Str;586 ], {587 Ok(Val::Arr((str.bytes().map(|b| Val::Num(b as f64)).collect::<Vec<Val>>()).into()))588 })589}590591fn builtin_decode_utf8(592 context: Context,593 _loc: Option<&ExprLocation>,594 args: &ArgsDesc,595) -> Result<Val> {596 parse_args!(context, "decodeUTF8", args, 1, [597 0, arr: ty!((Array<ubyte>)) => Val::Arr;598 ], {599 let data: Result<Vec<u8>> = arr.iter().map(|v| v.map(|v| match v{600 Val::Num(n) => n as u8,601 _ => unreachable!(),602 })).collect();603 let data = data?;604 Ok(Val::Str(String::from_utf8(data).map_err(|_| RuntimeError("bad utf8".into()))?.into()))605 })606}607608fn builtin_md5(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {609 parse_args!(context, "md5", args, 1, [610 0, str: ty!(string) => Val::Str;611 ], {612 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))613 })614}615616fn builtin_trace(context: Context, loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {617 parse_args!(context, "trace", args, 2, [618 0, str: ty!(string) => Val::Str;619 1, rest: ty!(any);620 ], {621 eprint!("TRACE:");622 if let Some(loc) = loc {623 with_state(|s|{624 let locs = s.map_source_locations(&loc.0, &[loc.1]);625 eprint!(" {}:{}", loc.0.file_name().unwrap().to_str().unwrap(), locs[0].line);626 });627 }628 eprintln!(" {}", str);629 Ok(rest)630 })631}632633fn builtin_base64(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {634 parse_args!(context, "base64", args, 1, [635 0, input: ty!((string | (Array<number>)));636 ], {637 Ok(Val::Str(match input {638 Val::Str(s) => {639 base64::encode(s.bytes().collect::<Vec<_>>()).into()640 },641 Val::Arr(a) => {642 base64::encode(a.iter().map(|v| {643 Ok(v?.unwrap_num()? as u8)644 }).collect::<Result<Vec<_>>>()?).into()645 },646 _ => unreachable!()647 }))648 })649}650651fn builtin_base64_decode_bytes(652 context: Context,653 _loc: Option<&ExprLocation>,654 args: &ArgsDesc,655) -> Result<Val> {656 parse_args!(context, "base64DecodeBytes", args, 1, [657 0, input: ty!(string) => Val::Str;658 ], {659 Ok(Val::Arr(660 base64::decode(&input.as_bytes())661 .map_err(|_| RuntimeError("bad base64".into()))?662 .iter()663 .map(|v| Val::Num(*v as f64)).collect::<Vec<_>>().into()664 ))665 })666}667668fn builtin_base64_decode(669 context: Context,670 _loc: Option<&ExprLocation>,671 args: &ArgsDesc,672) -> Result<Val> {673 parse_args!(context, "base64Decode", args, 1, [674 0, input: ty!(string) => Val::Str;675 ], {676 Ok(Val::Str(677 String::from_utf8(base64::decode(&input.as_bytes())678 .map_err(|_| RuntimeError("bad base64".into()))?)679 .map_err(|_| RuntimeError("bad utf8".into()))?.into()680 ))681 })682}683684fn builtin_join(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {685 parse_args!(context, "join", args, 2, [686 0, sep: ty!((string | array));687 1, arr: ty!(array) => Val::Arr;688 ], {689 Ok(match sep {690 Val::Arr(joiner_items) => {691 let mut out = Vec::new();692693 let mut first = true;694 for item in arr.iter() {695 let item = item?.clone();696 if let Val::Arr(items) = item {697 if !first {698 out.reserve(joiner_items.len());699 // TODO: extend700 for item in joiner_items.iter() {701 out.push(item?);702 }703 }704 first = false;705 out.reserve(items.len());706 // TODO: extend707 for item in items.iter() {708 out.push(item?);709 }710 } else {711 throw!(RuntimeError("in std.join all items should be arrays".into()));712 }713 }714715 Val::Arr(out.into())716 },717 Val::Str(sep) => {718 let mut out = String::new();719720 let mut first = true;721 for item in arr.iter() {722 let item = item?.clone();723 if let Val::Str(item) = item {724 if !first {725 out += &sep;726 }727 first = false;728 out += &item;729 } else {730 throw!(RuntimeError("in std.join all items should be strings".into()));731 }732 }733734 Val::Str(out.into())735 },736 _ => unreachable!()737 })738 })739}740741fn builtin_escape_string_json(742 context: Context,743 _loc: Option<&ExprLocation>,744 args: &ArgsDesc,745) -> Result<Val> {746 parse_args!(context, "escapeStringJson", args, 1, [747 0, str_: ty!(string) => Val::Str;748 ], {749 Ok(Val::Str(escape_string_json(&str_).into()))750 })751}752753fn builtin_manifest_json_ex(754 context: Context,755 _loc: Option<&ExprLocation>,756 args: &ArgsDesc,757) -> Result<Val> {758 parse_args!(context, "manifestJsonEx", args, 2, [759 0, value: ty!(any);760 1, indent: ty!(string) => Val::Str;761 ], {762 Ok(Val::Str(manifest_json_ex(&value, &ManifestJsonOptions {763 padding: &indent,764 mtype: ManifestType::Std,765 })?.into()))766 })767}768769fn builtin_reverse(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {770 parse_args!(context, "reverse", args, 1, [771 0, value: ty!(array) => Val::Arr;772 ], {773 Ok(Val::Arr(value.reversed()))774 })775}776777fn builtin_id(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {778 parse_args!(context, "id", args, 1, [779 0, v: ty!(any);780 ], {781 Ok(v)782 })783}784785fn builtin_str_replace(786 context: Context,787 _loc: Option<&ExprLocation>,788 args: &ArgsDesc,789) -> Result<Val> {790 parse_args!(context, "strReplace", args, 3, [791 0, str: ty!(string) => Val::Str;792 1, from: ty!(string) => Val::Str;793 2, to: ty!(string) => Val::Str;794 ], {795 let mut out = String::new();796 let mut last_idx = 0;797 while let Some(idx) = (&str[last_idx..]).find(&from as &str) {798 out.push_str(&str[last_idx..last_idx+idx]);799 out.push_str(&to);800 last_idx += idx + from.len();801 }802 if last_idx == 0 {803 return Ok(Val::Str(str))804 }805 out.push_str(&str[last_idx..]);806 Ok(Val::Str(out.into()))807 })808}809810fn builtin_splitlimit(811 context: Context,812 _loc: Option<&ExprLocation>,813 args: &ArgsDesc,814) -> Result<Val> {815 parse_args!(context, "splitLimit", args, 3, [816 0, str: ty!(string) => Val::Str;817 1, c: ty!(char) => Val::Str;818 2, maxsplits: ty!(number) => Val::Num;819 ], {820 let maxsplits = maxsplits as isize;821 let c = c.chars().next().unwrap();822823 let out: Vec<Val> = if maxsplits == -1 {824 str.split(c).map(|s| Val::Str(s.into())).collect()825 } else {826 str.splitn(maxsplits as usize + 1, c).map(|s| Val::Str(s.into())).collect()827 };828829 Ok(Val::Arr(out.into()))830 })831}832833fn builtin_ascii_upper(834 context: Context,835 _loc: Option<&ExprLocation>,836 args: &ArgsDesc,837) -> Result<Val> {838 parse_args!(context, "asciiUpper", args, 1, [839 0, str: ty!(string) => Val::Str;840 ], {841 Ok(Val::Str(str.to_ascii_uppercase().into()))842 })843}844845fn builtin_ascii_lower(846 context: Context,847 _loc: Option<&ExprLocation>,848 args: &ArgsDesc,849) -> Result<Val> {850 parse_args!(context, "asciiLower", args, 1, [851 0, str: ty!(string) => Val::Str;852 ], {853 Ok(Val::Str(str.to_ascii_lowercase().into()))854 })855}856857pub fn call_builtin(858 context: Context,859 loc: Option<&ExprLocation>,860 name: &str,861 args: &ArgsDesc,862) -> Result<Val> {863 BUILTINS864 .with(|builtins| builtins.get(name).copied())865 .ok_or_else(|| IntrinsicNotFound(name.into()))?(context, loc, args)866}crates/jrsonnet-evaluator/src/lib.rsdiffbeforeafterboth--- a/crates/jrsonnet-evaluator/src/lib.rs
+++ b/crates/jrsonnet-evaluator/src/lib.rs
@@ -1051,4 +1051,21 @@
assert_eval!(r#"std.assertEqual(std.asciiUpper("aBc😀"), "ABC😀")"#);
assert_eval!(r#"std.assertEqual(std.asciiLower("aBc😀"), "abc😀")"#);
}
+
+ #[test]
+ fn test_member() {
+ assert_eval!(r#"!std.member("", "")"#);
+ assert_eval!(r#"std.member("abc", "a")"#);
+ assert_eval!(r#"!std.member("abc", "d")"#);
+ assert_eval!(r#"!std.member([], "")"#);
+ assert_eval!(r#"std.member(["a", "b", "c"], "a")"#);
+ assert_eval!(r#"!std.member(["a", "b", "c"], "d")"#);
+ }
+
+ #[test]
+ fn test_count() {
+ assert_eval!(r#"std.assertEqual(std.count([], ""), 0)"#);
+ assert_eval!(r#"std.assertEqual(std.count(["a", "b", "a"], "d"), 0)"#);
+ assert_eval!(r#"std.assertEqual(std.count(["a", "b", "a"], "a"), 2)"#);
+ }
}
crates/jrsonnet-stdlib/src/std.jsonnetdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/std.jsonnet
+++ b/crates/jrsonnet-stdlib/src/std.jsonnet
@@ -143,14 +143,9 @@
slice:: $intrinsic(slice),
- member(arr, x)::
- if std.isArray(arr) then
- std.count(arr, x) > 0
- else if std.isString(arr) then
- std.length(std.findSubstr(x, arr)) > 0
- else error 'std.member first argument must be an array or a string',
+ member:: $intrinsic(member),
- count(arr, x):: std.length(std.filter(function(v) v == x, arr)),
+ count:: $intrinsic(count),
mod:: $intrinsic(mod),