git.delta.rocks / jrsonnet / refs/commits / 8a16e5a437f6

1#![allow(clippy::similar_names)]23use std::{4	cell::{Ref, RefCell, RefMut},5	collections::HashMap,6	f64,7	rc::Rc,8};910pub use arrays::*;11pub use compat::*;12pub use encoding::*;13pub use hash::*;14use jrsonnet_evaluator::{15	error::Result,16	function::{builtin_id, CallLocation, FuncVal},17	tla::TlaArg,18	trace::PathResolver,19	typed::SerializeTypedObj as _,20	val::NumValue,21	ContextBuilder, IStr, ObjValue, ObjValueBuilder, Thunk, Val,22};23use jrsonnet_gcmodule::{Acyclic, Cc, Trace};24use jrsonnet_ir::Source;25use jrsonnet_macros::{IntoUntyped, Typed};26pub use manifest::*;27pub use math::*;28pub use misc::*;29pub use objects::*;30pub use operator::*;31pub use parse::*;32pub use sets::*;33pub use sort::*;34pub use strings::*;35pub use types::*;3637#[cfg(feature = "exp-regex")]38pub use crate::regex::*;3940mod arrays;41mod compat;42mod encoding;43mod hash;44mod keyf;45mod manifest;46mod math;47mod misc;48mod objects;49mod operator;50mod parse;51#[cfg(feature = "exp-regex")]52mod regex;53mod sets;54mod sort;55mod strings;56mod types;5758#[derive(Typed, IntoUntyped, Default)]59#[allow(non_snake_case)]60struct Builtins {61	#[typed(method)]62	id: builtin_id,63	// Types64	#[typed(method, rename = "type")]65	r#type: builtin_type,66	#[typed(method)]67	isString: builtin_is_string,68	#[typed(method)]69	isNumber: builtin_is_number,70	#[typed(method)]71	isBoolean: builtin_is_boolean,72	#[typed(method)]73	isObject: builtin_is_object,74	#[typed(method)]75	isArray: builtin_is_array,76	#[typed(method)]77	isFunction: builtin_is_function,78	#[typed(method)]79	isNull: builtin_is_null,80	// Arrays81	#[typed(method)]82	makeArray: builtin_make_array,83	#[typed(method)]84	repeat: builtin_repeat,85	#[typed(method)]86	slice: builtin_slice,87	#[typed(method)]88	map: builtin_map,89	#[typed(method)]90	mapWithIndex: builtin_map_with_index,91	#[typed(method)]92	mapWithKey: builtin_map_with_key,93	#[typed(method)]94	flatMap: builtin_flatmap,95	#[typed(method)]96	filter: builtin_filter,97	#[typed(method)]98	foldl: builtin_foldl,99	#[typed(method)]100	foldr: builtin_foldr,101	#[typed(method)]102	range: builtin_range,103	#[typed(method)]104	join: builtin_join,105	#[typed(method)]106	lines: builtin_lines,107	#[typed(method)]108	resolvePath: builtin_resolve_path,109	#[typed(method)]110	deepJoin: builtin_deep_join,111	#[typed(method)]112	reverse: builtin_reverse,113	#[typed(method)]114	any: builtin_any,115	#[typed(method)]116	all: builtin_all,117	#[typed(method)]118	member: builtin_member,119	#[typed(method)]120	find: builtin_find,121	#[typed(method)]122	contains: builtin_contains,123	#[typed(method)]124	count: builtin_count,125	#[typed(method)]126	avg: builtin_avg,127	#[typed(method)]128	removeAt: builtin_remove_at,129	#[typed(method)]130	remove: builtin_remove,131	#[typed(method)]132	flattenArrays: builtin_flatten_arrays,133	#[typed(method)]134	flattenDeepArray: builtin_flatten_deep_array,135	#[typed(method)]136	prune: builtin_prune,137	#[typed(method)]138	filterMap: builtin_filter_map,139	// Math140	#[typed(method)]141	abs: builtin_abs,142	#[typed(method)]143	sign: builtin_sign,144	#[typed(method)]145	max: builtin_max,146	#[typed(method)]147	min: builtin_min,148	#[typed(method)]149	clamp: builtin_clamp,150	#[typed(method)]151	sum: builtin_sum,152	#[typed(method)]153	modulo: builtin_modulo,154	#[typed(method)]155	floor: builtin_floor,156	#[typed(method)]157	ceil: builtin_ceil,158	#[typed(method)]159	log: builtin_log,160	#[typed(method)]161	log2: builtin_log2,162	#[typed(method)]163	log10: builtin_log10,164	#[typed(method)]165	pow: builtin_pow,166	#[typed(method)]167	sqrt: builtin_sqrt,168	#[typed(method)]169	sin: builtin_sin,170	#[typed(method)]171	cos: builtin_cos,172	#[typed(method)]173	tan: builtin_tan,174	#[typed(method)]175	asin: builtin_asin,176	#[typed(method)]177	acos: builtin_acos,178	#[typed(method)]179	atan: builtin_atan,180	#[typed(method)]181	atan2: builtin_atan2,182	#[typed(method)]183	exp: builtin_exp,184	#[typed(method)]185	mantissa: builtin_mantissa,186	#[typed(method)]187	exponent: builtin_exponent,188	#[typed(method)]189	round: builtin_round,190	#[typed(method)]191	isEven: builtin_is_even,192	#[typed(method)]193	isOdd: builtin_is_odd,194	#[typed(method)]195	isInteger: builtin_is_integer,196	#[typed(method)]197	isDecimal: builtin_is_decimal,198	#[typed(method)]199	deg2rad: builtin_deg2rad,200	#[typed(method)]201	rad2deg: builtin_rad2deg,202	#[typed(method)]203	hypot: builtin_hypot,204	// Operator205	#[typed(rename = "mod", method)]206	r#mod: builtin_mod,207	#[typed(method)]208	primitiveEquals: builtin_primitive_equals,209	#[typed(method)]210	equals: builtin_equals,211	#[typed(method)]212	xor: builtin_xor,213	#[typed(method)]214	xnor: builtin_xnor,215	#[typed(method)]216	format: builtin_format,217	// Sort218	#[typed(method)]219	sort: builtin_sort,220	#[typed(method)]221	uniq: builtin_uniq,222	#[typed(method)]223	set: builtin_set,224	#[typed(method)]225	minArray: builtin_min_array,226	#[typed(method)]227	maxArray: builtin_max_array,228	// Hash229	#[typed(method)]230	md5: builtin_md5,231	#[typed(method)]232	sha1: builtin_sha1,233	#[typed(method)]234	sha256: builtin_sha256,235	#[typed(method)]236	sha512: builtin_sha512,237	#[typed(method)]238	sha3: builtin_sha3,239	// Encoding240	#[typed(method)]241	encodeUTF8: builtin_encode_utf8,242	#[typed(method)]243	decodeUTF8: builtin_decode_utf8,244	#[typed(method)]245	base64: builtin_base64,246	#[typed(method)]247	base64Decode: builtin_base64_decode,248	#[typed(method)]249	base64DecodeBytes: builtin_base64_decode_bytes,250	// Objects251	#[typed(method)]252	objectFieldsEx: builtin_object_fields_ex,253	#[typed(method)]254	objectFields: builtin_object_fields,255	#[typed(method)]256	objectFieldsAll: builtin_object_fields_all,257	#[typed(method)]258	objectValues: builtin_object_values,259	#[typed(method)]260	objectValuesAll: builtin_object_values_all,261	#[typed(method)]262	objectKeysValues: builtin_object_keys_values,263	#[typed(method)]264	objectKeysValuesAll: builtin_object_keys_values_all,265	#[typed(method)]266	objectHasEx: builtin_object_has_ex,267	#[typed(method)]268	objectHas: builtin_object_has,269	#[typed(method)]270	objectHasAll: builtin_object_has_all,271	#[typed(method)]272	objectRemoveKey: builtin_object_remove_key,273	// Manifest274	#[typed(method)]275	escapeStringJson: builtin_escape_string_json,276	#[typed(method)]277	escapeStringPython: builtin_escape_string_python,278	#[typed(method)]279	escapeStringXML: builtin_escape_string_xml,280	#[typed(method)]281	manifestJsonEx: builtin_manifest_json_ex,282	#[typed(method)]283	manifestJson: builtin_manifest_json,284	#[typed(method)]285	manifestJsonMinified: builtin_manifest_json_minified,286	#[typed(method)]287	manifestYamlDoc: builtin_manifest_yaml_doc,288	#[typed(method)]289	manifestYamlStream: builtin_manifest_yaml_stream,290	#[typed(method)]291	manifestTomlEx: builtin_manifest_toml_ex,292	#[typed(method)]293	manifestToml: builtin_manifest_toml,294	#[typed(method)]295	toString: builtin_to_string,296	#[typed(method)]297	manifestPython: builtin_manifest_python,298	#[typed(method)]299	manifestPythonVars: builtin_manifest_python_vars,300	#[typed(method)]301	manifestXmlJsonml: builtin_manifest_xml_jsonml,302	#[typed(method)]303	manifestIni: builtin_manifest_ini,304	// Parse305	#[typed(method)]306	parseJson: builtin_parse_json,307	#[typed(method)]308	parseYaml: builtin_parse_yaml,309	// Strings310	#[typed(method)]311	codepoint: builtin_codepoint,312	#[typed(method)]313	substr: builtin_substr,314	#[typed(method)]315	char: builtin_char,316	#[typed(method)]317	strReplace: builtin_str_replace,318	#[typed(method)]319	escapeStringBash: builtin_escape_string_bash,320	#[typed(method)]321	escapeStringDollars: builtin_escape_string_dollars,322	#[typed(method)]323	isEmpty: builtin_is_empty,324	#[typed(method)]325	equalsIgnoreCase: builtin_equals_ignore_case,326	#[typed(method)]327	splitLimit: builtin_splitlimit,328	#[typed(method)]329	splitLimitR: builtin_splitlimitr,330	#[typed(method)]331	split: builtin_split,332	#[typed(method)]333	asciiUpper: builtin_ascii_upper,334	#[typed(method)]335	asciiLower: builtin_ascii_lower,336	#[typed(method)]337	findSubstr: builtin_find_substr,338	#[typed(method)]339	parseInt: builtin_parse_int,340	#[cfg(feature = "exp-bigint")]341	#[typed(method)]342	bigint: builtin_bigint,343	#[typed(method)]344	parseOctal: builtin_parse_octal,345	#[typed(method)]346	parseHex: builtin_parse_hex,347	#[typed(method)]348	stringChars: builtin_string_chars,349	#[typed(method)]350	lstripChars: builtin_lstrip_chars,351	#[typed(method)]352	rstripChars: builtin_rstrip_chars,353	#[typed(method)]354	stripChars: builtin_strip_chars,355	#[typed(method)]356	trim: builtin_trim,357	// Misc358	#[typed(method)]359	length: builtin_length,360	#[typed(method)]361	get: builtin_get,362	#[typed(method)]363	startsWith: builtin_starts_with,364	#[typed(method)]365	endsWith: builtin_ends_with,366	#[typed(method)]367	assertEqual: builtin_assert_equal,368	#[typed(method)]369	mergePatch: builtin_merge_patch,370	// Sets371	#[typed(method)]372	setMember: builtin_set_member,373	#[typed(method)]374	setInter: builtin_set_inter,375	#[typed(method)]376	setDiff: builtin_set_diff,377	#[typed(method)]378	setUnion: builtin_set_union,379	// Regex380	#[cfg(feature = "exp-regex")]381	#[typed(method)]382	regexQuoteMeta: builtin_regex_quote_meta,383	// Compat384	#[typed(method)]385	__compare: builtin___compare,386	#[typed(method)]387	__compare_array: builtin___compare_array,388	#[typed(method)]389	__array_less: builtin___array_less,390	#[typed(method)]391	__array_greater: builtin___array_greater,392	#[typed(method)]393	__array_less_or_equal: builtin___array_less_or_equal,394	#[typed(method)]395	__array_greater_or_equal: builtin___array_greater_or_equal,396}397398#[allow(clippy::too_many_lines)]399pub fn stdlib_uncached(settings: Cc<RefCell<Settings>>) -> ObjValue {400	let mut builder = ObjValueBuilder::new();401402	let builtins = Builtins::default();403	builtins.serialize(&mut builder).expect("no conflicts");404405	builder.method(406		"extVar",407		builtin_ext_var {408			settings: settings.clone(),409		},410	);411	builder.method(412		"native",413		builtin_native {414			settings: settings.clone(),415		},416	);417	builder.method("trace", builtin_trace { settings });418419	builder.field("pi").hide().value(Val::Num(420		NumValue::new(f64::consts::PI).expect("pi is finite"),421	));422423	#[cfg(feature = "exp-regex")]424	{425		// Regex426		let regex_cache = RegexCache::default();427		builder.method(428			"regexFullMatch",429			builtin_regex_full_match {430				cache: regex_cache.clone(),431			},432		);433		builder.method(434			"regexPartialMatch",435			builtin_regex_partial_match {436				cache: regex_cache.clone(),437			},438		);439		builder.method(440			"regexReplace",441			builtin_regex_replace {442				cache: regex_cache.clone(),443			},444		);445		builder.method(446			"regexGlobalReplace",447			builtin_regex_global_replace { cache: regex_cache },448		);449	};450451	builder.build()452}453454pub trait TracePrinter: Acyclic {455	fn print_trace(&self, loc: CallLocation, value: IStr);456}457458#[derive(Acyclic)]459pub struct StdTracePrinter {460	resolver: PathResolver,461}462impl StdTracePrinter {463	pub fn new(resolver: PathResolver) -> Self {464		Self { resolver }465	}466}467impl TracePrinter for StdTracePrinter {468	fn print_trace(&self, loc: CallLocation, value: IStr) {469		eprint!("TRACE:");470		if let Some(loc) = loc.0 {471			let locs = loc.0.map_source_locations(&[loc.1]);472			eprint!(473				" {}:{}",474				loc.0.source_path().path().map_or_else(475					|| loc.0.source_path().to_string(),476					|p| self.resolver.resolve(p)477				),478				locs[0].line479			);480		}481		eprintln!(" {value}");482	}483}484485#[derive(Clone, Trace)]486pub struct Settings {487	/// Used for `std.extVar`488	pub ext_vars: HashMap<IStr, TlaArg>,489	/// Used for `std.native`490	pub ext_natives: HashMap<IStr, FuncVal>,491	/// Used for `std.trace`492	pub trace_printer: Rc<dyn TracePrinter>,493	/// Used for `std.thisFile`494	pub path_resolver: PathResolver,495}496497#[derive(Trace, Clone)]498pub struct ContextInitializer {499	/// std without applied thisFile overlay500	stdlib_obj: ObjValue,501	settings: Cc<RefCell<Settings>>,502}503impl ContextInitializer {504	pub fn new(resolver: PathResolver) -> Self {505		let settings = Settings {506			ext_vars: HashMap::new(),507			ext_natives: HashMap::new(),508			trace_printer: Rc::new(StdTracePrinter::new(resolver.clone())),509			path_resolver: resolver,510		};511		let settings = Cc::new(RefCell::new(settings));512		let stdlib_obj = stdlib_uncached(settings.clone());513		Self {514			stdlib_obj,515			settings,516		}517	}518	pub fn settings(&self) -> Ref<'_, Settings> {519		self.settings.borrow()520	}521	pub fn settings_mut(&self) -> RefMut<'_, Settings> {522		self.settings.borrow_mut()523	}524	pub fn add_ext_var(&self, name: IStr, value: Val) {525		self.settings_mut()526			.ext_vars527			.insert(name, TlaArg::Val(value));528	}529	pub fn add_ext_str(&self, name: IStr, value: IStr) {530		self.settings_mut()531			.ext_vars532			.insert(name, TlaArg::String(value));533	}534	pub fn add_ext_code(&self, name: &str, code: impl AsRef<str>) -> Result<()> {535		// self.data_mut().volatile_files.insert(source_name, code);536		self.settings_mut()537			.ext_vars538			.insert(name.into(), TlaArg::InlineCode(code.as_ref().to_owned()));539		Ok(())540	}541	pub fn add_native(&self, name: impl Into<IStr>, cb: impl Into<FuncVal>) {542		self.settings_mut()543			.ext_natives544			.insert(name.into(), cb.into());545	}546}547impl jrsonnet_evaluator::ContextInitializer for ContextInitializer {548	fn reserve_vars(&self) -> usize {549		1550	}551	fn populate(&self, source: Source, builder: &mut ContextBuilder) {552		let mut std = ObjValueBuilder::new();553		std.with_super(self.stdlib_obj.clone());554		std.field("thisFile").hide().value({555			let source_path = source.source_path();556			source_path.path().map_or_else(557				|| source_path.to_string(),558				|p| self.settings().path_resolver.resolve(p),559			)560		});561		let stdlib_with_this_file = std.build();562563		builder.bind("std", Thunk::evaluated(Val::Obj(stdlib_with_this_file)));564	}565	fn as_any(&self) -> &dyn std::any::Any {566		self567	}568}