json-writer 0.4.0

#![doc = include_str!("../README.md")]

use core::fmt;

///
/// Helper for appending a JSON object to the borrowed buffer.
///
/// Appends '{' on creation.
/// Appends '}' when dropped.
///
pub struct JSONObjectWriter<'a, Writer: JSONWriter = String> {
	///
	/// Mutable borrow of buffer
	///
	/// Consider using the methods instead of using this field directly.
	/// This field should not be used unless you know what you are doing.
	///
	pub writer: &'a mut Writer,
	empty: bool,
}

///
/// Helper for appending a JSON array to the borrowed buffer.
///
/// Appends '[' on creation.
/// Appends ']' when dropped.
///
pub struct JSONArrayWriter<'a, Writer: JSONWriter = String> {
	///
	/// Mutable borrow of buffer
	///
	/// Consider using the methods instead of using this field directly.
	/// This field should not be used unless you know what you are doing.
	///
	pub writer: &'a mut Writer,
	empty: bool,
}

/// Build a string using the `fmt::Write` impl
pub struct JSONStringWriter<'a, Writer: JSONWriter = String> {
	/// The generic writer
	pub writer: &'a mut Writer,
}

impl<'a, Writer: JSONWriter> JSONStringWriter<'a, Writer> {
	/// Creates a new JSONStringWriter that writes to the given buffer. Writes '"' to the buffer immediately.
	#[inline(always)]
	pub fn new(writer: &mut Writer) -> JSONStringWriter<'_, Writer> {
		writer.json_begin_string();
		JSONStringWriter { writer }
	}

	///
	/// Drops the JSONStringWriter.
	/// Dropping causes '"' to be appended to the buffer.
	///
	#[inline(always)]
	pub fn end(self) {
		drop(self)
	}
}

impl<'a, Writer: JSONWriter> fmt::Write for JSONStringWriter<'a, Writer> {
	#[inline(always)]
	fn write_str(&mut self, s: &str) -> fmt::Result {
		self.writer.json_string_part(s);
		Ok(())
	}
}

impl<Writer: JSONWriter> Drop for JSONStringWriter<'_, Writer> {
	#[inline(always)]
	fn drop(&mut self) {
		self.writer.json_end_string()
	}
}

#[doc(hidden)]
#[derive(Debug, Copy, Clone)]
pub struct Null();

///
/// Writer trait for custom formatting and output
///
/// You most likely want to use `JSONObjectWriter` or `JSONArrayWriter` instead of using this directly
///
pub trait JSONWriter {
	/// Writes null
	#[inline(always)]
	fn json_null(&mut self) {
		self.json_fragment("null");
	}

	/// Writes true or false
	fn json_bool(&mut self, value: bool) {
		self.json_fragment(if value { "true" } else { "false" });
	}

	/// Quotes and escapes the given string and writes the result to output with delimiting quotes.
	fn json_string(&mut self, value: &str);

	/// Quotes and escapes the given string and writes the result to output without delimiting quotes.
	fn json_string_part(&mut self, value: &str);

	/// Converts number to string and writes it. Writes null for NaN and infinity
	#[inline(never)]
	fn json_number_f64(&mut self, value: f64) {
		if !value.is_finite() {
			// JSON does not allow infinite or nan values. In browsers JSON.stringify(Number.NaN) = "null"
			self.json_null();
			return;
		}

		let mut buf = ryu::Buffer::new();
		let mut result = buf.format_finite(value);
		if result.ends_with(".0") {
			result = unsafe { result.get_unchecked(..result.len() - 2) };
		}
		self.json_number_str(result);
	}

	/// Writes a number that has already been converted to string
	#[inline(always)]
	fn json_number_str(&mut self, value: &str) {
		self.json_fragment(value);
	}

	/// Called at the start of writing an object. Writes the opening bracket
	#[inline(always)]
	fn json_begin_object(&mut self) {
		self.json_fragment("{");
	}

	/// Called after writing all key-value pairs of an object.
	///
	/// `empty` is `true` when the object contains no key-value pairs.
	#[inline(always)]
	fn json_end_object(&mut self, _empty: bool) {
		self.json_fragment("}");
	}

