From 144155259d841dc510082b3e72e6147e22bf03ea Mon Sep 17 00:00:00 2001
From: Sheldon Lee <sheldon.m.lee@gmail.com>
Date: Sat, 25 Sep 2021 21:17:31 +0000
Subject: [PATCH] Added experimentation with reactions.

---
 bot.py | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 54 insertions(+), 5 deletions(-)

diff --git a/bot.py b/bot.py
index e2621e4..31330fc 100644
--- a/bot.py
+++ b/bot.py
@@ -1,21 +1,40 @@
 import discord
 import random
+import asyncio
 from hangfish import Hangfish
 from command import Command as cmd
+from queue import Queue
 
 AQUATIC_EMOJIS = ( "octopus", "squid", "shrimp", "lobster", "oyster", "crab", "blowfish", "tropical_fish", "fish", "dolphin", "whale", "whale2", "shark", )
 AQUATIC_ANIMALS = ( "octopus", "squid", "shrimp", "lobster", "oyster", "crab", "blowfish", "fish", "dolphin", "whale", "shark", )
+AQUATIC_EMOJIS_RAW = (
+'\U0001F40B',
+'\U0001F419',
+'\U0001F41A',
+'\U0001F41F',
+'\U0001F42C',
+'\U0001F420',
+'\U0001F421',
+'\U0001F433',
+'\U0001F980',
+'\U0001F988',
+'\U0001F990',
+'\U0001F991',
+'\U0001F99E',
+'\U0001F9AA',
+'\U0001F9AD'
+)
 
 def strToEmoji(string):
     return ":"+string+":"
 
 def randomAquatic():
-    return strToEmoji(AQUATIC_EMOJIS[random.randint(0, len(AQUATIC_EMOJIS)-1)])
+    return AQUATIC_EMOJIS_RAW[random.randint(0, len(AQUATIC_EMOJIS)-1)]
 
 def aquaticAll():
     string = ""
-    for animal in AQUATIC_EMOJIS:
-            string += strToEmoji(animal)
+    for animal in AQUATIC_EMOJIS_RAW:
+            string += animal
     return string
 
 #
@@ -25,18 +44,33 @@ def aquaticAll():
 class Bot(discord.Client):
 
     hangfish_instances = {}
+    last_reactions = Queue()
 
     async def on_ready(self):
         print("We have logged in as {}".format(self.user))
 
     async def on_message(self, message):
         if message.author == self.user:
-                return
+            return
         
         print ("{} sent {}".format(message.author, message.content))
 
         await cmd.call(message.channel, message.content)
 
+    async def on_raw_reaction_add(self, payload):
+        #print("Reaction Add:\n Channel: {}\n Message: {}\n User: {}".format(payload.channel_id, payload.message_id, payload.user_id))
+        message = self.get_channel(payload.channel_id).get_partial_message(payload.message_id)
+        reaction = randomAquatic()
+        self.last_reactions.put(reaction)
+        await message.add_reaction(reaction)
+        await asyncio.sleep(4)
+        await message.remove_reaction(self.last_reactions.get(), self.user)
+        return
+
+    async def on_raw_reaction_remove(self, payload):
+        #print("Reaction Remove:\n Channel: {}\n Message: {}\n User: {}".format(payload.channel_id, payload.message_id, payload.user_id))
+        return
+
 #
 # Define callbacks
 #
@@ -75,12 +109,22 @@ class Bot(discord.Client):
         instance = Bot.getHangfishInstance(channel)
         if instance is None:
             await channel.send("Hangfish instance not created.")
+            return
         instance.guess(guess)
         await channel.send("```"+instance.getString()+"```")
         if not instance.running:
             del Bot.hangfish_instances[channel.id]
             await channel.send("Game over.")
 
+    @staticmethod
+    async def pong(channel):
+        await channel.send("pong")
+        return
+
+    @staticmethod
+    async def setRulesChannel(channel):
+        await channel.send("<insert message>")
+        return
 #
 # Register callbacks
 #
@@ -88,17 +132,22 @@ class Bot(discord.Client):
 cmd.registerCallback("aquatic", Bot.aquatic)
 cmd.registerCallback("hangfish", Bot.createHangfishInstance)
 cmd.registerCallback("guess", Bot.updateHangfishInstance)
+cmd.registerCallback("ping", Bot.pong)
+cmd.registerCallback("ruleschannel",  Bot.setRulesChannel)
 
 #
 # Ensure token on first line, with no whitespaces at the end/beginning
 #
 
+intents = discord.Intents.default()
+intents.reactions = True
+
 try:
     token_file = open("token.txt", "r")
     token = token_file.readline()
     token_file.close()
     print("\'"+token+"\'")
-    bot = Bot()
+    bot = Bot(intents=intents)
     bot.run(token)
 except discord.errors.LoginFailure:
     print("Invalid token")