use crate::severity::Severity;
#[derive(Copy, Clone, Debug, Default, PartialEq, PartialOrd)]
pub struct Score(f64);
impl Score {
pub fn new(score: f64) -> Score {
Score(score)
}
pub fn value(self) -> f64 {
self.0
}
#[cfg(feature = "std")]
pub fn roundup(self) -> Score {
let score_int = (self.0 * 100_000.0) as u64;
if score_int % 10000 == 0 {
Score((score_int as f64) / 100_000.0)
} else {
let score_floor = ((score_int as f64) / 10_000.0).floor();
Score((score_floor + 1.0) / 10.0)
}
}
pub fn severity(self) -> Severity {
if self.0 < 0.1 {
Severity::None
} else if self.0 < 4.0 {
Severity::Low
} else if self.0 < 7.0 {
Severity::Medium
} else if self.0 < 9.0 {
Severity::High
} else {
Severity::Critical
}
}
}
impl From<f64> for Score {
fn from(score: f64) -> Score {
Score(score)
}
}
impl From<Score> for f64 {
fn from(score: Score) -> f64 {
score.value()
}
}
impl From<Score> for Severity {
fn from(score: Score) -> Severity {
score.severity()
}
}