	/// Called at the start of writing an array.
	#[inline(always)]
	fn json_begin_array(&mut self) {
		self.json_fragment("[");
	}

	/// Called after writing all items of an array.
	///
	/// `empty` is `true` when the array contains no items.
	#[inline(always)]
	fn json_end_array(&mut self, _empty: bool) {
		self.json_fragment("]");
	}

	/// Writes a double-quote.
	///
	/// Called when creating a JSONStringWriter
	#[inline(always)]
	fn json_begin_string(&mut self) {
		self.json_fragment("\"");
	}

	/// Writes a double-quote.
	///
	/// Called when dropping a JSONStringWriter
	#[inline(always)]
	fn json_end_string(&mut self) {
		self.json_fragment("\"");
	}

	/// Called before each key-value pair in an object and each item in an array.
	///
	#[inline]
	fn json_begin_array_value(&mut self, first: bool) {
		if !first {
			self.json_fragment(",");
		}
	}

	/// writes a comma when not first entry, escapes and writes the key and a colon
	fn json_object_key(&mut self, key: &str, first: bool) {
		if !first {
			self.json_fragment(",");
		}
		self.json_string(key);
		self.json_fragment(":");
	}

	/// write a raw json fragment
	fn json_fragment(&mut self, value: &str);
}

///
/// Represents the null value in json.
///
/// **Note**: Option::None may be used instead in most cases.
///
pub static NULL: Null = Null();

impl<W: JSONWriter> JSONObjectWriter<'_, W> {
	///
	/// Creates a new JSONObjectWriter that writes to the given buffer. Writes '{' to the buffer immediately.
	///
	#[inline(always)]
	pub fn new(writer: &mut W) -> JSONObjectWriter<'_, W> {
		writer.json_begin_object();
		JSONObjectWriter {
			writer,
			empty: true,
		}
	}

	///
	/// Starts writing a nested object with given key:
	///
	/// Esacapes key, writes "\"key\":{" and returns a JSONObjectWriter
	///
	#[inline(always)]
	pub fn object<'a>(&'a mut self, key: &str) -> JSONObjectWriter<'a, W> {
		self.write_key(key);
		JSONObjectWriter::new(self.writer)
	}

	///
	/// Starts writing a nested array with given key:
	///
	/// Esacapes key, writes "\"key\":[" and returns a JSONArrayWriter.
	///
	#[inline(always)]
	pub fn array<'a>(&'a mut self, key: &str) -> JSONArrayWriter<'a, W> {
		self.write_key(key);
		JSONArrayWriter::new(self.writer)
	}

	///
	/// Escapes and appends key:value to the buffer
	///
	#[inline(always)]
	pub fn value<T: JSONWriterValue>(&mut self, key: &str, value: T) {
		self.write_key(key);
		value.write_json(self.writer);
	}

	/// Write string with the given key, where the body of the string is built up using the
	/// `JSONStringWriter` that impls the `fmt::Write` trait and so can be used in the `write!` macro.
	#[inline(always)]
	pub fn string_writer(&mut self, key: &str) -> JSONStringWriter<'_, W> {
		self.write_key(key);
		JSONStringWriter::new(self.writer)
	}

	///
	/// Writes a key without any value.
	///
	/// Consider using the methods value(key, value), object(key) and array(key) instead of using this method directly.
	///
	/// <p style="background:rgba(255,181,77,0.16);padding:0.75em;">
	/// <strong>Warning:</strong>
	/// If you use this method, you will have to write the value to the buffer yourself afterwards.
	/// </p>
	///
	pub fn write_key(&mut self, key: &str) {
		self.writer.json_object_key(key, self.empty);
		self.empty = false;
	}

	///
	/// Drops the writer.
	/// Dropping causes '}' to be appended to the buffer.
	///
	#[inline(always)]
	pub fn end(self) {
		drop(self);
	}
}

