use std::fmt;
#[derive(Clone)]
pub struct Token(String);
impl Token {
pub fn new(raw: impl Into<String>) -> Self {
Self(raw.into())
}
pub fn as_str(&self) -> &str {
&self.0
}
}
impl fmt::Debug for Token {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let prefix = match self.0.split_once('_') {
Some((p, _)) => p,
None => "",
};
if prefix.is_empty() {
write!(f, "Token(***redacted)")
} else {
write!(f, "Token({prefix}_***redacted)")
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn debug_redacts_secret() {
let secret = "wk_abcdef1234567890";
let t = Token::new(secret);
let dbg = format!("{t:?}");
assert!(dbg.contains("***"), "{dbg} should contain ***");
assert!(!dbg.contains(secret), "{dbg} leaked the secret");
assert!(dbg.contains("wk_"), "{dbg} should keep the prefix");
}
#[test]
fn debug_handles_no_prefix() {
let t = Token::new("rawvalue");
let dbg = format!("{t:?}");
assert!(!dbg.contains("rawvalue"));
assert!(dbg.contains("***"));
}
#[test]
fn as_str_returns_raw() {
let t = Token::new("wk_xyz");
assert_eq!(t.as_str(), "wk_xyz");
}
}