aboutsummaryrefslogtreecommitdiffstats
path: root/src/table.rs
blob: 531d1e82ba04e347f7c2e391376c5a66878beae9 (plain) (blame)
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
#[derive(Clone, Copy, Eq, Hash, PartialEq)]
pub enum TxnColumn {
    Date,
    Payee,
    Amount,
}

type TableView =
    cursive_table_view::TableView<super::ynab::Transaction, TxnColumn>;

impl cursive_table_view::TableViewItem<TxnColumn>
    for super::ynab::Transaction
{
    fn to_column(&self, column: TxnColumn) -> String {
        match column {
            TxnColumn::Date => self.date.clone(),
            TxnColumn::Payee => self.payee.clone(),
            TxnColumn::Amount => super::ynab::format_amount(self.amount),
        }
    }

    fn cmp(&self, other: &Self, column: TxnColumn) -> std::cmp::Ordering
    where
        Self: Sized,
    {
        match column {
            TxnColumn::Date => self.date.cmp(&other.date),
            TxnColumn::Payee => self.payee.cmp(&other.payee),
            TxnColumn::Amount => self.amount.cmp(&other.amount),
        }
    }
}

pub fn inflows_table(budget: &super::ynab::Budget) -> TableView {
    let inflows = budget
        .reimbursables()
        .iter()
        .filter(|t| !t.reimbursed && t.amount > 0)
        .cloned()
        .collect();
    txn_table(inflows)
}

pub fn outflows_table(budget: &super::ynab::Budget) -> TableView {
    let outflows = budget
        .reimbursables()
        .iter()
        .filter(|t| !t.reimbursed && t.amount <= 0)
        .cloned()
        .collect();
    txn_table(outflows)
}

fn txn_table(txns: Vec<super::ynab::Transaction>) -> TableView {
    let mut table = cursive_table_view::TableView::new()
        .column(TxnColumn::Date, "Date", |c| c.width(10))
        .column(TxnColumn::Payee, "Payee", |c| c)
        .column(TxnColumn::Amount, "Amount", |c| {
            c.align(cursive::align::HAlign::Right).width(10)
        })
        .default_column(TxnColumn::Date);
    table.set_items(txns);
    table
}