1 files changed, 109 insertions, 29 deletions

diff --git a/src/main.rs b/src/main.rs
index fccc7fd..65de274 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,37 +1,117 @@
-use syn::manager::Manager;
+use clap::{Args, Parser, Subcommand, ValueEnum};
+use syn::{manager::Manager, PrintResult};
 
-#[tokio::main(flavor = "current_thread")]
-async fn main() {
-    let mut manager = Manager::default();
-
-    let feeds = vec![
-        "https://peppe.rs/index.xml",
-        "https://jvns.ca/atom.xml",
-        // "https://www.youtube.com/feeds/videos.xml?channel_id=UCuTaETsuCOkJ0H_GAztWt0Q",
-    ];
-
-    for f in feeds {
-        match manager.add_feed(f).await {
-            Ok(s) => println!("{s}"),
-            Err(e) => println!("{e}"),
-        }
-    }
+// #[tokio::main(flavor = "current_thread")]
+// async fn main() {
+//     let mut manager = Manager::load().unwrap();
+//
+//     let feeds = vec![
+//         "https://peppe.rs/index.xml",
+//         "https://jvns.ca/atom.xml",
+//         // "https://www.youtube.com/feeds/videos.xml?channel_id=UCuTaETsuCOkJ0H_GAztWt0Q",
+//     ];
+//
+//     for f in feeds {
+//         match manager.add_feed(f).await {
+//             Ok(s) => println!("{s}"),
+//             Err(e) => println!("{e}"),
+//         }
+//     }
+//
+//     for entry in manager.list_entries() {
+//         println!("{entry}");
+//     }
+//
+//     match manager.store() {
+//         Ok(s) => println!("{s}"),
+//         Err(e) => eprintln!("{e}"),
+//     }
+//
+//     // let mut feed = Feed::new(url);
+//
+//     // feed.resolve().await.unwrap();
+//
+//     // let last_read = DateTime::parse_from_rfc2822("Mon, 16 Mar 2020 18:30:00 +0000")
+//     //     .unwrap()
+//     //     .with_timezone(&Utc);
+//
+//     // feed.last_read = last_read;
+//
+//     // for i in feed.unread().unwrap() {
+//     //     println!("{}", i.title.as_ref().unwrap().content)
+//     // }
+// }
 
-    for entry in manager.list_entries() {
-        println!("{entry}");
-    }
+#[derive(Parser)]
+#[command(author, version, about, long_about = None)]
+struct Cli {
+    #[command(subcommand)]
+    command: Option<Command>,
+}
 
-    // let mut feed = Feed::new(url);
+#[derive(Subcommand)]
+enum Command {
+    Add(AddCommand),
+    List(ListCommand),
+    Pull(PullCommand),
+}
 
-    // feed.resolve().await.unwrap();
+#[derive(Args)]
+struct AddCommand {
+    url: String,
+}
 
-    // let last_read = DateTime::parse_from_rfc2822("Mon, 16 Mar 2020 18:30:00 +0000")
-    //     .unwrap()
-    //     .with_timezone(&Utc);
+#[derive(Args)]
+struct ListCommand {
+    #[arg(value_enum)]
+    target: ListTarget,
+}
 
-    // feed.last_read = last_read;
+#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
+enum ListTarget {
+    Feeds,
+    Entries,
+}
 
-    // for i in feed.unread().unwrap() {
-    //     println!("{}", i.title.as_ref().unwrap().content)
-    // }
+#[derive(Args)]
+struct PullCommand {
+    target: Option<String>,
+}
+
+#[tokio::main(flavor = "current_thread")]
+async fn main() {
+    let opts = Cli::parse();
+    match &opts.command {
+        Some(Command::Add(AddCommand { url })) => {
+            let mut manager = Manager::load().unwrap_or_else(|e| {
+                eprintln!("{e}");
+                Manager::default()
+            });
+
+            manager.add_feed(&url).await.print();
+            manager.store().print();
+        }
+        Some(Command::List(ListCommand { target })) => {
+            let manager = Manager::load().unwrap_or_else(|e| {
+                eprintln!("{e}");
+                Manager::default()
+            });
+            match target {
+                ListTarget::Feeds => manager.list_feeds().for_each(|f| println!("{f}")),
+                ListTarget::Entries => manager.list_entries().for_each(|f| println!("{f}")),
+            }
+        }
+        Some(Command::Pull(PullCommand { .. })) => {
+            let mut manager = Manager::load().unwrap_or_else(|e| {
+                eprintln!("{e}");
+                Manager::default()
+            });
+            let (status, errors): (Vec<_>, Vec<_>) =
+                manager.pull().await.into_iter().partition(Result::is_ok);
+            status.iter().for_each(PrintResult::print);
+            errors.iter().for_each(PrintResult::print);
+            manager.store().print();
+        }
+        _ => {}
+    }
 }