qif 0.1.0

library that can read and write QIF (Quicken Interchange File).
Documentation 
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use std::fmt;
use regex::Regex;

use crate::{Type, Transaction, DateFormat};

/** 
 * structure that houses the type and transactions in a QIF file 
*/
#[derive(Clone, Debug)]
pub struct Section {
    pub qif_type: Type,
    pub transactions: Vec<Transaction>
}

impl Section {
    /**
     * builder method use to create a QIF section.
     * 
     * The creation is done like this:
     * 
     * let expected_section = Section::builder()
        .set_type("Bank")
        .add_transaction(sam_hill)
        .add_transaction(fake_street)
        .add_transaction(velociraptor_entertainment)
        .build().unwrap();
     */
    pub fn builder() -> SectionBuilder {
        SectionBuilder::new()
    }

    pub fn to_string(&self, df: &DateFormat) -> String {
        let mut content = format!("!Type:{}\r\n", self.qif_type.to_str());

        for transaction in self.transactions.clone() {
            let transaction_string = format!("{}\r\n\r\n", transaction.to_string(df));
            content.push_str(&transaction_string);
        }

        content
    }

    pub fn from_str(s: &str, df: &DateFormat) -> Option<Self> {
        let mut builder = Section::builder();
        
        builder.set_type(&extract_type(s));
        
        if let Ok(transaction) = Transaction::from_str(s, df) {
            builder.add_transaction(transaction);
        }

        builder.build()
    }

    pub fn add_transaction_if_not_exists(&mut self, transaction: &Transaction) {
        if let None = self.transactions.iter().find(|&e| e == transaction) {
            self.transactions.push(transaction.to_owned())
        }
    }
}

fn extract_type(s: &str) -> String {
    if let Ok(regex) = Regex::new("!Type:([A-Z|a-z]{4,9})") {
        if let Some(captures) = regex.captures(s) {
            let (_, [account_type]) = captures.extract();

            account_type.to_string()
        } else {
            String::default()
        }
    } else {
        String::default()
    }
}

impl fmt::Display for Section {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{}", self.to_string(&DateFormat::MonthDayFullYear))
    }
}

impl PartialEq for Section {
    fn eq(&self, other: &Self) -> bool {
        self.qif_type == other.qif_type &&
        self.transactions == other.transactions
    }
}

pub struct SectionBuilder {
    qif_type: Option<Type>,
    transactions: Vec<Transaction>
}

impl SectionBuilder {
    pub fn new() -> Self {
        SectionBuilder {
            qif_type: None,
            transactions: vec![]
        }
    }

    pub fn set_type(&mut self, t: &str) -> &mut Self {
        self.qif_type = Type::from(t);
        self
    }

    pub fn add_transaction(&mut self, transaction: Transaction) -> &mut Self {
        self.transactions.push(transaction);
        self
    }

    pub fn build(&self) -> Option<Section> {
        if let Some(qif_type) = self.qif_type.clone() {
            Some(Section { 
                qif_type, 
                transactions: self.transactions.clone() 
            })
        } else {
            None
        }
    }
}