use serde::{Deserialize, Deserializer, Serialize, Serializer};
#[non_exhaustive]
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum VulnerabilityStatus {
OPEN,
MUTED,
REMEDIATED,
INPROGRESS,
AUTOCLOSED,
UnparsedObject(crate::datadog::UnparsedObject),
}
impl ToString for VulnerabilityStatus {
fn to_string(&self) -> String {
match self {
Self::OPEN => String::from("Open"),
Self::MUTED => String::from("Muted"),
Self::REMEDIATED => String::from("Remediated"),
Self::INPROGRESS => String::from("InProgress"),
Self::AUTOCLOSED => String::from("AutoClosed"),
Self::UnparsedObject(v) => v.value.to_string(),
}
}
}
impl Serialize for VulnerabilityStatus {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
match self {
Self::UnparsedObject(v) => v.serialize(serializer),
_ => serializer.serialize_str(self.to_string().as_str()),
}
}
}
impl<'de> Deserialize<'de> for VulnerabilityStatus {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
let s: String = String::deserialize(deserializer)?;
Ok(match s.as_str() {
"Open" => Self::OPEN,
"Muted" => Self::MUTED,
"Remediated" => Self::REMEDIATED,
"InProgress" => Self::INPROGRESS,
"AutoClosed" => Self::AUTOCLOSED,
_ => Self::UnparsedObject(crate::datadog::UnparsedObject {
value: serde_json::Value::String(s.into()),
}),
})
}
}