em/echbot
1
0
Fork 0
mirror of https://github.com/ethanrusz/echbot.git synced 2024-11-25 13:17:45 -05:00
Code Issues Projects Releases Packages Wiki Activity

Update team random setup

Browse source
This commit is contained in:
Em (Ethan) Ruszanowski 2023-01-21 19:46:49 -05:00
parent 5d38b69f48
commit b349263fd0
No known key found for this signature in database
GPG key ID: C3E7A3C0B1491DFE
1 changed files with 31 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/commands/team.rs
View file

@ -1,13 +1,15 @@
use std::collections::HashMap; use std::collections::HashMap;
use crate::{Context, Error};
use crate::serenity;
use poise::serenity_prelude::{Guild, Member, UserId, VoiceState};
use rand::seq::SliceRandom;
use std::string::String; use std::string::String;
/// Return a string of pingable IDs from a slice of UserIds use poise::serenity_prelude::{Guild, Member, UserId, VoiceState};
fn team_to_ping(team: &[&&UserId]) -> String { use rand::seq::SliceRandom;
return team.iter().map(|o| format!("<@{}>", o.to_string())).collect::<Vec<String>>().join(", ");
use crate::{Context, Error};
use crate::serenity;
/// Return a string of pingable IDs from a slice of string UserIds
fn team_to_ping(team: &[&String]) -> String {
return team.iter().map(|o| format!("<@{}>", o)).collect::<Vec<String>>().join(", ");
} }
/// Splits up players for custom matches /// Splits up players for custom matches
@ -37,9 +39,8 @@ pub(crate) async fn team(
} }
let uuid_team: u64 = ctx.id(); // Grab context ID for action row let uuid_team: u64 = ctx.id(); // Grab context ID for action row
let users: Vec<&UserId> = Vec::from_iter(voice_states.keys()); // Get vec of PIDs let users: Vec<String> = Vec::from_iter(voice_states.keys().map(|u| u.to_string())); // Get vec of PIDs
let players: Vec<&&UserId> = users.choose_multiple( let players: Vec<&String> = users.choose_multiple(&mut rand::thread_rng(), size as usize * 2).collect(); // Pick players randomly into slice
&mut rand::thread_rng(), size as usize * 2).collect(); // Pick players randomly into slice
let (order, chaos) = players.split_at(players.len() / 2); // Split slice into two teams let (order, chaos) = players.split_at(players.len() / 2); // Split slice into two teams
ctx.send(|f| f ctx.send(|f| f
@ -61,14 +62,15 @@ pub(crate) async fn team(
while let Some(mci) = serenity::CollectComponentInteraction::new(ctx) // Handle the interaction while let Some(mci) = serenity::CollectComponentInteraction::new(ctx) // Handle the interaction
.await { .await {
let guild: Guild = ctx.guild().unwrap(); let guild: Guild = ctx.guild().unwrap(); // Grab guild from context
for user in chaos { for user in chaos {
let member: Member = guild.member(ctx, UserId(*user.as_u64())).await?; // Get the member in the correct guild let member: Member = guild.member(ctx, UserId(user.parse().unwrap())).await?; // Get the member in the correct guild
member.move_to_voice_channel(ctx, chaos_channel.id()).await?; // Move the member to the correct voice channel member.move_to_voice_channel(ctx, chaos_channel.id()).await?; // Move the member to the correct voice channel
} }
mci.create_interaction_response(ctx, |ir| { // Update embed mci.create_interaction_response(ctx, |ir| { // Edit embed
ir.kind(serenity::InteractionResponseType::UpdateMessage).interaction_response_data(|f| f ir.kind(serenity::InteractionResponseType::UpdateMessage)
.interaction_response_data(|f| f
.embed(|f| f .embed(|f| f
.title(format!("Custom {}v{} Teams", size, size)) .title(format!("Custom {}v{} Teams", size, size))
.description("VVGO VVW VVX") .description("VVGO VVW VVX")