From 89eb19cd9adf27eaea2ad40302dde26cd0e10bcd Mon Sep 17 00:00:00 2001
From: Ethan Ruszanowski <ethan@rusz.dev>
Date: Tue, 10 Jan 2023 19:34:35 -0500
Subject: [PATCH] Add team_up command framework

---
 src/main.rs | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 1886728..757cafb 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,7 +2,7 @@ use poise::serenity_prelude as serenity;
 use rand::seq::IteratorRandom;
 use std::{
     fs::File,
-    io::{BufRead, BufReader}
+    io::{BufRead, BufReader},
 };
 
 struct Data {}
@@ -10,25 +10,38 @@ struct Data {}
 type Error = Box<dyn std::error::Error + Send + Sync>;
 type Context<'a> = poise::Context<'a, Data, Error>;
 
-// Basically a ping command. Should be moved to a mod.
+/// Basically a ping command.
 #[poise::command(slash_command, prefix_command)]
 async fn slur(
     ctx: Context<'_>,
 ) -> Result<(), Error> {
-    let file = File::open("quotes.txt").unwrap_or_else(|_e| panic!("Quote file missing."));
-    let file = BufReader::new(file);
+    let file = File::open("quotes.txt").unwrap_or_else(|_e| panic!("Quote file missing.")); // Open quotes file
+    let file = BufReader::new(file); // Read quotes file
     let quotes = file.lines().map(|res| res.expect("Failed to read line."));
-    let quote = quotes.choose(&mut rand::thread_rng()).expect("No lines in file.");
+    let quote = quotes.choose(&mut rand::thread_rng()).expect("No lines in file."); // Pick random quote
 
     ctx.say(quote).await?;
     Ok(())
 }
 
+#[poise::command(slash_command, prefix_command)]
+async fn team_up(
+    ctx: Context<'_>,
+    #[description = "Your voice channel"] channel: Option<serenity::Channel>,
+) -> Result<(), Error> {
+    let c = channel.as_ref().unwrap(); // Get channel info from object
+    let mut v = ctx.guild().unwrap().voice_states; // Get hashmap of users' voice states within the guild
+    v.retain(|_, s| s.channel_id == Some(c.id())); // Drop users not active in requested voice channel from hashmap
+    let res = format!("Channel {} has {} active users", c.id(), v.keys().len());
+    ctx.say(res).await?;
+    Ok(())
+}
+
 #[tokio::main]
 async fn main() {
     let framework = poise::Framework::builder()
         .options(poise::FrameworkOptions {
-            commands: vec![slur()], // IntelliJ doesn't like this, but it's fine.
+            commands: vec![slur(), team_up()], // IntelliJ doesn't like this, but it's fine.
             ..Default::default()
         })
         .token(std::env::var("DISCORD_TOKEN").expect("missing DISCORD_TOKEN"))