use crate::directions::error::Error as DirectionsError;
use crate::error::Error as GoogleMapsError;
use phf::phf_map;
use serde::{Deserialize, Deserializer, Serialize, Serializer};
#[derive(Clone, Debug, Default, Eq, Hash, Ord, PartialEq, PartialOrd)]
#[repr(u8)]
pub enum TrafficModel {
#[default]
BestGuess = 0,
Optimistic = 1,
Pessimistic = 2,
}
impl<'de> Deserialize<'de> for TrafficModel {
fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
let string = String::deserialize(deserializer)?;
match Self::try_from(string.as_str()) {
Ok(variant) => Ok(variant),
Err(error) => Err(serde::de::Error::custom(error.to_string())),
} } }
impl Serialize for TrafficModel {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
serializer.serialize_str(std::convert::Into::<&str>::into(self))
} }
impl std::convert::From<&TrafficModel> for &str {
fn from(traffic_model: &TrafficModel) -> Self {
match traffic_model {
TrafficModel::BestGuess => "best_guess",
TrafficModel::Optimistic => "optimistic",
TrafficModel::Pessimistic => "pessimistic",
} } }
impl std::fmt::Display for TrafficModel {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", std::convert::Into::<&str>::into(self))
} }
impl std::convert::From<&TrafficModel> for String {
fn from(traffic_model: &TrafficModel) -> Self {
std::convert::Into::<&str>::into(traffic_model).to_string()
} }
static TRAFFIC_MODELS_BY_CODE: phf::Map<&'static str, TrafficModel> = phf_map! {
"best_guess" => TrafficModel::BestGuess,
"optimistic" => TrafficModel::Optimistic,
"pessimistic" => TrafficModel::Pessimistic,
};
impl std::convert::TryFrom<&str> for TrafficModel {
type Error = GoogleMapsError;
fn try_from(traffic_model_code: &str) -> Result<Self, Self::Error> {
Ok(TRAFFIC_MODELS_BY_CODE
.get(traffic_model_code)
.cloned()
.ok_or_else(|| {
DirectionsError::InvalidTrafficModelCode(traffic_model_code.to_string())
})?)
} }
impl std::str::FromStr for TrafficModel {
type Err = GoogleMapsError;
fn from_str(traffic_model_code: &str) -> Result<Self, Self::Err> {
Ok(TRAFFIC_MODELS_BY_CODE
.get(traffic_model_code)
.cloned()
.ok_or_else(|| {
DirectionsError::InvalidTrafficModelCode(traffic_model_code.to_string())
})?)
} }
impl TrafficModel {
#[must_use]
pub const fn display(&self) -> &str {
match self {
Self::BestGuess => "Best Guess",
Self::Optimistic => "Optimistic",
Self::Pessimistic => "Pessimistic",
} } }