wip

config.py.sample

@@ -8,4 +8,8 @@ config = {
 	'joinWarningCount': 5,
 	'joinWarningSeconds': 5,
 	'commandPrefix': '$',
+	'kickEmojiName': 'boot',
+	'kickEmoji': '👢',
+	'banEmojiName': 'no_entry_sign',
+	'banEmoji': '🚫',
 }

rocketbot.py

@@ -78,6 +78,31 @@ def saveGuildContext(gc):
 
 # -- Classes ----------------------------------------------------------------
 
+class JoinRaid:
+    def __init__(self, joins):
+        self.joins = joins[:]
+
+    async def addJoin(self, member):
+        for join in self.joins:
+            if join.member.id == member.id:
+                # TODO: Move to front of list
+                return # Already added
+        self.joins.append(JoinRecord(member))
+
+    async def kickAll(self):
+        for join in self.joins:
+            if join.isKicked or join.isBanned:
+                continue
+            await join.member.kick()
+            join.isKicked = True
+
+    async def banAll(self):
+        for join in self.joins:
+            if join.isBanned:
+                continue
+            await join.member.ban()
+            join.isBanned = True
+
 class JoinRecord:
     def __init__(self, member):
         self.member = member
@@ -108,6 +133,34 @@ class GuildContext:
         self.joins.append(JoinRecord(member))
         await self.__checkJoins()
 
+    async def handleReactionAdd(self, message, member, emoji):
+        global config
+        if member.bot:
+            return
+        print('User {0} added emoji {1}'.format(member, emoji))
+        if not member.permissions_in(message.channel).ban_members:
+            print('Reactor does not have ban permissions')
+            return
+        if self.lastWarningMessage == None or message.id != self.lastWarningMessage.id:
+            print('Reacted to a non-warning message')
+            return
+        if emoji.name == config['kickEmoji']:
+            print('Kicking these users:')
+            for join in self.joins:
+                await join.member.kick(reason='Kicked by rocketbot for join raiding')
+                join.isKicked = True
+                print('  ' + join.member.name)
+        elif emoji.name == config['banEmoji']:
+            print('Banning these users:')
+            for join in self.joins:
+                await join.member.ban(reason='Banned by rocketbot for join raiding', delete_message_days=0)
+                join.isBanned = True
+                print('  ' + join.member.name)
+        else:
+            print('Unhandled emoji. Doing nothing.')
+            return
+        await self.__updateLastJoinRaidMessage()
+
     async def handleSetWarningChannel(self, context):
         print('{0.guild.name}: Warning channel set to {0.channel.name}'.format(context))
         self.warningChannel = context.channel
@@ -162,25 +215,52 @@ class GuildContext:
             else:
                 i += 1
 
+    def __joinRaidMessage(self):
+        global config
+        # TODO: Mention mod role
+        message = 'A join raid has been detected! It includes these users:'
+        canKick = False
+        canBan = False
+        for join in self.joins:
+            message += '\n• '
+            if join.isBanned:
+                message += '~~' + join.member.mention + '~~ - banned'
+            elif join.isKicked:
+                message += '~~' + join.member.mention + '~~ - kicked'
+                canBan = True
+            else:
+                message += join.member.mention
+                canKick = True
+                canBan = True
+        if canKick:
+            message += '\nTo kick all these users, react with :' + config['kickEmojiName'] + ':'
+        if canBan:
+            message += '\nTo ban all these users, react with :' + config['banEmojiName'] + ':'
+        return (message, canKick, canBan)
+
+    async def __updateLastJoinRaidMessage(self):
+        if self.lastWarningMessage == None:
+            print('No previous warning message to update')
+            return
+        (message, canKick, canBan) = self.__joinRaidMessage()
+        await self.lastWarningMessage.edit(content=message)
+
     async def __onJoinRaidBegin(self):
+        global config
         print('A join raid has begun!')
         if self.warningChannel == None:
             print('No warning channel set')
             return
-        # TODO: Mention mod role
-        message = 'A join raid has been detected! It includes these users:'
-        for join in self.joins:
-            message += '\n• ' + join.member.mention
+        (message, canKick, canBan) = self.__joinRaidMessage()
         self.lastWarningMessage = await self.warningChannel.send(message)
+        if canKick:
+            await self.lastWarningMessage.add_reaction(config['kickEmoji'])
+        if canBan:
+            await self.lastWarningMessage.add_reaction(config['banEmoji'])
 
     async def __onJoinRaidUpdated(self):
         print('Join raid still occurring')
-        if self.lastWarningMessage == None:
-            return
-        message = 'A join raid has been detected! It includes these users:'
-        for join in self.joins:
-            message += '\n• ' + join.member.mention
-        await self.lastWarningMessage.edit(content=message)
+        await self.__updateLastJoinRaidMessage()
 
     async def __onJoinRaidEnd(self):
         print('Join raid has ended')
@@ -190,6 +270,7 @@ guildIdToGuildContext = {}
 
 def getOrCreateGuildContext(obj, save=True):
     gid = None
+    guild = None
     if obj == None:
         return None
     if isinstance(obj, int):
@@ -198,8 +279,10 @@ def getOrCreateGuildContext(obj, save=True):
         return obj
     elif isinstance(obj, discord.Guild):
         gid = obj.id
+        guild = obj
     elif isinstance(obj, (discord.Message, discord.Member, discord.TextChannel, discord.ext.commands.context.Context)):
         gid = obj.guild.id
+        guild = obj.guild
     if gid == None:
         print('Unhandled datatype', type(obj))
         return None
@@ -207,6 +290,7 @@ def getOrCreateGuildContext(obj, save=True):
     if lookedUp != None:
         return lookedUp
     g = GuildContext(gid)
+    g.guild = guild or g.guild
     guildIdToGuildContext[gid] = g
     if save:
         saveGuildContext(g)
@@ -224,7 +308,7 @@ bot = commands.Bot(command_prefix='$', intents=intents)
 async def hello(ctx):
     message = ctx.message
     print('Got message from {0.author.name} in {0.channel.id}'.format(message))
-    await message.channel.send('Hello, {0.author.mention}!'.format(message))
+    m = await message.channel.send('Hello, {0.author.mention}!'.format(message))
     print('Replied "Hello!"')
 bot.add_command(hello)
 
@@ -248,7 +332,7 @@ bot.add_command(setwarningchannel)
 
 @commands.command()
 @commands.has_permissions(manage_messages=True)
-async def setwarningrole(ctx):
+async def setwarningrole(ctx, role: str):
     g = getOrCreateGuildContext(ctx)
     if g == None:
         return
@@ -291,6 +375,24 @@ async def on_member_join(member):
 async def on_member_remove(member):
     print('User {0.name} left'.format(member))
 
+@bot.listen()
+async def on_raw_reaction_add(payload: discord.RawReactionActionEvent):
+    # payload.message_id: int
+    # payload.user_id: int
+    # payload.channel_id: int
+    # payload.guild_id: int
+    # payload.emoji: PartialEmoji
+    # payload.member: Member
+    # payload.event_type: str
+    global bot
+    guild = bot.get_guild(payload.guild_id)
+    channel = guild.get_channel(payload.channel_id)
+    message = await channel.fetch_message(payload.message_id)
+    member = payload.member
+    emoji = payload.emoji
+    gc = getOrCreateGuildContext(guild)
+    await gc.handleReactionAdd(message, member, emoji)
+
 print('Starting bot')
 bot.run(config['clientToken'])
 print('Bot done')