add: view subcommand implemented

1 files changed, 13 insertions, 14 deletions

diff --git a/src/main.rs b/src/main.rs
index 8ef62b3..be23ac3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -22,26 +22,25 @@ fn main() -> Result<()> {
             }
         },
         Commands::Register {person, amount, note, payment} => {
-            let register = Database::new(person, amount, note, !payment);
-            if let Some(datapath) = cli.database_path {
-                register.add_register(&datapath)?;
-            } else {
-                panic!("database file not found");
-            }
+            let register = Database::new(person, amount as i32, note, !payment);
+            let datapath = cli.database_path.unwrap();
+            register.add_register(&datapath)?;
         },
         Commands::View {history, filter, total} => {
-            println!("{}", filter.unwrap_or(String::from("None")));
-            println!("{}", history);
-            println!("{}", total);
+            let datapath = cli.database_path.expect("database path not found");
+
+            match (history, total) {
+                (true, false) => Database::view_history(&datapath, filter)?,
+                (false, true) => Database::view_total(&datapath)?,
+                (false, false) => Database::view_total(&datapath)?,
+                (true, true) => unreachable!()
+            }
         },
 
         Commands::Add {name} => {
             let register = Database::new(name, 0, None, false);
-            if let Some(datapath) = cli.database_path {
-                register.add_agent(&datapath)?;
-            } else {
-                panic!("database file not found");
-            }
+            let datapath = cli.database_path.unwrap();
+            register.add_agent(&datapath)?;
         }
     };
     Ok(())