serde_v8vs 0.1.0

use alloc::collections::BTreeSet;
#[cfg(any(test, feature = "std"))]
use std::collections::HashSet;

use serde_core::ser::{Impossible, Serialize, Serializer};

use crate::{
	Error, SerializationTag,
	segmented_buf::SegmentedBuf,
	ser::{V8Serializer, Write}
};

pub(crate) const SET_MAGIC: &str = "{a1e80447-dceb-4f1a-b322-8173037601d0}";

/// Serialize a "set-ish" type as a JS [`Set`](https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Set).
/// The default behavior would serialize them as an array.
///
/// Supported "set-ish" types are:
/// - [`HashSet<T>`]
/// - [`BTreeSet<T>`]
pub fn as_set<T: AsSet, S>(set: &T, serializer: S) -> Result<S::Ok, S::Error>
where
	S: Serializer
{
	set.serialize_as_set(serializer)
}

pub trait AsSet {
	fn serialize_as_set<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
	where
		S: Serializer;
}

#[cfg(any(test, feature = "std"))]
impl<T: Serialize, St> AsSet for HashSet<T, St> {
	fn serialize_as_set<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
	where
		S: Serializer
	{
		serializer.serialize_newtype_struct(SET_MAGIC, self)
	}
}

impl<T: Serialize> AsSet for BTreeSet<T> {
	fn serialize_as_set<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
	where
		S: Serializer
	{
		serializer.serialize_newtype_struct(SET_MAGIC, self)
	}
}

pub(crate) struct SetSerializer<'a, W: Write>
where
	Error: From<W::Error>
{
	pub(crate) ser: &'a mut V8Serializer<W>
}

impl<'a, W: Write> Serializer for SetSerializer<'a, W>
where
	Error: From<W::Error>
{
	type Ok = ();
	type Error = Error;

	type SerializeSeq = SerializeSet<'a, W>;
	type SerializeMap = Impossible<Self::Ok, Self::Error>;
	type SerializeStruct = Impossible<Self::Ok, Self::Error>;
	type SerializeStructVariant = Impossible<Self::Ok, Self::Error>;
	type SerializeTuple = Impossible<Self::Ok, Self::Error>;
	type SerializeTupleStruct = Impossible<Self::Ok, Self::Error>;
	type SerializeTupleVariant = Impossible<Self::Ok, Self::Error>;

	fn serialize_seq(self, _size_hint: Option<usize>) -> Result<Self::SerializeSeq, Self::Error> {
		self.ser.writer.push(SerializationTag::BeginJSSet as u8)?;
		Ok(SerializeSet {
			subser: V8Serializer::<SegmentedBuf>::new(SegmentedBuf::new(self.ser.options.grow_strategy.clone()), self.ser.options.clone()),
			ser: self.ser,
			len: 0
		})
	}

	fn serialize_bool(self, _: bool) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_bytes(self, _: &[u8]) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_char(self, _: char) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_f32(self, _: f32) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_f64(self, _: f64) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_i16(self, _: i16) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_i128(self, _: i128) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_i32(self, _: i32) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_i64(self, _: i64) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_i8(self, _: i8) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_map(self, _: Option<usize>) -> Result<Self::SerializeMap, Self::Error> {
		unreachable!()
	}
	fn serialize_newtype_struct<T>(self, _: &'static str, _: &T) -> Result<Self::Ok, Self::Error>
	where
		T: ?Sized + Serialize
	{
		unreachable!()
	}
	fn serialize_newtype_variant<T>(self, _: &'static str, _: u32, _: &'static str, _: &T) -> Result<Self::Ok, Self::Error>
	where
		T: ?Sized + Serialize
	{
		unreachable!()
	}
	fn serialize_none(self) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_some<T>(self, _: &T) -> Result<Self::Ok, Self::Error>
	where
		T: ?Sized + Serialize
	{
		unreachable!()
	}
	fn serialize_str(self, _: &str) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_struct(self, _: &'static str, _: usize) -> Result<Self::SerializeStruct, Self::Error> {
		unreachable!()
	}
	fn serialize_struct_variant(self, _: &'static str, _: u32, _: &'static str, _: usize) -> Result<Self::SerializeStructVariant, Self::Error> {
		unreachable!()
	}
	fn serialize_tuple(self, _: usize) -> Result<Self::SerializeTuple, Self::Error> {
		unreachable!()
	}
	fn serialize_tuple_struct(self, _: &'static str, _: usize) -> Result<Self::SerializeTupleStruct, Self::Error> {
		unreachable!()
	}
	fn serialize_tuple_variant(self, _: &'static str, _: u32, _: &'static str, _: usize) -> Result<Self::SerializeTupleVariant, Self::Error> {
		unreachable!()
	}
	fn serialize_u128(self, _: u128) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_u16(self, _: u16) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_u32(self, _: u32) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_u64(self, _: u64) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_u8(self, _: u8) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_unit(self) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_unit_struct(self, _: &'static str) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
	fn serialize_unit_variant(self, _: &'static str, _: u32, _: &'static str) -> Result<Self::Ok, Self::Error> {
		unreachable!()
	}
}

pub struct SerializeSet<'a, W: Write>
where
	Error: From<W::Error>
{
	ser: &'a mut V8Serializer<W>,
	subser: V8Serializer<SegmentedBuf>,
	len: usize
}

impl<'a, W: Write> serde_core::ser::SerializeSeq for SerializeSet<'a, W>
where
	Error: From<W::Error>
{
	type Ok = ();
	type Error = Error;

	fn serialize_element<T>(&mut self, value: &T) -> Result<(), Self::Error>
	where
		T: ?Sized + serde_core::Serialize
	{
		value.serialize(&mut self.subser)?;
		self.len += 1;
		Ok(())
	}

	fn end(self) -> Result<Self::Ok, Self::Error> {
		self.ser.writer.take_from(self.subser.writer)?;
		self.ser.writer.push(SerializationTag::EndJSSet as _)?;
		self.ser.write_var_int(self.len as _)?;
		Ok(())
	}
}