impl JSONObjectWriter<'_, String> {
	///
	/// Writes the entire buffer to given writer and clears entire buffer on success.
	///
	#[inline(always)]
	pub fn output_buffered_data<Writer: std::io::Write>(
		&mut self,
		writer: &mut Writer,
	) -> Result<usize, std::io::Error> {
		output_buffer_to(self.writer, writer)
	}

	///
	/// Returns buffer length in bytes
	///
	#[inline(always)]
	pub fn buffer_len(&self) -> usize {
		self.writer.len()
	}
}

impl<'a, W: JSONWriter> Drop for JSONObjectWriter<'a, W> {
	#[inline(always)]
	fn drop(&mut self) {
		self.writer.json_end_object(self.empty);
	}
}

impl<W: JSONWriter> JSONArrayWriter<'_, W> {
	///
	/// Creates a new JSONArrayWriter that writes to the given buffer. Writes '[' to the buffer immediately.
	///
	#[inline(always)]
	pub fn new(writer: &mut W) -> JSONArrayWriter<'_, W> {
		writer.json_begin_array();
		JSONArrayWriter {
			writer,
			empty: true,
		}
	}

	///
	/// Starts writing a nested object as array entry.
	///
	/// Writes '{' and returns a JSONObjectWriter
	///
	#[inline(always)]
	pub fn object(&mut self) -> JSONObjectWriter<'_, W> {
		self.write_comma();
		JSONObjectWriter::new(self.writer)
	}

	///
	/// Starts writing a nested array as array entry.
	///
	/// Writes '[' and returns a JSONArrayWriter
	///
	#[inline(always)]
	pub fn array(&mut self) -> JSONArrayWriter<'_, W> {
		self.write_comma();
		JSONArrayWriter::new(self.writer)
	}

	///
	/// Writes given value as array entry
	///
	#[inline(always)]
	pub fn value<T: JSONWriterValue>(&mut self, value: T) {
		self.write_comma();
		value.write_json(self.writer);
	}

	/// Write string with the given key, where the body of the string is built up using the
	/// `JSONStringWriter` that impls the `fmt::Write` trait and so can be used in the `write!` macro.
	#[inline(always)]
	pub fn string_writer(&mut self) -> JSONStringWriter<'_, W> {
		self.write_comma();
		JSONStringWriter::new(self.writer)
	}

	///
	/// Writes a comma unless at the beginning of the array
	///
	/// <p style="background:rgba(255,181,77,0.16);padding:0.75em;">
	/// <strong>Warning:</strong>
	/// If you use this method, you will have to write the value to the buffer yourself afterwards.
	/// </p>
	///
	#[inline]
	pub fn write_comma(&mut self) {
		self.writer.json_begin_array_value(self.empty);
		self.empty = false;
	}

	///
	/// Drops the writer.
	/// Dropping causes ']' to be appended to the buffer.
	///
	#[inline(always)]
	pub fn end(self) {
		drop(self);
	}
}

impl<W: JSONWriter> Drop for JSONArrayWriter<'_, W> {
	#[inline(always)]
	fn drop(&mut self) {
		self.writer.json_end_array(self.empty);
	}
}

impl JSONArrayWriter<'_, String> {
	///
	/// Writes the entire buffer to given writer and clears entire buffer on success.
	///
	#[inline(always)]
	pub fn output_buffered_data<Writer: std::io::Write>(
		&mut self,
		writer: &mut Writer,
	) -> Result<usize, std::io::Error> {
		output_buffer_to(self.writer, writer)
	}

	///
	/// Returns buffer length in bytes
	///
	#[inline(always)]
	pub fn buffer_len(&self) -> usize {
		self.writer.len()
	}
}

impl JSONWriter for String {
	#[inline(always)]
	fn json_string(&mut self, value: &str) {
		write_string(self, value);
	}

	#[inline(always)]
	fn json_string_part(&mut self, value: &str) {
		write_part_of_string(self, value);
	}

	#[inline(always)]
	fn json_fragment(&mut self, value: &str) {
		self.push_str(value);
	}

	/// Called at the start of writing an object.
	#[inline(always)]
	fn json_begin_object(&mut self) {
		self.push('{');
	}

