basic io implementation finished

2 files changed, 91 insertions, 13 deletions

diff --git a/src/commands/io.rs b/src/commands/io.rs
index dc1f408..47991c5 100644
--- a/src/commands/io.rs
+++ b/src/commands/io.rs
@@ -1,12 +1,81 @@
-use crate::{input::{Input, Subcmd}, config::Cfg};
+use std::fs::File;
+use std::io::prelude::*;
+use serde::{Serialize, Deserialize};
 
-pub fn io(_config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
+use crate::{input::{Input, Subcmd}, db::{lists_get, userlist_get_all_ids, lists_get_all_ids, lists_insert}, config::Cfg, Modloader, mod_add, List};
+
+#[derive(Debug, Serialize, Deserialize)]
+struct Export {
+    lists: Vec<ExportList>
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+struct ExportList {
+    id: String,
+    mods: String,
+    launcher: String,
+    mc_version: String,
+    download_folder: Option<String>,
+}
+
+impl ExportList {
+    pub fn from(config: Cfg, list_id: String, download: bool) -> Result<Self, Box<dyn std::error::Error>> {
+
+        let list = lists_get(config.clone(), String::from(&list_id))?;
+
+        let mut dl_folder = None;
+        if download == true { dl_folder = Some(list.download_folder) };
+
+        let mods = userlist_get_all_ids(config, list_id)?.join("|");
+
+        Ok(Self { id: list.id, mods, launcher: list.modloader.stringify(), mc_version: list.mc_version, download_folder: dl_folder })
+    }
+}
+
+pub async fn io(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
     
     match input.subcommand.ok_or("INVALID_INPUT")? {
-        Subcmd::Export => {},
-        Subcmd::Import => {},
-        _ => {},
+        Subcmd::Export => { export(config, input.args)? },
+        Subcmd::Import => { import(config).await? },
+        _ => { },
+    }
+
+    Ok(())
+}
+
+fn export(config: Cfg, _args: Option<Vec<String>>) -> Result<(), Box<dyn std::error::Error>> {
+    let list_ids = lists_get_all_ids(config.clone())?;
+    let mut lists: Vec<ExportList> = vec![];
+    for list_id in list_ids {
+        lists.push(ExportList::from(config.clone(), String::from(list_id), true)?);
     }
+    
+    let toml = toml::to_string( &Export { lists } )?;
+
+    let mut file = File::create("export.toml")?;
+    file.write_all(&toml.as_bytes())?;
 
     Ok(())
 }
+
+async fn import(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
+
+    let mut file = File::open("export.toml")?;
+    let mut content = String::new();
+    file.read_to_string(&mut content)?;
+    let export: Export = toml::from_str(&content)?;
+
+    println!("{:#?}", export);
+
+    for exportlist in export.lists {
+        let list = List { id: exportlist.id, mc_version: exportlist.mc_version, modloader: Modloader::from(&exportlist.launcher)?, download_folder: exportlist.download_folder.ok_or("NO_DL")? };
+        lists_insert(config.clone(), list.id.clone(), list.mc_version.clone(), list.modloader.clone(), String::from(&list.download_folder))?;
+        //TODO currently workaround, too many requests
+        let mods: Vec<&str> = exportlist.mods.split("|").collect();
+        for mod_id in mods {
+            println!("Adding {}", mod_id);
+            mod_add(config.clone(), mod_id, list.clone(), false).await?;
+        }
+    }
+    Ok(())
+}
diff --git a/src/commands/modification.rs b/src/commands/modification.rs
index 91243fc..f66ce28 100644
--- a/src/commands/modification.rs
+++ b/src/commands/modification.rs
@@ -1,6 +1,6 @@
 use std::io::{Error, ErrorKind};
 
-use crate::{modrinth::{project, versions, extract_current_version, Version}, config::Cfg, db::{mods_insert, userlist_remove, mods_get_id, userlist_insert, mods_get_all_ids, userlist_get_all_ids, userlist_get_current_version, lists_get_all_ids, mods_remove}, input::{Input, Subcmd}, get_current_list, files::{delete_version, download_versions}};
+use crate::{modrinth::{project, versions, extract_current_version, Version}, config::Cfg, db::{mods_insert, userlist_remove, mods_get_id, userlist_insert, mods_get_all_ids, userlist_get_all_ids, userlist_get_current_version, lists_get_all_ids, mods_remove}, input::{Input, Subcmd}, get_current_list, files::{delete_version, download_versions}, List};
 
 pub async fn modification(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
     
@@ -22,10 +22,19 @@ async fn add(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>
     if args.is_empty() { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_FEW_ARGUMENTS"))); };
     
     let current_list = get_current_list(config.clone())?;
+    
+    mod_add(config, &args[0], current_list, input.disable_download).await?;
+
+    Ok(())
+}
 
-    let project = project(String::from(&config.apis.modrinth), &args[0]).await;
+pub async fn mod_add(config: Cfg, mod_id: &str, list: List, disable_download: bool) -> Result<(), Box<dyn std::error::Error>> {
+    
+    println!("Adding mod {}", mod_id);
+
+    let project = project(String::from(&config.apis.modrinth), &mod_id).await;
 
-    let available_versions = versions(String::from(&config.apis.modrinth), String::from(&project.id), current_list.clone()).await;
+    let available_versions = versions(String::from(&config.apis.modrinth), String::from(&project.id), list.clone()).await;
 
     let mut available_versions_vec: Vec<String> = Vec::new();
     let current_version: Option<Version>;
@@ -52,15 +61,15 @@ async fn add(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>
     }
 
     //add to current list and mod table
-    match userlist_get_all_ids(config.clone(), current_list.clone().id) {
+    match userlist_get_all_ids(config.clone(), list.clone().id) {
         Ok(mods) => {
             if mods.contains(&project.id) {
                 return Err(Box::new(Error::new(ErrorKind::Other, "MOD_ALREADY_ON_LIST"))); } 
             else {
-                userlist_insert(config.clone(), String::from(&current_list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?;
+                userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?;
             } 
         },
-        Err(..) => userlist_insert(config.clone(), String::from(&current_list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?,
+        Err(..) => userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?,
     };
     
     match mods_get_all_ids(config.clone()) {
@@ -75,8 +84,8 @@ async fn add(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>
             mods_insert(config.clone(), String::from(&project.id), String::from(&project.title), project.versions)?;
         },
     };
-    
-    if !input.disable_download && current_version.is_some() { download_versions(current_list, vec![current_version.unwrap()]).await?; }
+
+    if !disable_download && current_version.is_some() { download_versions(list, vec![current_version.unwrap()]).await?; };
 
     Ok(())
 }