time_macros/
datetime.rs

1use std::iter::Peekable;
2
3use proc_macro::{token_stream, TokenStream};
4
5use crate::date::Date;
6use crate::error::Error;
7use crate::offset::Offset;
8use crate::time::Time;
9use crate::to_tokens::ToTokenStream;
10use crate::{date, offset, time};
11
12pub(crate) struct DateTime {
13    date: Date,
14    time: Time,
15    offset: Option<Offset>,
16}
17
18pub(crate) fn parse(chars: &mut Peekable<token_stream::IntoIter>) -> Result<DateTime, Error> {
19    let date = date::parse(chars)?;
20    let time = time::parse(chars)?;
21    let offset = match offset::parse(chars) {
22        Ok(offset) => Some(offset),
23        Err(Error::UnexpectedEndOfInput | Error::MissingComponent { name: "sign", .. }) => None,
24        Err(err) => return Err(err),
25    };
26
27    if let Some(token) = chars.peek() {
28        return Err(Error::UnexpectedToken {
29            tree: token.clone(),
30        });
31    }
32
33    Ok(DateTime { date, time, offset })
34}
35
36impl ToTokenStream for DateTime {
37    fn append_to(self, ts: &mut TokenStream) {
38        let maybe_offset = match self.offset {
39            Some(offset) => quote_! { .assume_offset(#S(offset)) },
40            None => quote_! {},
41        };
42
43        quote_append! { ts
44            ::time::PrimitiveDateTime::new(
45                #S(self.date),
46                #S(self.time),
47            ) #S(maybe_offset)
48        }
49    }
50}
time_macros/datetime.rs

time_macros/
datetime.rs
