1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use chrono::prelude::*;
use crate::is_proper_date_format;

/// trait that allows object to become a datetime object.
pub trait LocalDateTimeExt {
    fn local_datetime(&self) -> Result<DateTime<Local>, String>;
}

impl LocalDateTimeExt for String {
    fn local_datetime(&self) -> Result<DateTime<Local>, String> {
        if is_proper_date_format(self) {
            let naive_date = NaiveDate::parse_from_str(self, crate::transaction::transaction_date_format::FORMAT).unwrap();

            let naive_datetime = naive_date.and_hms_opt(0, 0, 0).unwrap();

            let local_datetime = Local.from_local_datetime(&naive_datetime).unwrap();

            Ok(local_datetime)
        } else {
            Err(String::from("String must be in the format of YYYY-MM-DD."))
        }
        
    }
}

impl LocalDateTimeExt for str {
    fn local_datetime(&self) -> Result<DateTime<Local>, String> {
        if is_proper_date_format(self) {
            let naive_date = NaiveDate::parse_from_str(self, crate::transaction::transaction_date_format::FORMAT).unwrap();

            let naive_datetime = naive_date.and_hms_opt(0, 0, 0).unwrap();

            let local_datetime = Local.from_local_datetime(&naive_datetime).unwrap();

            Ok(local_datetime)
        } else {
            Err(String::from("String must be in the format of YYYY-MM-DD."))
        }
        
    }
}