Merge branch 'main' into #17-automate-onboarding-mk-ii

# Conflicts:
#	.gitignore
#	src/commands/committee.rs

src/lib.rs

@@ -3,7 +3,7 @@ use serde::{Deserialize, Serialize};
 use serenity::{
   model::{
     guild,
-    id::{GuildId, RoleId},
+    id::{ChannelId, GuildId, RoleId},
   },
   prelude::TypeMapKey,
 };
@@ -65,7 +65,7 @@ pub fn get_config() -> Config {
     wolves_url: "".to_string(),
   };
 
-  if let Ok(x) = env::var("HOME") {
+  if let Ok(x) = env::var("DATABASE_HOME") {
     config.home = x.trim().to_string();
   }
   if let Ok(x) = env::var("DATABASE") {
@@ -214,9 +214,13 @@ pub struct Servers {
   pub server: GuildId,
   pub wolves_api: String,
   pub role_past: Option<RoleId>,
-  pub role_current: Option<RoleId>,
+  pub role_current: RoleId,
   pub member_past: i64,
   pub member_current: i64,
+  pub bot_channel_id: ChannelId,
+  // these can be removed in teh future with an API update
+  pub server_name: String,
+  pub wolves_link: String,
 }
 impl<'r> FromRow<'r, SqliteRow> for Servers {
   fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
@@ -236,15 +240,14 @@ impl<'r> FromRow<'r, SqliteRow> for Servers {
     let role_current = match row.try_get("role_current") {
       Ok(x) => {
         let tmp: i64 = x;
-        if tmp == 0 {
-          None
-        } else {
-          Some(RoleId::from(tmp as u64))
-        }
+        RoleId::from(tmp as u64)
       }
-      _ => None,
+      _ => RoleId::from(0u64),
     };
 
+    let bot_channel_tmp: i64 = row.try_get("bot_channel_id")?;
+    let bot_channel_id = ChannelId::from(bot_channel_tmp as u64);
+
     Ok(Self {
       server,
       wolves_api: row.try_get("wolves_api")?,
@@ -252,6 +255,9 @@ impl<'r> FromRow<'r, SqliteRow> for Servers {
       role_current,
       member_past: row.try_get("member_past")?,
       member_current: row.try_get("member_current")?,
+      bot_channel_id,
+      server_name: row.try_get("server_name")?,
+      wolves_link: row.try_get("wolves_link")?,
     })
   }
 }
@@ -273,6 +279,37 @@ impl<'r> FromRow<'r, SqliteRow> for Minecraft {
   }
 }
 
+#[derive(Debug, Clone, Deserialize, Serialize)]
+pub struct RoleAdder {
+  pub server: GuildId,
+  pub role_a: RoleId,
+  pub role_b: RoleId,
+  pub role_c: RoleId,
+}
+impl<'r> FromRow<'r, SqliteRow> for RoleAdder {
+  fn from_row(row: &'r SqliteRow) -> Result<Self, Error> {
+    let server_tmp: i64 = row.try_get("server")?;
+    let server = GuildId::from(server_tmp as u64);
+
+    Ok(Self {
+      server,
+      role_a: get_role_from_row(row, "role_a"),
+      role_b: get_role_from_row(row, "role_b"),
+      role_c: get_role_from_row(row, "role_c"),
+    })
+  }
+}
+
+fn get_role_from_row(row: &SqliteRow, col: &str) -> RoleId {
+  match row.try_get(col) {
+    Ok(x) => {
+      let tmp: i64 = x;
+      RoleId(tmp as u64)
+    }
+    _ => RoleId::from(0u64),
+  }
+}
+
 pub async fn db_init(config: &Config) -> Result<Pool<Sqlite>, Error> {
   let database = format!("{}/{}", &config.home, &config.database);
 
@@ -286,7 +323,7 @@ pub async fn db_init(config: &Config) -> Result<Pool<Sqlite>, Error> {
     .await?;
 
   // migrations are amazing!
-  sqlx::migrate!("./db/migrations").run(&pool).await.unwrap();
+  sqlx::migrate!("./db/migrations").run(&pool).await?;
 
   Ok(pool)
 }
@@ -389,11 +426,9 @@ pub mod set_roles {
             }
           }
 
-          if let Some(role) = &role_current {
-            if !member.roles.contains(role) {
-              roles_set[1] += 1;
-              roles.push(role.to_owned());
-            }
+          if !member.roles.contains(role_current) {
+            roles_set[1] += 1;
+            roles.push(role_current.to_owned());
           }
 
           if let Err(e) = member.add_roles(ctx, &roles).await {
@@ -408,13 +443,11 @@ pub mod set_roles {
             }
           }
 
-          if let Some(role) = &role_current {
-            if member.roles.contains(role) {
-              roles_set[2] += 1;
-              // if theya re not a current member and have the role then remove it
-              if let Err(e) = member.remove_role(ctx, role).await {
-                println!("{:?}", e);
-              }
+          if member.roles.contains(role_current) {
+            roles_set[2] += 1;
+            // if theya re not a current member and have the role then remove it
+            if let Err(e) = member.remove_role(ctx, role_current).await {
+              println!("{:?}", e);
             }
           }
         }