use crate::*;
use serde::de::{self, Visitor};
use serde::{Deserialize, Deserializer, Serialize, Serializer};
impl Serialize for SecretValue {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_str(
String::from_utf8(self.ref_sensitive_value().clone())
.map_err(serde::ser::Error::custom)?
.as_str(),
)
}
}
struct SecretValueVisitor;
impl<'de> Visitor<'de> for SecretValueVisitor {
type Value = SecretValue;
fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
formatter.write_str("a string expected as a secret value")
}
fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
where
E: de::Error,
{
Ok(SecretValue::new(value.as_bytes().to_vec()))
}
fn visit_string<E>(self, value: String) -> Result<Self::Value, E>
where
E: de::Error,
{
Ok(SecretValue::new(value.as_bytes().to_vec()))
}
fn visit_bytes<E>(self, value: &[u8]) -> Result<Self::Value, E>
where
E: de::Error,
{
Ok(SecretValue::new(value.to_vec()))
}
fn visit_byte_buf<E>(self, value: Vec<u8>) -> Result<Self::Value, E>
where
E: de::Error,
{
Ok(SecretValue::new(value))
}
}
impl<'de> Deserialize<'de> for SecretValue {
fn deserialize<D>(deserializer: D) -> Result<SecretValue, D::Error>
where
D: Deserializer<'de>,
{
deserializer.deserialize_string(SecretValueVisitor)
}
}
#[cfg(test)]
mod test {
use super::*;
use proptest::prelude::*;
proptest! {
#[test]
fn serialize_as_string(mock_secret in "[a-zA-Z0-9]*") {
let mock_secret_vec = mock_secret.as_bytes().to_vec();
let secret_value = SecretValue::new(mock_secret_vec);
let test_serialized_string = serde_json::to_string(&secret_value).unwrap();
assert_eq!(test_serialized_string, format!("\"{}\"",mock_secret));
}
#[test]
fn deserialize_from_string(mock_secret in "[a-zA-Z0-9]*") {
let mock_secret_quoted = format!("\"{}\"",mock_secret);
let secret_value: SecretValue = serde_json::from_str(&mock_secret_quoted).unwrap();
assert_eq!(String::from_utf8(secret_value.ref_sensitive_value().clone()).unwrap(), mock_secret);
}
}
}