From d66f7895bce1d97c87861098227a4b32d57cfdc2 Mon Sep 17 00:00:00 2001
From: Brendan Golden <git@brendan.ie>
Date: Sat, 26 Aug 2023 16:26:39 +0100
Subject: [PATCH] feat: now have a member count across all threads

---
 src/main.rs | 44 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 43 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index fbf62ee..769b4cc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,6 @@
 use std::env;
 use std::future::Future;
+use std::sync::Arc;
 
 use serenity::async_trait;
 use serenity::model::channel::Message;
@@ -46,8 +47,22 @@ struct Config {
     member_role_past: RoleId,
 }
 
+#[derive(Default)]
+struct Members {
+    members: i32,
+    members_current: i32,
+}
+struct MemberCounter;
+impl TypeMapKey for MemberCounter {
+    type Value = Arc<RwLock<Members>>;
+}
+
 async fn bulk_check(ctx: &Context, config: &Config){
     let mut roles_set = [0,0,0];
+    let mut res = Members {
+        members: 0,
+        members_current: 0,
+    };
     if let Ok(x) = config.server.members(ctx, None, None).await {
         for mut member in x {
             if config.members.contains(&member.user.name) {
@@ -72,10 +87,31 @@ async fn bulk_check(ctx: &Context, config: &Config){
                     println!("{:?}", e);
                 }
             }
+            
+            if member.roles.contains(&config.member_role_past){
+                res.members += 1;
+            }
+            if member.roles.contains(&config.member_role_current){
+                res.members_current += 1;
+            }
         }
     }
     // small bit of logging to note changes over time
     println!("Changes: New: +{}, Current: +{}/-{}", roles_set[0], roles_set[1], roles_set[2]);
+
+    let counter_lock = {
+        let data_read = ctx.data.read().await;
+        data_read.get::<MemberCounter>().expect("Expected MemberCounter in TypeMap.").clone()
+    };
+    {
+        // The HashMap of CommandCounter is wrapped in an RwLock; since we want to write to it, we will
+        // open the lock in write mode.
+        let mut counter = counter_lock.write().await;
+
+        // And we write the amount of times the command has been called to it.
+        counter.members_current = res.members_current;
+        counter.members = res.members;
+    }
 }
 
 #[tokio::main]
@@ -91,7 +127,13 @@ async fn main() {
       .event_handler(Handler)
       .await
       .expect("Error creating client");
-
+    
+    {
+        let mut data = client.data.write().await;
+        data.insert::<MemberCounter>(Arc::new(RwLock::new(Members::default())));
+    }
+    
+    
     // Finally, start a single shard, and start listening to events.
     //
     // Shards will automatically attempt to reconnect, and will perform