97 lines
3.3 KiB
Python
97 lines
3.3 KiB
Python
|
from bot.cogs.cog_manager import BaseCog
|
||
|
|
import discord
|
||
|
|
import datetime
|
||
|
|
import re
|
||
|
|
|
||
|
|
class AFKCog(BaseCog):
|
||
|
|
def __init__(self, bot):
|
||
|
|
super().__init__(bot)
|
||
|
|
self.afk = False
|
||
|
|
self.afk_since = None
|
||
|
|
self.afk_reason = None
|
||
|
|
|
||
|
|
def cleanup(self):
|
||
|
|
self.afk = False
|
||
|
|
self.afk_since = None
|
||
|
|
self.afk_reason = None
|
||
|
|
|
||
|
|
async def cmd_afk(self, message):
|
||
|
|
"""
|
||
|
|
Set AFK status
|
||
|
|
Usage: .afk [reason]
|
||
|
|
"""
|
||
|
|
parts = message.content.split(' ', 1)
|
||
|
|
reason = parts[1] if len(parts) > 1 else "AFK"
|
||
|
|
|
||
|
|
self.afk = True
|
||
|
|
self.afk_since = datetime.datetime.now()
|
||
|
|
self.afk_reason = reason
|
||
|
|
|
||
|
|
await message.edit(content=f"You are now AFK: {reason}")
|
||
|
|
|
||
|
|
async def cmd_unafk(self, message):
|
||
|
|
"""
|
||
|
|
Remove AFK status
|
||
|
|
Usage: .unafk
|
||
|
|
"""
|
||
|
|
if not self.afk:
|
||
|
|
await message.edit(content="You are not AFK.")
|
||
|
|
return
|
||
|
|
|
||
|
|
self.afk = False
|
||
|
|
afk_duration = datetime.datetime.now() - self.afk_since
|
||
|
|
hours, remainder = divmod(afk_duration.seconds, 3600)
|
||
|
|
minutes, seconds = divmod(remainder, 60)
|
||
|
|
|
||
|
|
duration_str = ""
|
||
|
|
if hours > 0:
|
||
|
|
duration_str += f"{hours} hour{'s' if hours != 1 else ''} "
|
||
|
|
if minutes > 0:
|
||
|
|
duration_str += f"{minutes} minute{'s' if minutes != 1 else ''} "
|
||
|
|
if seconds > 0 or (hours == 0 and minutes == 0):
|
||
|
|
duration_str += f"{seconds} second{'s' if seconds != 1 else ''}"
|
||
|
|
|
||
|
|
await message.edit(content=f"Welcome back! You were AFK for {duration_str.strip()}.")
|
||
|
|
|
||
|
|
self.afk_since = None
|
||
|
|
self.afk_reason = None
|
||
|
|
|
||
|
|
async def handle_afk_dm(self, message):
|
||
|
|
"""
|
||
|
|
Handle incoming DMs when in AFK mode
|
||
|
|
"""
|
||
|
|
if not self.afk:
|
||
|
|
return
|
||
|
|
|
||
|
|
# Don't respond to self
|
||
|
|
if message.author == self.bot.user:
|
||
|
|
return
|
||
|
|
|
||
|
|
# Check if we already replied recently
|
||
|
|
# This prevents spam if someone sends multiple messages
|
||
|
|
async for msg in message.channel.history(limit=10):
|
||
|
|
if msg.author == self.bot.user and "I'm currently AFK" in msg.content:
|
||
|
|
# Only reply once every 5 minutes to the same person
|
||
|
|
time_diff = datetime.datetime.now(datetime.timezone.utc) - msg.created_at
|
||
|
|
if time_diff.total_seconds() < 300: # 5 minutes
|
||
|
|
return
|
||
|
|
|
||
|
|
# Calculate AFK duration
|
||
|
|
afk_duration = datetime.datetime.now() - self.afk_since
|
||
|
|
hours, remainder = divmod(afk_duration.seconds, 3600)
|
||
|
|
minutes, seconds = divmod(remainder, 60)
|
||
|
|
|
||
|
|
duration_str = ""
|
||
|
|
if hours > 0:
|
||
|
|
duration_str += f"{hours} hour{'s' if hours != 1 else ''} "
|
||
|
|
if minutes > 0:
|
||
|
|
duration_str += f"{minutes} minute{'s' if minutes != 1 else ''} "
|
||
|
|
if seconds > 0 or (hours == 0 and minutes == 0):
|
||
|
|
duration_str += f"{seconds} second{'s' if seconds != 1 else ''}"
|
||
|
|
|
||
|
|
# Send AFK message
|
||
|
|
await message.channel.send(
|
||
|
|
f"I'm currently AFK ({duration_str.strip()}): {self.afk_reason}\n"
|
||
|
|
f"I'll respond when I return."
|
||
|
|
)
|