	/// Called after writing all key-value pairs of an object.
	///
	/// `empty` is `true` when the object contains no key-value pairs.
	#[inline(always)]
	fn json_end_object(&mut self, _empty: bool) {
		self.push('}');
	}

	/// Called at the start of writing an array.
	#[inline(always)]
	fn json_begin_array(&mut self) {
		self.push('[');
	}

	/// Called after writing all items of an array.
	///
	/// `empty` is `true` when the array contains no items.
	#[inline(always)]
	fn json_end_array(&mut self, _empty: bool) {
		self.push(']');
	}

	/// Called before each key-value pair in an object and each item in an array.
	///
	#[inline]
	fn json_begin_array_value(&mut self, first: bool) {
		if !first {
			self.push(',');
		}
	}

	/// Called before each key-value pair in an object and each item in an array.
	///
	fn json_object_key(&mut self, key: &str, first: bool) {
		if !first {
			self.push(',');
		}
		write_string(self, key);
		self.push(':');
	}

	#[inline(always)]
	fn json_begin_string(&mut self) {
		self.push('"');
	}

	#[inline(always)]
	fn json_end_string(&mut self) {
		self.push('"');
	}
}

/// Formats JSON in a human-readable format with whitespace, newlines, and indentation.
pub struct PrettyJSONWriter<'a> {
	/// Result
	pub buffer: &'a mut String,
	indent: &'a str,
	depth: usize,
}

impl PrettyJSONWriter<'_> {
	/// Creates a new human-readable formatter with two spaces for indentation.
	pub fn new(buffer: &mut String) -> PrettyJSONWriter<'_> {
		// Same default as serde_json::ser::PrettyFormatter
		PrettyJSONWriter {
			buffer,
			indent: "  ",
			depth: 0,
		}
	}

	/// Creates a new formatter using `indent` for indentation.
	pub fn with_indent<'a>(buffer: &'a mut String, indent: &'a str) -> PrettyJSONWriter<'a> {
		PrettyJSONWriter {
			buffer,
			indent,
			depth: 0,
		}
	}

	fn write_indent(&mut self) {
		for _ in 0..self.depth {
			self.buffer.push_str(self.indent);
		}
	}
}

impl JSONWriter for PrettyJSONWriter<'_> {
	fn json_begin_object(&mut self) {
		self.depth += 1;
		self.buffer.push('{');
	}

	fn json_end_object(&mut self, empty: bool) {
		self.depth -= 1;
		if !empty {
			self.buffer.push('\n');
			self.write_indent();
		}
		self.buffer.push('}');
	}

	fn json_begin_array(&mut self) {
		self.depth += 1;
		self.buffer.push('[');
	}

	fn json_end_array(&mut self, empty: bool) {
		self.depth -= 1;
		if !empty {
			self.buffer.push('\n');
			self.write_indent();
		}
		self.buffer.push(']');
	}

	fn json_begin_array_value(&mut self, first: bool) {
		self.buffer.push_str(if first { "\n" } else { ",\n" });
		self.write_indent();
	}

	fn json_object_key(&mut self, key: &str, first: bool) {
		self.buffer.push_str(if first { "\n" } else { ",\n" });
		self.write_indent();
		write_string(self.buffer, key);
		self.buffer.push_str(": ");
	}

	fn json_string(&mut self, value: &str) {
		write_string(self.buffer, value);
	}

	fn json_string_part(&mut self, value: &str) {
		write_part_of_string(self.buffer, value);
	}

	fn json_fragment(&mut self, value: &str) {
		self.buffer.push_str(value);
	}
}

///
/// Types with this trait can be converted to JSON
///
pub trait JSONWriterValue {
	///
	/// Appends a JSON representation of self to the output buffer
	///
	fn write_json<W: JSONWriter>(self, writer: &mut W);
}

impl JSONWriterValue for &str {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		writer.json_string(self);
	}
}

impl JSONWriterValue for &std::borrow::Cow<'_, str> {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		writer.json_string(AsRef::as_ref(self));
	}
}

