git.delta.rocks / jrsonnet / refs/commits / 609e5ef85d49

1use std::{cell::RefCell, rc::Rc};23use jrsonnet_evaluator::{4	bail,5	error::{ErrorKind::*, Result},6	function::{builtin, ArgLike, CallLocation, FuncVal},7	manifest::JsonFormat,8	typed::{Either2, Either4},9	val::{equals, ArrValue},10	Context, Either, IStr, ObjValue, Thunk, Val,11};1213use crate::{extvar_source, Settings};1415#[builtin]16pub fn builtin_length(x: Either![IStr, ArrValue, ObjValue, FuncVal]) -> usize {17	use Either4::*;18	match x {19		A(x) => x.chars().count(),20		B(x) => x.len(),21		C(x) => x.len(),22		D(f) => f.params_len(),23	}24}2526#[builtin(fields(27	settings: Rc<RefCell<Settings>>,28))]29pub fn builtin_ext_var(this: &builtin_ext_var, ctx: Context, x: IStr) -> Result<Val> {30	let ctx = ctx.state().create_default_context(extvar_source(&x, ""));31	this.settings32		.borrow()33		.ext_vars34		.get(&x)35		.cloned()36		.ok_or_else(|| UndefinedExternalVariable(x))?37		.evaluate_arg(ctx, true)?38		.evaluate()39}4041#[builtin(fields(42	settings: Rc<RefCell<Settings>>,43))]44pub fn builtin_native(this: &builtin_native, x: IStr) -> Val {45	this.settings46		.borrow()47		.ext_natives48		.get(&x)49		.cloned()50		.map_or(Val::Null, |v| Val::Func(v))51}5253#[builtin(fields(54	settings: Rc<RefCell<Settings>>,55))]56pub fn builtin_trace(57	this: &builtin_trace,58	loc: CallLocation,59	str: Val,60	rest: Option<Thunk<Val>>,61) -> Result<Val> {62	this.settings.borrow().trace_printer.print_trace(63		loc,64		match &str {65			Val::Str(s) => s.clone().into_flat(),66			Val::Func(f) => format!("{f:?}").into(),67			v => v68				.manifest(JsonFormat::std_to_json(69					String::from("  "),70					"\n",71					": ",72					#[cfg(feature = "exp-preserve-order")]73					true,74				))?75				.into(),76		},77	);78	if let Some(rest) = rest {79		rest.evaluate()80	} else {81		Ok(str)82	}83}8485#[allow(clippy::comparison_chain)]86#[builtin]87pub fn builtin_starts_with(a: Either![IStr, ArrValue], b: Either![IStr, ArrValue]) -> Result<bool> {88	Ok(match (a, b) {89		(Either2::A(a), Either2::A(b)) => a.starts_with(b.as_str()),90		(Either2::B(a), Either2::B(b)) => {91			if b.len() > a.len() {92				return Ok(false);93			} else if b.len() == a.len() {94				return equals(&Val::Arr(a), &Val::Arr(b));95			} else {96				for (a, b) in a.iter().take(b.len()).zip(b.iter()) {97					let a = a?;98					let b = b?;99					if !equals(&a, &b)? {100						return Ok(false);101					}102				}103				true104			}105		}106		_ => bail!("both arguments should be of the same type"),107	})108}109110#[allow(clippy::comparison_chain)]111#[builtin]112pub fn builtin_ends_with(a: Either![IStr, ArrValue], b: Either![IStr, ArrValue]) -> Result<bool> {113	Ok(match (a, b) {114		(Either2::A(a), Either2::A(b)) => a.ends_with(b.as_str()),115		(Either2::B(a), Either2::B(b)) => {116			if b.len() > a.len() {117				return Ok(false);118			} else if b.len() == a.len() {119				return equals(&Val::Arr(a), &Val::Arr(b));120			} else {121				let a_len = a.len();122				for (a, b) in a.iter().skip(a_len - b.len()).zip(b.iter()) {123					let a = a?;124					let b = b?;125					if !equals(&a, &b)? {126						return Ok(false);127					}128				}129				true130			}131		}132		_ => bail!("both arguments should be of the same type"),133	})134}