difftreelog
feat std.{sin,cos,tan,asin,acos,atan}
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 ("extVar".into(), builtin_ext_var),94 ("native".into(), builtin_native),95 ("filter".into(), builtin_filter),96 ("map".into(), builtin_map),97 ("flatMap".into(), builtin_flatmap),98 ("foldl".into(), builtin_foldl),99 ("foldr".into(), builtin_foldr),100 ("sortImpl".into(), builtin_sort_impl),101 ("format".into(), builtin_format),102 ("range".into(), builtin_range),103 ("char".into(), builtin_char),104 ("encodeUTF8".into(), builtin_encode_utf8),105 ("md5".into(), builtin_md5),106 ("base64".into(), builtin_base64),107 ("trace".into(), builtin_trace),108 ("join".into(), builtin_join),109 ("escapeStringJson".into(), builtin_escape_string_json),110 ("manifestJsonEx".into(), builtin_manifest_json_ex),111 ("reverse".into(), builtin_reverse),112 ("id".into(), builtin_id),113 ("strReplace".into(), builtin_str_replace),114 ("parseJson".into(), builtin_parse_json),115 ].iter().cloned().collect()116 };117}118119fn builtin_length(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {120 parse_args!(context, "length", args, 1, [121 0, x: ty!((string | object | array));122 ], {123 Ok(match x {124 Val::Str(n) => Val::Num(n.chars().count() as f64),125 Val::Arr(a) => Val::Num(a.len() as f64),126 Val::Obj(o) => Val::Num(127 o.fields_visibility()128 .into_iter()129 .filter(|(_k, v)| *v)130 .count() as f64,131 ),132 _ => unreachable!(),133 })134 })135}136137fn builtin_type(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {138 parse_args!(context, "type", args, 1, [139 0, x: ty!(any);140 ], {141 Ok(Val::Str(x.value_type().name().into()))142 })143}144145fn builtin_make_array(146 context: Context,147 _loc: Option<&ExprLocation>,148 args: &ArgsDesc,149) -> Result<Val> {150 parse_args!(context, "makeArray", args, 2, [151 0, sz: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;152 1, func: ty!(function) => Val::Func;153 ], {154 let mut out = Vec::with_capacity(sz as usize);155 for i in 0..sz as usize {156 out.push(LazyVal::new_resolved(func.evaluate_values(157 context.clone(),158 &[Val::Num(i as f64)]159 )?))160 }161 Ok(Val::Arr(out.into()))162 })163}164165fn builtin_codepoint(166 context: Context,167 _loc: Option<&ExprLocation>,168 args: &ArgsDesc,169) -> Result<Val> {170 parse_args!(context, "codepoint", args, 1, [171 0, str: ty!(char) => Val::Str;172 ], {173 Ok(Val::Num(str.chars().next().unwrap() as u32 as f64))174 })175}176177fn builtin_object_fields_ex(178 context: Context,179 _loc: Option<&ExprLocation>,180 args: &ArgsDesc,181) -> Result<Val> {182 parse_args!(context, "objectFieldsEx", args, 2, [183 0, obj: ty!(object) => Val::Obj;184 1, inc_hidden: ty!(boolean) => Val::Bool;185 ], {186 let out = obj.fields_ex(inc_hidden);187 Ok(Val::Arr(out.into_iter().map(Val::Str).collect::<Vec<_>>().into()))188 })189}190191fn builtin_object_has_ex(192 context: Context,193 _loc: Option<&ExprLocation>,194 args: &ArgsDesc,195) -> Result<Val> {196 parse_args!(context, "objectHasEx", args, 3, [197 0, obj: ty!(object) => Val::Obj;198 1, f: ty!(string) => Val::Str;199 2, inc_hidden: ty!(boolean) => Val::Bool;200 ], {201 Ok(Val::Bool(obj.has_field_ex(f, inc_hidden)))202 })203}204205fn builtin_parse_json(206 context: Context,207 _loc: Option<&ExprLocation>,208 args: &ArgsDesc,209) -> Result<Val> {210 parse_args!(context, "parseJson", args, 1, [211 0, s: ty!(string) => Val::Str;212 ], {213 let state = EvaluationState::default();214 let path = PathBuf::from("std.parseJson").into();215 state.evaluate_snippet_raw(path ,s)216 })217}218219fn builtin_slice(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {220 parse_args!(context, "slice", args, 4, [221 0, indexable: ty!((string | array));222 1, index: ty!((number | null));223 2, end: ty!((number | null));224 3, step: ty!((number | null));225 ], {226 std_slice(227 indexable.into_indexable()?,228 index.try_cast_nullable_num("index")?.map(|v| v as usize),229 end.try_cast_nullable_num("end")?.map(|v| v as usize),230 step.try_cast_nullable_num("step")?.map(|v| v as usize),231 )232 })233}234235fn builtin_primitive_equals(236 context: Context,237 _loc: Option<&ExprLocation>,238 args: &ArgsDesc,239) -> Result<Val> {240 parse_args!(context, "primitiveEquals", args, 2, [241 0, a: ty!(any);242 1, b: ty!(any);243 ], {244 Ok(Val::Bool(primitive_equals(&a, &b)?))245 })246}247248fn builtin_equals(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {249 parse_args!(context, "equals", args, 2, [250 0, a: ty!(any);251 1, b: ty!(any);252 ], {253 Ok(Val::Bool(equals(&a, &b)?))254 })255}256257fn builtin_modulo(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {258 parse_args!(context, "modulo", args, 2, [259 0, a: ty!(number) => Val::Num;260 1, b: ty!(number) => Val::Num;261 ], {262 Ok(Val::Num(a % b))263 })264}265266fn builtin_mod(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {267 parse_args!(context, "mod", args, 2, [268 0, a: ty!((number | string));269 1, b: ty!(any);270 ], {271 evaluate_mod_op(&a, &b)272 })273}274275fn builtin_floor(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {276 parse_args!(context, "floor", args, 1, [277 0, x: ty!(number) => Val::Num;278 ], {279 Ok(Val::Num(x.floor()))280 })281}282283fn builtin_ceil(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {284 parse_args!(context, "ceil", args, 1, [285 0, x: ty!(number) => Val::Num;286 ], {287 Ok(Val::Num(x.ceil()))288 })289}290291fn builtin_log(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {292 parse_args!(context, "log", args, 1, [293 0, n: ty!(number) => Val::Num;294 ], {295 Ok(Val::Num(n.ln()))296 })297}298299fn builtin_pow(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {300 parse_args!(context, "pow", args, 2, [301 0, x: ty!(number) => Val::Num;302 1, n: ty!(number) => Val::Num;303 ], {304 Ok(Val::Num(x.powf(n)))305 })306}307308fn builtin_sqrt(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {309 parse_args!(context, "sqrt", args, 1, [310 0, x: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;311 ], {312 Ok(Val::Num(x.sqrt()))313 })314}315316fn builtin_ext_var(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {317 parse_args!(context, "extVar", args, 1, [318 0, x: ty!(string) => Val::Str;319 ], {320 Ok(with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or(UndefinedExternalVariable(x))?)321 })322}323324fn builtin_native(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {325 parse_args!(context, "native", args, 1, [326 0, x: ty!(string) => Val::Str;327 ], {328 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))?)329 })330}331332fn builtin_filter(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {333 parse_args!(context, "filter", args, 2, [334 0, func: ty!(function) => Val::Func;335 1, arr: ty!(array) => Val::Arr;336 ], {337 Ok(Val::Arr(arr.filter(|val| func338 .evaluate_values(context.clone(), &[val.clone()])?339 .try_cast_bool("filter predicate"))?))340 })341}342343fn builtin_map(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {344 parse_args!(context, "map", args, 2, [345 0, func: ty!(function) => Val::Func;346 1, arr: ty!(array) => Val::Arr;347 ], {348 Ok(Val::Arr(arr.map(|val| func349 .evaluate_values(context.clone(), &[val]))?))350 })351}352353fn builtin_flatmap(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {354 parse_args!(context, "flatMap", args, 2, [355 0, func: ty!(function) => Val::Func;356 1, arr: ty!((array | string));357 ], {358 match arr {359 Val::Str(s) => {360 let mut out = String::new();361 for c in s.chars() {362 match func.evaluate_values(context.clone(), &[Val::Str(c.to_string().into())])? {363 Val::Str(o) => out.push_str(&o),364 _ => throw!(RuntimeError("in std.join all items should be strings".into())),365 };366 }367 Ok(Val::Str(out.into()))368 },369 Val::Arr(a) => {370 let mut out = Vec::new();371 for el in a.iter() {372 let el = el?;373 match func.evaluate_values(context.clone(), &[el])? {374 Val::Arr(o) => for oe in o.iter() {375 out.push(oe?)376 },377 _ => throw!(RuntimeError("in std.join all items should be arrays".into())),378 };379 }380 Ok(Val::Arr(out.into()))381 },382 _ => unreachable!(),383 }384 })385}386387fn builtin_foldl(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {388 parse_args!(context, "foldl", args, 3, [389 0, func: ty!(function) => Val::Func;390 1, arr: ty!(array) => Val::Arr;391 2, init: ty!(any);392 ], {393 let mut acc = init;394 for i in arr.iter() {395 acc = func.evaluate_values(context.clone(), &[acc, i?])?;396 }397 Ok(acc)398 })399}400401fn builtin_foldr(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {402 parse_args!(context, "foldr", args, 3, [403 0, func: ty!(function) => Val::Func;404 1, arr: ty!(array) => Val::Arr;405 2, init: ty!(any);406 ], {407 let mut acc = init;408 for i in arr.iter().rev() {409 acc = func.evaluate_values(context.clone(), &[acc, i?])?;410 }411 Ok(acc)412 })413}414415#[allow(non_snake_case)]416fn builtin_sort_impl(417 context: Context,418 _loc: Option<&ExprLocation>,419 args: &ArgsDesc,420) -> Result<Val> {421 parse_args!(context, "sort", args, 2, [422 0, arr: ty!(array) => Val::Arr;423 1, keyF: ty!(function) => Val::Func;424 ], {425 if arr.len() <= 1 {426 return Ok(Val::Arr(arr))427 }428 Ok(Val::Arr(ArrValue::Eager(sort::sort(context, arr.evaluated()?, &keyF)?)))429 })430}431432fn builtin_format(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {433 parse_args!(context, "format", args, 2, [434 0, str: ty!(string) => Val::Str;435 1, vals: ty!(any)436 ], {437 std_format(str, vals)438 })439}440441fn builtin_range(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {442 parse_args!(context, "range", args, 2, [443 0, from: ty!(number) => Val::Num;444 1, to: ty!(number) => Val::Num;445 ], {446 if to < from {447 return Ok(Val::Arr(ArrValue::new_eager()))448 }449 let mut out = Vec::with_capacity((1+to as usize-from as usize).max(0));450 for i in from as usize..=to as usize {451 out.push(Val::Num(i as f64));452 }453 Ok(Val::Arr(out.into()))454 })455}456457fn builtin_char(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {458 parse_args!(context, "char", args, 1, [459 0, n: ty!(number) => Val::Num;460 ], {461 let mut out = String::new();462 out.push(std::char::from_u32(n as u32).ok_or_else(||463 InvalidUnicodeCodepointGot(n as u32)464 )?);465 Ok(Val::Str(out.into()))466 })467}468469fn builtin_encode_utf8(470 context: Context,471 _loc: Option<&ExprLocation>,472 args: &ArgsDesc,473) -> Result<Val> {474 parse_args!(context, "encodeUTF8", args, 1, [475 0, str: ty!(string) => Val::Str;476 ], {477 Ok(Val::Arr((str.bytes().map(|b| Val::Num(b as f64)).collect::<Vec<Val>>()).into()))478 })479}480481fn builtin_md5(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {482 parse_args!(context, "md5", args, 1, [483 0, str: ty!(string) => Val::Str;484 ], {485 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))486 })487}488489fn builtin_trace(context: Context, loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {490 parse_args!(context, "trace", args, 2, [491 0, str: ty!(string) => Val::Str;492 1, rest: ty!(any);493 ], {494 eprint!("TRACE:");495 if let Some(loc) = loc {496 with_state(|s|{497 let locs = s.map_source_locations(&loc.0, &[loc.1]);498 eprint!(" {}:{}", loc.0.file_name().unwrap().to_str().unwrap(), locs[0].line);499 });500 }501 eprintln!(" {}", str);502 Ok(rest)503 })504}505506fn builtin_base64(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {507 parse_args!(context, "base64", args, 1, [508 0, input: ty!((string | (Array<number>)));509 ], {510 Ok(Val::Str(match input {511 Val::Str(s) => {512 base64::encode(s.bytes().collect::<Vec<_>>()).into()513 },514 Val::Arr(a) => {515 base64::encode(a.iter().map(|v| {516 Ok(v?.unwrap_num()? as u8)517 }).collect::<Result<Vec<_>>>()?).into()518 },519 _ => unreachable!()520 }))521 })522}523524fn builtin_join(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {525 parse_args!(context, "join", args, 2, [526 0, sep: ty!((string | array));527 1, arr: ty!(array) => Val::Arr;528 ], {529 Ok(match sep {530 Val::Arr(joiner_items) => {531 let mut out = Vec::new();532533 let mut first = true;534 for item in arr.iter() {535 let item = item?.clone();536 if let Val::Arr(items) = item {537 if !first {538 out.reserve(joiner_items.len());539 // TODO: extend540 for item in joiner_items.iter() {541 out.push(item?);542 }543 }544 first = false;545 out.reserve(items.len());546 // TODO: extend547 for item in items.iter() {548 out.push(item?);549 }550 } else {551 throw!(RuntimeError("in std.join all items should be arrays".into()));552 }553 }554555 Val::Arr(out.into())556 },557 Val::Str(sep) => {558 let mut out = String::new();559560 let mut first = true;561 for item in arr.iter() {562 let item = item?.clone();563 if let Val::Str(item) = item {564 if !first {565 out += &sep;566 }567 first = false;568 out += &item;569 } else {570 throw!(RuntimeError("in std.join all items should be strings".into()));571 }572 }573574 Val::Str(out.into())575 },576 _ => unreachable!()577 })578 })579}580581fn builtin_escape_string_json(582 context: Context,583 _loc: Option<&ExprLocation>,584 args: &ArgsDesc,585) -> Result<Val> {586 parse_args!(context, "escapeStringJson", args, 1, [587 0, str_: ty!(string) => Val::Str;588 ], {589 Ok(Val::Str(escape_string_json(&str_).into()))590 })591}592593fn builtin_manifest_json_ex(594 context: Context,595 _loc: Option<&ExprLocation>,596 args: &ArgsDesc,597) -> Result<Val> {598 parse_args!(context, "manifestJsonEx", args, 2, [599 0, value: ty!(any);600 1, indent: ty!(string) => Val::Str;601 ], {602 Ok(Val::Str(manifest_json_ex(&value, &ManifestJsonOptions {603 padding: &indent,604 mtype: ManifestType::Std,605 })?.into()))606 })607}608609fn builtin_reverse(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {610 parse_args!(context, "reverse", args, 1, [611 0, value: ty!(array) => Val::Arr;612 ], {613 Ok(Val::Arr(value.reversed()))614 })615}616617fn builtin_id(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {618 parse_args!(context, "id", args, 1, [619 0, v: ty!(any);620 ], {621 Ok(v)622 })623}624625fn builtin_str_replace(626 context: Context,627 _loc: Option<&ExprLocation>,628 args: &ArgsDesc,629) -> Result<Val> {630 parse_args!(context, "strReplace", args, 3, [631 0, str: ty!(string) => Val::Str;632 1, from: ty!(string) => Val::Str;633 2, to: ty!(string) => Val::Str;634 ], {635 let mut out = String::new();636 let mut last_idx = 0;637 while let Some(idx) = (&str[last_idx..]).find(&from as &str) {638 out.push_str(&str[last_idx..last_idx+idx]);639 out.push_str(&to);640 last_idx += idx + from.len();641 }642 if last_idx == 0 {643 return Ok(Val::Str(str))644 }645 out.push_str(&str[last_idx..]);646 Ok(Val::Str(out.into()))647 })648}649650pub fn call_builtin(651 context: Context,652 loc: Option<&ExprLocation>,653 name: &str,654 args: &ArgsDesc,655) -> Result<Val> {656 BUILTINS657 .with(|builtins| builtins.get(name).copied())658 .ok_or_else(|| IntrinsicNotFound(name.into()))?(context, loc, args)659}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 ("extVar".into(), builtin_ext_var),100 ("native".into(), builtin_native),101 ("filter".into(), builtin_filter),102 ("map".into(), builtin_map),103 ("flatMap".into(), builtin_flatmap),104 ("foldl".into(), builtin_foldl),105 ("foldr".into(), builtin_foldr),106 ("sortImpl".into(), builtin_sort_impl),107 ("format".into(), builtin_format),108 ("range".into(), builtin_range),109 ("char".into(), builtin_char),110 ("encodeUTF8".into(), builtin_encode_utf8),111 ("md5".into(), builtin_md5),112 ("base64".into(), builtin_base64),113 ("trace".into(), builtin_trace),114 ("join".into(), builtin_join),115 ("escapeStringJson".into(), builtin_escape_string_json),116 ("manifestJsonEx".into(), builtin_manifest_json_ex),117 ("reverse".into(), builtin_reverse),118 ("id".into(), builtin_id),119 ("strReplace".into(), builtin_str_replace),120 ("parseJson".into(), builtin_parse_json),121 ].iter().cloned().collect()122 };123}124125fn builtin_length(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {126 parse_args!(context, "length", args, 1, [127 0, x: ty!((string | object | array));128 ], {129 Ok(match x {130 Val::Str(n) => Val::Num(n.chars().count() as f64),131 Val::Arr(a) => Val::Num(a.len() as f64),132 Val::Obj(o) => Val::Num(133 o.fields_visibility()134 .into_iter()135 .filter(|(_k, v)| *v)136 .count() as f64,137 ),138 _ => unreachable!(),139 })140 })141}142143fn builtin_type(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {144 parse_args!(context, "type", args, 1, [145 0, x: ty!(any);146 ], {147 Ok(Val::Str(x.value_type().name().into()))148 })149}150151fn builtin_make_array(152 context: Context,153 _loc: Option<&ExprLocation>,154 args: &ArgsDesc,155) -> Result<Val> {156 parse_args!(context, "makeArray", args, 2, [157 0, sz: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;158 1, func: ty!(function) => Val::Func;159 ], {160 let mut out = Vec::with_capacity(sz as usize);161 for i in 0..sz as usize {162 out.push(LazyVal::new_resolved(func.evaluate_values(163 context.clone(),164 &[Val::Num(i as f64)]165 )?))166 }167 Ok(Val::Arr(out.into()))168 })169}170171fn builtin_codepoint(172 context: Context,173 _loc: Option<&ExprLocation>,174 args: &ArgsDesc,175) -> Result<Val> {176 parse_args!(context, "codepoint", args, 1, [177 0, str: ty!(char) => Val::Str;178 ], {179 Ok(Val::Num(str.chars().next().unwrap() as u32 as f64))180 })181}182183fn builtin_object_fields_ex(184 context: Context,185 _loc: Option<&ExprLocation>,186 args: &ArgsDesc,187) -> Result<Val> {188 parse_args!(context, "objectFieldsEx", args, 2, [189 0, obj: ty!(object) => Val::Obj;190 1, inc_hidden: ty!(boolean) => Val::Bool;191 ], {192 let out = obj.fields_ex(inc_hidden);193 Ok(Val::Arr(out.into_iter().map(Val::Str).collect::<Vec<_>>().into()))194 })195}196197fn builtin_object_has_ex(198 context: Context,199 _loc: Option<&ExprLocation>,200 args: &ArgsDesc,201) -> Result<Val> {202 parse_args!(context, "objectHasEx", args, 3, [203 0, obj: ty!(object) => Val::Obj;204 1, f: ty!(string) => Val::Str;205 2, inc_hidden: ty!(boolean) => Val::Bool;206 ], {207 Ok(Val::Bool(obj.has_field_ex(f, inc_hidden)))208 })209}210211fn builtin_parse_json(212 context: Context,213 _loc: Option<&ExprLocation>,214 args: &ArgsDesc,215) -> Result<Val> {216 parse_args!(context, "parseJson", args, 1, [217 0, s: ty!(string) => Val::Str;218 ], {219 let state = EvaluationState::default();220 let path = PathBuf::from("std.parseJson").into();221 state.evaluate_snippet_raw(path ,s)222 })223}224225fn builtin_slice(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {226 parse_args!(context, "slice", args, 4, [227 0, indexable: ty!((string | array));228 1, index: ty!((number | null));229 2, end: ty!((number | null));230 3, step: ty!((number | null));231 ], {232 std_slice(233 indexable.into_indexable()?,234 index.try_cast_nullable_num("index")?.map(|v| v as usize),235 end.try_cast_nullable_num("end")?.map(|v| v as usize),236 step.try_cast_nullable_num("step")?.map(|v| v as usize),237 )238 })239}240241fn builtin_primitive_equals(242 context: Context,243 _loc: Option<&ExprLocation>,244 args: &ArgsDesc,245) -> Result<Val> {246 parse_args!(context, "primitiveEquals", args, 2, [247 0, a: ty!(any);248 1, b: ty!(any);249 ], {250 Ok(Val::Bool(primitive_equals(&a, &b)?))251 })252}253254fn builtin_equals(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {255 parse_args!(context, "equals", args, 2, [256 0, a: ty!(any);257 1, b: ty!(any);258 ], {259 Ok(Val::Bool(equals(&a, &b)?))260 })261}262263fn builtin_modulo(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {264 parse_args!(context, "modulo", args, 2, [265 0, a: ty!(number) => Val::Num;266 1, b: ty!(number) => Val::Num;267 ], {268 Ok(Val::Num(a % b))269 })270}271272fn builtin_mod(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {273 parse_args!(context, "mod", args, 2, [274 0, a: ty!((number | string));275 1, b: ty!(any);276 ], {277 evaluate_mod_op(&a, &b)278 })279}280281fn builtin_floor(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {282 parse_args!(context, "floor", args, 1, [283 0, x: ty!(number) => Val::Num;284 ], {285 Ok(Val::Num(x.floor()))286 })287}288289fn builtin_ceil(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {290 parse_args!(context, "ceil", args, 1, [291 0, x: ty!(number) => Val::Num;292 ], {293 Ok(Val::Num(x.ceil()))294 })295}296297fn builtin_log(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {298 parse_args!(context, "log", args, 1, [299 0, n: ty!(number) => Val::Num;300 ], {301 Ok(Val::Num(n.ln()))302 })303}304305fn builtin_pow(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {306 parse_args!(context, "pow", args, 2, [307 0, x: ty!(number) => Val::Num;308 1, n: ty!(number) => Val::Num;309 ], {310 Ok(Val::Num(x.powf(n)))311 })312}313314fn builtin_sqrt(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {315 parse_args!(context, "sqrt", args, 1, [316 0, x: ty!(BoundedNumber<(Some(0.0)), (None)>) => Val::Num;317 ], {318 Ok(Val::Num(x.sqrt()))319 })320}321322fn builtin_sin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {323 parse_args!(context, "sin", args, 1, [324 0, x: ty!(number) => Val::Num;325 ], {326 Ok(Val::Num(x.sin()))327 })328}329330fn builtin_cos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {331 parse_args!(context, "cos", args, 1, [332 0, x: ty!(number) => Val::Num;333 ], {334 Ok(Val::Num(x.cos()))335 })336}337338fn builtin_tan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {339 parse_args!(context, "tan", args, 1, [340 0, x: ty!(number) => Val::Num;341 ], {342 Ok(Val::Num(x.tan()))343 })344}345346fn builtin_asin(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {347 parse_args!(context, "asin", args, 1, [348 0, x: ty!(number) => Val::Num;349 ], {350 Ok(Val::Num(x.asin()))351 })352}353354fn builtin_acos(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {355 parse_args!(context, "acos", args, 1, [356 0, x: ty!(number) => Val::Num;357 ], {358 Ok(Val::Num(x.acos()))359 })360}361362fn builtin_atan(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {363 parse_args!(context, "atan", args, 1, [364 0, x: ty!(number) => Val::Num;365 ], {366 Ok(Val::Num(x.atan()))367 })368}369370fn builtin_ext_var(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {371 parse_args!(context, "extVar", args, 1, [372 0, x: ty!(string) => Val::Str;373 ], {374 Ok(with_state(|s| s.settings().ext_vars.get(&x).cloned()).ok_or(UndefinedExternalVariable(x))?)375 })376}377378fn builtin_native(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {379 parse_args!(context, "native", args, 1, [380 0, x: ty!(string) => Val::Str;381 ], {382 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))?)383 })384}385386fn builtin_filter(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {387 parse_args!(context, "filter", args, 2, [388 0, func: ty!(function) => Val::Func;389 1, arr: ty!(array) => Val::Arr;390 ], {391 Ok(Val::Arr(arr.filter(|val| func392 .evaluate_values(context.clone(), &[val.clone()])?393 .try_cast_bool("filter predicate"))?))394 })395}396397fn builtin_map(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {398 parse_args!(context, "map", args, 2, [399 0, func: ty!(function) => Val::Func;400 1, arr: ty!(array) => Val::Arr;401 ], {402 Ok(Val::Arr(arr.map(|val| func403 .evaluate_values(context.clone(), &[val]))?))404 })405}406407fn builtin_flatmap(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {408 parse_args!(context, "flatMap", args, 2, [409 0, func: ty!(function) => Val::Func;410 1, arr: ty!((array | string));411 ], {412 match arr {413 Val::Str(s) => {414 let mut out = String::new();415 for c in s.chars() {416 match func.evaluate_values(context.clone(), &[Val::Str(c.to_string().into())])? {417 Val::Str(o) => out.push_str(&o),418 _ => throw!(RuntimeError("in std.join all items should be strings".into())),419 };420 }421 Ok(Val::Str(out.into()))422 },423 Val::Arr(a) => {424 let mut out = Vec::new();425 for el in a.iter() {426 let el = el?;427 match func.evaluate_values(context.clone(), &[el])? {428 Val::Arr(o) => for oe in o.iter() {429 out.push(oe?)430 },431 _ => throw!(RuntimeError("in std.join all items should be arrays".into())),432 };433 }434 Ok(Val::Arr(out.into()))435 },436 _ => unreachable!(),437 }438 })439}440441fn builtin_foldl(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {442 parse_args!(context, "foldl", args, 3, [443 0, func: ty!(function) => Val::Func;444 1, arr: ty!(array) => Val::Arr;445 2, init: ty!(any);446 ], {447 let mut acc = init;448 for i in arr.iter() {449 acc = func.evaluate_values(context.clone(), &[acc, i?])?;450 }451 Ok(acc)452 })453}454455fn builtin_foldr(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {456 parse_args!(context, "foldr", args, 3, [457 0, func: ty!(function) => Val::Func;458 1, arr: ty!(array) => Val::Arr;459 2, init: ty!(any);460 ], {461 let mut acc = init;462 for i in arr.iter().rev() {463 acc = func.evaluate_values(context.clone(), &[acc, i?])?;464 }465 Ok(acc)466 })467}468469#[allow(non_snake_case)]470fn builtin_sort_impl(471 context: Context,472 _loc: Option<&ExprLocation>,473 args: &ArgsDesc,474) -> Result<Val> {475 parse_args!(context, "sort", args, 2, [476 0, arr: ty!(array) => Val::Arr;477 1, keyF: ty!(function) => Val::Func;478 ], {479 if arr.len() <= 1 {480 return Ok(Val::Arr(arr))481 }482 Ok(Val::Arr(ArrValue::Eager(sort::sort(context, arr.evaluated()?, &keyF)?)))483 })484}485486fn builtin_format(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {487 parse_args!(context, "format", args, 2, [488 0, str: ty!(string) => Val::Str;489 1, vals: ty!(any)490 ], {491 std_format(str, vals)492 })493}494495fn builtin_range(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {496 parse_args!(context, "range", args, 2, [497 0, from: ty!(number) => Val::Num;498 1, to: ty!(number) => Val::Num;499 ], {500 if to < from {501 return Ok(Val::Arr(ArrValue::new_eager()))502 }503 let mut out = Vec::with_capacity((1+to as usize-from as usize).max(0));504 for i in from as usize..=to as usize {505 out.push(Val::Num(i as f64));506 }507 Ok(Val::Arr(out.into()))508 })509}510511fn builtin_char(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {512 parse_args!(context, "char", args, 1, [513 0, n: ty!(number) => Val::Num;514 ], {515 let mut out = String::new();516 out.push(std::char::from_u32(n as u32).ok_or_else(||517 InvalidUnicodeCodepointGot(n as u32)518 )?);519 Ok(Val::Str(out.into()))520 })521}522523fn builtin_encode_utf8(524 context: Context,525 _loc: Option<&ExprLocation>,526 args: &ArgsDesc,527) -> Result<Val> {528 parse_args!(context, "encodeUTF8", args, 1, [529 0, str: ty!(string) => Val::Str;530 ], {531 Ok(Val::Arr((str.bytes().map(|b| Val::Num(b as f64)).collect::<Vec<Val>>()).into()))532 })533}534535fn builtin_md5(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {536 parse_args!(context, "md5", args, 1, [537 0, str: ty!(string) => Val::Str;538 ], {539 Ok(Val::Str(format!("{:x}", md5::compute(&str.as_bytes())).into()))540 })541}542543fn builtin_trace(context: Context, loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {544 parse_args!(context, "trace", args, 2, [545 0, str: ty!(string) => Val::Str;546 1, rest: ty!(any);547 ], {548 eprint!("TRACE:");549 if let Some(loc) = loc {550 with_state(|s|{551 let locs = s.map_source_locations(&loc.0, &[loc.1]);552 eprint!(" {}:{}", loc.0.file_name().unwrap().to_str().unwrap(), locs[0].line);553 });554 }555 eprintln!(" {}", str);556 Ok(rest)557 })558}559560fn builtin_base64(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {561 parse_args!(context, "base64", args, 1, [562 0, input: ty!((string | (Array<number>)));563 ], {564 Ok(Val::Str(match input {565 Val::Str(s) => {566 base64::encode(s.bytes().collect::<Vec<_>>()).into()567 },568 Val::Arr(a) => {569 base64::encode(a.iter().map(|v| {570 Ok(v?.unwrap_num()? as u8)571 }).collect::<Result<Vec<_>>>()?).into()572 },573 _ => unreachable!()574 }))575 })576}577578fn builtin_join(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {579 parse_args!(context, "join", args, 2, [580 0, sep: ty!((string | array));581 1, arr: ty!(array) => Val::Arr;582 ], {583 Ok(match sep {584 Val::Arr(joiner_items) => {585 let mut out = Vec::new();586587 let mut first = true;588 for item in arr.iter() {589 let item = item?.clone();590 if let Val::Arr(items) = item {591 if !first {592 out.reserve(joiner_items.len());593 // TODO: extend594 for item in joiner_items.iter() {595 out.push(item?);596 }597 }598 first = false;599 out.reserve(items.len());600 // TODO: extend601 for item in items.iter() {602 out.push(item?);603 }604 } else {605 throw!(RuntimeError("in std.join all items should be arrays".into()));606 }607 }608609 Val::Arr(out.into())610 },611 Val::Str(sep) => {612 let mut out = String::new();613614 let mut first = true;615 for item in arr.iter() {616 let item = item?.clone();617 if let Val::Str(item) = item {618 if !first {619 out += &sep;620 }621 first = false;622 out += &item;623 } else {624 throw!(RuntimeError("in std.join all items should be strings".into()));625 }626 }627628 Val::Str(out.into())629 },630 _ => unreachable!()631 })632 })633}634635fn builtin_escape_string_json(636 context: Context,637 _loc: Option<&ExprLocation>,638 args: &ArgsDesc,639) -> Result<Val> {640 parse_args!(context, "escapeStringJson", args, 1, [641 0, str_: ty!(string) => Val::Str;642 ], {643 Ok(Val::Str(escape_string_json(&str_).into()))644 })645}646647fn builtin_manifest_json_ex(648 context: Context,649 _loc: Option<&ExprLocation>,650 args: &ArgsDesc,651) -> Result<Val> {652 parse_args!(context, "manifestJsonEx", args, 2, [653 0, value: ty!(any);654 1, indent: ty!(string) => Val::Str;655 ], {656 Ok(Val::Str(manifest_json_ex(&value, &ManifestJsonOptions {657 padding: &indent,658 mtype: ManifestType::Std,659 })?.into()))660 })661}662663fn builtin_reverse(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {664 parse_args!(context, "reverse", args, 1, [665 0, value: ty!(array) => Val::Arr;666 ], {667 Ok(Val::Arr(value.reversed()))668 })669}670671fn builtin_id(context: Context, _loc: Option<&ExprLocation>, args: &ArgsDesc) -> Result<Val> {672 parse_args!(context, "id", args, 1, [673 0, v: ty!(any);674 ], {675 Ok(v)676 })677}678679fn builtin_str_replace(680 context: Context,681 _loc: Option<&ExprLocation>,682 args: &ArgsDesc,683) -> Result<Val> {684 parse_args!(context, "strReplace", args, 3, [685 0, str: ty!(string) => Val::Str;686 1, from: ty!(string) => Val::Str;687 2, to: ty!(string) => Val::Str;688 ], {689 let mut out = String::new();690 let mut last_idx = 0;691 while let Some(idx) = (&str[last_idx..]).find(&from as &str) {692 out.push_str(&str[last_idx..last_idx+idx]);693 out.push_str(&to);694 last_idx += idx + from.len();695 }696 if last_idx == 0 {697 return Ok(Val::Str(str))698 }699 out.push_str(&str[last_idx..]);700 Ok(Val::Str(out.into()))701 })702}703704pub fn call_builtin(705 context: Context,706 loc: Option<&ExprLocation>,707 name: &str,708 args: &ArgsDesc,709) -> Result<Val> {710 BUILTINS711 .with(|builtins| builtins.get(name).copied())712 .ok_or_else(|| IntrinsicNotFound(name.into()))?(context, loc, args)713}crates/jrsonnet-stdlib/src/std.jsonnetdiffbeforeafterboth--- a/crates/jrsonnet-stdlib/src/std.jsonnet
+++ b/crates/jrsonnet-stdlib/src/std.jsonnet
@@ -27,6 +27,13 @@
pow:: $intrinsic(pow),
sqrt:: $intrinsic(sqrt),
+ sin:: $intrinsic(sin),
+ cos:: $intrinsic(cos),
+ tan:: $intrinsic(tan),
+ asin:: $intrinsic(asin),
+ acos:: $intrinsic(acos),
+ atan:: $intrinsic(atan),
+
isString(v):: std.type(v) == 'string',
isNumber(v):: std.type(v) == 'number',
isBoolean(v):: std.type(v) == 'boolean',