1 files changed, 15 insertions, 35 deletions

diff --git a/src/commands/update.rs b/src/commands/update.rs
index 85630f5..eba5e91 100644
--- a/src/commands/update.rs
+++ b/src/commands/update.rs
@@ -1,10 +1,6 @@
-use std::{io::{Error, ErrorKind, Write}, fs::File};
+use std::io::{Error, ErrorKind};
 
-use reqwest::Client;
-
-use futures_util::StreamExt;
-
-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}, List, input::Input};
+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}, List, input::Input, download_file};
 
 pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
 
@@ -29,16 +25,16 @@ pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error:
 
         //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) {
-            updatestack.push(match specific_update(config.clone(), input.clone(), current_list.clone(), project).await {
+            updatestack.push(match specific_update(config.clone(), input.clone(), current_list.clone(), project.clone()).await {
                 Ok(ver) => ver,
                 //TODO handle errors (only continue on "NO_UPDATE_AVAILABLE")
-                Err(_) => { continue; },
+                Err(_) => { println!("({}) No new version found for the specified minecraft version", project.title); continue; },
             });
+        } else {
+            println!("({}) No new version found", project.title);
         };
     };
-    //println!("{:?}", updatestack);
-    
-    
+
     if input.clean {
         let dl_path = &current_list.download_folder;
         println!("Cleaning {}", dl_path);
@@ -47,8 +43,8 @@ pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error:
             std::fs::remove_file(entry.path())?;
         }
     }
-
-    download_updates(config, current_list, updatestack).await?;
+    
+    if input.direct_download { download_updates(current_list, updatestack).await?; };
 
     Ok(())
 }
@@ -82,41 +78,25 @@ async fn specific_update(config: Cfg, input: Input, list: List, project: Project
     Ok(current[0].clone())
 }
 
-async fn download_updates(config: Cfg, current_list: List, versions: Vec<Version>) -> Result<String, Box<dyn std::error::Error>> {
+async fn download_updates(current_list: List, versions: Vec<Version>) -> Result<String, Box<dyn std::error::Error>> {
 
     let dl_path = String::from(&current_list.download_folder);
 
     for ver in versions {
         let primary_file = ver.files.into_iter().find(|file| file.primary).unwrap();
-        let dl_path_file = format!("{}/{}", config.downloads, primary_file.filename);
-        println!("Downloading {}", primary_file.url);
-
-        let res = Client::new()
-            .get(String::from(&primary_file.url))
-            .send()
-            .await
-            .or(Err(format!("Failed to GET from '{}'", &primary_file.url)))?;
-        
-        // download chunks
-        let mut file = File::create(String::from(&dl_path_file)).or(Err(format!("Failed to create file '{}'", dl_path_file)))?;
-        let mut stream = res.bytes_stream();
-
-        while let Some(item) = stream.next().await {
-            let chunk = item.or(Err("Error while downloading file"))?;
-            file.write_all(&chunk)
-                .or(Err("Error while writing to file"))?;
-        }
+        download_file(primary_file.url, current_list.clone().download_folder, primary_file.filename).await?;
     }
 
+
     Ok(dl_path)
 }
 
 #[tokio::test]
 async fn download_updates_test() {
 
-    use crate::{modrinth::{Version, VersionFile, Hash, VersionType}, config::{Cfg, Apis}};
+    use crate::{modrinth::{Version, VersionFile, Hash, VersionType}, Modloader, List};
     
-    let config = Cfg { data: "...".to_string(), clean_remove: false, downloads: "./dl".to_string(), apis: Apis { modrinth: "...".to_string() } };
+    let current_list = List { id: String::from("..."), mc_version: String::from("..."), modloader: Modloader::Forge, download_folder: String::from("./dl") }; 
 
     let versions = vec![Version { 
             id: "dEqtGnT9".to_string(),
@@ -148,5 +128,5 @@ async fn download_updates_test() {
                 "fabric".to_string()
             ]
         }];
-    assert_eq!(download_updates(config, versions).await.unwrap(), "./dl")
+    assert!(download_updates(current_list, versions).await.is_ok())
 }