use chrono::NaiveDate;

use crate::files::commands::{DoneDate, Time};

#[derive(Debug, Clone, Copy)]
struct Times {
    root: Time,
    other: Time,
}

#[derive(Debug, Clone, Copy)]
pub struct Dates {
    root: NaiveDate,
    other: NaiveDate,
    times: Option<Times>,
}

impl Dates {
    pub fn new(root: NaiveDate, other: NaiveDate) -> Self {
        Self {
            root,
            other,
            times: None,
        }
    }

    pub fn new_with_time(
        root: NaiveDate,
        root_time: Time,
        other: NaiveDate,
        other_time: Time,
    ) -> Self {
        Self {
            root,
            other,
            times: Some(Times {
                root: root_time,
                other: other_time,
            }),
        }
    }

    pub fn root(&self) -> NaiveDate {
        self.root
    }

    pub fn other(&self) -> NaiveDate {
        self.other
    }

    pub fn root_time(&self) -> Option<Time> {
        self.times.map(|times| times.root)
    }

    pub fn other_time(&self) -> Option<Time> {
        self.times.map(|times| times.other)
    }

    fn start_end(&self) -> (NaiveDate, NaiveDate) {
        if self.root <= self.other {
            (self.root, self.other)
        } else {
            (self.other, self.root)
        }
    }

    pub fn start(&self) -> NaiveDate {
        self.start_end().0
    }

    pub fn end(&self) -> NaiveDate {
        self.start_end().1
    }

    pub fn start_end_time(&self) -> Option<(Time, Time)> {
        if let Some(times) = self.times {
            if self.root < self.other || (self.root == self.other && times.root <= times.other) {
                Some((times.root, times.other))
            } else {
                Some((times.other, times.root))
            }
        } else {
            None
        }
    }

    pub fn start_time(&self) -> Option<Time> {
        self.start_end_time().map(|times| times.0)
    }

    pub fn end_time(&self) -> Option<Time> {
        self.start_end_time().map(|times| times.1)
    }
}

impl From<DoneDate> for Dates {
    fn from(date: DoneDate) -> Self {
        match date {
            DoneDate::Date { root } => Self::new(root, root),
            DoneDate::DateWithTime { root, root_time } => {
                Self::new_with_time(root, root_time, root, root_time)
            }
            DoneDate::DateToDate { root, other } => {
                if root <= other {
                    Self::new(root, other)
                } else {
                    Self::new(other, root)
                }
            }
            DoneDate::DateToDateWithTime {
                root,
                root_time,
                other,
                other_time,
            } => {
                if root < other || (root == other && root_time <= other_time) {
                    Self::new_with_time(root, root_time, other, other_time)
                } else {
                    Self::new_with_time(other, other_time, root, root_time)
                }
            }
        }
    }
}