git.delta.rocks / jrsonnet / refs/commits / bc97bc99f5fd

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.unwrap_if_lazy()? {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) => buf.push_str(&escape_string_json(&s)),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 {78				buf.push(' ');79			}80			buf.push(']');81		}82		Val::Obj(obj) => {83			buf.push('{');84			let fields = obj.visible_fields();85			if !fields.is_empty() {86				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {87					buf.push('\n');88				}8990				let old_len = cur_padding.len();91				cur_padding.push_str(options.padding);92				for (i, field) in fields.into_iter().enumerate() {93					if i != 0 {94						buf.push(',');95						if mtype == ManifestType::ToString {96							buf.push(' ');97						} else if mtype != ManifestType::Minify {98							buf.push('\n');99						}100					}101					buf.push_str(cur_padding);102					buf.push_str(&escape_string_json(&field));103					buf.push_str(": ");104					manifest_json_ex_buf(&obj.get(field)?.unwrap(), buf, cur_padding, options)?;105				}106				cur_padding.truncate(old_len);107108				if mtype != ManifestType::ToString && mtype != ManifestType::Minify {109					buf.push('\n');110					buf.push_str(cur_padding);111				}112			} else if mtype == ManifestType::Std {113				buf.push_str("\n\n");114				buf.push_str(cur_padding);115			} else if mtype == ManifestType::ToString {116				buf.push(' ');117			}118			buf.push('}');119		}120		Val::Func(_) => throw!(RuntimeError("tried to manifest function".into())),121		Val::Lazy(_) => unreachable!(),122	};123	Ok(())124}125pub fn escape_string_json(s: &str) -> String {126	use std::fmt::Write;127	let mut out = String::new();128	out.push('"');129	for c in s.chars() {130		match c {131			'"' => out.push_str("\\\""),132			'\\' => out.push_str("\\\\"),133			'\u{0008}' => out.push_str("\\b"),134			'\u{000c}' => out.push_str("\\f"),135			'\n' => out.push_str("\\n"),136			'\r' => out.push_str("\\r"),137			'\t' => out.push_str("\\t"),138			c if c < 32 as char || (c >= 127 as char && c <= 159 as char) => {139				write!(out, "\\u{:04x}", c as u32).unwrap()140			}141			c => out.push(c),142		}143	}144	out.push('"');145	out146}147148#[test]149fn json_test() {150	assert_eq!(escape_string_json("\u{001f}"), "\"\\u001f\"")151}