1 files changed, 60 insertions, 55 deletions

diff --git a/src/commands/modification.rs b/src/commands/modification.rs
index f66ce28..7d4be8d 100644
--- a/src/commands/modification.rs
+++ b/src/commands/modification.rs
@@ -1,9 +1,8 @@
 use std::io::{Error, ErrorKind};
 
-use crate::{modrinth::{project, versions, extract_current_version, Version}, config::Cfg, db::{mods_insert, userlist_remove, mods_get_id, userlist_insert, mods_get_all_ids, userlist_get_all_ids, userlist_get_current_version, lists_get_all_ids, mods_remove}, input::{Input, Subcmd}, get_current_list, files::{delete_version, download_versions}, List};
+use crate::{modrinth::{project, versions, extract_current_version, Version, projects}, config::Cfg, db::{mods_insert, userlist_remove, mods_get_id, userlist_insert, mods_get_all_ids, userlist_get_all_ids, userlist_get_current_version, lists_get_all_ids, mods_remove}, input::{Input, Subcmd}, get_current_list, files::{delete_version, download_versions}, List};
 
 pub async fn modification(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
-    
     match input.subcommand.as_ref().ok_or("")? {
         Subcmd::Add => {
             add(config, input).await
@@ -23,69 +22,75 @@ async fn add(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>
     
     let current_list = get_current_list(config.clone())?;
     
-    mod_add(config, &args[0], current_list, input.disable_download).await?;
+    mod_add(config, vec![String::from(&args[0])], current_list, input.disable_download).await?;
 
     Ok(())
 }
 
-pub async fn mod_add(config: Cfg, mod_id: &str, list: List, disable_download: bool) -> Result<(), Box<dyn std::error::Error>> {
+pub async fn mod_add(config: Cfg, mod_id: Vec<String>, list: List, disable_download: bool) -> Result<(), Box<dyn std::error::Error>> {
     
-    println!("Adding mod {}", mod_id);
-
-    let project = project(String::from(&config.apis.modrinth), &mod_id).await;
-
-    let available_versions = versions(String::from(&config.apis.modrinth), String::from(&project.id), list.clone()).await;
-
-    let mut available_versions_vec: Vec<String> = Vec::new();
-    let current_version: Option<Version>;
-    let current_version_id: String;
-    let file: String;
-    if !available_versions.is_empty() {
-        let current_id = extract_current_version(available_versions.clone())?;
-
-        current_version = Some(available_versions.clone().into_iter().find(|v| v.id == current_id).unwrap());
-
-        current_version_id = current_version.clone().unwrap().id;
-
-        file = current_version.clone().ok_or("VERSION_CORRUPTED")?.files.into_iter().find(|f| f.primary).unwrap().url;
-
-        for ver in available_versions {
-            available_versions_vec.push(ver.id);
-        };
+    println!("Adding mod(s) {:?}", mod_id);
+    let projects = if mod_id.len() == 1 {
+        vec![project(String::from(&config.apis.modrinth), &mod_id[0]).await]
     } else {
-        println!("There's currently no mod version for your specified target");
-        current_version = None;
-        current_version_id = String::from("NONE");
-        file = String::from("NONE");
-        available_versions_vec.push(String::from("NONE"));
-    }
-
-    //add to current list and mod table
-    match userlist_get_all_ids(config.clone(), list.clone().id) {
-        Ok(mods) => {
-            if mods.contains(&project.id) {
-                return Err(Box::new(Error::new(ErrorKind::Other, "MOD_ALREADY_ON_LIST"))); } 
-            else {
-                userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?;
-            } 
-        },
-        Err(..) => userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?,
+        projects(String::from(&config.apis.modrinth), mod_id).await
     };
     
-    match mods_get_all_ids(config.clone()) {
-        Ok(mods) => {
-            if mods.contains(&project.id) {
-                //return Err(Box::new(Error::new(ErrorKind::Other, "MOD_ALREADY_IN_DATABASE")))
-            } else {
+    for project in projects {
+        let available_versions = versions(String::from(&config.apis.modrinth), String::from(&project.id), list.clone()).await;
+
+        let mut available_versions_vec: Vec<String> = Vec::new();
+        let current_version: Option<Version>;
+        let current_version_id: String;
+        let file: String;
+        if !available_versions.is_empty() {
+            let current_id = extract_current_version(available_versions.clone())?;
+
+            current_version = Some(available_versions.clone().into_iter().find(|v| v.id == current_id).unwrap());
+
+            current_version_id = current_version.clone().unwrap().id;
+
+            file = current_version.clone().ok_or("VERSION_CORRUPTED")?.files.into_iter().find(|f| f.primary).unwrap().url;
+
+            for ver in available_versions {
+                available_versions_vec.push(ver.id);
+            };
+        } else {
+            println!("There's currently no mod version for your specified target");
+            current_version = None;
+            current_version_id = String::from("NONE");
+            file = String::from("NONE");
+            available_versions_vec.push(String::from("NONE"));
+        }
+
+        //add to current list and mod table
+        match userlist_get_all_ids(config.clone(), list.clone().id) {
+            Ok(mods) => {
+                if mods.contains(&project.id) {
+                    return Err(Box::new(Error::new(ErrorKind::Other, "MOD_ALREADY_ON_LIST"))); } 
+                else {
+                    userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?;
+                } 
+            },
+            Err(..) => userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?,
+        };
+        
+        match mods_get_all_ids(config.clone()) {
+            Ok(mods) => {
+                if mods.contains(&project.id) {
+                    //return Err(Box::new(Error::new(ErrorKind::Other, "MOD_ALREADY_IN_DATABASE")))
+                } else {
+                    mods_insert(config.clone(), String::from(&project.id), String::from(&project.title), project.versions)?;
+                } 
+            },
+            Err(..) => {
                 mods_insert(config.clone(), String::from(&project.id), String::from(&project.title), project.versions)?;
-            } 
-        },
-        Err(..) => {
-            mods_insert(config.clone(), String::from(&project.id), String::from(&project.title), project.versions)?;
-        },
-    };
+            },
+        };
 
-    if !disable_download && current_version.is_some() { download_versions(list, vec![current_version.unwrap()]).await?; };
+        if !disable_download && current_version.is_some() { download_versions(list.clone(), vec![current_version.unwrap()]).await?; };
+            
+    }
 
     Ok(())
 }