impl JSONWriterValue for &String {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		writer.json_string(self);
	}
}

impl JSONWriterValue for f64 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		writer.json_number_f64(self);
	}
}

impl JSONWriterValue for f32 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		writer.json_number_f64(self as f64);
	}
}

impl JSONWriterValue for u128 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for i128 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for u64 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for i64 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for u32 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for i32 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for u16 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for i16 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for u8 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for i8 {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut buf = itoa::Buffer::new();
		writer.json_number_str(buf.format(self));
	}
}

impl JSONWriterValue for bool {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		writer.json_bool(self);
	}
}

impl JSONWriterValue for Null {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		writer.json_null();
	}
}

impl<T: JSONWriterValue + Copy> JSONWriterValue for &T {
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		(*self).write_json(writer);
	}
}

impl<T: JSONWriterValue> JSONWriterValue for Option<T> {
	#[inline]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		match self {
			None => {
				writer.json_null();
			}
			Some(value) => {
				value.write_json(writer);
			}
		}
	}
}

impl<Item> JSONWriterValue for &Vec<Item>
where
	for<'b> &'b Item: JSONWriterValue,
{
	#[inline(always)]
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		self.as_slice().write_json(writer);
	}
}

impl<Item> JSONWriterValue for &[Item]
where
	for<'b> &'b Item: JSONWriterValue,
{
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut array = JSONArrayWriter::new(writer);
		for item in self.iter() {
			array.value(item);
		}
	}
}

impl<Key: AsRef<str>, Item> JSONWriterValue for &std::collections::HashMap<Key, Item>
where
	for<'b> &'b Item: JSONWriterValue,
{
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut obj = JSONObjectWriter::new(writer);
		for (key, value) in self.iter() {
			obj.value(key.as_ref(), value);
		}
	}
}

impl<Key: AsRef<str>, Item> JSONWriterValue for &std::collections::BTreeMap<Key, Item>
where
	for<'b> &'b Item: JSONWriterValue,
{
	fn write_json<W: JSONWriter>(self, writer: &mut W) {
		let mut obj = JSONObjectWriter::new(writer);
		for (key, value) in self.iter() {
			obj.value(key.as_ref(), value);
		}
	}
}

///
/// Converts given value to a json string.
///
#[inline]
pub fn to_json_string<T: JSONWriterValue>(v: T) -> String {
	let mut result = String::new();
	v.write_json(&mut result);
	return result;
}

fn output_buffer_to<Writer: std::io::Write>(
	buffer: &mut String,
	writer: &mut Writer,
) -> Result<usize, std::io::Error> {
	match writer.write_all(buffer.as_bytes()) {
		Ok(_) => {
			let len = buffer.len();
			buffer.clear();
			Ok(len)
		}
		Err(err) => Err(err),
	}
}

///
/// Quotes and escapes input and appends result to output buffer
///
#[inline(never)]
pub fn write_string(output_buffer: &mut String, input: &str) {
	output_buffer.push('"');
	write_part_of_string_impl(output_buffer, input);
	output_buffer.push('"');
}

///
/// Escapes input and appends result to output buffer without adding quotes.
///
#[inline(never)]
pub fn write_part_of_string(output_buffer: &mut String, input: &str) {
	write_part_of_string_impl(output_buffer, input);
}

const fn get_replacements() -> [u8; 256] {
	// NOTE: Only characters smaller than 128 are allowed here.
	// Trying to escape values above 128 would generate invalid utf-8 output
	// -----
	// see https://www.json.org/json-en.html
	let mut result = [0u8; 256];
	// Escape everything from 0 to 0x1F
	let mut i = 0;
	while i < 0x20 {
		result[i] = b'u';
		i += 1;
	}
	result[b'\"' as usize] = b'"';
	result[b'\\' as usize] = b'\\';
	result[b'/' as usize] = b'/';
	result[8] = b'b';
	result[0xc] = b'f';
	result[b'\n' as usize] = b'n';
	result[b'\r' as usize] = b'r';
	result[b'\t' as usize] = b't';
	result[0] = b'u';

	return result;
}

