difftreelog
feat std.{mantissa,exponent} builtins
in: master
2 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 ("primitiveEquals".into(), builtin_primitive_equals),85 ("equals".into(), builtin_equals),86 ("modulo".into(), builtin_modulo),87 ("mod".into(), builtin_mod),88 ("floor".into(), builtin_floor),89 ("ceil".into(), builtin_ceil),90 ("log".into(), builtin_log),91 ("pow".into(), builtin_pow),92 ("sqrt".into(), builtin_sqrt),93 ("sin".into(), builtin_sin),94 ("cos".into(), builtin_cos),95 ("tan".into(), builtin_tan),96 ("asin".into(), builtin_asin),97 ("acos".into(), builtin_acos),98 ("atan".into(), builtin_atan),99 ("exp".into(), builtin_exp),100 ("extVar".into(), builtin_ext_var),101 ("native".into(), builtin_native),102 ("filter".into(), builtin_filter),103 ("map".into(), builtin_map),104 ("flatMap".into(), builtin_flatmap),105 ("foldl".into(), builtin_foldl),106 ("foldr".into(), builtin_foldr),107 ("sortImpl".into(), builtin_sort_impl),108 ("format".into(), builtin_format),109 ("range".into(), builtin_range),110 ("char".into(), builtin_char),111 ("encodeUTF8".into(), builtin_encode_utf8),112 ("md5".into(), builtin_md5),113 ("base64".into(), builtin_base64),114 ("trace".into(), builtin_trace),115 ("join".into(), builtin_join),116 ("escapeStringJson".into(), builtin_escape_string_json),117 ("manifestJsonEx".into(), builtin_manifest_json_ex),118 ("reverse".into(), builtin_reverse),119 ("id".into(), builtin_id),120 ("strReplace".into(), builtin_str_replace),121 ("parseJson".into(), builtin_parse_json),122 ].iter().cloned().collect()123 };124}125126fn builtin_length(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {127 parse_args!(context, "length", args, 1, [128 0, x: ty!((string | object | array));129 ], {130 Ok(match x {131 Val::Str(n) => Val::Num(n.chars().count() as f64),132 Val::Arr(a) => Val::Num(a.len() as f64),133 Val::Obj(o) => Val::Num(134 o.fields_visibility()135 .into_iter()136 .filter(|(_k, v)| *v)137 .count() as f64,138 ),139 _ => unreachable!(),140 })141 })142}143144fn builtin_type(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {145 parse_args!(context, "type", args, 1, [146 0, x: ty!(any);147 ], {148 Ok(Val::Str(x.value_type().name().into()))149 })150}151152fn builtin_make_array(153 context: Context,154 _loc: Option<&ExprLocation>,155 args: &ArgsDesc,156) -> Result<Val> {157 parse_args!(context, "makeArray", args, 2, [158 0, sz: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;159 1, func: ty!(function) => Val::Func;160 ], {161 let mut out = Vec::with_capacity(sz as usize);162 for i in 0..sz as usize {163 out.push(LazyVal::new_resolved(func.evaluate_values(164 context.clone(),165 &[Val::Num(i as f64)]166 )?))167 }168 Ok(Val::Arr(out.into()))169 })170}171172fn builtin_codepoint(173 context: Context,174 _loc: Option<&ExprLocation>,175 args: &ArgsDesc,176) -> Result<Val> {177 parse_args!(context, "codepoint", args, 1, [178 0, str: ty!(char) => Val::Str;179 ], {180 Ok(Val::Num(str.chars().next().unwrap() as u32 as f64))181 })182}183184fn builtin_object_fields_ex(185 context: Context,186 _loc: Option<&ExprLocation>,187 args: &ArgsDesc,188) -> Result<Val> {189 parse_args!(context, "objectFieldsEx", args, 2, [190 0, obj: ty!(object) => Val::Obj;191 1, inc_hidden: ty!(boolean) => Val::Bool;192 ], {193 let out = obj.fields_ex(inc_hidden);194 Ok(Val::Arr(out.into_iter().map(Val::Str).collect::<Vec<_>>().into()))195 })196}197198fn builtin_object_has_ex(199 context: Context,200 _loc: Option<&ExprLocation>,201 args: &ArgsDesc,202) -> Result<Val> {203 parse_args!(context, "objectHasEx", args, 3, [204 0, obj: ty!(object) => Val::Obj;205 1, f: ty!(string) => Val::Str;206 2, inc_hidden: ty!(boolean) => Val::Bool;207 ], {208 Ok(Val::Bool(obj.has_field_ex(f, inc_hidden)))209 })210}211212fn builtin_parse_json(213 context: Context,214 _loc: Option<&ExprLocation>,215 args: &ArgsDesc,216) -> Result<Val> {217 parse_args!(context, "parseJson", args, 1, [218 0, s: ty!(string) => Val::Str;219 ], {220 let state = EvaluationState::default();221 let path = PathBuf::from("std.parseJson").into();222 state.evaluate_snippet_raw(path ,s)223 })224}225226fn builtin_slice(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {227 parse_args!(context, "slice", args, 4, [228 0, indexable: ty!((string | array));229 1, index: ty!((number | null));230 2, end: ty!((number | null));231 3, step: ty!((number | null));232 ], {233 std_slice(234 indexable.into_indexable()?,235 index.try_cast_nullable_num("index")?.map(|v| v as usize),236 end.try_cast_nullable_num("end")?.map(|v| v as usize),237 step.try_cast_nullable_num("step")?.map(|v| v as usize),238 )239 })240}241242fn builtin_primitive_equals(243 context: Context,244 _loc: Option<&ExprLocation>,245 args: &ArgsDesc,246) -> Result<Val> {247 parse_args!(context, "primitiveEquals", args, 2, [248 0, a: ty!(any);249 1, b: ty!(any);250 ], {251 Ok(Val::Bool(primitive_equals(&a, &b)?))252 })253}254255fn builtin_equals(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {256 parse_args!(context, "equals", args, 2, [257 0, a: ty!(any);258 1, b: ty!(any);259 ], {260 Ok(Val::Bool(equals(&a, &b)?))261 })262}263264fn builtin_modulo(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {265 parse_args!(context, "modulo", args, 2, [266 0, a: ty!(number) => Val::Num;267 1, b: ty!(number) => Val::Num;268 ], {269 Ok(Val::Num(a % b))270 })271}272273fn builtin_mod(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {274 parse_args!(context, "mod", args, 2, [275 0, a: ty!((number | string));276 1, b: ty!(any);277 ], {278 evaluate_mod_op(&a, &b)279 })280}281282fn builtin_floor(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {283 parse_args!(context, "floor", args, 1, [284 0, x: ty!(number) => Val::Num;285 ], {286 Ok(Val::Num(x.floor()))287 })288}289290fn builtin_ceil(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {291 parse_args!(context, "ceil", args, 1, [292 0, x: ty!(number) => Val::Num;293 ], {294 Ok(Val::Num(x.ceil()))295 })296}297298fn builtin_log(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {299 parse_args!(context, "log", args, 1, [300 0, n: ty!(number) => Val::Num;301 ], {302 Ok(Val::Num(n.ln()))303 })304}305306fn builtin_pow(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {307 parse_args!(context, "pow", args, 2, [308 0, x: ty!(number) => Val::Num;309 1, n: ty!(number) => Val::Num;310 ], {311 Ok(Val::Num(x.powf(n)))312 })313}314315fn builtin_sqrt(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {316 parse_args!(context, "sqrt", args, 1, [317 0, x: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;318 ], {319 Ok(Val::Num(x.sqrt()))320 })321}322323fn builtin_sin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {324 parse_args!(context, "sin", args, 1, [325 0, x: ty!(number) => Val::Num;326 ], {327 Ok(Val::Num(x.sin()))328 })329}330331fn builtin_cos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {332 parse_args!(context, "cos", args, 1, [333 0, x: ty!(number) => Val::Num;334 ], {335 Ok(Val::Num(x.cos()))336 })337}338339fn builtin_tan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {340 parse_args!(context, "tan", args, 1, [341 0, x: ty!(number) => Val::Num;342 ], {343 Ok(Val::Num(x.tan()))344 })345}346347fn builtin_asin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {348 parse_args!(context, "asin", args, 1, [349 0, x: ty!(number) => Val::Num;350 ], {351 Ok(Val::Num(x.asin()))352 })353}354355fn builtin_acos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {356 parse_args!(context, "acos", args, 1, [357 0, x: ty!(number) => Val::Num;358 ], {359 Ok(Val::Num(x.acos()))360 })361}362363fn builtin_atan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {364 parse_args!(context, "atan", args, 1, [365 0, x: ty!(number) => Val::Num;366 ], {367 Ok(Val::Num(x.atan()))368 })369}370371fn builtin_exp(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {372 parse_args!(context, "exp", args, 1, [373 0, x: ty!(number) => Val::Num;374 ], {375 Ok(Val::Num(x.exp()))376 })377}378379fn builtin_ext_var(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {380 parse_args!(context, "extVar", args, 1, [381 0, x: ty!(string) => Val::Str;382 ], {383 Ok(with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or(UndefinedExternalVariable(x))?)384 })385}386387fn builtin_native(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {388 parse_args!(context, "native", args, 1, [389 0, x: ty!(string) => Val::Str;390 ], {391 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))?)392 })393}394395fn builtin_filter(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {396 parse_args!(context, "filter", args, 2, [397 0, func: ty!(function) => Val::Func;398 1, arr: ty!(array) => Val::Arr;399 ], {400 Ok(Val::Arr(arr.filter(|val| func401 .evaluate_values(context.clone(), &[val.clone()])?402 .try_cast_bool("filter predicate"))?))403 })404}405406fn builtin_map(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {407 parse_args!(context, "map", args, 2, [408 0, func: ty!(function) => Val::Func;409 1, arr: ty!(array) => Val::Arr;410 ], {411 Ok(Val::Arr(arr.map(|val| func412 .evaluate_values(context.clone(), &[val]))?))413 })414}415416fn builtin_flatmap(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {417 parse_args!(context, "flatMap", args, 2, [418 0, func: ty!(function) => Val::Func;419 1, arr: ty!((array | string));420 ], {421 match arr {422 Val::Str(s) => {423 let mut out = String::new();424 for c in s.chars() {425 match func.evaluate_values(context.clone(), &[Val::Str(c.to_string().into())])? {426 Val::Str(o) => out.push_str(&o),427 _ => throw!(RuntimeError("in std.join all items should be strings".into())),428 };429 }430 Ok(Val::Str(out.into()))431 },432 Val::Arr(a) => {433 let mut out = Vec::new();434 for el in a.iter() {435 let el = el?;436 match func.evaluate_values(context.clone(), &[el])? {437 Val::Arr(o) => for oe in o.iter() {438 out.push(oe?)439 },440 _ => throw!(RuntimeError("in std.join all items should be arrays".into())),441 };442 }443 Ok(Val::Arr(out.into()))444 },445 _ => unreachable!(),446 }447 })448}449450fn builtin_foldl(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {451 parse_args!(context, "foldl", args, 3, [452 0, func: ty!(function) => Val::Func;453 1, arr: ty!(array) => Val::Arr;454 2, init: ty!(any);455 ], {456 let mut acc = init;457 for i in arr.iter() {458 acc = func.evaluate_values(context.clone(), &[acc, i?])?;459 }460 Ok(acc)461 })462}463464fn builtin_foldr(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {465 parse_args!(context, "foldr", args, 3, [466 0, func: ty!(function) => Val::Func;467 1, arr: ty!(array) => Val::Arr;468 2, init: ty!(any);469 ], {470 let mut acc = init;471 for i in arr.iter().rev() {472 acc = func.evaluate_values(context.clone(), &[acc, i?])?;473 }474 Ok(acc)475 })476}477478#[allow(non_snake_case)]479fn builtin_sort_impl(480 context: Context,481 _loc: Option<&ExprLocation>,482 args: &ArgsDesc,483) -> Result<Val> {484 parse_args!(context, "sort", args, 2, [485 0, arr: ty!(array) => Val::Arr;486 1, keyF: ty!(function) => Val::Func;487 ], {488 if arr.len() <= 1 {489 return Ok(Val::Arr(arr))490 }491 Ok(Val::Arr(ArrValue::Eager(sort::sort(context, arr.evaluated()?, &keyF)?)))492 })493}494495fn builtin_format(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {496 parse_args!(context, "format", args, 2, [497 0, str: ty!(string) => Val::Str;498 1, vals: ty!(any)499 ], {500 std_format(str, vals)501 })502}503504fn builtin_range(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {505 parse_args!(context, "range", args, 2, [506 0, from: ty!(number) => Val::Num;507 1, to: ty!(number) => Val::Num;508 ], {509 if to < from {510 return Ok(Val::Arr(ArrValue::new_eager()))511 }512 let mut out = Vec::with_capacity((1+to as usize-from as usize).max(0));513 for i in from as usize..=to as usize {514 out.push(Val::Num(i as f64));515 }516 Ok(Val::Arr(out.into()))517 })518}519520fn builtin_char(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {521 parse_args!(context, "char", args, 1, [522 0, n: ty!(number) => Val::Num;523 ], {524 let mut out = String::new();525 out.push(std::char::from_u32(n as u32).ok_or_else(||526 InvalidUnicodeCodepointGot(n as u32)527 )?);528 Ok(Val::Str(out.into()))529 })530}531532fn builtin_encode_utf8(533 context: Context,534 _loc: Option<&ExprLocation>,535 args: &ArgsDesc,536) -> Result<Val> {537 parse_args!(context, "encodeUTF8", args, 1, [538 0, str: ty!(string) => Val::Str;539 ], {540 Ok(Val::Arr((str.bytes().map(|b| Val::Num(b as f64)).collect::<Vec<Val>>()).into()))541 })542}543544fn builtin_md5(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {545 parse_args!(context, "md5", args, 1, [546 0, str: ty!(string) => Val::Str;547 ], {548 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))549 })550}551552fn builtin_trace(context: Context, loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {553 parse_args!(context, "trace", args, 2, [554 0, str: ty!(string) => Val::Str;555 1, rest: ty!(any);556 ], {557 eprint!("TRACE:");558 if let Some(loc) = loc {559 with_state(|s|{560 let locs = s.map_source_locations(&loc.0, &[loc.1]);561 eprint!(" {}:{}", loc.0.file_name().unwrap().to_str().unwrap(), locs[0].line);562 });563 }564 eprintln!(" {}", str);565 Ok(rest)566 })567}568569fn builtin_base64(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {570 parse_args!(context, "base64", args, 1, [571 0, input: ty!((string | (Array<number>)));572 ], {573 Ok(Val::Str(match input {574 Val::Str(s) => {575 base64::encode(s.bytes().collect::<Vec<_>>()).into()576 },577 Val::Arr(a) => {578 base64::encode(a.iter().map(|v| {579 Ok(v?.unwrap_num()? as u8)580 }).collect::<Result<Vec<_>>>()?).into()581 },582 _ => unreachable!()583 }))584 })585}586587fn builtin_join(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {588 parse_args!(context, "join", args, 2, [589 0, sep: ty!((string | array));590 1, arr: ty!(array) => Val::Arr;591 ], {592 Ok(match sep {593 Val::Arr(joiner_items) => {594 let mut out = Vec::new();595596 let mut first = true;597 for item in arr.iter() {598 let item = item?.clone();599 if let Val::Arr(items) = item {600 if !first {601 out.reserve(joiner_items.len());602 // TODO: extend603 for item in joiner_items.iter() {604 out.push(item?);605 }606 }607 first = false;608 out.reserve(items.len());609 // TODO: extend610 for item in items.iter() {611 out.push(item?);612 }613 } else {614 throw!(RuntimeError("in std.join all items should be arrays".into()));615 }616 }617618 Val::Arr(out.into())619 },620 Val::Str(sep) => {621 let mut out = String::new();622623 let mut first = true;624 for item in arr.iter() {625 let item = item?.clone();626 if let Val::Str(item) = item {627 if !first {628 out += &sep;629 }630 first = false;631 out += &item;632 } else {633 throw!(RuntimeError("in std.join all items should be strings".into()));634 }635 }636637 Val::Str(out.into())638 },639 _ => unreachable!()640 })641 })642}643644fn builtin_escape_string_json(645 context: Context,646 _loc: Option<&ExprLocation>,647 args: &ArgsDesc,648) -> Result<Val> {649 parse_args!(context, "escapeStringJson", args, 1, [650 0, str_: ty!(string) => Val::Str;651 ], {652 Ok(Val::Str(escape_string_json(&str_).into()))653 })654}655656fn builtin_manifest_json_ex(657 context: Context,658 _loc: Option<&ExprLocation>,659 args: &ArgsDesc,660) -> Result<Val> {661 parse_args!(context, "manifestJsonEx", args, 2, [662 0, value: ty!(any);663 1, indent: ty!(string) => Val::Str;664 ], {665 Ok(Val::Str(manifest_json_ex(&value, &ManifestJsonOptions {666 padding: &indent,667 mtype: ManifestType::Std,668 })?.into()))669 })670}671672fn builtin_reverse(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {673 parse_args!(context, "reverse", args, 1, [674 0, value: ty!(array) => Val::Arr;675 ], {676 Ok(Val::Arr(value.reversed()))677 })678}679680fn builtin_id(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {681 parse_args!(context, "id", args, 1, [682 0, v: ty!(any);683 ], {684 Ok(v)685 })686}687688fn builtin_str_replace(689 context: Context,690 _loc: Option<&ExprLocation>,691 args: &ArgsDesc,692) -> Result<Val> {693 parse_args!(context, "strReplace", args, 3, [694 0, str: ty!(string) => Val::Str;695 1, from: ty!(string) => Val::Str;696 2, to: ty!(string) => Val::Str;697 ], {698 let mut out = String::new();699 let mut last_idx = 0;700 while let Some(idx) = (&str[last_idx..]).find(&from as &str) {701 out.push_str(&str[last_idx..last_idx+idx]);702 out.push_str(&to);703 last_idx += idx + from.len();704 }705 if last_idx == 0 {706 return Ok(Val::Str(str))707 }708 out.push_str(&str[last_idx..]);709 Ok(Val::Str(out.into()))710 })711}712713pub fn call_builtin(714 context: Context,715 loc: Option<&ExprLocation>,716 name: &str,717 args: &ArgsDesc,718) -> Result<Val> {719 BUILTINS720 .with(|builtins| builtins.get(name).copied())721 .ok_or_else(|| IntrinsicNotFound(name.into()))?(context, loc, args)722}1use 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 ("primitiveEquals".into(), builtin_primitive_equals),85 ("equals".into(), builtin_equals),86 ("modulo".into(), builtin_modulo),87 ("mod".into(), builtin_mod),88 ("floor".into(), builtin_floor),89 ("ceil".into(), builtin_ceil),90 ("log".into(), builtin_log),91 ("pow".into(), builtin_pow),92 ("sqrt".into(), builtin_sqrt),93 ("sin".into(), builtin_sin),94 ("cos".into(), builtin_cos),95 ("tan".into(), builtin_tan),96 ("asin".into(), builtin_asin),97 ("acos".into(), builtin_acos),98 ("atan".into(), builtin_atan),99 ("exp".into(), builtin_exp),100 ("mantissa".into(), builtin_mantissa),101 ("exponent".into(), builtin_exponent),102 ("extVar".into(), builtin_ext_var),103 ("native".into(), builtin_native),104 ("filter".into(), builtin_filter),105 ("map".into(), builtin_map),106 ("flatMap".into(), builtin_flatmap),107 ("foldl".into(), builtin_foldl),108 ("foldr".into(), builtin_foldr),109 ("sortImpl".into(), builtin_sort_impl),110 ("format".into(), builtin_format),111 ("range".into(), builtin_range),112 ("char".into(), builtin_char),113 ("encodeUTF8".into(), builtin_encode_utf8),114 ("md5".into(), builtin_md5),115 ("base64".into(), builtin_base64),116 ("trace".into(), builtin_trace),117 ("join".into(), builtin_join),118 ("escapeStringJson".into(), builtin_escape_string_json),119 ("manifestJsonEx".into(), builtin_manifest_json_ex),120 ("reverse".into(), builtin_reverse),121 ("id".into(), builtin_id),122 ("strReplace".into(), builtin_str_replace),123 ("parseJson".into(), builtin_parse_json),124 ].iter().cloned().collect()125 };126}127128fn builtin_length(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {129 parse_args!(context, "length", args, 1, [130 0, x: ty!((string | object | array));131 ], {132 Ok(match x {133 Val::Str(n) => Val::Num(n.chars().count() as f64),134 Val::Arr(a) => Val::Num(a.len() as f64),135 Val::Obj(o) => Val::Num(136 o.fields_visibility()137 .into_iter()138 .filter(|(_k, v)| *v)139 .count() as f64,140 ),141 _ => unreachable!(),142 })143 })144}145146fn builtin_type(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {147 parse_args!(context, "type", args, 1, [148 0, x: ty!(any);149 ], {150 Ok(Val::Str(x.value_type().name().into()))151 })152}153154fn builtin_make_array(155 context: Context,156 _loc: Option<&ExprLocation>,157 args: &ArgsDesc,158) -> Result<Val> {159 parse_args!(context, "makeArray", args, 2, [160 0, sz: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;161 1, func: ty!(function) => Val::Func;162 ], {163 let mut out = Vec::with_capacity(sz as usize);164 for i in 0..sz as usize {165 out.push(LazyVal::new_resolved(func.evaluate_values(166 context.clone(),167 &[Val::Num(i as f64)]168 )?))169 }170 Ok(Val::Arr(out.into()))171 })172}173174fn builtin_codepoint(175 context: Context,176 _loc: Option<&ExprLocation>,177 args: &ArgsDesc,178) -> Result<Val> {179 parse_args!(context, "codepoint", args, 1, [180 0, str: ty!(char) => Val::Str;181 ], {182 Ok(Val::Num(str.chars().next().unwrap() as u32 as f64))183 })184}185186fn builtin_object_fields_ex(187 context: Context,188 _loc: Option<&ExprLocation>,189 args: &ArgsDesc,190) -> Result<Val> {191 parse_args!(context, "objectFieldsEx", args, 2, [192 0, obj: ty!(object) => Val::Obj;193 1, inc_hidden: ty!(boolean) => Val::Bool;194 ], {195 let out = obj.fields_ex(inc_hidden);196 Ok(Val::Arr(out.into_iter().map(Val::Str).collect::<Vec<_>>().into()))197 })198}199200fn builtin_object_has_ex(201 context: Context,202 _loc: Option<&ExprLocation>,203 args: &ArgsDesc,204) -> Result<Val> {205 parse_args!(context, "objectHasEx", args, 3, [206 0, obj: ty!(object) => Val::Obj;207 1, f: ty!(string) => Val::Str;208 2, inc_hidden: ty!(boolean) => Val::Bool;209 ], {210 Ok(Val::Bool(obj.has_field_ex(f, inc_hidden)))211 })212}213214fn builtin_parse_json(215 context: Context,216 _loc: Option<&ExprLocation>,217 args: &ArgsDesc,218) -> Result<Val> {219 parse_args!(context, "parseJson", args, 1, [220 0, s: ty!(string) => Val::Str;221 ], {222 let state = EvaluationState::default();223 let path = PathBuf::from("std.parseJson").into();224 state.evaluate_snippet_raw(path ,s)225 })226}227228fn builtin_slice(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {229 parse_args!(context, "slice", args, 4, [230 0, indexable: ty!((string | array));231 1, index: ty!((number | null));232 2, end: ty!((number | null));233 3, step: ty!((number | null));234 ], {235 std_slice(236 indexable.into_indexable()?,237 index.try_cast_nullable_num("index")?.map(|v| v as usize),238 end.try_cast_nullable_num("end")?.map(|v| v as usize),239 step.try_cast_nullable_num("step")?.map(|v| v as usize),240 )241 })242}243244fn builtin_primitive_equals(245 context: Context,246 _loc: Option<&ExprLocation>,247 args: &ArgsDesc,248) -> Result<Val> {249 parse_args!(context, "primitiveEquals", args, 2, [250 0, a: ty!(any);251 1, b: ty!(any);252 ], {253 Ok(Val::Bool(primitive_equals(&a, &b)?))254 })255}256257fn builtin_equals(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {258 parse_args!(context, "equals", args, 2, [259 0, a: ty!(any);260 1, b: ty!(any);261 ], {262 Ok(Val::Bool(equals(&a, &b)?))263 })264}265266fn builtin_modulo(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {267 parse_args!(context, "modulo", args, 2, [268 0, a: ty!(number) => Val::Num;269 1, b: ty!(number) => Val::Num;270 ], {271 Ok(Val::Num(a % b))272 })273}274275fn builtin_mod(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {276 parse_args!(context, "mod", args, 2, [277 0, a: ty!((number | string));278 1, b: ty!(any);279 ], {280 evaluate_mod_op(&a, &b)281 })282}283284fn builtin_floor(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {285 parse_args!(context, "floor", args, 1, [286 0, x: ty!(number) => Val::Num;287 ], {288 Ok(Val::Num(x.floor()))289 })290}291292fn builtin_ceil(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {293 parse_args!(context, "ceil", args, 1, [294 0, x: ty!(number) => Val::Num;295 ], {296 Ok(Val::Num(x.ceil()))297 })298}299300fn builtin_log(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {301 parse_args!(context, "log", args, 1, [302 0, n: ty!(number) => Val::Num;303 ], {304 Ok(Val::Num(n.ln()))305 })306}307308fn builtin_pow(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {309 parse_args!(context, "pow", args, 2, [310 0, x: ty!(number) => Val::Num;311 1, n: ty!(number) => Val::Num;312 ], {313 Ok(Val::Num(x.powf(n)))314 })315}316317fn builtin_sqrt(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {318 parse_args!(context, "sqrt", args, 1, [319 0, x: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;320 ], {321 Ok(Val::Num(x.sqrt()))322 })323}324325fn builtin_sin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {326 parse_args!(context, "sin", args, 1, [327 0, x: ty!(number) => Val::Num;328 ], {329 Ok(Val::Num(x.sin()))330 })331}332333fn builtin_cos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {334 parse_args!(context, "cos", args, 1, [335 0, x: ty!(number) => Val::Num;336 ], {337 Ok(Val::Num(x.cos()))338 })339}340341fn builtin_tan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {342 parse_args!(context, "tan", args, 1, [343 0, x: ty!(number) => Val::Num;344 ], {345 Ok(Val::Num(x.tan()))346 })347}348349fn builtin_asin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {350 parse_args!(context, "asin", args, 1, [351 0, x: ty!(number) => Val::Num;352 ], {353 Ok(Val::Num(x.asin()))354 })355}356357fn builtin_acos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {358 parse_args!(context, "acos", args, 1, [359 0, x: ty!(number) => Val::Num;360 ], {361 Ok(Val::Num(x.acos()))362 })363}364365fn builtin_atan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {366 parse_args!(context, "atan", args, 1, [367 0, x: ty!(number) => Val::Num;368 ], {369 Ok(Val::Num(x.atan()))370 })371}372373fn builtin_exp(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {374 parse_args!(context, "exp", args, 1, [375 0, x: ty!(number) => Val::Num;376 ], {377 Ok(Val::Num(x.exp()))378 })379}380381fn frexp(s: f64) -> (f64, i16) {382 if 0.0 == s {383 return (s, 0);384 } else {385 let lg = s.abs().log2();386 let x = (lg - lg.floor() - 1.0).exp2();387 let exp = lg.floor() + 1.0;388 (s.signum() * x, exp as i16)389 }390}391392fn builtin_mantissa(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {393 parse_args!(context, "mantissa", args, 1, [394 0, x: ty!(number) => Val::Num;395 ], {396 Ok(Val::Num(frexp(x).0))397 })398}399400fn builtin_exponent(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {401 parse_args!(context, "exponent", args, 1, [402 0, x: ty!(number) => Val::Num;403 ], {404 Ok(Val::Num(frexp(x).1.into()))405 })406}407408fn builtin_ext_var(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {409 parse_args!(context, "extVar", args, 1, [410 0, x: ty!(string) => Val::Str;411 ], {412 Ok(with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or(UndefinedExternalVariable(x))?)413 })414}415416fn builtin_native(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {417 parse_args!(context, "native", args, 1, [418 0, x: ty!(string) => Val::Str;419 ], {420 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))?)421 })422}423424fn builtin_filter(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {425 parse_args!(context, "filter", args, 2, [426 0, func: ty!(function) => Val::Func;427 1, arr: ty!(array) => Val::Arr;428 ], {429 Ok(Val::Arr(arr.filter(|val| func430 .evaluate_values(context.clone(), &[val.clone()])?431 .try_cast_bool("filter predicate"))?))432 })433}434435fn builtin_map(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {436 parse_args!(context, "map", args, 2, [437 0, func: ty!(function) => Val::Func;438 1, arr: ty!(array) => Val::Arr;439 ], {440 Ok(Val::Arr(arr.map(|val| func441 .evaluate_values(context.clone(), &[val]))?))442 })443}444445fn builtin_flatmap(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {446 parse_args!(context, "flatMap", args, 2, [447 0, func: ty!(function) => Val::Func;448 1, arr: ty!((array | string));449 ], {450 match arr {451 Val::Str(s) => {452 let mut out = String::new();453 for c in s.chars() {454 match func.evaluate_values(context.clone(), &[Val::Str(c.to_string().into())])? {455 Val::Str(o) => out.push_str(&o),456 _ => throw!(RuntimeError("in std.join all items should be strings".into())),457 };458 }459 Ok(Val::Str(out.into()))460 },461 Val::Arr(a) => {462 let mut out = Vec::new();463 for el in a.iter() {464 let el = el?;465 match func.evaluate_values(context.clone(), &[el])? {466 Val::Arr(o) => for oe in o.iter() {467 out.push(oe?)468 },469 _ => throw!(RuntimeError("in std.join all items should be arrays".into())),470 };471 }472 Ok(Val::Arr(out.into()))473 },474 _ => unreachable!(),475 }476 })477}478479fn builtin_foldl(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {480 parse_args!(context, "foldl", args, 3, [481 0, func: ty!(function) => Val::Func;482 1, arr: ty!(array) => Val::Arr;483 2, init: ty!(any);484 ], {485 let mut acc = init;486 for i in arr.iter() {487 acc = func.evaluate_values(context.clone(), &[acc, i?])?;488 }489 Ok(acc)490 })491}492493fn builtin_foldr(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {494 parse_args!(context, "foldr", args, 3, [495 0, func: ty!(function) => Val::Func;496 1, arr: ty!(array) => Val::Arr;497 2, init: ty!(any);498 ], {499 let mut acc = init;500 for i in arr.iter().rev() {501 acc = func.evaluate_values(context.clone(), &[acc, i?])?;502 }503 Ok(acc)504 })505}506507#[allow(non_snake_case)]508fn builtin_sort_impl(509 context: Context,510 _loc: Option<&ExprLocation>,511 args: &ArgsDesc,512) -> Result<Val> {513 parse_args!(context, "sort", args, 2, [514 0, arr: ty!(array) => Val::Arr;515 1, keyF: ty!(function) => Val::Func;516 ], {517 if arr.len() <= 1 {518 return Ok(Val::Arr(arr))519 }520 Ok(Val::Arr(ArrValue::Eager(sort::sort(context, arr.evaluated()?, &keyF)?)))521 })522}523524fn builtin_format(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {525 parse_args!(context, "format", args, 2, [526 0, str: ty!(string) => Val::Str;527 1, vals: ty!(any)528 ], {529 std_format(str, vals)530 })531}532533fn builtin_range(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {534 parse_args!(context, "range", args, 2, [535 0, from: ty!(number) => Val::Num;536 1, to: ty!(number) => Val::Num;537 ], {538 if to < from {539 return Ok(Val::Arr(ArrValue::new_eager()))540 }541 let mut out = Vec::with_capacity((1+to as usize-from as usize).max(0));542 for i in from as usize..=to as usize {543 out.push(Val::Num(i as f64));544 }545 Ok(Val::Arr(out.into()))546 })547}548549fn builtin_char(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {550 parse_args!(context, "char", args, 1, [551 0, n: ty!(number) => Val::Num;552 ], {553 let mut out = String::new();554 out.push(std::char::from_u32(n as u32).ok_or_else(||555 InvalidUnicodeCodepointGot(n as u32)556 )?);557 Ok(Val::Str(out.into()))558 })559}560561fn builtin_encode_utf8(562 context: Context,563 _loc: Option<&ExprLocation>,564 args: &ArgsDesc,565) -> Result<Val> {566 parse_args!(context, "encodeUTF8", args, 1, [567 0, str: ty!(string) => Val::Str;568 ], {569 Ok(Val::Arr((str.bytes().map(|b| Val::Num(b as f64)).collect::<Vec<Val>>()).into()))570 })571}572573fn builtin_md5(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {574 parse_args!(context, "md5", args, 1, [575 0, str: ty!(string) => Val::Str;576 ], {577 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))578 })579}580581fn builtin_trace(context: Context, loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {582 parse_args!(context, "trace", args, 2, [583 0, str: ty!(string) => Val::Str;584 1, rest: ty!(any);585 ], {586 eprint!("TRACE:");587 if let Some(loc) = loc {588 with_state(|s|{589 let locs = s.map_source_locations(&loc.0, &[loc.1]);590 eprint!(" {}:{}", loc.0.file_name().unwrap().to_str().unwrap(), locs[0].line);591 });592 }593 eprintln!(" {}", str);594 Ok(rest)595 })596}597598fn builtin_base64(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {599 parse_args!(context, "base64", args, 1, [600 0, input: ty!((string | (Array<number>)));601 ], {602 Ok(Val::Str(match input {603 Val::Str(s) => {604 base64::encode(s.bytes().collect::<Vec<_>>()).into()605 },606 Val::Arr(a) => {607 base64::encode(a.iter().map(|v| {608 Ok(v?.unwrap_num()? as u8)609 }).collect::<Result<Vec<_>>>()?).into()610 },611 _ => unreachable!()612 }))613 })614}615616fn builtin_join(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {617 parse_args!(context, "join", args, 2, [618 0, sep: ty!((string | array));619 1, arr: ty!(array) => Val::Arr;620 ], {621 Ok(match sep {622 Val::Arr(joiner_items) => {623 let mut out = Vec::new();624625 let mut first = true;626 for item in arr.iter() {627 let item = item?.clone();628 if let Val::Arr(items) = item {629 if !first {630 out.reserve(joiner_items.len());631 // TODO: extend632 for item in joiner_items.iter() {633 out.push(item?);634 }635 }636 first = false;637 out.reserve(items.len());638 // TODO: extend639 for item in items.iter() {640 out.push(item?);641 }642 } else {643 throw!(RuntimeError("in std.join all items should be arrays".into()));644 }645 }646647 Val::Arr(out.into())648 },649 Val::Str(sep) => {650 let mut out = String::new();651652 let mut first = true;653 for item in arr.iter() {654 let item = item?.clone();655 if let Val::Str(item) = item {656 if !first {657 out += &sep;658 }659 first = false;660 out += &item;661 } else {662 throw!(RuntimeError("in std.join all items should be strings".into()));663 }664 }665666 Val::Str(out.into())667 },668 _ => unreachable!()669 })670 })671}672673fn builtin_escape_string_json(674 context: Context,675 _loc: Option<&ExprLocation>,676 args: &ArgsDesc,677) -> Result<Val> {678 parse_args!(context, "escapeStringJson", args, 1, [679 0, str_: ty!(string) => Val::Str;680 ], {681 Ok(Val::Str(escape_string_json(&str_).into()))682 })683}684685fn builtin_manifest_json_ex(686 context: Context,687 _loc: Option<&ExprLocation>,688 args: &ArgsDesc,689) -> Result<Val> {690 parse_args!(context, "manifestJsonEx", args, 2, [691 0, value: ty!(any);692 1, indent: ty!(string) => Val::Str;693 ], {694 Ok(Val::Str(manifest_json_ex(&value, &ManifestJsonOptions {695 padding: &indent,696 mtype: ManifestType::Std,697 })?.into()))698 })699}700701fn builtin_reverse(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {702 parse_args!(context, "reverse", args, 1, [703 0, value: ty!(array) => Val::Arr;704 ], {705 Ok(Val::Arr(value.reversed()))706 })707}708709fn builtin_id(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {710 parse_args!(context, "id", args, 1, [711 0, v: ty!(any);712 ], {713 Ok(v)714 })715}716717fn builtin_str_replace(718 context: Context,719 _loc: Option<&ExprLocation>,720 args: &ArgsDesc,721) -> Result<Val> {722 parse_args!(context, "strReplace", args, 3, [723 0, str: ty!(string) => Val::Str;724 1, from: ty!(string) => Val::Str;725 2, to: ty!(string) => Val::Str;726 ], {727 let mut out = String::new();728 let mut last_idx = 0;729 while let Some(idx) = (&str[last_idx..]).find(&from as &str) {730 out.push_str(&str[last_idx..last_idx+idx]);731 out.push_str(&to);732 last_idx += idx + from.len();733 }734 if last_idx == 0 {735 return Ok(Val::Str(str))736 }737 out.push_str(&str[last_idx..]);738 Ok(Val::Str(out.into()))739 })740}741742pub fn call_builtin(743 context: Context,744 loc: Option<&ExprLocation>,745 name: &str,746 args: &ArgsDesc,747) -> Result<Val> {748 BUILTINS749 .with(|builtins| builtins.get(name).copied())750 .ok_or_else(|| IntrinsicNotFound(name.into()))?(context, loc, args)751}crates/jrsonnet-stdlib/src/std.jsonnetdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/std.jsonnet
+++ b/crates/jrsonnet-stdlib/src/std.jsonnet
@@ -35,6 +35,8 @@
atan:: $intrinsic(atan),
exp:: $intrinsic(exp),
+ mantissa:: $intrinsic(mantissa),
+ exponent:: $intrinsic(exponent),
isString(v):: std.type(v) == 'string',
isNumber(v):: std.type(v) == 'number',