use std::fmt;
use serde::{
Deserialize, Deserializer,
de::{self, MapAccess, SeqAccess, Visitor},
};
use serde_json::{Map, Number, Value};
pub(super) struct UniqueValue(pub Value);
struct UniqueVisitor;
impl<'de> Deserialize<'de> for UniqueValue {
fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
deserializer.deserialize_any(UniqueVisitor).map(Self)
}
}
impl<'de> Visitor<'de> for UniqueVisitor {
type Value = Value;
fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
formatter.write_str("configuration JSON with unique object keys")
}
fn visit_bool<E: de::Error>(self, value: bool) -> Result<Value, E> {
Ok(Value::Bool(value))
}
fn visit_i64<E: de::Error>(self, value: i64) -> Result<Value, E> {
Ok(Value::Number(value.into()))
}
fn visit_u64<E: de::Error>(self, value: u64) -> Result<Value, E> {
Ok(Value::Number(value.into()))
}
fn visit_f64<E: de::Error>(self, value: f64) -> Result<Value, E> {
Number::from_f64(value)
.map(Value::Number)
.ok_or_else(|| E::custom("nonfinite configuration number"))
}
fn visit_str<E: de::Error>(self, value: &str) -> Result<Value, E> {
Ok(Value::String(value.into()))
}
fn visit_string<E: de::Error>(self, value: String) -> Result<Value, E> {
Ok(Value::String(value))
}
fn visit_unit<E: de::Error>(self) -> Result<Value, E> {
Ok(Value::Null)
}
fn visit_seq<A: SeqAccess<'de>>(self, mut sequence: A) -> Result<Value, A::Error> {
let mut items = Vec::new();
while let Some(UniqueValue(value)) = sequence.next_element()? {
items.push(value);
}
Ok(Value::Array(items))
}
fn visit_map<A: MapAccess<'de>>(self, mut object: A) -> Result<Value, A::Error> {
let mut fields = Map::new();
while let Some(key) = object.next_key::<String>()? {
if fields.contains_key(&key) {
return Err(de::Error::custom("duplicate configuration JSON field"));
}
let UniqueValue(value) = object.next_value()?;
fields.insert(key, value);
}
Ok(Value::Object(fields))
}
}