static REPLACEMENTS: [u8; 256] = get_replacements();
static HEX: [u8; 16] = *b"0123456789ABCDEF";

///
/// Escapes and append part of string
///
#[inline(always)]
fn write_part_of_string_impl(output_buffer: &mut String, input: &str) {
	// All of the relevant characters are in the ansi range (<128).
	// This means we can safely ignore any utf-8 characters and iterate over the bytes directly
	let mut num_bytes_written: usize = 0;
	let mut index: usize = 0;
	let bytes = input.as_bytes();
	while index < bytes.len() {
		let cur_byte = bytes[index];
		let replacement = REPLACEMENTS[cur_byte as usize];
		if replacement != 0 {
			if num_bytes_written < index {
				// Checks can be omitted here:
				// We know that index is smaller than the output_buffer length.
				// We also know that num_bytes_written is smaller than index
				// We also know that the boundaries are not in the middle of an utf-8 multi byte sequence, because those characters are not escaped
				output_buffer.push_str(unsafe { input.get_unchecked(num_bytes_written..index) });
			}
			if replacement == b'u' {
				let bytes: [u8; 6] = [
					b'\\',
					b'u',
					b'0',
					b'0',
					HEX[((cur_byte / 16) & 0xF) as usize],
					HEX[(cur_byte & 0xF) as usize],
				];
				// Checks can be omitted here: We know bytes is a valid utf-8 string (see above)
				output_buffer.push_str(unsafe { std::str::from_utf8_unchecked(&bytes) });
			} else {
				let bytes: [u8; 2] = [b'\\', replacement];
				// Checks can be omitted here: We know bytes is a valid utf-8 string, because the replacement table only contains characters smaller than 128
				output_buffer.push_str(unsafe { std::str::from_utf8_unchecked(&bytes) });
			}
			num_bytes_written = index + 1;
		}
		index += 1;
	}
	if num_bytes_written < bytes.len() {
		// Checks can be omitted here:
		// We know that num_bytes_written is smaller than index
		// We also know that num_bytes_written not in the middle of an utf-8 multi byte sequence, because those are not escaped
		output_buffer.push_str(unsafe { input.get_unchecked(num_bytes_written..bytes.len()) });
	}
}

#[cfg(test)]
mod tests {
	use super::*;

	#[test]
	fn test_array() {
		let mut buffer = String::new();
		let mut array = JSONArrayWriter::new(&mut buffer);
		array.value(0u8);
		array.value(1i32);
		array.value("2");
		array.value("\"<script>1/2</script>\"");
		let mut nested_arr = array.array();
		nested_arr.value("nested");
		nested_arr.end();
		let mut nested_obj = array.object();
		nested_obj.value("ä\töü", "ä\töü");
		nested_obj.end();
		let nested_obj2 = array.object();
		nested_obj2.end();
		drop(array);

		assert_eq!(buffer, "[0,1,\"2\",\"\\\"<script>1\\/2<\\/script>\\\"\",[\"nested\"],{\"ä\\töü\":\"ä\\töü\"},{}]");
	}

	#[test]
	fn test_array_range() {
		let bytes = b"ABC";
		assert_eq!(to_json_string(&bytes[..]), "[65,66,67]");

		let mut v = Vec::<u8>::new();
		v.extend_from_slice(bytes);
		assert_eq!(to_json_string(&v), "[65,66,67]");
	}

	#[test]
	fn test_object() {
		let mut map = std::collections::HashMap::<String, String>::new();
		map.insert("a".to_owned(), "a".to_owned());
		assert_eq!(to_json_string(&map), "{\"a\":\"a\"}");
	}

