use super::Object;
#[cfg(feature = "chrono_time")]
use chrono::prelude::*;
use time::{format_description::FormatItem, OffsetDateTime, Time};
#[cfg(feature = "chrono_time")]
impl From<DateTime<Local>> for Object {
fn from(date: DateTime<Local>) -> Self {
let mut timezone_str = date.format("D:%Y%m%d%H%M%S%:z'").to_string().into_bytes();
convert_utc_offset(&mut timezone_str);
Object::string_literal(timezone_str)
}
}
#[allow(dead_code)]
fn convert_utc_offset(bytes: &mut [u8]) {
let mut index = bytes.len();
while let Some(last) = bytes[..index].last_mut() {
if *last == b':' {
*last = b'\'';
break;
}
index -= 1;
}
}
#[cfg(feature = "chrono_time")]
impl From<DateTime<Utc>> for Object {
fn from(date: DateTime<Utc>) -> Self {
Object::string_literal(date.format("D:%Y%m%d%H%M%SZ").to_string())
}
}
impl From<Time> for Object {
fn from(date: Time) -> Self {
Object::string_literal(
format!(
"D:{}",
date.format(&FormatItem::Literal("%Y%m%d%H%M%SZ".as_bytes())).unwrap()
)
.into_bytes(),
)
}
}
impl From<OffsetDateTime> for Object {
fn from(date: OffsetDateTime) -> Self {
Object::string_literal({
let format = time::format_description::parse(
"D:[year][month][day][hour][minute][second][offset_hour sign:mandatory]'[offset_minute]'",
)
.unwrap();
date.format(&format).unwrap()
})
}
}
impl Object {
fn datetime_string(&self) -> Option<String> {
if let Object::String(ref bytes, _) = self {
String::from_utf8(
bytes
.iter()
.filter(|b| ![b'D', b':', b'\''].contains(b))
.cloned()
.collect(),
)
.ok()
} else {
None
}
}
#[cfg(feature = "chrono_time")]
pub fn as_datetime(&self) -> Option<DateTime<Local>> {
let text = self.datetime_string()?;
let from_date = |date: NaiveDate| {
FixedOffset::east_opt(0)
.unwrap()
.from_utc_datetime(&date.and_time(NaiveTime::from_hms_opt(0, 0, 0).unwrap()))
};
DateTime::parse_from_str(&text, "%Y%m%d%H%M%S%#z")
.or_else(|_| DateTime::parse_from_str(&text, "%Y%m%d%H%M%#z"))
.or_else(|_| NaiveDate::parse_from_str(&text, "%Y%m%d").map(from_date))
.map(|date| date.with_timezone(&Local))
.ok()
}
#[cfg(not(feature = "chrono_time"))]
pub fn as_datetime(&self) -> Option<OffsetDateTime> {
let format = time::format_description::parse(
"[year][month][day][hour][minute][second][offset_hour sign:mandatory][offset_minute]",
)
.unwrap();
let text = self.datetime_string()?;
OffsetDateTime::parse(&text, &format).ok()
}
}
#[cfg(feature = "chrono_time")]
#[test]
fn parse_datetime_local() {
let time = Local::now().with_nanosecond(0).unwrap();
let text: Object = time.into();
let time2 = text.as_datetime();
assert_eq!(time2, Some(time));
}
#[cfg(feature = "chrono_time")]
#[test]
fn parse_datetime_utc() {
let time = Utc::now().with_nanosecond(0).unwrap();
let text: Object = time.into();
let time2 = text.as_datetime();
assert_eq!(time2, Some(time.with_timezone(&Local)));
}
#[cfg(feature = "chrono_time")]
#[test]
fn parse_datetime_seconds_missing() {
let text = Object::string_literal("D:199812231952-08'00'");
assert!(text.as_datetime().is_some());
}
#[cfg(feature = "chrono_time")]
#[test]
fn parse_datetime_time_missing() {
let text = Object::string_literal("D:20040229");
assert!(text.as_datetime().is_some());
}
#[cfg(not(feature = "chrono_time"))]
#[test]
fn parse_datetime() {
let time = time::OffsetDateTime::now_utc();
let text: Object = time.into();
let time2 = text.as_datetime().unwrap();
assert_eq!(time2.date(), time.date());
assert_eq!(time2.time().hour(), time.time().hour());
assert_eq!(time2.time().minute(), time.time().minute());
assert_eq!(time2.time().second(), time.time().second());
}