use std::error;
use std::fmt;
use std::str::{self, FromStr};
#[cfg(feature = "serde")]
use serde::{de, ser};
#[derive(PartialEq, Eq, PartialOrd, Ord, Copy, Clone, Debug)]
pub struct Datetime {
pub date: Option<Date>,
pub time: Option<Time>,
pub offset: Option<Offset>,
}
#[doc(hidden)]
#[cfg(feature = "serde")]
pub const FIELD: &str = "$__toml_private_datetime";
#[doc(hidden)]
#[cfg(feature = "serde")]
pub const NAME: &str = "$__toml_private_Datetime";
#[derive(PartialEq, Eq, PartialOrd, Ord, Copy, Clone, Debug)]
pub struct Date {
pub year: u16,
pub month: u8,
pub day: u8,
}
#[derive(PartialEq, Eq, PartialOrd, Ord, Copy, Clone, Debug)]
pub struct Time {
pub hour: u8,
pub minute: u8,
pub second: u8,
pub nanosecond: u32,
}
#[derive(PartialEq, Eq, PartialOrd, Ord, Copy, Clone, Debug)]
pub enum Offset {
Z,
Custom {
minutes: i16,
},
}
impl Datetime {
#[cfg(feature = "serde")]
fn type_name(&self) -> &'static str {
match (
self.date.is_some(),
self.time.is_some(),
self.offset.is_some(),
) {
(true, true, true) => "offset datetime",
(true, true, false) => "local datetime",
(true, false, false) => Date::type_name(),
(false, true, false) => Time::type_name(),
_ => unreachable!("unsupported datetime combination"),
}
}
}
impl Date {
#[cfg(feature = "serde")]
fn type_name() -> &'static str {
"local date"
}
}
impl Time {
#[cfg(feature = "serde")]
fn type_name() -> &'static str {
"local time"
}
}
impl From<Date> for Datetime {
fn from(other: Date) -> Self {
Datetime {
date: Some(other),
time: None,
offset: None,
}
}
}
impl From<Time> for Datetime {
fn from(other: Time) -> Self {
Datetime {
date: None,
time: Some(other),
offset: None,
}
}
}
impl fmt::Display for Datetime {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
if let Some(ref date) = self.date {
write!(f, "{date}")?;
}
if let Some(ref time) = self.time {
if self.date.is_some() {
write!(f, "T")?;
}
write!(f, "{time}")?;
}
if let Some(ref offset) = self.offset {
write!(f, "{offset}")?;
}
Ok(())
}
}
impl fmt::Display for Date {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{:04}-{:02}-{:02}", self.year, self.month, self.day)
}
}
impl fmt::Display for Time {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{:02}:{:02}:{:02}", self.hour, self.minute, self.second)?;
if self.nanosecond != 0 {
let s = format!("{:09}", self.nanosecond);
write!(f, ".{}", s.trim_end_matches('0'))?;
}
Ok(())
}
}
impl fmt::Display for Offset {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match *self {
Offset::Z => write!(f, "Z"),
Offset::Custom { mut minutes } => {
let mut sign = '+';
if minutes < 0 {
minutes *= -1;
sign = '-';
}
let hours = minutes / 60;
let minutes = minutes % 60;
write!(f, "{sign}{hours:02}:{minutes:02}")
}
}
}
}
impl FromStr for Datetime {
type Err = DatetimeParseError;
fn from_str(date: &str) -> Result<Datetime, DatetimeParseError> {
let mut result = Datetime {
date: None,
time: None,
offset: None,
};
let mut lexer = Lexer::new(date);
let digits = lexer
.next()
.ok_or(DatetimeParseError::new().expected("year or hour"))?;
digits
.is(TokenKind::Digits)
.map_err(|err| err.expected("year or hour"))?;
let sep = lexer
.next()
.ok_or(DatetimeParseError::new().expected("`-` (YYYY-MM) or `:` (HH:MM)"))?;
match sep.kind {
TokenKind::Dash => {
let year = digits;
let month = lexer
.next()
.ok_or_else(|| DatetimeParseError::new().what("date").expected("month"))?;
month
.is(TokenKind::Digits)
.map_err(|err| err.what("date").expected("month"))?;
let sep = lexer.next().ok_or(
DatetimeParseError::new()
.what("date")
.expected("`-` (MM-DD)"),
)?;
sep.is(TokenKind::Dash)
.map_err(|err| err.what("date").expected("`-` (MM-DD)"))?;
let day = lexer
.next()
.ok_or(DatetimeParseError::new().what("date").expected("day"))?;
day.is(TokenKind::Digits)
.map_err(|err| err.what("date").expected("day"))?;
if year.raw.len() != 4 {
return Err(DatetimeParseError::new()
.what("date")
.expected("a four-digit year (YYYY)"));
}
if month.raw.len() != 2 {
return Err(DatetimeParseError::new()
.what("date")
.expected("a two-digit month (MM)"));
}
if day.raw.len() != 2 {
return Err(DatetimeParseError::new()
.what("date")
.expected("a two-digit day (DD)"));
}
let date = Date {
year: year.raw.parse().map_err(|_err| DatetimeParseError::new())?,
month: month
.raw
.parse()
.map_err(|_err| DatetimeParseError::new())?,
day: day.raw.parse().map_err(|_err| DatetimeParseError::new())?,
};
if date.month < 1 || date.month > 12 {
return Err(DatetimeParseError::new()
.what("date")
.expected("month between 01 and 12"));
}
let is_leap_year =
(date.year % 4 == 0) && ((date.year % 100 != 0) || (date.year % 400 == 0));
let (max_days_in_month, expected_day) = match date.month {
2 if is_leap_year => (29, "day between 01 and 29"),
2 => (28, "day between 01 and 28"),
4 | 6 | 9 | 11 => (30, "day between 01 and 30"),
_ => (31, "day between 01 and 31"),
};
if date.day < 1 || date.day > max_days_in_month {
return Err(DatetimeParseError::new()
.what("date")
.expected(expected_day));
}
result.date = Some(date);
}
TokenKind::Colon => lexer = Lexer::new(date),
_ => {
return Err(DatetimeParseError::new().expected("`-` (YYYY-MM) or `:` (HH:MM)"));
}
}
let partial_time = if result.date.is_some() {
let sep = lexer.next();
match sep {
Some(token) if matches!(token.kind, TokenKind::T | TokenKind::Space) => true,
Some(_token) => {
return Err(DatetimeParseError::new()
.what("date-time")
.expected("`T` between date and time"));
}
None => false,
}
} else {
result.date.is_none()
};
if partial_time {
let hour = lexer
.next()
.ok_or_else(|| DatetimeParseError::new().what("time").expected("hour"))?;
hour.is(TokenKind::Digits)
.map_err(|err| err.what("time").expected("hour"))?;
let sep = lexer.next().ok_or(
DatetimeParseError::new()
.what("time")
.expected("`:` (HH:MM)"),
)?;
sep.is(TokenKind::Colon)
.map_err(|err| err.what("time").expected("`:` (HH:MM)"))?;
let minute = lexer
.next()
.ok_or(DatetimeParseError::new().what("time").expected("minute"))?;
minute
.is(TokenKind::Digits)
.map_err(|err| err.what("time").expected("minute"))?;
let sep = lexer.next().ok_or(
DatetimeParseError::new()
.what("time")
.expected("`:` (MM:SS)"),
)?;
sep.is(TokenKind::Colon)
.map_err(|err| err.what("time").expected("`:` (MM:SS)"))?;
let second = lexer
.next()
.ok_or(DatetimeParseError::new().what("time").expected("second"))?;
second
.is(TokenKind::Digits)
.map_err(|err| err.what("time").expected("second"))?;
let nanosecond = if lexer.clone().next().map(|t| t.kind) == Some(TokenKind::Dot) {
let sep = lexer.next().ok_or(DatetimeParseError::new())?;
sep.is(TokenKind::Dot)?;
let nanosecond = lexer.next().ok_or(
DatetimeParseError::new()
.what("time")
.expected("nanosecond"),
)?;
nanosecond
.is(TokenKind::Digits)
.map_err(|err| err.what("time").expected("nanosecond"))?;
Some(nanosecond)
} else {
None
};
if hour.raw.len() != 2 {
return Err(DatetimeParseError::new()
.what("time")
.expected("a two-digit hour (HH)"));
}
if minute.raw.len() != 2 {
return Err(DatetimeParseError::new()
.what("time")
.expected("a two-digit minute (MM)"));
}
if second.raw.len() != 2 {
return Err(DatetimeParseError::new()
.what("time")
.expected("a two-digit second (SS)"));
}
let time = Time {
hour: hour.raw.parse().map_err(|_err| DatetimeParseError::new())?,
minute: minute
.raw
.parse()
.map_err(|_err| DatetimeParseError::new())?,
second: second
.raw
.parse()
.map_err(|_err| DatetimeParseError::new())?,
nanosecond: nanosecond.map(|t| s_to_nanoseconds(t.raw)).unwrap_or(0),
};
if time.hour > 23 {
return Err(DatetimeParseError::new()
.what("time")
.expected("hour between 00 and 23"));
}
if time.minute > 59 {
return Err(DatetimeParseError::new()
.what("time")
.expected("minute between 00 and 59"));
}
if time.second > 60 {
return Err(DatetimeParseError::new()
.what("time")
.expected("second between 00 and 60"));
}
if time.nanosecond > 999_999_999 {
return Err(DatetimeParseError::new()
.what("time")
.expected("nanoseconds overflowed"));
}
result.time = Some(time);
}
if result.date.is_some() && result.time.is_some() {
match lexer.next() {
Some(token) if token.kind == TokenKind::Z => {
result.offset = Some(Offset::Z);
}
Some(token) if matches!(token.kind, TokenKind::Plus | TokenKind::Dash) => {
let sign = if token.kind == TokenKind::Plus { 1 } else { -1 };
let hours = lexer
.next()
.ok_or(DatetimeParseError::new().what("offset").expected("hour"))?;
hours
.is(TokenKind::Digits)
.map_err(|err| err.what("offset").expected("hour"))?;
let sep = lexer.next().ok_or(
DatetimeParseError::new()
.what("offset")
.expected("`:` (HH:MM)"),
)?;
sep.is(TokenKind::Colon)
.map_err(|err| err.what("offset").expected("`:` (HH:MM)"))?;
let minutes = lexer
.next()
.ok_or(DatetimeParseError::new().what("offset").expected("minute"))?;
minutes
.is(TokenKind::Digits)
.map_err(|err| err.what("offset").expected("minute"))?;
if hours.raw.len() != 2 {
return Err(DatetimeParseError::new()
.what("offset")
.expected("a two-digit hour (HH)"));
}
if minutes.raw.len() != 2 {
return Err(DatetimeParseError::new()
.what("offset")
.expected("a two-digit minute (MM)"));
}
let hours = hours
.raw
.parse::<u8>()
.map_err(|_err| DatetimeParseError::new())?;
let minutes = minutes
.raw
.parse::<u8>()
.map_err(|_err| DatetimeParseError::new())?;
if hours > 23 {
return Err(DatetimeParseError::new()
.what("offset")
.expected("hours between 00 and 23"));
}
if minutes > 59 {
return Err(DatetimeParseError::new()
.what("offset")
.expected("minutes between 00 and 59"));
}
let total_minutes = sign * (hours as i16 * 60 + minutes as i16);
if !((-24 * 60)..=(24 * 60)).contains(&total_minutes) {
return Err(DatetimeParseError::new().what("offset"));
}
result.offset = Some(Offset::Custom {
minutes: total_minutes,
});
}
Some(_token) => {
return Err(DatetimeParseError::new()
.what("offset")
.expected("`Z`, +OFFSET, -OFFSET"));
}
None => {}
}
}
if lexer.unknown().is_some() {
return Err(DatetimeParseError::new());
}
Ok(result)
}
}
fn s_to_nanoseconds(input: &str) -> u32 {
let mut nanosecond = 0;
for (i, byte) in input.bytes().enumerate() {
if byte.is_ascii_digit() {
if i < 9 {
let p = 10_u32.pow(8 - i as u32);
nanosecond += p * u32::from(byte - b'0');
}
} else {
panic!("invalid nanoseconds {input:?}");
}
}
nanosecond
}
#[derive(Copy, Clone)]
struct Token<'s> {
kind: TokenKind,
raw: &'s str,
}
impl Token<'_> {
fn is(&self, kind: TokenKind) -> Result<(), DatetimeParseError> {
if self.kind == kind {
Ok(())
} else {
Err(DatetimeParseError::new())
}
}
}
#[derive(Copy, Clone, PartialEq, Eq)]
enum TokenKind {
Digits,
Dash,
Colon,
Dot,
T,
Space,
Z,
Plus,
Unknown,
}
#[derive(Copy, Clone)]
struct Lexer<'s> {
stream: &'s str,
}
impl<'s> Lexer<'s> {
fn new(input: &'s str) -> Self {
Self { stream: input }
}
fn unknown(&mut self) -> Option<Token<'s>> {
let remaining = self.stream.len();
if remaining == 0 {
return None;
}
let raw = self.stream;
self.stream = &self.stream[remaining..remaining];
Some(Token {
kind: TokenKind::Unknown,
raw,
})
}
}
impl<'s> Iterator for Lexer<'s> {
type Item = Token<'s>;
fn next(&mut self) -> Option<Self::Item> {
let (kind, end) = match self.stream.as_bytes().first()? {
b'0'..=b'9' => {
let end = self
.stream
.as_bytes()
.iter()
.position(|b| !b.is_ascii_digit())
.unwrap_or(self.stream.len());
(TokenKind::Digits, end)
}
b'-' => (TokenKind::Dash, 1),
b':' => (TokenKind::Colon, 1),
b'T' | b't' => (TokenKind::T, 1),
b' ' => (TokenKind::Space, 1),
b'Z' | b'z' => (TokenKind::Z, 1),
b'+' => (TokenKind::Plus, 1),
b'.' => (TokenKind::Dot, 1),
_ => (TokenKind::Unknown, self.stream.len()),
};
let (raw, rest) = self.stream.split_at(end);
self.stream = rest;
Some(Token { kind, raw })
}
}
#[derive(Debug, Clone)]
#[non_exhaustive]
pub struct DatetimeParseError {
what: Option<&'static str>,
expected: Option<&'static str>,
}
impl DatetimeParseError {
fn new() -> Self {
Self {
what: None,
expected: None,
}
}
fn what(mut self, what: &'static str) -> Self {
self.what = Some(what);
self
}
fn expected(mut self, expected: &'static str) -> Self {
self.expected = Some(expected);
self
}
}
impl fmt::Display for DatetimeParseError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
if let Some(what) = self.what {
write!(f, "invalid {what}")?;
} else {
"invalid datetime".fmt(f)?;
}
if let Some(expected) = self.expected {
write!(f, ", expected {expected}")?;
}
Ok(())
}
}
impl error::Error for DatetimeParseError {}
#[cfg(feature = "serde")]
impl ser::Serialize for Datetime {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: ser::Serializer,
{
use serde::ser::SerializeStruct;
let mut s = serializer.serialize_struct(NAME, 1)?;
s.serialize_field(FIELD, &self.to_string())?;
s.end()
}
}
#[cfg(feature = "serde")]
impl ser::Serialize for Date {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: ser::Serializer,
{
Datetime::from(*self).serialize(serializer)
}
}
#[cfg(feature = "serde")]
impl ser::Serialize for Time {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: ser::Serializer,
{
Datetime::from(*self).serialize(serializer)
}
}
#[cfg(feature = "serde")]
impl<'de> de::Deserialize<'de> for Datetime {
fn deserialize<D>(deserializer: D) -> Result<Datetime, D::Error>
where
D: de::Deserializer<'de>,
{
struct DatetimeVisitor;
impl<'de> de::Visitor<'de> for DatetimeVisitor {
type Value = Datetime;
fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
formatter.write_str("a TOML datetime")
}
fn visit_map<V>(self, mut visitor: V) -> Result<Datetime, V::Error>
where
V: de::MapAccess<'de>,
{
let value = visitor.next_key::<DatetimeKey>()?;
if value.is_none() {
return Err(de::Error::custom("datetime key not found"));
}
let v: DatetimeFromString = visitor.next_value()?;
Ok(v.value)
}
}
static FIELDS: [&str; 1] = [FIELD];
deserializer.deserialize_struct(NAME, &FIELDS, DatetimeVisitor)
}
}
#[cfg(feature = "serde")]
impl<'de> de::Deserialize<'de> for Date {
fn deserialize<D>(deserializer: D) -> Result<Date, D::Error>
where
D: de::Deserializer<'de>,
{
match Datetime::deserialize(deserializer)? {
Datetime {
date: Some(date),
time: None,
offset: None,
} => Ok(date),
datetime => Err(de::Error::invalid_type(
de::Unexpected::Other(datetime.type_name()),
&Self::type_name(),
)),
}
}
}
#[cfg(feature = "serde")]
impl<'de> de::Deserialize<'de> for Time {
fn deserialize<D>(deserializer: D) -> Result<Time, D::Error>
where
D: de::Deserializer<'de>,
{
match Datetime::deserialize(deserializer)? {
Datetime {
date: None,
time: Some(time),
offset: None,
} => Ok(time),
datetime => Err(de::Error::invalid_type(
de::Unexpected::Other(datetime.type_name()),
&Self::type_name(),
)),
}
}
}
#[cfg(feature = "serde")]
struct DatetimeKey;
#[cfg(feature = "serde")]
impl<'de> de::Deserialize<'de> for DatetimeKey {
fn deserialize<D>(deserializer: D) -> Result<DatetimeKey, D::Error>
where
D: de::Deserializer<'de>,
{
struct FieldVisitor;
impl de::Visitor<'_> for FieldVisitor {
type Value = ();
fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
formatter.write_str("a valid datetime field")
}
fn visit_str<E>(self, s: &str) -> Result<(), E>
where
E: de::Error,
{
if s == FIELD {
Ok(())
} else {
Err(de::Error::custom("expected field with custom name"))
}
}
}
deserializer.deserialize_identifier(FieldVisitor)?;
Ok(DatetimeKey)
}
}
#[doc(hidden)]
#[cfg(feature = "serde")]
pub struct DatetimeFromString {
pub value: Datetime,
}
#[cfg(feature = "serde")]
impl<'de> de::Deserialize<'de> for DatetimeFromString {
fn deserialize<D>(deserializer: D) -> Result<DatetimeFromString, D::Error>
where
D: de::Deserializer<'de>,
{
struct Visitor;
impl de::Visitor<'_> for Visitor {
type Value = DatetimeFromString;
fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
formatter.write_str("string containing a datetime")
}
fn visit_str<E>(self, s: &str) -> Result<DatetimeFromString, E>
where
E: de::Error,
{
match s.parse() {
Ok(date) => Ok(DatetimeFromString { value: date }),
Err(e) => Err(de::Error::custom(e)),
}
}
}
deserializer.deserialize_str(Visitor)
}
}