git.delta.rocks / jrsonnet / refs/commits / 5e5ab6bd1d3a

difftreelog

Merge pull request #56 from messense/member-count-builtin

Yaroslav Bolyukin2021-07-14parents: #07ae295 #859b044.patch.diff
in: master

3 files changed

modifiedcrates/jrsonnet-evaluator/src/builtin/mod.rsdiffbeforeafterboth
before · crates/jrsonnet-evaluator/src/builtin/mod.rs
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			("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}
modifiedcrates/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)"#);
+	}
 }
modifiedcrates/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),