feat: will update users when there are changes made to teh server config

src/lib.rs

@@ -10,6 +10,7 @@ use serenity::{
 
 use chrono::{Datelike, SecondsFormat, Utc};
 use rand::{distributions::Alphanumeric, thread_rng, Rng};
+use serenity::client::Context;
 use serenity::model::id::UserId;
 use sqlx::{
   sqlite::{SqliteConnectOptions, SqlitePoolOptions, SqliteRow},
@@ -377,3 +378,118 @@ pub fn get_now_iso(short: bool) -> String {
 pub fn random_string(len: usize) -> String {
   thread_rng().sample_iter(&Alphanumeric).take(len).map(char::from).collect()
 }
+
+pub async fn update_server(db: &Pool<Sqlite>, ctx: &Context, server: &Servers, remove_roles: &[Option<RoleId>]) {
+  let Servers {
+    server,
+    role_past,
+    role_current,
+    ..
+  } = server;
+
+  let mut roles_set = [0, 0, 0];
+  let mut members = vec![];
+
+  for member in get_server_member_bulk(db, server).await {
+    if let Some(x) = member.discord {
+      members.push(x);
+    }
+  }
+  let mut members_all = members.len();
+
+  if let Ok(x) = server.members(ctx, None, None).await {
+    for mut member in x {
+      if members.contains(&member.user.id) {
+        let mut roles = vec![];
+
+        if let Some(role) = &role_past {
+          if !member.roles.contains(role) {
+            roles_set[0] += 1;
+            roles.push(role.to_owned());
+          }
+        }
+
+        if let Some(role) = &role_current {
+          if !member.roles.contains(role) {
+            roles_set[1] += 1;
+            roles.push(role.to_owned());
+          }
+        }
+
+        if let Err(e) = member.add_roles(ctx, &roles).await {
+          println!("{:?}", e);
+        }
+      } else {
+        // old and never
+
+        if let Some(role) = &role_past {
+          if member.roles.contains(role) {
+            members_all += 1;
+          }
+        }
+
+        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);
+            }
+          }
+        }
+      }
+      for role in remove_roles.iter().flatten() {
+        if let Err(e) = member.remove_role(ctx, role).await {
+          println!("{:?}", e);
+        }
+      }
+    }
+  }
+
+  set_server_numbers(db, server, members_all as i64, members.len() as i64).await;
+
+  // small bit of logging to note changes over time
+  println!("{:?} Changes: New: +{}, Current: +{}/-{}", server.as_u64(), roles_set[0], roles_set[1], roles_set[2]);
+}
+
+async fn get_server_member_bulk(db: &Pool<Sqlite>, server: &GuildId) -> Vec<ServerMembersWolves> {
+  sqlx::query_as::<_, ServerMembersWolves>(
+    r#"
+      SELECT * 
+      FROM server_members
+      JOIN wolves USING (id_wolves)
+      WHERE (
+        server = ? 
+        AND discord IS NOT NULL
+        AND expiry > ?
+      )
+      "#,
+  )
+  .bind(*server.as_u64() as i64)
+  .bind(get_now_iso(true))
+  .fetch_all(db)
+  .await
+  .unwrap_or_default()
+}
+
+async fn set_server_numbers(db: &Pool<Sqlite>, server: &GuildId, past: i64, current: i64) {
+  match sqlx::query_as::<_, Wolves>(
+    "
+      UPDATE servers
+      SET member_past = ?, member_current = ?
+      WHERE server = ?
+    ",
+  )
+  .bind(past)
+  .bind(current)
+  .bind(*server.as_u64() as i64)
+  .fetch_optional(db)
+  .await
+  {
+    Ok(_) => {}
+    Err(e) => {
+      println!("Failure to insert into {}", server.as_u64());
+      println!("{:?}", e);
+    }
+  }
+}