aboutsummaryrefslogtreecommitdiffstats
path: root/src/views
diff options
context:
space:
mode:
Diffstat (limited to 'src/views')
-rw-r--r--src/views/txn_table.rs151
-rw-r--r--src/views/txn_tables.rs353
-rw-r--r--src/views/util.rs5
3 files changed, 0 insertions, 509 deletions
diff --git a/src/views/txn_table.rs b/src/views/txn_table.rs
deleted file mode 100644
index b71a92e..0000000
--- a/src/views/txn_table.rs
+++ /dev/null
@@ -1,151 +0,0 @@
-use cursive::view::{Identifiable, View, ViewWrapper};
-
-#[derive(Clone, Copy, Eq, Hash, PartialEq)]
-pub enum TxnColumn {
- Selected,
- Date,
- Account,
- Payee,
- Amount,
- TotalAmount,
-}
-
-pub type TxnTableView =
- cursive_table_view::TableView<crate::ynab::Transaction, TxnColumn>;
-pub struct TxnTable {
- view: super::util::FullView<TxnTableView>,
-}
-
-impl cursive::view::ViewWrapper for TxnTable {
- cursive::wrap_impl!(self.view: super::util::FullView<TxnTableView>);
-}
-
-impl TxnTable {
- pub fn new(
- txns: Vec<crate::ynab::Transaction>,
- id: &'static str,
- ) -> Self {
- let mut table = cursive_table_view::TableView::new()
- .column(TxnColumn::Selected, "Sel", |c| c.width(3))
- .column(TxnColumn::Date, "Date", |c| c.width(10))
- .column(TxnColumn::Account, "Account", |c| c.width(15))
- .column(TxnColumn::Payee, "Payee", |c| c)
- .column(TxnColumn::Amount, "Amount", |c| {
- c.align(cursive::align::HAlign::Right).width(10)
- })
- .column(TxnColumn::TotalAmount, "", |c| {
- c.align(cursive::align::HAlign::Right).width(10)
- })
- .default_column(TxnColumn::Date)
- .on_submit(|s, _, _| {
- s.on_event(cursive::event::Event::Key(
- cursive::event::Key::F0,
- ));
- });
- table.set_items(txns);
- let view = cursive::views::OnEventView::new(table)
- .on_event_inner(' ', |v: &mut TxnTableView, _| {
- if let Some(idx) = v.item() {
- let txn = v.borrow_item_mut(idx).unwrap();
- txn.selected = !txn.selected;
- }
- None
- })
- .on_event_inner('h', |v: &mut TxnTableView, _| {
- v.on_event(cursive::event::Event::Key(
- cursive::event::Key::Left,
- ));
- None
- })
- .on_event_inner('j', |v: &mut TxnTableView, _| {
- v.on_event(cursive::event::Event::Key(
- cursive::event::Key::Down,
- ));
- None
- })
- .on_event_inner('k', |v: &mut TxnTableView, _| {
- v.on_event(cursive::event::Event::Key(
- cursive::event::Key::Up,
- ));
- None
- })
- .on_event_inner('l', |v: &mut TxnTableView, _| {
- v.on_event(cursive::event::Event::Key(
- cursive::event::Key::Right,
- ));
- None
- })
- .on_event_inner('g', |v: &mut TxnTableView, _| {
- v.set_selected_row(0);
- None
- })
- .on_event_inner('G', |v: &mut TxnTableView, _| {
- v.set_selected_row(v.len() - 1);
- None
- })
- .with_id(id);
- TxnTable { view }
- }
-
- pub fn len(&self) -> usize {
- self.view
- .with_view(|v| v.with_view(|v| v.len()).unwrap())
- .unwrap()
- }
-
- // XXX why does borrow_items require &mut self?
- pub fn amount(&mut self) -> i64 {
- self.view
- .get_mut()
- .get_inner_mut()
- .borrow_items()
- .iter()
- .map(|t| t.amount)
- .sum()
- }
-}
-
-impl cursive_table_view::TableViewItem<TxnColumn>
- for crate::ynab::Transaction
-{
- fn to_column(&self, column: TxnColumn) -> String {
- match column {
- TxnColumn::Selected => {
- if self.selected {
- "[x]".to_string()
- } else {
- "[ ]".to_string()
- }
- }
- TxnColumn::Date => self.date.clone(),
- TxnColumn::Account => {
- self.account.clone().unwrap_or_else(|| "".to_string())
- }
- TxnColumn::Payee => {
- self.payee.clone().unwrap_or_else(|| "".to_string())
- }
- TxnColumn::Amount => crate::ynab::format_amount(self.amount),
- TxnColumn::TotalAmount => {
- if self.amount == self.total_amount {
- "".to_string()
- } else {
- crate::ynab::format_amount(self.total_amount)
- }
- }
- }
- }
-
- fn cmp(&self, other: &Self, column: TxnColumn) -> std::cmp::Ordering
- where
- Self: Sized,
- {
- match column {
- TxnColumn::Selected => std::cmp::Ordering::Equal,
- TxnColumn::Date => self.date.cmp(&other.date),
- TxnColumn::Account => self.account.cmp(&other.account),
- TxnColumn::Payee => self.payee.cmp(&other.payee),
- TxnColumn::Amount => self.amount.cmp(&other.amount),
- TxnColumn::TotalAmount => self.amount.cmp(&other.total_amount),
- }
- }
-}
diff --git a/src/views/txn_tables.rs b/src/views/txn_tables.rs
deleted file mode 100644
index 62539ac..0000000
--- a/src/views/txn_tables.rs
+++ /dev/null
@@ -1,353 +0,0 @@
-use cursive::view::{Identifiable, View};
-
-const SELECTED_TOTAL_ID: &str = "selected_total";
-const INFLOWS_TABLE_ID: &str = "inflows_table";
-const OUTFLOWS_TABLE_ID: &str = "outflows_table";
-
-pub struct TxnTables {
- view: super::util::FullView<cursive::views::LinearLayout>,
-}
-
-impl cursive::view::ViewWrapper for TxnTables {
- cursive::wrap_impl!(
- self.view: super::util::FullView<cursive::views::LinearLayout>
- );
-}
-
-impl TxnTables {
- pub fn new(id: &'static str, budget: &crate::ynab::Budget) -> Self {
- let mut layout = cursive::views::LinearLayout::vertical();
-
- layout.add_child(
- cursive::views::TextView::new("Selected: $0.00 (0 transactions)")
- .h_align(cursive::align::HAlign::Right)
- .with_id(SELECTED_TOTAL_ID),
- );
-
- let inflows = budget
- .reimbursables()
- .iter()
- .filter(|t| !t.reimbursed && t.amount > 0)
- .cloned()
- .collect();
- let mut inflows_table =
- super::txn_table::TxnTable::new(inflows, INFLOWS_TABLE_ID);
- layout.add_child(cursive::views::TextView::new(format!(
- "Inflows: {} ({} transaction{}",
- crate::ynab::format_amount(inflows_table.amount()),
- inflows_table.len(),
- if inflows_table.len() == 1 { ") " } else { "s)" }
- )));
- layout.add_child(cursive::views::CircularFocus::wrap_arrows(
- cursive::views::BoxView::with_min_height(
- std::cmp::min(std::cmp::max(inflows_table.len(), 1), 5) + 2,
- cursive::views::BoxView::with_full_width(inflows_table),
- ),
- ));
-
- layout.add_child(cursive::views::TextView::new(" "));
-
- let outflows = budget
- .reimbursables()
- .iter()
- .filter(|t| !t.reimbursed && t.amount <= 0)
- .cloned()
- .collect();
- let mut outflows_table =
- super::txn_table::TxnTable::new(outflows, OUTFLOWS_TABLE_ID);
- layout.add_child(cursive::views::TextView::new(format!(
- "Outflows: {} ({} transaction{}",
- crate::ynab::format_amount(outflows_table.amount()),
- outflows_table.len(),
- if outflows_table.len() == 1 {
- ") "
- } else {
- "s)"
- }
- )));
- layout.add_child(cursive::views::CircularFocus::wrap_arrows(
- cursive::views::BoxView::with_full_screen(outflows_table),
- ));
-
- let event_view = cursive::views::OnEventView::new(layout)
- .on_event(cursive::event::Key::F0, move |s| {
- submit(s);
- })
- .on_event('r', move |s| {
- refresh(s);
- })
- .on_pre_event_inner(' ', |v, _| select(v))
- .with_id(id);
-
- TxnTables { view: event_view }
- }
-}
-
-fn submit(s: &mut cursive::Cursive) {
- let inflows: Vec<_> = s
- .call_on_id(
- INFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- v.get_inner_mut()
- .borrow_items()
- .iter()
- .filter(|t| t.selected)
- .cloned()
- .collect()
- },
- )
- .unwrap();
- let outflows: Vec<_> = s
- .call_on_id(
- OUTFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- v.get_inner_mut()
- .borrow_items()
- .iter()
- .filter(|t| t.selected)
- .cloned()
- .collect()
- },
- )
- .unwrap();
- let total_inflow: i64 = inflows.iter().map(|t| t.amount).sum();
- let total_outflow: i64 = outflows.iter().map(|t| t.amount).sum();
- let total_amount = total_inflow + total_outflow;
- if total_amount == 0 && (inflows.len() + outflows.len() > 0) {
- let budget: &mut crate::ynab::Budget = s.user_data().unwrap();
- let txns: Vec<_> = inflows.iter().chain(outflows.iter()).collect();
- let err = budget.reconcile_transactions(&txns);
- if let Err(err) = err {
- s.add_layer(super::util::dialog(&format!("Error: {}", err)))
- } else {
- s.add_layer(super::util::dialog(&format!(
- "Successfully updated {} transactions",
- txns.len()
- )));
- s.call_on_id(
- INFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- let v = v.get_inner_mut();
- let all_txns = v.borrow_items_mut();
- for id in txns.iter().map(|t| t.id.clone()) {
- if let Some(idx) =
- all_txns.iter().position(|t| t.id == id)
- {
- all_txns.remove(idx);
- }
- }
- if let Some(row) = v.row() {
- if row >= v.len() {
- v.set_selected_row(v.len() - 1);
- }
- }
- },
- )
- .unwrap();
- s.call_on_id(
- OUTFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- let v = v.get_inner_mut();
- let all_txns = v.borrow_items_mut();
- for id in txns.iter().map(|t| t.id.clone()) {
- if let Some(idx) =
- all_txns.iter().position(|t| t.id == id)
- {
- all_txns.remove(idx);
- }
- }
- if let Some(row) = v.row() {
- if row >= v.len() {
- v.set_selected_row(v.len() - 1);
- }
- }
- },
- )
- .unwrap();
- }
- } else if total_amount != 0 {
- s.add_layer(super::util::dialog(&format!(
- "Selected amount is {}, must be 0",
- crate::ynab::format_amount(total_amount)
- )))
- }
-}
-
-fn refresh(s: &mut cursive::Cursive) {
- let budget: &mut crate::ynab::Budget = s.user_data().unwrap();
- let err = budget.refresh();
- if let Err(err) = err {
- s.add_layer(super::util::dialog(&format!("Error: {}", err)));
- return;
- }
-
- let mut inflows: Vec<_> = budget
- .reimbursables()
- .iter()
- .filter(|t| !t.reimbursed && t.amount > 0)
- .cloned()
- .collect();
- s.call_on_id(
- INFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- let v = v.get_inner_mut();
- let selected: std::collections::HashSet<_> = v
- .borrow_items()
- .iter()
- .filter(|t| t.selected)
- .map(|t| t.id.clone())
- .collect();
- let row = v
- .item()
- .and_then(|idx| v.borrow_item(idx).map(|t| t.id.clone()));
- for mut t in inflows.iter_mut() {
- if selected.contains(&t.id) {
- t.selected = true;
- }
- }
- let idx =
- row.and_then(|id| inflows.iter().position(|t| t.id == id));
- v.set_items(inflows);
- if let Some(idx) = idx {
- v.set_selected_item(idx);
- }
- },
- )
- .unwrap();
-
- let budget: &mut crate::ynab::Budget = s.user_data().unwrap();
- let mut outflows: Vec<_> = budget
- .reimbursables()
- .iter()
- .filter(|t| !t.reimbursed && t.amount <= 0)
- .cloned()
- .collect();
- s.call_on_id(
- OUTFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- let v = v.get_inner_mut();
- let selected: std::collections::HashSet<_> = v
- .borrow_items()
- .iter()
- .filter(|t| t.selected)
- .map(|t| t.id.clone())
- .collect();
- let row = v
- .item()
- .and_then(|idx| v.borrow_item(idx).map(|t| t.id.clone()));
- for mut t in outflows.iter_mut() {
- if selected.contains(&t.id) {
- t.selected = true;
- }
- }
- let idx =
- row.and_then(|id| outflows.iter().position(|t| t.id == id));
- v.set_items(outflows);
- if let Some(idx) = idx {
- v.set_selected_item(idx);
- }
- },
- )
- .unwrap();
-
- render_selected_total(s);
-}
-
-fn select(
- v: &mut cursive::views::LinearLayout,
-) -> Option<cursive::event::EventResult> {
- let idx = v.get_focus_index();
- let child = v.get_child_mut(idx).unwrap();
- child.call_on_any(
- &cursive::view::Selector::Id(INFLOWS_TABLE_ID),
- Box::new(|v| {
- v.downcast_mut::<cursive::views::IdView<
- cursive::views::OnEventView<super::txn_table::TxnTableView>,
- >>()
- .map(|v| v.on_event(cursive::event::Event::Char(' ')));
- }),
- );
- child.call_on_any(
- &cursive::view::Selector::Id(OUTFLOWS_TABLE_ID),
- Box::new(|v| {
- v.downcast_mut::<cursive::views::IdView<
- cursive::views::OnEventView<super::txn_table::TxnTableView>,
- >>()
- .map(|v| v.on_event(cursive::event::Event::Char(' ')));
- }),
- );
- Some(cursive::event::EventResult::with_cb(|s| {
- render_selected_total(s);
- }))
-}
-
-fn render_selected_total(s: &mut cursive::Cursive) {
- let inflows: Vec<_> = s
- .call_on_id(
- INFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- v.get_inner_mut()
- .borrow_items()
- .iter()
- .filter(|t| t.selected)
- .map(|t| t.amount)
- .collect()
- },
- )
- .unwrap();
- let outflows: Vec<_> = s
- .call_on_id(
- OUTFLOWS_TABLE_ID,
- |v: &mut cursive::views::OnEventView<
- super::txn_table::TxnTableView,
- >| {
- v.get_inner_mut()
- .borrow_items()
- .iter()
- .filter(|t| t.selected)
- .map(|t| t.amount)
- .collect()
- },
- )
- .unwrap();
- let outflow: i64 = outflows.iter().sum();
- let inflow: i64 = inflows.iter().sum();
- let amount = outflow + inflow;
- s.call_on_id(SELECTED_TOTAL_ID, |v: &mut cursive::views::TextView| {
- let mut sstr =
- cursive::utils::markup::StyledString::plain("Selected: ");
- let color = if amount == 0 && outflows.len() + inflows.len() != 0 {
- cursive::theme::Color::Dark(cursive::theme::BaseColor::Green)
- } else {
- cursive::theme::Color::TerminalDefault
- };
- sstr.append(cursive::utils::markup::StyledString::styled(
- crate::ynab::format_amount(amount),
- color,
- ));
- sstr.append(format!(
- " ({} transaction{}",
- outflows.len() + inflows.len(),
- if outflows.len() + inflows.len() == 1 {
- ") "
- } else {
- "s)"
- }
- ));
- v.set_content(sstr);
- });
-}
diff --git a/src/views/util.rs b/src/views/util.rs
deleted file mode 100644
index eb22976..0000000
--- a/src/views/util.rs
+++ /dev/null
@@ -1,5 +0,0 @@
-pub type FullView<T> = cursive::views::IdView<cursive::views::OnEventView<T>>;
-
-pub fn dialog(s: &str) -> impl cursive::view::View {
- cursive::views::Panel::new(cursive::views::Dialog::info(s))
-}