extracted filedownload to fn;

fixed some tests;
added direct-dl to update

5 files changed, 52 insertions, 75 deletions

diff --git a/src/apis/modrinth.rs b/src/apis/modrinth.rs
index ec8d203..c99cfbf 100644
--- a/src/apis/modrinth.rs
+++ b/src/apis/modrinth.rs
@@ -4,7 +4,7 @@ use serde::Deserialize;
 
 use crate::{Modloader, List};
 
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Deserialize, Clone)]
 pub struct Project {
     pub slug: String,
     pub title: String,
@@ -29,21 +29,21 @@ pub struct Project {
     pub versions: Vec<String>,
 }
 
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Deserialize, Clone)]
 pub struct License {
     pub id: String,
     pub name: String,
     pub url: Option<String>,
 }
 
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Deserialize, Clone)]
 pub struct ModeratorMessage {
     pub message: String,
     pub body: Option<String>,
 }
 
 #[allow(non_camel_case_types)]
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Deserialize, Clone)]
 pub enum Side {
     required,
     optional,
@@ -51,7 +51,7 @@ pub enum Side {
 }
 
 #[allow(non_camel_case_types)]
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Deserialize, Clone)]
 pub enum Type {
     r#mod,
     modpack,
@@ -59,7 +59,7 @@ pub enum Type {
 }
 
 #[allow(non_camel_case_types)]
-#[derive(Debug, Deserialize)]
+#[derive(Debug, Deserialize, Clone)]
 pub enum Status {
     approved,
     rejected,
diff --git a/src/commands/download.rs b/src/commands/download.rs
index db0fc93..82d6b02 100644
--- a/src/commands/download.rs
+++ b/src/commands/download.rs
@@ -1,9 +1,3 @@
-use std::{io::Write, fs::File};
-
-use reqwest::Client;
-
-use futures_util::StreamExt;
-
 use crate::{List, get_current_list, config::Cfg, db::userlist_get_all_downloads, input::Input};
 
 pub async fn download(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
@@ -16,9 +10,11 @@ pub async fn download(config: Cfg, input: Input) -> Result<(), Box<dyn std::erro
     Ok(())
 }
 
-async fn download_links(config: Cfg, input: Input, current_list: List, links: Vec<String>) -> Result<String, Box<dyn std::error::Error>> {
-
-    let dl_path = String::from(&config.downloads);
+async fn download_links(_config: Cfg, _input: Input, _current_list: List, _links: Vec<String>) -> Result<String, Box<dyn std::error::Error>> {
+    println!("NO DL IMPLEMENTATION FOR DOWNLOAD YET");
+    //TODO copy dl from update if possible
+    /*
+    let dl_path = String::from(&current_list.download_folder);
 
     if input.clean {
         let dl_path = &current_list.download_folder;
@@ -28,28 +24,7 @@ async fn download_links(config: Cfg, input: Input, current_list: List, links: Ve
             std::fs::remove_file(entry.path())?;
         }
     }
+    */
 
-    for link in links {
-        let filename = link.split('/').last().unwrap();
-        let dl_path_file = format!("{}/{}", config.downloads, filename);
-        println!("Downloading {}", link);
-
-        let res = Client::new()
-            .get(String::from(&link))
-            .send()
-            .await
-            .or(Err(format!("Failed to GET from '{}'", &link)))?;
-        
-        // 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"))?;
-        }
-    }
-
-    Ok(dl_path)
+    Ok(String::new())
 }
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())
 }
diff --git a/src/input.rs b/src/input.rs
index 19aa2c2..0b616d4 100644
--- a/src/input.rs
+++ b/src/input.rs
@@ -129,7 +129,7 @@ pub async fn get_input(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
 
 #[test]
 fn input_from() {
-    let string = "lis add test 1.19.2 fabric";
+    let string = "list add test 1.19.2 fabric";
     let input = Input{ command: Cmd::List, subcommand: Some(Subcmd::Add), args: Some(vec![String::from("test"), String::from("1.19.2"), String::from("fabric")]), force_download: false, direct_download: false, all_lists: false, clean: false, delete_old: false };
     assert_eq!(Input::from(string).unwrap(), input);
 }
diff --git a/src/lib.rs b/src/lib.rs
index 51b4487..e4ebf76 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -5,10 +5,12 @@ pub mod input;
 pub mod db;
 pub mod error;
 
-use std::io::{Error, ErrorKind};
+use std::{io::{Error, ErrorKind, Write}, fs::File};
 
 pub use apis::*;
 pub use commands::*;
+use futures_util::StreamExt;
+use reqwest::Client;
 
 #[derive(Debug, Clone, PartialEq, Eq)]
 pub enum Modloader {
@@ -32,3 +34,23 @@ impl Modloader {
         }
     }
 }
+
+pub async fn download_file(url: String, path: String, name: String) -> Result<(), Box<dyn std::error::Error>> {
+    println!("Downloading {}", url);
+    let dl_path_file = format!("{}/{}", path, name);
+    let res = Client::new()
+        .get(String::from(&url))
+        .send()
+        .await?;
+    
+    // download chunks
+    let mut file = File::create(String::from(&dl_path_file))?;
+    let mut stream = res.bytes_stream();
+
+    while let Some(item) = stream.next().await {
+        let chunk = item?;
+        file.write_all(&chunk)?;
+    }
+
+    Ok(())
+}