1 files changed, 26 insertions, 12 deletions

diff --git a/src/commands/modification.rs b/src/commands/modification.rs
index 8e39d11..ac23970 100644
--- a/src/commands/modification.rs
+++ b/src/commands/modification.rs
@@ -1,6 +1,6 @@
 use std::io::{Error, ErrorKind};
 
-use crate::{modrinth::{project, versions, extract_current_version}, config::Cfg, db::{mods_insert, userlist_remove, mods_get_id, userlist_insert, mods_get_all_ids, userlist_get_all_ids}, input::{Input, Subcmd}, get_current_list, download_versions};
+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}, input::{Input, Subcmd}, get_current_list, download_versions};
 
 pub async fn modification(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
     
@@ -26,27 +26,41 @@ async fn add(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>
     let project = project(String::from(&config.apis.modrinth), &args[0]).await;
 
     let available_versions = versions(String::from(&config.apis.modrinth), String::from(&project.id), current_list.clone()).await;
-    
-    let current_id = extract_current_version(available_versions.clone())?;
-
-    let current_version = available_versions.clone().into_iter().find(|v| v.id == current_id).unwrap();
 
-    let file = current_version.clone().files.into_iter().find(|f| f.primary).unwrap().url;
-    
     let mut available_versions_vec: Vec<String> = Vec::new();
-    for ver in available_versions {
-        available_versions_vec.push(ver.id);
+    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(), current_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(&current_list.id), String::from(&project.id), String::from(&current_version.id), available_versions_vec, file)?;
+                userlist_insert(config.clone(), String::from(&current_list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?;
             } 
         },
-        Err(..) => userlist_insert(config.clone(), String::from(&current_list.id), String::from(&project.id), String::from(&current_version.id), available_versions_vec, file)?,
+        Err(..) => userlist_insert(config.clone(), String::from(&current_list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?,
     };
     
     match mods_get_all_ids(config.clone()) {
@@ -62,7 +76,7 @@ async fn add(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>
         },
     };
     
-    if !input.disable_download { download_versions(current_list, vec![current_version]).await?; }
+    if !input.disable_download && current_version.is_some() { download_versions(current_list, vec![current_version.unwrap()]).await?; }
 
     Ok(())
 }