1 files changed, 23 insertions, 5 deletions

diff --git a/src/commands/add.rs b/src/commands/add.rs
index 67f63de..ed4a6d8 100644
--- a/src/commands/add.rs
+++ b/src/commands/add.rs
@@ -1,11 +1,28 @@
 use std::io::{Error, ErrorKind};
 
-use crate::{modrinth::{project, versions}, config::Cfg, db::insert_mod, Modloader};
+use crate::{modrinth::{project, versions}, config::Cfg, db::insert_mod, Modloader, input::Input};
 
-pub async fn add(config: Cfg, mc_mod: String) -> Result<(), Box<dyn std::error::Error>> {
-    println!("Adding");
+pub async fn modification(config: Cfg, args: Option<Vec<String>>) -> Result<(), Box<dyn std::error::Error>> {
     
-    let project = project(String::from(&config.apis.modrinth), &mc_mod).await;
+    if args.is_none() { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_FEW_ARGUMENTS"))) }
+
+    let arguments = Input::from(args.unwrap().join(" "))?;
+
+    if arguments.args.is_none() { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_FEW_ARGUMENTS"))); };
+
+    match arguments.command.as_str() {
+        "add" => {
+            add(config, arguments.args.unwrap()).await
+        },
+        _ => Err(Box::new(Error::new(ErrorKind::InvalidInput, "UNKNOWN_SUBCOMMAND")))
+    }
+}
+
+pub async fn add(config: Cfg, args: Vec<String>) -> Result<(), Box<dyn std::error::Error>> {
+    
+    if args.len() < 1 { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_FEW_ARGUMENTS"))); };
+
+    let project = project(String::from(&config.apis.modrinth), &args[0]).await;
  
     dbg!(&project);
 
@@ -14,7 +31,8 @@ pub async fn add(config: Cfg, mc_mod: String) -> Result<(), Box<dyn std::error::
     if project.versions.is_empty() { panic!("This should never happen"); };
 
     let current_version = get_current(config, String::from(&project.id)).await?;
-
+    
+    //add to current list and mod table
     match insert_mod(project.id, project.title, current_version, project.versions, loader, String::from("1.19.2")) {
         Err(err) => { Err(Box::new(err)) },
         Ok(()) => Ok(()),