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

/*!
 * The domain model.
 */

use std::{fmt::Display, str::FromStr};

use chrono::{NaiveDateTime, NaiveDate};
use rust_decimal::Decimal;

use crate::{flex_query::CashTransaction, ISO_DATE_FORMAT, flex_enums::{self, CashAction}};

/**
 * The ledger transaction record.
 * Used for comparison between IB (translated) and Ledger records.
 * This transaction has only one posting. The posting is merged together with the
 * transaction details because the Flex report has individual lines only.
 */
#[derive(Debug, Default, PartialEq, Eq)]
pub struct CommonTransaction {
    pub date: NaiveDateTime,
    pub report_date: String,
    // effective_date: str = None
    pub payee: String,
    pub account: String,
    pub amount: Decimal,
    pub currency: String,
    pub symbol: String, // required for IB Cash Transactions
    pub r#type: String,
    pub description: String,
}

// const ISO_DATE_FMT: &str = "%Y-%m-%d";

impl From<&CashTransaction> for CommonTransaction {
    fn from(value: &CashTransaction) -> Self {
        log::debug!("converting ib tx: {:?}", value);

        // prepare symbol
        let mut symbol = value.symbol.to_owned();
        // Eliminate 'd' at the end of the symbol.
        if symbol.ends_with('d') {
            symbol = symbol[..symbol.len() - 1].to_string();
        }

        // Add the exchange
        symbol = format!("{}:{}", value.listing_exchange, symbol);
        // TODO: lookup the ledger symbol

        CommonTransaction {
            date: match value.date_time.len() {
                10 => {
                    log::debug!("the date is {}", value.date_time);

                    let tx_date = NaiveDate::parse_from_str(&value.date_time.as_str(), ISO_DATE_FORMAT)
                    .expect("valid date expected");
                    NaiveDateTime::from(tx_date.and_hms_opt(0, 0, 0).unwrap())
                },
                19 => NaiveDateTime::parse_from_str(&value.date_time.as_str(), "%Y-%m-%d;%H:%M:%S")
                    .expect("valid date/time expected"),
                _ => panic!("Invalid date/time"),
            },
            report_date: value.report_date.to_owned(),
            payee: String::default(),   // not used
            account: String::default(), // not used
            amount: Decimal::from_str(value.amount.as_str()).unwrap(),
            currency: value.currency.to_owned(),
            symbol,
            r#type: get_simplified_tx_type(&value.r#type),
            description: value.description.to_owned(),
        }
    }
}

fn get_simplified_tx_type(ib_tx_type: &String) -> String {
    let tx_type = flex_enums::cash_action(ib_tx_type.as_str()).to_owned();

    // mask in-lieu payments as regular dividends (?!) for comparison
    if tx_type == CashAction::PaymentInLieu.to_string() {
        CashAction::Dividend.to_string()
    } else {
        tx_type
    }
}

impl Display for CommonTransaction {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{}/{} {:7} {:8} {:>7} {}, {}",
            self.report_date,
            self.date.date(),
            self.symbol,
            self.r#type,
            self.amount,
            self.currency,
            self.description
        )
    }
}

#[cfg(test)]
mod tests {
    use rstest::{fixture, rstest};
    use rust_decimal::Decimal;
    use std::str::FromStr;

    use crate::{flex_query::CashTransaction, model::CommonTransaction};

    #[fixture]
    fn cash_transactions() -> Vec<CashTransaction> {
        crate::test_fixtures::cash_transactions()
    }

    #[rstest]
    fn conversion_test(cash_transactions: Vec<CashTransaction>) {
        let t1 = CommonTransaction::from(&cash_transactions[0]);

        // assert
        assert_eq!(String::default(), t1.account);
        assert_eq!(Decimal::from_str("-0.91").unwrap(), t1.amount);
    }
}