use crate::Source;
use serde::{
Deserialize, Deserializer, Serialize, Serializer,
de::{self, Visitor},
};
use std::fmt;
impl Serialize for Source {
fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
serializer.serialize_str(&self.to_string())
}
}
impl<'de> Deserialize<'de> for Source {
fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
deserializer.deserialize_str(SourceVisitor)
}
}
struct SourceVisitor;
impl<'de> Visitor<'de> for SourceVisitor {
type Value = Source;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
formatter.write_str("a configuration source string such as `env(prefix=APP)`")
}
fn visit_str<E: de::Error>(self, value: &str) -> Result<Self::Value, E> {
Source::parse(value).map_err(|error| E::custom(error.to_string()))
}
fn visit_string<E: de::Error>(self, value: String) -> Result<Self::Value, E> {
self.visit_str(value.as_str())
}
}