filter on amount compares

1 files changed, 39 insertions, 1 deletions

diff --git a/cli/src/search.rs b/cli/src/search.rs
index 86d057b..c19437f 100644
--- a/cli/src/search.rs
+++ b/cli/src/search.rs
@@ -1,4 +1,6 @@
 use chrono::{naive::NaiveDate, Duration};
+use rust_decimal::Decimal;
+use std::str::FromStr;
 
 use crate::transaction::{Category, Transaction};
 
@@ -40,11 +42,20 @@ impl DateIsh {
     }
 }
 
+pub enum Comparison {
+    Equal,
+    Greater,
+    GreaterOrEqual,
+    Less,
+    LessOrEqual,
+}
+
 pub enum Constraint {
     Category(Category),
     Before(DateIsh),
     After(DateIsh),
-    On(DateIsh)
+    On(DateIsh),
+    AmountCompare(Decimal, Comparison),
 }
 
 impl Constraint {
@@ -54,6 +65,13 @@ impl Constraint {
             Constraint::Before(date) => transaction.date < date.clone().get(),
             Constraint::After(date) => transaction.date >= date.clone().get(),
             Constraint::On(date) => transaction.date == date.clone().get(),
+            Constraint::AmountCompare(amount, comparison) => match comparison {
+                Comparison::Equal => &transaction.amount == amount,
+                Comparison::Greater => &transaction.amount > amount,
+                Comparison::GreaterOrEqual => &transaction.amount >= amount,
+                Comparison::Less => &transaction.amount < amount,
+                Comparison::LessOrEqual => &transaction.amount <= amount,
+            }
         }
     }
 }
@@ -137,6 +155,10 @@ impl<'t> Search<'t> {
                 ("before", date_ish) => Constraint::Before(DateIsh::parse(date_ish)),
                 ("after", date_ish) => Constraint::After(DateIsh::parse(date_ish)),
                 ("on", date_ish) => Constraint::On(DateIsh::parse(date_ish)),
+                ("amount", comparison) => {
+                    let (amount, comparison) = parse_comparison(comparison).unwrap();
+                    Constraint::AmountCompare(amount, comparison)
+                }
                 _ => panic!(),
             };
 
@@ -145,3 +167,19 @@ impl<'t> Search<'t> {
         self
     }
 }
+
+fn parse_comparison(s: &str) -> Option<(Decimal, Comparison)> {
+    if s.starts_with(">=") {
+        Some((Decimal::from_str(&s[2..]).ok()?, Comparison::GreaterOrEqual))
+    } else if s.starts_with("<=") {
+        Some((Decimal::from_str(&s[2..]).ok()?, Comparison::LessOrEqual))
+    } else if s.starts_with("=") {
+        Some((Decimal::from_str(&s[1..]).ok()?, Comparison::Equal))
+    } else if s.starts_with(">") {
+        Some((Decimal::from_str(&s[1..]).ok()?, Comparison::Greater))
+    } else if s.starts_with("<") {
+        Some((Decimal::from_str(&s[1..]).ok()?, Comparison::Less))
+    } else {
+        None
+    }
+}