1 files changed, 48 insertions, 19 deletions

diff --git a/src/commands/update.rs b/src/commands/update.rs
index 42d19aa..c8f0880 100644
--- a/src/commands/update.rs
+++ b/src/commands/update.rs
@@ -1,9 +1,9 @@
-use std::io::{Error, ErrorKind};
+use std::{io::{Error, ErrorKind}, fs::{rename, remove_file}};
 
-use crate::{config::Cfg, modrinth::{projects, Project, versions, extract_current_version, Version}, get_current_list, db::{userlist_get_all_ids, mods_get_versions, userlist_get_applicable_versions, userlist_change_versions, lists_get_all_ids, lists_get}, List, input::Input, download_file};
+use crate::{config::Cfg, modrinth::{projects, Project, versions, extract_current_version, Version}, get_current_list, db::{userlist_get_all_ids, mods_get_versions, userlist_get_applicable_versions, userlist_change_versions, lists_get_all_ids, lists_get, userlist_get_current_version, userlist_add_disabled_versions, mods_change_versions}, List, input::Input, files::get_file_path, download_versions};
 
 pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
-    
+
     let mut liststack: Vec<List> = vec![];
     if input.all_lists {
         let list_ids = lists_get_all_ids(config.clone())?;
@@ -11,18 +11,22 @@ pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error:
             liststack.push(lists_get(config.clone(), id)?);
         }
     } else {
-        liststack.push(get_current_list(config.clone())?)
+        let current = get_current_list(config.clone())?;
+        println!("Checking for updates of mods in {}", current.id);
+        liststack.push(current)
     }
 
     for current_list in liststack {
         let mods = userlist_get_all_ids(config.clone(), current_list.clone().id)?;
+        
+        let mut current_versions: Vec<(String, String)> = vec![];
 
         let mut versions = mods_get_versions(config.clone(), mods.clone())?;
         versions.sort_by_key(|ver| ver.mod_id.clone());
 
         let mut projects = projects(String::from(&config.apis.modrinth), mods).await;
         projects.sort_by_key(|pro| pro.id.clone());
-        
+
         let mut updatestack: Vec<Version> = vec![];
         for (index, project) in projects.into_iter().enumerate() {
             //Get versions for project and check if they match up
@@ -30,14 +34,26 @@ pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error:
             let p_id = String::from(&project.id);
             let v_id = &current_version.mod_id;
             if &p_id != v_id { return Err(Box::new(Error::new(ErrorKind::Other, "SORTING_ERROR"))) };
-
+            
+            //Getting current installed version for disable or delete
+            let disable_version = userlist_get_current_version(config.clone(), String::from(&current_list.id), String::from(&project.id))?;
+            
+            let version_db_string = project.versions.join("|");
 
             //Adding to stack if not the same versions in the list OR if clean == true
-            if input.clone().clean || (project.versions.join("|") != current_version.versions) {
+            if input.clone().clean || (version_db_string != current_version.versions) {
                 updatestack.push(match specific_update(config.clone(), input.clone(), current_list.clone(), project.clone()).await {
-                    Ok(ver) => ver,
+                    Ok(ver) => {
+                        current_versions.push((disable_version, p_id));
+                        ver
+                    },
                     //TODO handle errors (only continue on "NO_UPDATE_AVAILABLE")
-                    Err(_) => { println!("({}) No new version found for the specified minecraft version", project.title); continue; },
+                    Err(e) => {
+                        //Updating versions in modlist for no repeating version calls
+                        mods_change_versions(config.clone(), version_db_string, project.id)?;
+                        println!("({}) No new version found for the specified minecraft version({})", project.title, e);
+                        continue;
+                    },
                 });
             } else {
                 println!("({}) No new version found", project.title);
@@ -53,14 +69,19 @@ pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error:
             }
         }
         
-        if input.direct_download { download_updates(current_list, updatestack).await?; };
+        if input.direct_download { download_versions(current_list.clone(), updatestack).await?; };
+
+        //Disable old versions
+        for ver in current_versions {
+            if input.delete_old { delete_old(current_list.clone(), ver.0, ver.1)? } else { disable_old(config.clone(), current_list.clone(), ver.0, ver.1)? };
+        }
     }
 
     Ok(())
 }
 
 async fn specific_update(config: Cfg, input: Input, list: List, project: Project) -> Result<Version, Box<dyn std::error::Error>> {
-    print!("Checking update for '{}' in {}", project.title, list.id);
+    println!("Checking update for '{}' in {}", project.title, list.id);
     
     let applicable_versions = versions(String::from(&config.apis.modrinth), String::from(&project.id), list.clone()).await;
     
@@ -88,17 +109,25 @@ async fn specific_update(config: Cfg, input: Input, list: List, project: Project
     Ok(current[0].clone())
 }
 
-async fn download_updates(current_list: List, versions: Vec<Version>) -> Result<String, Box<dyn std::error::Error>> {
+fn disable_old(config: Cfg, current_list: List, versionid: String, mod_id: String) -> Result<(), Box<dyn std::error::Error>> {
+    println!("Disabling version {} for mod {}", versionid, mod_id);
+    let file = get_file_path(current_list.clone(), String::from(&versionid))?;
+    let disabled = format!("{}.disabled", file);
 
-    let dl_path = String::from(&current_list.download_folder);
+    rename(file, disabled)?;
 
-    for ver in versions {
-        let primary_file = ver.files.into_iter().find(|file| file.primary).unwrap();
-        download_file(primary_file.url, current_list.clone().download_folder, primary_file.filename).await?;
-    }
+    userlist_add_disabled_versions(config, current_list.id, versionid, mod_id)?;
+
+    Ok(())
+}
 
+fn delete_old(current_list: List, versionid: String, mod_id: String) -> Result<(), Box<dyn std::error::Error>> {
+    println!("Deleting version {} for mod {}", versionid, mod_id);
+    let file = get_file_path(current_list, String::from(&versionid))?;
+    
+    remove_file(file)?;
 
-    Ok(dl_path)
+    Ok(())
 }
 
 #[tokio::test]
@@ -138,5 +167,5 @@ async fn download_updates_test() {
                 "fabric".to_string()
             ]
         }];
-    assert!(download_updates(current_list, versions).await.is_ok())
+    assert!(download_versions(current_list, versions).await.is_ok())
 }