chore: add support for allowing commands in multiple guilds

1 files changed, 5 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index 2e9a2f3..61a00c1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -17,6 +17,7 @@ mod consts;
 mod utils;
 
 const TEAWIE_GUILD: GuildId = GuildId(1055663552679137310);
+const ALLOWED_GUILDS: [GuildId; 2] = [TEAWIE_GUILD, GuildId(1091969030694375444)];
 const BOT: u64 = 1056467120986271764;
 
 #[group]
@@ -33,7 +34,9 @@ impl EventHandler for Handler {
 	async fn message(&self, ctx: Context, msg: Message) {
 		let author = msg.author.id.as_u64();
 
-		if author == &BOT || msg.guild_id.unwrap_or_else(|| GuildId::from(0)) != TEAWIE_GUILD {
+		if author == &BOT
+			|| !ALLOWED_GUILDS.contains(&msg.guild_id.unwrap_or_else(|| GuildId::from(0)))
+		{
 			return;
 		}
 
@@ -182,7 +185,7 @@ async fn random_teawie(ctx: &Context, msg: &Message) -> CommandResult {
 
 #[command]
 async fn teawiespam(ctx: &Context, msg: &Message) -> CommandResult {
-	if msg.guild_id.unwrap_or_else(|| GuildId::from(0)) != TEAWIE_GUILD {
+	if !ALLOWED_GUILDS.contains(&msg.guild_id.unwrap_or_else(|| GuildId::from(0))) {
 		return Ok(());
 	}