Merge pull request #3 from FxQnLr/new_input

New input, fuck it

1 files changed, 49 insertions, 3 deletions

diff --git a/src/main.rs b/src/main.rs
index 59d41c5..d177c3e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,13 +1,59 @@
-use modlist::{config::Cfg, input::get_input};
+use std::{env, process};
+
+use modlist::{config::Cfg, input::{get_input, Cmd}, update, download, list, io, modification};
 
 #[tokio::main]
 async fn main() {
     let config = Cfg::init("modlist.toml").unwrap();
 
-    match get_input(config).await {
-        Ok(..) => (),
+    let mut args: Vec<String> = env::args().collect();
+    args.reverse();
+    args.pop();
+    args.reverse();
+    
+    let input = match get_input(config.clone(), args).await {
+        Ok(i) => i,
         Err(e) => {
             println!("{}", e);
+            process::exit(1);
         }
     };
+
+    //dbg!(&input);
+
+    match input.clone().command.unwrap() {
+        Cmd::Mod => {
+            modification(config, input).await
+        },
+        Cmd::List => {
+            list(config, input).await
+        },
+        Cmd::Update => {
+            update(config, input).await
+        },
+        Cmd::Download => {
+            download(config, input).await
+        },
+        Cmd::Io => {
+            io(config, input).await
+        },
+        Cmd::Version => {
+            show_version();
+            Ok(())
+        },
+    }.unwrap()
+}
+
+fn show_version() {
+    match std::env::var("DEV") {
+        Ok(dev) => {
+            let devint = dev.parse::<i32>().unwrap();
+            if devint >= 1 {
+                println!("Modlist by FxQnLr v{} (DEV)", env!("CARGO_PKG_VERSION"));
+            } else {
+                println!("Modlist by FxQnLr v{}", env!("CARGO_PKG_VERSION"));
+            }
+        },
+        Err(..) => println!("Modlist by FxQnLr v{}", env!("CARGO_PKG_VERSION")),
+    }
 }