git.delta.rocks / jrsonnet / refs/commits / 7b01ecbd8fbc

1use crate::{2	error::{Error::*, Result},3	throw, State, Val,4};56#[derive(PartialEq, Eq, Clone, Copy)]7pub enum ManifestType {8	// Applied in manifestification9	Manifest,10	/// Used for std.manifestJson11	/// Empty array/objects extends to "[\n\n]" instead of "[ ]" as in manifest12	Std,13	/// No line breaks, used in `obj+''`14	ToString,15	/// Minified json16	Minify,17}1819pub struct ManifestJsonOptions<'s> {20	pub padding: &'s str,21	pub mtype: ManifestType,22	pub newline: &'s str,23	pub key_val_sep: &'s str,24	#[cfg(feature = "exp-preserve-order")]25	pub preserve_order: bool,26}2728pub fn manifest_json_ex(val: &Val, options: &ManifestJsonOptions<'_>) -> Result<String> {29	let mut out = String::new();30	manifest_json_ex_buf(val, &mut out, &mut String::new(), options)?;31	Ok(out)32}33fn manifest_json_ex_buf(34	val: &Val,35	buf: &mut String,36	cur_padding: &mut String,37	options: &ManifestJsonOptions<'_>,38) -> Result<()> {39	use std::fmt::Write;40	let mtype = options.mtype;41	match val {42		Val::Bool(v) => {43			if *v {44				buf.push_str("true");45			} else {46				buf.push_str("false");47			}48		}49		Val::Null => buf.push_str("null"),50		Val::Str(s) => escape_string_json_buf(s, buf),51		Val::Num(n) => write!(buf, "{n}").unwrap(),52		Val::Arr(items) => {53			buf.push('[');54			if !items.is_empty() {55				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {56					buf.push_str(options.newline);57				}5859				let old_len = cur_padding.len();60				cur_padding.push_str(options.padding);61				for (i, item) in items.iter().enumerate() {62					if i != 0 {63						buf.push(',');64						if mtype == ManifestType::ToString {65							buf.push(' ');66						} else if mtype != ManifestType::Minify {67							buf.push_str(options.newline);68						}69					}70					buf.push_str(cur_padding);71					manifest_json_ex_buf(&item?, buf, cur_padding, options)?;72				}73				cur_padding.truncate(old_len);7475				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {76					buf.push_str(options.newline);77					buf.push_str(cur_padding);78				}79			} else if mtype == ManifestType::Std {80				buf.push_str("\n\n");81				buf.push_str(cur_padding);82			} else if mtype == ManifestType::ToString || mtype == ManifestType::Manifest {83				buf.push(' ');84			}85			buf.push(']');86		}87		Val::Obj(obj) => {88			obj.run_assertions()?;89			buf.push('{');90			let fields = obj.fields(91				#[cfg(feature = "exp-preserve-order")]92				options.preserve_order,93			);94			if !fields.is_empty() {95				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {96					buf.push_str(options.newline);97				}9899				let old_len = cur_padding.len();100				cur_padding.push_str(options.padding);101				for (i, field) in fields.into_iter().enumerate() {102					if i != 0 {103						buf.push(',');104						if mtype == ManifestType::ToString {105							buf.push(' ');106						} else if mtype != ManifestType::Minify {107							buf.push_str(options.newline);108						}109					}110					buf.push_str(cur_padding);111					escape_string_json_buf(&field, buf);112					buf.push_str(options.key_val_sep);113					State::push_description(114						|| format!("field <{}> manifestification", field.clone()),115						|| {116							let value = obj.get(field.clone())?.unwrap();117							manifest_json_ex_buf(&value, buf, cur_padding, options)?;118							Ok(())119						},120					)?;121				}122				cur_padding.truncate(old_len);123124				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {125					buf.push_str(options.newline);126					buf.push_str(cur_padding);127				}128			} else if mtype == ManifestType::Std {129				buf.push_str("\n\n");130				buf.push_str(cur_padding);131			} else if mtype == ManifestType::ToString || mtype == ManifestType::Manifest {132				buf.push(' ');133			}134			buf.push('}');135		}136		Val::Func(_) => throw!(RuntimeError("tried to manifest function".into())),137	};138	Ok(())139}140141pub fn escape_string_json(s: &str) -> String {142	let mut buf = String::new();143	escape_string_json_buf(s, &mut buf);144	buf145}146147fn escape_string_json_buf(s: &str, buf: &mut String) {148	use std::fmt::Write;149	buf.push('"');150	for c in s.chars() {151		match c {152			'"' => buf.push_str("\\\""),153			'\\' => buf.push_str("\\\\"),154			'\u{0008}' => buf.push_str("\\b"),155			'\u{000c}' => buf.push_str("\\f"),156			'\n' => buf.push_str("\\n"),157			'\r' => buf.push_str("\\r"),158			'\t' => buf.push_str("\\t"),159			c if c < 32 as char || (c >= 127 as char && c <= 159 as char) => {160				write!(buf, "\\u{:04x}", c as u32).unwrap();161			}162			c => buf.push(c),163		}164	}165	buf.push('"');166}167168pub struct ManifestYamlOptions<'s> {169	/// Padding before fields, i.e170	/// ```yaml171	/// a:172	///   b:173	/// ## <- this174	/// ```175	pub padding: &'s str,176	/// Padding before array elements in objects177	/// ```yaml178	/// a:179	///   - 1180	/// ## <- this181	/// ```182	pub arr_element_padding: &'s str,183	/// Should yaml keys appear unescaped, when possible184	/// ```yaml185	/// "safe_key": 1186	/// # vs187	/// safe_key: 1188	/// ```189	pub quote_keys: bool,190	/// If true - then order of fields is preserved as written,191	/// instead of sorting alphabetically192	#[cfg(feature = "exp-preserve-order")]193	pub preserve_order: bool,194}195196/// From <https://github.com/chyh1990/yaml-rust/blob/da52a68615f2ecdd6b7e4567019f280c433c1521/src/emitter.rs#L289>197/// With added date check198fn yaml_needs_quotes(string: &str) -> bool {199	fn need_quotes_spaces(string: &str) -> bool {200		string.starts_with(' ') || string.ends_with(' ')201	}202203	string.is_empty()204		|| need_quotes_spaces(string)205		|| string.starts_with(|c| matches!(c, '&' | '*' | '?' | '|' | '-' | '<' | '>' | '=' | '!' | '%' | '@'))206		|| string.contains(|c| matches!(c, ':' | '{' | '}' | '[' | ']' | ',' | '#' | '`' | '\"' | '\'' | '\\' | '\0'..='\x06' | '\t' | '\n' | '\r' | '\x0e'..='\x1a' | '\x1c'..='\x1f'))207		|| [208			// http://yaml.org/type/bool.html209			// Note: 'y', 'Y', 'n', 'N', is not quoted deliberately, as in libyaml. PyYAML also parse210			// them as string, not booleans, although it is violating the YAML 1.1 specification.211			// See https://github.com/dtolnay/serde-yaml/pull/83#discussion_r152628088.212			"yes", "Yes", "YES", "no", "No", "NO", "True", "TRUE", "true", "False", "FALSE", "false",213			"on", "On", "ON", "off", "Off", "OFF", // http://yaml.org/type/null.html214			"null", "Null", "NULL", "~",215		].contains(&string)216		|| (string.chars().all(|c| matches!(c, '0'..='9' | '-'))217			&& string.chars().filter(|c| *c == '-').count() == 2)218		|| string.starts_with('.')219		|| string.starts_with("0x")220		|| string.parse::<i64>().is_ok()221		|| string.parse::<f64>().is_ok()222}223224pub fn manifest_yaml_ex(val: &Val, options: &ManifestYamlOptions<'_>) -> Result<String> {225	let mut out = String::new();226	manifest_yaml_ex_buf(val, &mut out, &mut String::new(), options)?;227	Ok(out)228}229230#[allow(clippy::too_many_lines)]231fn manifest_yaml_ex_buf(232	val: &Val,233	buf: &mut String,234	cur_padding: &mut String,235	options: &ManifestYamlOptions<'_>,236) -> Result<()> {237	use std::fmt::Write;238	match val {239		Val::Bool(v) => {240			if *v {241				buf.push_str("true");242			} else {243				buf.push_str("false");244			}245		}246		Val::Null => buf.push_str("null"),247		Val::Str(s) => {248			if s.is_empty() {249				buf.push_str("\"\"");250			} else if let Some(s) = s.strip_suffix('\n') {251				buf.push('|');252				for line in s.split('\n') {253					buf.push('\n');254					buf.push_str(cur_padding);255					buf.push_str(options.padding);256					buf.push_str(line);257				}258			} else if !options.quote_keys && !yaml_needs_quotes(s) {259				buf.push_str(s);260			} else {261				escape_string_json_buf(s, buf);262			}263		}264		Val::Num(n) => write!(buf, "{}", *n).unwrap(),265		Val::Arr(a) => {266			if a.is_empty() {267				buf.push_str("[]");268			} else {269				for (i, item) in a.iter().enumerate() {270					if i != 0 {271						buf.push('\n');272						buf.push_str(cur_padding);273					}274					let item = item?;275					buf.push('-');276					match &item {277						Val::Arr(a) if !a.is_empty() => {278							buf.push('\n');279							buf.push_str(cur_padding);280							buf.push_str(options.padding);281						}282						_ => buf.push(' '),283					}284					let extra_padding = match &item {285						Val::Arr(a) => !a.is_empty(),286						Val::Obj(o) => !o.is_empty(),287						_ => false,288					};289					let prev_len = cur_padding.len();290					if extra_padding {291						cur_padding.push_str(options.padding);292					}293					manifest_yaml_ex_buf(&item, buf, cur_padding, options)?;294					cur_padding.truncate(prev_len);295				}296			}297		}298		Val::Obj(o) => {299			if o.is_empty() {300				buf.push_str("{}");301			} else {302				for (i, key) in o303					.fields(304						#[cfg(feature = "exp-preserve-order")]305						options.preserve_order,306					)307					.iter()308					.enumerate()309				{310					if i != 0 {311						buf.push('\n');312						buf.push_str(cur_padding);313					}314					if !options.quote_keys && !yaml_needs_quotes(key) {315						buf.push_str(key);316					} else {317						escape_string_json_buf(key, buf);318					}319					buf.push(':');320					let prev_len = cur_padding.len();321					let item = o.get(key.clone())?.expect("field exists");322					match &item {323						Val::Arr(a) if !a.is_empty() => {324							buf.push('\n');325							buf.push_str(cur_padding);326							buf.push_str(options.arr_element_padding);327							cur_padding.push_str(options.arr_element_padding);328						}329						Val::Obj(o) if !o.is_empty() => {330							buf.push('\n');331							buf.push_str(cur_padding);332							buf.push_str(options.padding);333							cur_padding.push_str(options.padding);334						}335						_ => buf.push(' '),336					}337					manifest_yaml_ex_buf(&item, buf, cur_padding, options)?;338					cur_padding.truncate(prev_len);339				}340			}341		}342		Val::Func(_) => throw!("tried to manifest function"),343	}344	Ok(())345}