	#[allow(clippy::approx_constant)]
	#[test]
	fn test_numbers() {
		// unsigned
		assert_eq!(to_json_string(1u8), "1");
		assert_eq!(to_json_string(1u16), "1");
		assert_eq!(to_json_string(1u32), "1");
		assert_eq!(to_json_string(u8::MAX), "255");
		assert_eq!(to_json_string(u16::MAX), "65535");
		assert_eq!(to_json_string(u32::MAX), "4294967295");
		assert_eq!(to_json_string(u64::MAX), "18446744073709551615");
		assert_eq!(
			to_json_string(u128::MAX),
			"340282366920938463463374607431768211455"
		);

		// signed
		assert_eq!(to_json_string(-1i8), "-1");
		assert_eq!(to_json_string(-1i16), "-1");
		assert_eq!(to_json_string(-1i32), "-1");
		assert_eq!(to_json_string(-1i64), "-1");
		assert_eq!(to_json_string(-1i128), "-1");

		// float
		assert_eq!(to_json_string(0f32), "0");
		assert_eq!(to_json_string(2f32), "2");
		assert_eq!(to_json_string(-2f32), "-2");

		assert_eq!(to_json_string(0f64), "0");
		assert_eq!(to_json_string(2f64), "2");
		assert_eq!(to_json_string(-2f64), "-2");
		assert_eq!(to_json_string(3.141592653589793), "3.141592653589793");
		assert_eq!(to_json_string(0.1f64), "0.1");
		assert_eq!(to_json_string(-0.1f64), "-0.1");
		//assert_eq!(to_json_string(-5.0/3.0), "-1.6666666666666667");
		assert_eq!(to_json_string(1.5e30f64), "1.5e30");
		assert_eq!(
			to_json_string(-2.220446049250313e-16f64),
			"-2.220446049250313e-16"
		);

		assert_eq!(to_json_string(1.0 / 0.0), "null");
		assert_eq!(to_json_string(f64::INFINITY), "null");
		assert_eq!(to_json_string(f64::NEG_INFINITY), "null");
		assert_eq!(to_json_string(f64::NAN), "null");
	}

	#[test]
	fn test_dtoa() {
		assert_dtoa(0.0);
		assert_dtoa(1.0);
		assert_dtoa(-1.0);
		assert_dtoa(2.0);
		//assert_dtoa(-5.0/3.0);
	}

	fn assert_dtoa(v: f64) {
		let a = v.to_string();
		let mut b = String::new();
		b.json_number_f64(v);
		assert_eq!(b, a);
	}

	#[test]
	fn test_strings() {
		assert_eq!(
			to_json_string("中文\0\x08\x09\"\\\n\r\t</script>"),
			"\"中文\\u0000\\b\\t\\\"\\\\\\n\\r\\t<\\/script>\""
		);
	}

	#[test]
	fn test_basic_example() {
		let mut object_str = String::new();
		{
			let mut object_writer = JSONObjectWriter::new(&mut object_str);
			object_writer.value("number", 42i32);
		}
		assert_eq!(&object_str, "{\"number\":42}");
	}

	#[allow(clippy::approx_constant)]
	#[test]
	fn test_misc_examples() {
		// Values
		assert_eq!(to_json_string("Hello World\n"), "\"Hello World\\n\"");
		assert_eq!(to_json_string(3.141592653589793f64), "3.141592653589793");
		assert_eq!(to_json_string(true), "true");
		assert_eq!(to_json_string(false), "false");
		assert_eq!(to_json_string(NULL), "null");

		// Options of values
		assert_eq!(to_json_string(Option::<u8>::Some(42)), "42");
		assert_eq!(to_json_string(Option::<u8>::None), "null");

		// Slices and vectors
		let numbers: [u8; 4] = [1, 2, 3, 4];
		assert_eq!(to_json_string(&numbers[..]), "[1,2,3,4]");
		let numbers_vec: Vec<u8> = vec![1u8, 2u8, 3u8, 4u8];
		assert_eq!(to_json_string(&numbers_vec), "[1,2,3,4]");
		let strings: [&str; 4] = ["a", "b", "c", "d"];
		assert_eq!(to_json_string(&strings[..]), "[\"a\",\"b\",\"c\",\"d\"]");

		// Hash-maps:
		let mut map = std::collections::HashMap::<String, String>::new();
		map.insert("Hello".to_owned(), "World".to_owned());
		assert_eq!(to_json_string(&map), "{\"Hello\":\"World\"}");

		// Objects:
		let mut object_str = String::new();
		let mut object_writer = JSONObjectWriter::new(&mut object_str);

		// Values
		object_writer.value("number", 42i32);
		object_writer.value("slice", &numbers[..]);

		// Nested arrays
		let mut nested_array = object_writer.array("array");
		nested_array.value(42u32);
		nested_array.value("?");
		nested_array.end();

		// Nested objects
		let nested_object = object_writer.object("object");
		nested_object.end();

		object_writer.end();
		assert_eq!(
			&object_str,
			"{\"number\":42,\"slice\":[1,2,3,4],\"array\":[42,\"?\"],\"object\":{}}"
		);
	}

