1 files changed, 26 insertions, 12 deletions
diff --git a/src/main.rs b/src/main.rs
index a791f29..b18cb1a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -6,35 +6,49 @@ mod space_repetition;
 mod sync;
 mod util;
 
-use crate::util::event::Events;
 use anyhow::Result;
+use clap::Parser;
 use std::path::PathBuf;
-use structopt::StructOpt;
 
-#[derive(StructOpt)]
-#[structopt()]
+#[derive(Parser)]
+#[clap()]
 struct Opt {
-    #[structopt(long, default_value = "deck.deck")]
+    /// Path to the deck
+    #[clap(long, default_value = "deck.deck")]
     deck: String,
+
+    /// Hide remaining card counts
+    #[clap(long)]
+    hide_remaining: bool,
 }
 
 fn main() -> Result<()> {
-    let deck_path = Opt::from_args().deck;
+    let args = Opt::parse();
+    let deck_path = args.deck;
     let mut conn = db::init(db_path(&deck_path))?;
     let deck_name = deck::pp_from_path(&deck_path).unwrap_or_else(|| "Deck".to_string());
 
     sync::run(&mut conn, &deck_path)?;
+    let deck_last_sync = util::time::seconds_since_unix_epoch()?;
+
+    let mut term = gui::setup_terminal()?;
 
-    let mut term = gui::terminal()?;
-    let events = Events::new();
-    match gui::start(&conn, &mut term, &events, &deck_name) {
-        Ok(()) => Ok(()),
+    match gui::start(
+        &mut conn,
+        &mut term,
+        &deck_path,
+        &deck_name,
+        deck_last_sync,
+        args.hide_remaining,
+    ) {
+        Ok(()) => (),
         Err(msg) => {
             // Show errors in TUI, otherwise they are hidden
-            gui::message::show(&mut term, &events, &deck_name, &format!("{msg}"), true)?;
-            Err(msg)
+            gui::message::show(&mut term, &deck_name, &format!("{msg}"), true)?
         }
     }
+
+    gui::restore_terminal(&mut term)
 }
 
 fn db_path(deck_path: &str) -> String {