Adding autokick option to only kick users who go offline seconds after joining. Added presence permission to intents to get member status.

bot.py

@@ -57,6 +57,7 @@ class Rocketbot(commands.Bot):
 intents = Intents.default()
 intents.messages = True  # pylint: disable=assigning-non-slot
 intents.members = True  # pylint: disable=assigning-non-slot
+intents.presences = True
 bot = Rocketbot(command_prefix=CONFIG['command_prefix'], intents=intents)
 
 # Core

rocketbot/cogs/autokickcog.py

@@ -1,12 +1,15 @@
 from datetime import datetime, timedelta
+from typing import cast
 
-from discord import Guild, Member
-from discord.ext import commands
+from discord import Guild, Member, Status
+from discord.ext import commands, tasks
+from discord.ext.tasks import Loop
 
 from config import CONFIG
 from rocketbot.cogs.basecog import BaseCog, BotMessage, CogSetting
 from rocketbot.collections import AgeBoundDict
 from rocketbot.storage import Storage
+from rocketbot.utils import bot_log
 
 class AutoKickContext:
 	"""
@@ -22,6 +25,11 @@ class AutoKickContext:
 		self.last_kick = time
 		self.kick_count += 1
 
+class StatusCheckContext:
+	def __init__(self, member: Member):
+		self.member = member
+		self.joined_at = datetime.now()
+
 class AutoKickCog(BaseCog, name='Auto Kick'):
 	"""
 	Cog for automatically kicking ALL new joins. For temporary use during join raids.
@@ -36,6 +44,13 @@ class AutoKickCog(BaseCog, name='Auto Kick'):
 					'disables this feature (only kick, never ban).',
 			usage='<count:int>',
 			min_value=0)
+	SETTING_OFFLINE_ONLY = CogSetting('offlineonly', bool,
+			brief='whether to only kick users whose status is offline',
+			description='Compromised accounts may have a status of offline. ' + \
+					'If this setting is enabled, the user\'s status will be ' + \
+					'checked a few seconds after joining. If it is offline ' + \
+					'they will be kicked.',
+			usage='<true|false>')
 
 	STATE_KEY_RECENT_KICKS = "AutoKickCog.recent_joins"
 
@@ -43,6 +58,10 @@ class AutoKickCog(BaseCog, name='Auto Kick'):
 		super().__init__(bot)
 		self.add_setting(AutoKickCog.SETTING_ENABLED)
 		self.add_setting(AutoKickCog.SETTING_BAN_COUNT)
+		self.add_setting(AutoKickCog.SETTING_OFFLINE_ONLY)
+		self.status_check_members = []
+		timer: Loop = cast(Loop, self.status_check_timer)
+		timer.start()
 
 	@commands.group(
 		brief='Automatically kicks all new users as soon as they join',
@@ -60,6 +79,36 @@ class AutoKickCog(BaseCog, name='Auto Kick'):
 		guild: Guild = member.guild
 		if not self.get_guild_setting(guild, self.SETTING_ENABLED):
 			return
+		if self.get_guild_setting(guild, self.SETTING_OFFLINE_ONLY):
+			self.log(guild, f'New member {member.name} status is {member.status}')
+			self.status_check_members.append(StatusCheckContext(member))
+			return
+		self.__kick_or_ban_if_needed(member)
+
+	@tasks.loop(seconds=5.0)
+	async def status_check_timer(self):
+		'Checks status of new members shortly after joining to see if they go offline'
+		contexts = self.status_check_members.copy()
+		self.status_check_members = []
+		now = datetime.now()
+		# bot_log(guild=None, cog_class=None, message=f'Found {len(contexts)} members to check')
+		for c in contexts:
+			context: StatusCheckContext = c
+			member: Member = context.member
+			guild: Guild = member.guild
+			if now - context.joined_at < timedelta(seconds=5.0):
+				# Too soon, check again later
+				self.status_check_members.append(context)
+				continue
+			if member.status != Status.offline:
+				# Online, ignore
+				self.log(guild, f'{member.name} status is {member.status}. Not kicking.')
+				continue
+			self.log(guild, f'{member.name} went offline 5s later')
+			await self.__kick_or_ban_if_needed(member)
+
+	async def __kick_or_ban_if_needed(self, member: Member):
+		guild: Guild = member.guild
 		recent_kicks: AgeBoundDict = Storage.get_state_value(guild, AutoKickCog.STATE_KEY_RECENT_KICKS)
 		if recent_kicks is None:
 			recent_kicks = AgeBoundDict(timedelta(seconds=3600), lambda i, context : context.last_kick)