124 lines
2.9 KiB
Rust
124 lines
2.9 KiB
Rust
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)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|