ticker/libticker/src/ics/mod.rs

use std::collections::HashMap;
use chrono::{NaiveDate, NaiveDateTime};

mod tokenizer;
mod parser;
pub use parser::Parser;

pub trait GetValue<'a, R: 'a> {
    fn get(&'a self, key: &'_ str) -> Option<R>;
}

pub type Props = Vec<(String, String)>;

#[derive(Debug, PartialEq)]
pub struct Object {
    pub name: String,
    pub content: HashMap<String, (Props, String)>,
    pub children: Vec<Box<Object>>,
}

impl<'a> GetValue<'a, &'a str> for Object {
    fn get(&'a self, key: &'_ str) -> Option<&'a str> {
        self.content.get(key)
            .map(|(_props, value)| value.as_ref())
    }
}

impl<'a> GetValue<'a, String> for Object {
    fn get(&self, key: &'_ str) -> Option<String> {
        self.content.get(key)
            .map(|(_props, value)| value.clone())
    }
}

// TODO: TZID, offset
impl<'a> GetValue<'a, Timestamp> for Object {
    fn get(&self, key: &'_ str) -> Option<Timestamp> {
        let s = self.get(key)?;
        NaiveDateTime::parse_from_str(s, "%Y%m%dT%H%M%SZ")
            .or_else(|_| NaiveDateTime::parse_from_str(s, "%Y%m%dT%H%M%S"))
            .map(Timestamp::DateTime)
            .or_else(|_|
                 NaiveDate::parse_from_str(s, "%Y%m%d")
                 .map(Timestamp::Date)
            )
            .map_err(|e| println!("Cannot parse date: {:?}", e))
            .ok()
    }
}

#[derive(Debug, Clone)]
pub enum Timestamp {
    Date(NaiveDate),
    DateTime(NaiveDateTime),
}

impl Timestamp {
    pub fn or_hms(self, hour: u32, min: u32, sec: u32) -> NaiveDateTime {
        match self {
            Timestamp::Date(date) => date.and_hms(hour, min, sec),
            Timestamp::DateTime(date_time) => date_time,
        }
    }
}