use serde::{Deserialize, Deserializer, Serialize, Serializer};
#[non_exhaustive]
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum VulnerabilitySeverity {
UNKNOWN,
NONE,
LOW,
MEDIUM,
HIGH,
CRITICAL,
UnparsedObject(crate::datadog::UnparsedObject),
}
impl ToString for VulnerabilitySeverity {
fn to_string(&self) -> String {
match self {
Self::UNKNOWN => String::from("Unknown"),
Self::NONE => String::from("None"),
Self::LOW => String::from("Low"),
Self::MEDIUM => String::from("Medium"),
Self::HIGH => String::from("High"),
Self::CRITICAL => String::from("Critical"),
Self::UnparsedObject(v) => v.value.to_string(),
}
}
}
impl Serialize for VulnerabilitySeverity {
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 VulnerabilitySeverity {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
let s: String = String::deserialize(deserializer)?;
Ok(match s.as_str() {
"Unknown" => Self::UNKNOWN,
"None" => Self::NONE,
"Low" => Self::LOW,
"Medium" => Self::MEDIUM,
"High" => Self::HIGH,
"Critical" => Self::CRITICAL,
_ => Self::UnparsedObject(crate::datadog::UnparsedObject {
value: serde_json::Value::String(s.into()),
}),
})
}
}