use std::fmt;
#[derive(Clone, PartialEq, Eq)]
pub struct Secret(String);
impl Secret {
pub fn new(value: impl Into<String>) -> Self {
Self(value.into())
}
pub fn reveal(&self) -> &str {
&self.0
}
pub fn is_empty(&self) -> bool {
self.0.is_empty()
}
}
impl fmt::Debug for Secret {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str("Secret(***)")
}
}
impl fmt::Display for Secret {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str("***")
}
}
impl From<String> for Secret {
fn from(s: String) -> Self {
Self(s)
}
}
impl From<&str> for Secret {
fn from(s: &str) -> Self {
Self(s.to_string())
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn debug_redacts_value() {
let s = Secret::new("hunter2");
assert_eq!(format!("{:?}", s), "Secret(***)");
assert!(!format!("{:?}", s).contains("hunter2"));
}
#[test]
fn display_redacts_value() {
let s = Secret::new("hunter2");
assert_eq!(format!("{}", s), "***");
}
#[test]
fn reveal_returns_underlying_value() {
let s = Secret::new("hunter2");
assert_eq!(s.reveal(), "hunter2");
}
#[test]
fn redaction_survives_nested_debug() {
#[derive(Debug)]
#[allow(dead_code)] struct Wrap {
token: Secret,
other: u32,
}
let w = Wrap {
token: Secret::new("hunter2"),
other: 42,
};
let printed = format!("{:?}", w);
assert!(!printed.contains("hunter2"));
assert!(printed.contains("Secret(***)"));
assert!(printed.contains("42"));
}
}