git.delta.rocks / jrsonnet / refs/commits / 3cd4c9197033

1use crate::error::Error::*;2use crate::error::Result;3use crate::{throw, Val};45#[derive(PartialEq, Clone, Copy)]6pub enum ManifestType {7	// Applied in manifestification8	Manifest,9	/// Used for std.manifestJson10	/// Empty array/objects extends to "[\n\n]" instead of "[ ]" as in manifest11	Std,12	/// No line breaks, used in `obj+''`13	ToString,14	/// Minified json15	Minify,16}1718pub struct ManifestJsonOptions<'s> {19	pub padding: &'s str,20	pub mtype: ManifestType,21}2223pub fn manifest_json_ex(val: &Val, options: &ManifestJsonOptions<'_>) -> Result<String> {24	let mut out = String::new();25	manifest_json_ex_buf(val, &mut out, &mut String::new(), options)?;26	Ok(out)27}28fn manifest_json_ex_buf(29	val: &Val,30	buf: &mut String,31	cur_padding: &mut String,32	options: &ManifestJsonOptions<'_>,33) -> Result<()> {34	use std::fmt::Write;35	let mtype = options.mtype;36	match val {37		Val::Bool(v) => {38			if *v {39				buf.push_str("true");40			} else {41				buf.push_str("false");42			}43		}44		Val::Null => buf.push_str("null"),45		Val::Str(s) => escape_string_json_buf(s, buf),46		Val::Num(n) => write!(buf, "{}", n).unwrap(),47		Val::Arr(items) => {48			buf.push('[');49			if !items.is_empty() {50				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {51					buf.push('\n');52				}5354				let old_len = cur_padding.len();55				cur_padding.push_str(options.padding);56				for (i, item) in items.iter().enumerate() {57					if i != 0 {58						buf.push(',');59						if mtype == ManifestType::ToString {60							buf.push(' ');61						} else if mtype != ManifestType::Minify {62							buf.push('\n');63						}64					}65					buf.push_str(cur_padding);66					manifest_json_ex_buf(&item?, buf, cur_padding, options)?;67				}68				cur_padding.truncate(old_len);6970				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {71					buf.push('\n');72					buf.push_str(cur_padding);73				}74			} else if mtype == ManifestType::Std {75				buf.push_str("\n\n");76				buf.push_str(cur_padding);77			} else if mtype == ManifestType::ToString || mtype == ManifestType::Manifest {78				buf.push(' ');79			}80			buf.push(']');81		}82		Val::Obj(obj) => {83			obj.run_assertions()?;84			buf.push('{');85			let fields = obj.fields();86			if !fields.is_empty() {87				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {88					buf.push('\n');89				}9091				let old_len = cur_padding.len();92				cur_padding.push_str(options.padding);93				for (i, field) in fields.into_iter().enumerate() {94					if i != 0 {95						buf.push(',');96						if mtype == ManifestType::ToString {97							buf.push(' ');98						} else if mtype != ManifestType::Minify {99							buf.push('\n');100						}101					}102					buf.push_str(cur_padding);103					escape_string_json_buf(&field, buf);104					buf.push_str(": ");105					crate::push(106						None,107						|| format!("field <{}> manifestification", field.clone()),108						|| {109							let value = obj.get(field.clone())?.unwrap();110							manifest_json_ex_buf(&value, buf, cur_padding, options)111						},112					)?;113				}114				cur_padding.truncate(old_len);115116				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {117					buf.push('\n');118					buf.push_str(cur_padding);119				}120			} else if mtype == ManifestType::Std {121				buf.push_str("\n\n");122				buf.push_str(cur_padding);123			} else if mtype == ManifestType::ToString || mtype == ManifestType::Manifest {124				buf.push(' ');125			}126			buf.push('}');127		}128		Val::Func(_) => throw!(RuntimeError("tried to manifest function".into())),129	};130	Ok(())131}132133pub fn escape_string_json(s: &str) -> String {134	let mut buf = String::new();135	escape_string_json_buf(s, &mut buf);136	buf137}138139fn escape_string_json_buf(s: &str, buf: &mut String) {140	use std::fmt::Write;141	buf.push('"');142	for c in s.chars() {143		match c {144			'"' => buf.push_str("\\\""),145			'\\' => buf.push_str("\\\\"),146			'\u{0008}' => buf.push_str("\\b"),147			'\u{000c}' => buf.push_str("\\f"),148			'\n' => buf.push_str("\\n"),149			'\r' => buf.push_str("\\r"),150			'\t' => buf.push_str("\\t"),151			c if c < 32 as char || (c >= 127 as char && c <= 159 as char) => {152				write!(buf, "\\u{:04x}", c as u32).unwrap()153			}154			c => buf.push(c),155		}156	}157	buf.push('"');158}159160pub struct ManifestYamlOptions<'s> {161	/// Padding before fields, i.e162	/// a:163	///   b:164	/// ^^ this165	pub padding: &'s str,166	/// Padding before array elements in objects167	/// a:168	///   - 1169	/// ^^ this170	pub arr_element_padding: &'s str,171}172173pub fn manifest_yaml_ex(val: &Val, options: &ManifestYamlOptions<'_>) -> Result<String> {174	let mut out = String::new();175	manifest_yaml_ex_buf(val, &mut out, &mut String::new(), options)?;176	Ok(out)177}178fn manifest_yaml_ex_buf(179	val: &Val,180	buf: &mut String,181	cur_padding: &mut String,182	options: &ManifestYamlOptions<'_>,183) -> Result<()> {184	use std::fmt::Write;185	match val {186		Val::Bool(v) => {187			if *v {188				buf.push_str("true")189			} else {190				buf.push_str("false")191			}192		}193		Val::Null => buf.push_str("null"),194		Val::Str(s) => {195			if s.is_empty() {196				buf.push_str("\"\"");197			} else if let Some(s) = s.strip_suffix('\n') {198				buf.push('|');199				for line in s.split('\n') {200					buf.push('\n');201					buf.push_str(options.padding);202					buf.push_str(line);203				}204			} else {205				escape_string_json_buf(s, buf)206			}207		}208		Val::Num(n) => write!(buf, "{}", *n).unwrap(),209		Val::Arr(a) => {210			if a.is_empty() {211				buf.push_str("[]");212			} else {213				for (i, item) in a.iter().enumerate() {214					if i != 0 {215						buf.push('\n');216						buf.push_str(cur_padding);217					}218					let item = item?;219					buf.push('-');220					if let Val::Arr(a) = &item {221						if !a.is_empty() {222							buf.push('\n');223							buf.push_str(cur_padding);224							buf.push_str(options.padding);225						} else {226							buf.push(' ');227						}228					} else {229						buf.push(' ');230					}231					let extra_padding = if let Val::Arr(a) = &item {232						!a.is_empty()233					} else if let Val::Obj(a) = &item {234						!a.is_empty()235					} else {236						false237					};238					let prev_len = cur_padding.len();239					if extra_padding {240						cur_padding.push_str(options.padding);241					}242					manifest_yaml_ex_buf(&item, buf, cur_padding, options)?;243					cur_padding.truncate(prev_len);244				}245			}246		}247		Val::Obj(o) => {248			if o.is_empty() {249				buf.push_str("{}");250			} else {251				for (i, key) in o.fields().iter().enumerate() {252					if i != 0 {253						buf.push('\n');254						buf.push_str(cur_padding);255					}256					escape_string_json_buf(key, buf);257					buf.push(':');258					let item = o.get(key.clone())?.expect("field exists");259					if let Val::Arr(a) = &item {260						if !a.is_empty() {261							buf.push('\n');262							buf.push_str(cur_padding);263							buf.push_str(options.arr_element_padding);264						} else {265							buf.push(' ');266						}267					} else if let Val::Obj(o) = &item {268						if !o.is_empty() {269							buf.push('\n');270							buf.push_str(cur_padding);271							buf.push_str(options.padding);272						} else {273							buf.push(' ');274						}275					} else {276						buf.push(' ');277					}278					let prev_len = cur_padding.len();279					if let Val::Arr(a) = &item {280						if !a.is_empty() {281							cur_padding.push_str(options.arr_element_padding);282						}283					} else if let Val::Obj(a) = &item {284						if !a.is_empty() {285							cur_padding.push_str(options.padding);286						}287					};288					manifest_yaml_ex_buf(&item, buf, cur_padding, options)?;289					cur_padding.truncate(prev_len);290				}291			}292		}293		Val::Func(_) => throw!(RuntimeError("tried to manifest function".into())),294	}295	Ok(())296}