From e3ee322adcc8fc4dfc5a99a789fc610e33216615 Mon Sep 17 00:00:00 2001
From: Flourish <flourish@goonstation.space>
Date: Sat, 16 Apr 2022 19:21:24 -0400
Subject: [PATCH] WIP uncool word alert reactions

---
 spacebeecentcom/spacebeecentcom.py | 51 ++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/spacebeecentcom/spacebeecentcom.py b/spacebeecentcom/spacebeecentcom.py
index 2d9a169..6586f19 100644
--- a/spacebeecentcom/spacebeecentcom.py
+++ b/spacebeecentcom/spacebeecentcom.py
@@ -159,6 +159,18 @@ async def game_broadcast_asay(
         }
         await goonservers.send_to_servers(servers, send_data, exception=exception)
 
+    async def discord_broadcast_alert(
+        self, channels, server_name, from_key, from_name, msg, exception=None
+    ):
+        if hasattr(channels, "channels"):
+            channels = channels.channels["alert"]
+        out_msg = f"\N{RED EXCLAMATION MARK} [{server_name}] {from_name} ({from_key}) {msg}"
+        messages = await server.subtype.channel_broadcast(self.bot, "alert", out_msg)
+        for message in messages:
+            await message.add_reaction("\N{BLUE SQUARE}")
+            await message.add_reaction("\N{YELLOW SQUARE}")
+            await message.add_reaction("\N{RED SQUARE}")
+
     async def server_dep(self, server: str, server_name: str, api_key: str):
         if api_key != (await self.bot.get_shared_api_tokens("spacebee"))["api_key"]:
             raise self.SpacebeeError("Invalid API key.", 403)
@@ -290,6 +302,15 @@ async def admin(
             await server.subtype.channel_broadcast(self.bot, "admin_misc", out)
             return self.SUCCESS_REPLY
 
+        @app.get("/alert")
+        async def alert(
+            msg: str, key: str = "", name: str = "", server=Depends(self.server_dep)
+        ):
+            await self.discord_broadcast_alert(
+                server.subtype, server.full_name, key, name, msg
+            )
+            return self.SUCCESS_REPLY
+
         @app.get("/admin_debug")
         async def admin_debug(
             msg: str, key: str = "", name: str = "", server=Depends(self.server_dep)
@@ -661,3 +682,33 @@ async def on_message_without_command(self, message: discord.Message):
             import traceback
 
             await self.bot.send_to_owners(traceback.format_exc())
+
+    @commands.Cog.listener()
+    async def on_reaction_add(reaction, user):
+        _id = payload.message_id
+        reaction_msg = reaction.message 
+        message
+
+        if reaction_msg.author.id != self.bot.user.id:
+            return
+        if reaction.me is False:
+            return
+
+        if str(reaction.emoji) == "\N{BLUE SQUARE}":
+            message = "notes"
+        elif str(reaction.emoji == "\N{YELLOW SQUARE}"):
+            message = "pm"
+        elif str(reaction.emoji == "\N{RED SQUARE}"):
+            message = "ban"
+            
+        await self.check_and_send_message(
+            channel_type,
+            message,
+            server_id,
+            {
+                "type": reply_type,
+                "nick": message.author.name,
+                "msg": message.content,
+                "target": target,
+            },
+        )
\ No newline at end of file