1 files changed, 16 insertions, 25 deletions

diff --git a/src/commands/update.rs b/src/commands/update.rs
index eeb5a4d..498b6a9 100644
--- a/src/commands/update.rs
+++ b/src/commands/update.rs
@@ -1,6 +1,6 @@
-use std::{io::{Error, ErrorKind}, fs::rename};
+use std::io::{Error, ErrorKind};
 
-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, delete_version, download_versions}};
+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, mods_change_versions}, List, input::Input, files::{delete_version, download_versions, disable_version}};
 
 pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
 
@@ -69,18 +69,21 @@ pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error:
             }
         }
         
-        if input.direct_download { download_versions(current_list.clone(), 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 {
+                    println!("Deleting version {} for mod {}", ver.0, ver.1);
+                    delete_version(current_list.clone(), ver.0)?;
+                } else if ver.0 != "NONE" { 
+                    println!("Disabling version {} for mod {}", ver.0, ver.1);
+                    disable_version(config.clone(), current_list.clone(), ver.0, ver.1)?;
+                };
+            }
+        };
         
-        //Disable old versions
-        for ver in current_versions {
-            if input.delete_old {
-                println!("Deleting version {} for mod {}", ver.0, ver.1);
-                delete_version(current_list.clone(), ver.0)?;
-            } else if ver.0 != "NONE" { 
-                println!("Disabling version {} for mod {}", ver.0, ver.1);
-                disable_old(config.clone(), current_list.clone(), ver.0, ver.1)?;
-            };
-        }
     }
 
     Ok(())
@@ -120,18 +123,6 @@ async fn specific_update(config: Cfg, input: Input, list: List, project: Project
     Ok(current[0].clone())
 }
 
-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);
-
-    rename(file, disabled)?;
-
-    userlist_add_disabled_versions(config, current_list.id, versionid, mod_id)?;
-
-    Ok(())
-}
-
 #[tokio::test]
 async fn download_updates_test() {