use std::fmt;
use std::str::FromStr;
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum Visibility {
#[default]
Public,
Authenticated,
Organization,
Private,
}
impl Visibility {
pub fn as_str(&self) -> &'static str {
match self {
Self::Public => "public",
Self::Authenticated => "authenticated",
Self::Organization => "organization",
Self::Private => "private",
}
}
pub fn is_public(&self) -> bool {
matches!(self, Self::Public)
}
pub fn is_redacted_on_dht(&self) -> bool {
!self.is_public()
}
}
impl fmt::Display for Visibility {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(self.as_str())
}
}
impl FromStr for Visibility {
type Err = String;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"public" => Ok(Self::Public),
"authenticated" => Ok(Self::Authenticated),
"organization" => Ok(Self::Organization),
"private" => Ok(Self::Private),
_ => Err(format!(
"unknown visibility '{s}'; expected one of: public, authenticated, organization, private"
)),
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn default_is_public() {
assert_eq!(Visibility::default(), Visibility::Public);
}
#[test]
fn public_is_not_redacted() {
assert!(!Visibility::Public.is_redacted_on_dht());
}
#[test]
fn non_public_is_redacted() {
assert!(Visibility::Authenticated.is_redacted_on_dht());
assert!(Visibility::Organization.is_redacted_on_dht());
assert!(Visibility::Private.is_redacted_on_dht());
}
#[test]
fn round_trip_all_variants() {
for vis in [
Visibility::Public,
Visibility::Authenticated,
Visibility::Organization,
Visibility::Private,
] {
let s = vis.as_str();
let parsed = Visibility::from_str(s);
assert!(parsed.is_ok());
assert_eq!(parsed.unwrap_or(Visibility::Public), vis);
}
}
#[test]
fn serde_round_trip() {
let vis = Visibility::Organization;
let json = serde_json::to_string(&vis).unwrap_or_default();
assert_eq!(json, r#""organization""#);
let parsed: Visibility = serde_json::from_str(&json).unwrap_or(Visibility::Public);
assert_eq!(parsed, vis);
}
#[test]
fn unknown_returns_error() {
assert!(Visibility::from_str("secret").is_err());
}
}