1 files changed, 9 insertions, 4 deletions

diff --git a/src/db.rs b/src/db.rs
index 3f05772..d823018 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -32,7 +32,6 @@ pub fn mods_get_all_ids(config: Cfg) -> Result<Vec<String>, Box<dyn std::error::
     })?;
 
     for id in id_iter {
-        //println!("Found id {:?}", id.as_ref().unwrap());
         mods.push(id?);
     }
 
@@ -53,7 +52,6 @@ pub fn mods_get_id(config: Cfg, name: String) -> Result<String, Box<dyn std::err
     })?;
 
     for id in id_iter {
-        //println!("Found id {:?}", id.as_ref().unwrap());
         mod_id = id?;
     };
 
@@ -74,7 +72,6 @@ pub fn mods_get_name(config: Cfg, id: String) -> Result<String, Box<dyn std::err
     })?;
 
     for name in name_iter {
-        //println!("Found id {:?}", id.as_ref().unwrap());
         mod_name = name?;
     };
 
@@ -394,6 +391,14 @@ pub fn lists_get(config: Cfg, list_id: String) -> Result<List, Box<dyn std::erro
     Ok(list)
 }
 
+pub fn lists_version(config: Cfg, list_id: String, version: String) -> Result<(), Box<dyn std::error::Error>> {
+    let data = devdir(format!("{}/data.db", config.data).as_str());
+    let connection = Connection::open(data).unwrap();
+
+    connection.execute("UPDATE lists SET mc_version = ? WHERE id = ?", [version, list_id])?;
+    Ok(())
+}
+
 pub fn lists_get_all_ids(config: Cfg) -> Result<Vec<String>, Box<dyn std::error::Error>> {
     let data = devdir(format!("{}/data.db", config.data).as_str());
     let connection = Connection::open(data).unwrap();
@@ -510,7 +515,7 @@ pub fn db_setup(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
         "CREATE TABLE 'user_config' ( 'id' TEXT, 'value' TEXT );
          CREATE TABLE 'mods' ( 'id' TEXT, 'name' TEXT, 'versions' TEXT );
          CREATE TABLE 'lists' ( 'id' TEXT, 'mc_version' TEXT, 'modloader' TEXT, 'download_folder' TEXT );
-         INSERT INTO 'user_config' VALUES ( 'db_version', '0.3' );
+         INSERT INTO 'user_config' VALUES ( 'db_version', '0.4' );
          INSERT INTO 'user_config' VALUES ( 'current_list', '...' )",
     )?;