1 files changed, 49 insertions, 39 deletions

diff --git a/src/commands/update.rs b/src/commands/update.rs
index eba5e91..42d19aa 100644
--- a/src/commands/update.rs
+++ b/src/commands/update.rs
@@ -1,50 +1,60 @@
 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}, 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}, List, input::Input, download_file};
 
 pub async fn update(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
-
-    let current_list = get_current_list(config.clone())?;
-
-    let mods = userlist_get_all_ids(config.clone(), current_list.clone().id)?;
-
-    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
-        let current_version = &versions[index];
-        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"))) };
-
-
-        //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.clone()).await {
-                Ok(ver) => ver,
-                //TODO handle errors (only continue on "NO_UPDATE_AVAILABLE")
-                Err(_) => { println!("({}) No new version found for the specified minecraft version", project.title); continue; },
-            });
-        } else {
-            println!("({}) No new version found", project.title);
+    let mut liststack: Vec<List> = vec![];
+    if input.all_lists {
+        let list_ids = lists_get_all_ids(config.clone())?;
+        for id in list_ids {
+            liststack.push(lists_get(config.clone(), id)?);
+        }
+    } else {
+        liststack.push(get_current_list(config.clone())?)
+    }
+
+    for current_list in liststack {
+        let mods = userlist_get_all_ids(config.clone(), current_list.clone().id)?;
+
+        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
+            let current_version = &versions[index];
+            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"))) };
+
+
+            //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.clone()).await {
+                    Ok(ver) => ver,
+                    //TODO handle errors (only continue on "NO_UPDATE_AVAILABLE")
+                    Err(_) => { println!("({}) No new version found for the specified minecraft version", project.title); continue; },
+                });
+            } else {
+                println!("({}) No new version found", project.title);
+            };
         };
-    };
-
-    if input.clean {
-        let dl_path = &current_list.download_folder;
-        println!("Cleaning {}", dl_path);
-        for entry in std::fs::read_dir(dl_path)? {
-            let entry = entry?;
-            std::fs::remove_file(entry.path())?;
+
+        if input.clean {
+            let dl_path = &current_list.download_folder;
+            println!("Cleaning {}", dl_path);
+            for entry in std::fs::read_dir(dl_path)? {
+                let entry = entry?;
+                std::fs::remove_file(entry.path())?;
+            }
         }
+        
+        if input.direct_download { download_updates(current_list, updatestack).await?; };
     }
-    
-    if input.direct_download { download_updates(current_list, updatestack).await?; };
 
     Ok(())
 }