use std::io::ErrorKind; use crate::{Modloader, config::Cfg, List, modrinth::Version, get_modloader}; //TODO use prepared statements / change to rusqlite //MODS pub fn insert_mod(config: Cfg, id: String, name: String, versions: Vec) -> Result<(), sqlite::Error> { println!("Inserting into modlist"); let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("INSERT INTO mods VALUES ('{}', '{}', '{}')", id, name, versions.join("|")); connection.execute(sql) } pub fn insert_mod_in_list(config: Cfg, list: List, id: String, current_version: String, applicable_versions: Vec) -> Result<(), sqlite::Error> { println!("Inserting into current list"); let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let mut applicable_versions_vec = vec![]; for ver in applicable_versions { applicable_versions_vec.push(ver.id); } let sql = format!("INSERT INTO {} VALUES ('{}', '{}', '{}')", list.id, id, current_version, applicable_versions_vec.join("|")); connection.execute(sql) } pub fn remove_mod_from_list(config: Cfg, list: List, mod_id: String) -> Result<(), sqlite::Error> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("DELETE FROM {} WHERE mod_id = '{}'", list.id, mod_id); dbg!(&sql); connection.execute(sql) } pub fn get_mods(config: Cfg) -> Result, Box> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = "SELECT id FROM mods"; let mut mods: Vec = Vec::new(); //TODO catch sql errors better connection.iterate(sql, |ids| { if ids.is_empty() { return false; }; for &(_column, value) in ids.iter() { mods.push(String::from(value.unwrap())); } true }).unwrap(); match mods.is_empty() { true => Err(Box::new(std::io::Error::new(ErrorKind::NotFound, "NO_MODS"))), false => Ok(mods), } } pub fn get_mods_from_list(config: Cfg, list: List) -> Result, Box> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("SELECT mod_id FROM {}", list.id); let mut mods: Vec = Vec::new(); //TODO catch sql errors better connection.iterate(sql, |ids| { if ids.is_empty() { return false; }; for &(_column, value) in ids.iter() { mods.push(String::from(value.unwrap())); } true }).unwrap(); match mods.is_empty() { true => Err(Box::new(std::io::Error::new(ErrorKind::NotFound, "NO_MODS"))), false => Ok(mods), } } pub fn get_mod_id(config: Cfg, name: String) -> Result> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("SELECT id FROM mods WHERE name = '{}'", name); dbg!(&sql); let mut modification = String::new(); //TODO catch sql errors better connection.iterate(sql, |id| { if id.is_empty() { return false; }; for &(_column, value) in id.iter() { dbg!(&(_column, value)); modification = String::from(value.unwrap()); } true }).unwrap(); dbg!(&modification); if modification.is_empty() { return Err(Box::new(std::io::Error::new(ErrorKind::NotFound, "MOD_NOT_IN_DATABASE"))) }; Ok(modification) } #[derive(Debug, Clone)] pub struct DBModlistVersions { pub mod_id: String, pub versions: String, } pub fn get_versions(config: Cfg, mods: Vec) -> Result, Box> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let mut wherestr = String::from("WHERE"); for (i, id) in mods.iter().enumerate() { let mut or = " OR"; if i == mods.len() - 1 { or = "" } println!("Pushing {}({}) | OR: '{}'", id, i, or); wherestr = format!("{} id = '{}'{}", wherestr, id, or); } let sql = format!("SELECT id, versions FROM mods {}", wherestr); dbg!(&sql); let mut versionmaps: Vec = Vec::new(); //TODO catch sql errors better let mut cursor = connection.prepare(sql).unwrap().into_cursor(); while let Some(Ok(row)) = cursor.next() { println!("{}: {}", row.get::(0), row.get::(1)); versionmaps.push(DBModlistVersions { mod_id: row.get::(0), versions: row.get::(1) }) }; if versionmaps.is_empty() { return Err(Box::new(std::io::Error::new(ErrorKind::Other, "NO_MODS_ON_LIST"))); }; Ok(versionmaps) } pub fn get_list_version(config: Cfg, list: List, mod_id: String) -> Result> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("SELECT applicable_versions FROM {} WHERE mod_id = '{}'", list.id, mod_id); //TODO catch sql errors better let mut version: String = String::new(); connection.iterate(sql, |ver| { if ver.is_empty() { return false; }; for &(_column, value) in ver.iter() { version = String::from(value.unwrap()); } true }).unwrap(); if version.is_empty() { return Err(Box::new(std::io::Error::new(ErrorKind::Other, "NO_MODS_ON_LIST"))); }; Ok(version) } //LIST pub fn insert_list(config: Cfg, id: String, mc_version: String, mod_loader: Modloader) -> Result<(), sqlite::Error> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql_list = format!("INSERT INTO lists VALUES ('{}', '{}', '{}')", id, mc_version, mod_loader.stringify()); let sql_table = format!("CREATE TABLE '{}' ( 'mod_id' TEXT, 'current_version' TEXT, 'applicable_versions' BLOB)", id); let sql = format!("{};{};", sql_list, sql_table); connection.execute(sql) } pub fn remove_list(config: Cfg, id: String) -> Result<(), sqlite::Error> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql_list = format!("DELETE FROM lists WHERE id = '{}'", id); let sql_table = format!("DROP TABLE '{}'", id); let sql = format!("{};{};", sql_list, sql_table); connection.execute(sql) } pub fn get_lists(config: Cfg) -> Result, Box> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = "SELECT id FROM lists"; let mut list: Vec = Vec::new(); //TODO catch sql errors better connection.iterate(sql, |ids| { if ids.is_empty() { return false; }; for &(_column, value) in ids.iter() { list.push(String::from(value.unwrap())); } true }).unwrap(); match list.is_empty() { true => Err(Box::new(std::io::Error::new(ErrorKind::NotFound, "NO_LISTS"))), false => Ok(list), } } pub fn get_list(config: Cfg, id: String) -> Result> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("SELECT mc_version, modloader FROM lists WHERE id = '{}'", id); let mut list = vec![]; //TODO catch sql errors better connection.iterate(sql, |ids| { if ids.is_empty() { return false; }; for &(_column, value) in ids.iter() { list.push(String::from(value.unwrap())); } true }).unwrap(); if list.len() != 2 { return Err(Box::new(std::io::Error::new(ErrorKind::InvalidData, "LIST_MISSING_DATA"))) }; Ok(List { id, mc_version: String::from(&list[0]), modloader: get_modloader(String::from(&list[1]))? }) } pub fn change_list_versions(config: Cfg, list: List, current_version: String, versions: Vec, mod_id: String) -> Result<(), sqlite::Error> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("UPDATE {} SET current_version = '{}', applicable_versions = '{}' WHERE mod_id = '{}'", list.id, current_version, versions.join("|"), mod_id); connection.execute(sql) } //config pub fn change_list(config: Cfg, id: String) -> Result<(), sqlite::Error> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = format!("UPDATE user_config SET value = '{}' WHERE id = 'current_list'", id); connection.execute(sql) } pub fn get_current_list_id(config: Cfg) -> Result> { let data = format!("{}/data.db", config.data); let connection = sqlite::open(data).unwrap(); let sql = "SELECT id FROM lists"; let mut list: String = String::new(); //TODO catch sql errors better connection.iterate(sql, |ids| { if ids.is_empty() { return false; }; for &(_column, value) in ids.iter() { list = String::from(value.unwrap()); } true }).unwrap(); if list.is_empty() { get_lists(config)?; panic!("current list field should never be empty if there are other lists"); }; Ok(list) }