From bbe7bd43c3cbf12b4e18e9e6882b215973451215 Mon Sep 17 00:00:00 2001
From: jakubmanczak <jakub@manczak.net>
Date: Thu, 1 May 2025 17:13:50 +0200
Subject: [PATCH] guild_icon -> events; also change guild name

---
 src/discordbot/{guild_icon.rs => events.rs} | 52 +++++++++++++++------
 src/discordbot/mod.rs                       |  4 +-
 2 files changed, 40 insertions(+), 16 deletions(-)
 rename src/discordbot/{guild_icon.rs => events.rs} (66%)

diff --git a/src/discordbot/guild_icon.rs b/src/discordbot/events.rs
similarity index 66%
rename from src/discordbot/guild_icon.rs
rename to src/discordbot/events.rs
index 9c70e1a..856549a 100644
--- a/src/discordbot/guild_icon.rs
+++ b/src/discordbot/events.rs
@@ -1,3 +1,4 @@
+use super::events;
 use chrono::{Datelike, Duration, Local, TimeZone, Utc};
 use chrono_tz::Europe::Warsaw;
 use serenity::all::{Context, CreateAttachment, EditGuild, Guild, GuildId};
@@ -6,7 +7,7 @@ use tracing::{error, info, warn};
 
 const MAIN_GUILD_ID: GuildId = GuildId::new(447075692664979466);
 
-enum Event {
+pub enum Event {
     Normal,
     PolskaGórą,
     PrideMonth,
@@ -22,7 +23,7 @@ enum Event {
 }
 
 impl Event {
-    fn icon(&self) -> &str {
+    pub fn icon(&self) -> &str {
         use Event as E;
         match self {
             E::PolskaGórą => "./assets/logo-x512-polish.png",
@@ -31,9 +32,28 @@ impl Event {
             _ => "./assets/logo-x512.png",
         }
     }
+    pub fn guild_name(&self) -> String {
+        use Event as E;
+        match self {
+            E::PolskaGórą => "Gractwo 🇵🇱".into(),
+            E::PrideMonth => "Gractwo 🏳️‍🌈".into(),
+            E::ValentineDay => "Gractwo 💗".into(),
+            E::Rogaliki => "Rogalictwo 🥐".into(),
+            E::Halloween => "Spooky Gractwo 🎃".into(),
+            E::Christmas => "Jolly Gractwo 🎄🎁☃️".into(),
+            E::NewYears => "New Gractwo, New Me 🎉".into(),
+            E::AnniversaryTF2 => "Polscy Gracze Team Fortress".into(),
+            E::AnniversaryGractwo => {
+                // TODO: this will conjugate badly in 17 years
+                let xlecie = Utc::now().year_ce().1 - 2020;
+                format!("{xlecie} lat Gractwa!")
+            }
+            _ => "Gractwo".into(),
+        }
+    }
 }
 
-fn get_current_event() -> Event {
+pub fn get_current_event() -> Event {
     let today = Local::now();
     let (month, day) = (today.month(), today.day());
 
@@ -58,26 +78,26 @@ fn get_current_event() -> Event {
 
 pub fn init_service(ctx: &Context, guild_id: &GuildId) {
     let (ctx, guild_id) = (ctx.clone(), guild_id.clone());
-    info!("Initialising guild name/icon automation service...");
+    info!("Initialising event automation service...");
 
     if guild_id != MAIN_GUILD_ID {
-        info!("Guild name/icon automation service not initialised; Bot not running on main guild.");
+        info!("Guild event automation service not initialised; Bot not running on main guild.");
         return;
     }
 
     tokio::spawn(async move {
-        run_icon_service(ctx, guild_id).await;
+        run_event_service(ctx, guild_id).await;
     });
 }
 
-pub async fn run_icon_service(ctx: Context, guild_id: GuildId) {
+pub async fn run_event_service(ctx: Context, guild_id: GuildId) {
     loop {
-        update_icon(&ctx, guild_id).await.ok();
+        update_guild(&ctx, guild_id).await.ok();
         sleep_until_next_midnight().await;
     }
 }
 
-async fn update_icon(ctx: &Context, guild_id: GuildId) -> Result<(), String> {
+async fn update_guild(ctx: &Context, guild_id: GuildId) -> Result<(), String> {
     let mut guild = match Guild::get(&ctx.http, guild_id).await {
         Ok(g) => g,
         Err(e) => {
@@ -85,7 +105,7 @@ async fn update_icon(ctx: &Context, guild_id: GuildId) -> Result<(), String> {
             return Err("Could not get guild info: {e}".to_string());
         }
     };
-    let event = get_current_event();
+    let event = events::get_current_event();
 
     let icon = match CreateAttachment::path(event.icon()).await {
         Ok(i) => i,
@@ -94,14 +114,18 @@ async fn update_icon(ctx: &Context, guild_id: GuildId) -> Result<(), String> {
             return Err("Could not create icon attachment: {e}".to_string());
         }
     };
+
     match guild
-        .edit(&ctx.http, EditGuild::new().icon(Some(&icon)))
+        .edit(
+            &ctx.http,
+            EditGuild::new().name(event.guild_name()).icon(Some(&icon)),
+        )
         .await
     {
-        Ok(_) => info!("Guild icon updated."),
+        Ok(_) => info!("Guild name/icon updated."),
         Err(e) => {
-            error!("Could not update guild icon...");
-            return Err("Could not update guild icon: {e}".to_string());
+            error!("Could not update guild name/icon...");
+            return Err("Could not update guild name/icon: {e}".to_string());
         }
     };
 
diff --git a/src/discordbot/mod.rs b/src/discordbot/mod.rs
index b3df5b3..2a30c5f 100644
--- a/src/discordbot/mod.rs
+++ b/src/discordbot/mod.rs
@@ -11,7 +11,7 @@ const GUILD_ID: &str = "DISCORD_SERVER_ID";
 struct Handler;
 
 mod commands;
-mod guild_icon;
+mod events;
 
 #[async_trait]
 impl EventHandler for Handler {
@@ -33,7 +33,7 @@ impl EventHandler for Handler {
             Err(why) => info!("Failed to register commands on the guild: {why:?}"),
         };
 
-        guild_icon::init_service(&ctx, &guild_id);
+        events::init_service(&ctx, &guild_id);
     }
 
     async fn interaction_create(&self, ctx: Context, interaction: Interaction) {