use crate::prelude::tz::*;
use chrono::offset::{Offset, TimeZone};
use chrono::{Datelike, FixedOffset, NaiveDate, NaiveDateTime};
use chrono_tz::Canada::*;
use std::str::FromStr;
use strum_macros::EnumString;
#[derive(Debug, EnumString, Clone)]
#[non_exhaustive]
pub enum Canada {
Atlantic,
Central,
Eastern,
Mountain,
Newfoundland,
Pacific,
Saskatchewan,
Yukon,
}
impl Canada {
pub(crate) fn try_from_path(p: &[&str]) -> Result<Self, Error> {
if p.len() != 1 {
return Err(Error::TooManyElements(p.len()));
}
Self::from_str(p[0]).map_err(|_| Error::WrongTimeZone(p[0].to_string()))
}
pub(crate) fn get_tz(&self, datetime: &NaiveDateTime) -> FixedOffset {
let p = match self {
Self::Atlantic => Atlantic.from_utc_datetime(datetime),
Self::Central => Central.from_utc_datetime(datetime),
Self::Eastern => Eastern.from_utc_datetime(datetime),
Self::Mountain => Mountain.from_utc_datetime(datetime),
Self::Newfoundland => Newfoundland.from_utc_datetime(datetime),
Self::Pacific => Pacific.from_utc_datetime(datetime),
Self::Saskatchewan => Saskatchewan.from_utc_datetime(datetime),
Self::Yukon => Yukon.from_utc_datetime(datetime),
};
p.timezone()
.offset_from_utc_date(&NaiveDate::from_ymd(
datetime.year(),
datetime.month(),
datetime.day(),
))
.fix()
}
}