121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
import discord
|
|
from discord import app_commands
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
import logging
|
|
|
|
# Temporary storage for moments (consider database in future)
|
|
saved_moments = []
|
|
incident_logs = {}
|
|
|
|
|
|
async def setup(client):
|
|
# Create command group
|
|
moments_group = app_commands.Group(
|
|
name="moments",
|
|
description="Capture and manage important moments"
|
|
)
|
|
|
|
# Funny moment command
|
|
@moments_group.command(
|
|
name="funny",
|
|
description="Tag a funny moment with reactions"
|
|
)
|
|
@app_commands.describe(
|
|
message_link="Link to the funny message",
|
|
description="Optional description"
|
|
)
|
|
async def funny_moment(
|
|
interaction: discord.Interaction,
|
|
message_link: str,
|
|
description: Optional[str] = None
|
|
):
|
|
"""Capture a funny moment with reaction tracking"""
|
|
try:
|
|
# Parse message ID from link
|
|
parts = message_link.split("/")
|
|
channel_id = int(parts[-2])
|
|
message_id = int(parts[-1])
|
|
|
|
channel = interaction.guild.get_channel(channel_id)
|
|
message = await channel.fetch_message(message_id)
|
|
|
|
# Add tracking reactions
|
|
await message.add_reaction("😂")
|
|
await message.add_reaction("🎉")
|
|
|
|
# Store moment
|
|
saved_moments.append({
|
|
"message": message.content,
|
|
"author": message.author.id,
|
|
"description": description,
|
|
"timestamp": datetime.now().isoformat()
|
|
})
|
|
|
|
await interaction.response.send_message(
|
|
"✅ Funny moment registered! Reactions added!",
|
|
ephemeral=True
|
|
)
|
|
|
|
except Exception as e:
|
|
logging.error(f"Funny moment error: {e}")
|
|
await interaction.response.send_message(
|
|
"❌ Failed to register funny moment. Check message link!",
|
|
ephemeral=True
|
|
)
|
|
|
|
# Incident tracking command
|
|
@moments_group.command(
|
|
name="incident",
|
|
description="Log an incident with recent messages"
|
|
)
|
|
@app_commands.describe(
|
|
message_count="Number of messages to capture (max 50)",
|
|
reason="Reason for logging"
|
|
)
|
|
@app_commands.checks.has_permissions(manage_messages=True)
|
|
async def incident_log(
|
|
interaction: discord.Interaction,
|
|
message_count: app_commands.Range[int, 1, 50],
|
|
reason: str
|
|
):
|
|
"""Log messages for moderator review"""
|
|
try:
|
|
# Fetch messages
|
|
messages = [
|
|
msg async for msg in interaction.channel.history(limit=message_count)
|
|
][::-1] # Reverse to get chronological order
|
|
|
|
# Format log
|
|
log_content = "\n".join(
|
|
f"{msg.author.display_name} ({msg.author.id}) | {msg.created_at}:\n"
|
|
f"{msg.content}\n{'-'*40}"
|
|
for msg in messages
|
|
)
|
|
|
|
# Store log
|
|
incident_id = f"incident_{datetime.now().timestamp()}"
|
|
incident_logs[incident_id] = {
|
|
"reason": reason,
|
|
"messages": log_content,
|
|
"moderator": interaction.user.id
|
|
}
|
|
|
|
# Send confirmation
|
|
await interaction.response.send_message(
|
|
f"✅ Incident logged with ID `{incident_id}`\n"
|
|
f"**Reason:** {reason}\n"
|
|
f"Captured {len(messages)} messages",
|
|
ephemeral=True
|
|
)
|
|
|
|
except Exception as e:
|
|
logging.error(f"Incident log error: {e}")
|
|
await interaction.response.send_message(
|
|
"❌ Failed to log incident. Check permissions!",
|
|
ephemeral=True
|
|
)
|
|
|
|
# Add commands to client
|
|
client.tree.add_command(moments_group)
|