	#[test]
	fn test_string_writer() {
		use core::fmt::Write;

		let mut object_str = String::new();
		let mut object_writer = JSONObjectWriter::new(&mut object_str);

		let name = r#"zenora "bariella""#;
		let color = "yellow";
		object_writer.value("name", name);
		let mut w = object_writer.string_writer("compound");
		write!(w, "{name} : {color}").unwrap();
		w.end();

		object_writer.value("number", 42i32);

		{
			let mut array = object_writer.array("tools");
			let prefix = "air";
			array.value("hammer");
			write!(array.string_writer(), "{prefix}-hammer").unwrap();
			write!(array.string_writer(), "{prefix}-saw").unwrap();
		}

		object_writer.end();

		eprintln!("{}", object_str);

		assert_eq!(
			&object_str,
			r#"{"name":"zenora \"bariella\"","compound":"zenora \"bariella\" : yellow","number":42,"tools":["hammer","air-hammer","air-saw"]}"#
		)
	}

	#[test]
	fn test_duplicate_keys() {
		let mut object_str = String::new();
		{
			let mut object_writer = JSONObjectWriter::new(&mut object_str);
			object_writer.value("number", 42i32);
			object_writer.value("number", 43i32);
		}
		// Duplicates are not checked, this is by design!
		assert_eq!(&object_str, "{\"number\":42,\"number\":43}");
	}

	#[test]
	fn test_flush() {
		// this could also be a file writer.
		let mut writer = Vec::<u8>::new();

		let mut buffer = String::new();
		let mut array = JSONArrayWriter::new(&mut buffer);
		for i in 1i32..=1000000i32 {
			array.value(i);
			if array.buffer_len() > 2000 {
				array.output_buffered_data(&mut writer).unwrap();
			}
		}
		array.end();
		std::io::Write::write_all(&mut writer, buffer.as_bytes()).unwrap();

		assert!(buffer.len() <= 4000, "Buffer too long");
		assert_eq!(
			&writer[writer.len() - b",999999,1000000]".len()..],
			b",999999,1000000]"
		);
	}

	#[test]
	fn test_encoding() {
		for c in 0x00..0x20 {
			let c = char::from(c);
			let json = to_json_string(c.to_string().as_str());
			assert!(&json[0..2] == "\"\\");
		}
		assert_eq!(
			to_json_string("</script >\0\x1F"),
			"\"<\\/script >\\u0000\\u001F\""
		);
	}

	#[test]
	fn test_pretty() {
		let mut buffer = String::new();
		let mut formatter = PrettyJSONWriter::with_indent(&mut buffer, "   ");
		let mut writer = JSONObjectWriter::new(&mut formatter);
		{
			let mut nested_writer = writer.object("nested");
			nested_writer.value("a", 3);
			nested_writer.value("b", &vec![0, 1, 4]);
		}
		writer.value("c", &vec![true, false, true]);
		writer.value("d", NULL);
		// empty object
		writer.object("e");
		writer.array("f");
		writer.end();
		assert_eq!(
			formatter.buffer,
			r#"{
   "nested": {
      "a": 3,
      "b": [
         0,
         1,
         4
      ]
   },
   "c": [
      true,
      false,
      true
   ],
   "d": null,
   "e": {},
   "f": []
}"#
		);
	}
}