Fixed: Incidents now have a red border instead of green when reporting it.
Fixed: Made the Funny Moments a context command instead of slash command Updated: Changed the status for now to be related to what's being worked on.
This commit is contained in:
parent
c4962f2d09
commit
6f2b13f055
@ -1,6 +1,6 @@
|
|||||||
import discord
|
import discord
|
||||||
from discord import app_commands
|
from discord import app_commands
|
||||||
from typing import Optional
|
from typing import Optional, List
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
import logging
|
import logging
|
||||||
from utils.database import Database
|
from utils.database import Database
|
||||||
@ -8,77 +8,33 @@ from utils.database import Database
|
|||||||
db = Database()
|
db = Database()
|
||||||
|
|
||||||
|
|
||||||
async def setup(client):
|
class IncidentModal(discord.ui.Modal):
|
||||||
moments_group = app_commands.Group(
|
def __init__(self):
|
||||||
name="moments",
|
super().__init__(title="Log Incident")
|
||||||
description="Capture and manage important moments"
|
self.reason = discord.ui.TextInput(
|
||||||
)
|
label="Reason for logging",
|
||||||
|
style=discord.TextStyle.long,
|
||||||
|
required=True
|
||||||
|
)
|
||||||
|
self.message_count = discord.ui.TextInput(
|
||||||
|
label="Messages to capture (1-50)",
|
||||||
|
default="10",
|
||||||
|
required=True
|
||||||
|
)
|
||||||
|
|
||||||
# Funny moment command
|
self.add_item(self.reason)
|
||||||
@moments_group.command(
|
self.add_item(self.message_count)
|
||||||
name="funny",
|
|
||||||
description="Tag a funny moment with reactions"
|
async def on_submit(self, interaction: discord.Interaction):
|
||||||
)
|
|
||||||
@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
|
|
||||||
):
|
|
||||||
try:
|
try:
|
||||||
parts = message_link.split("/")
|
count = int(self.message_count.value)
|
||||||
channel_id = int(parts[-2])
|
if not 1 <= count <= 50:
|
||||||
message_id = int(parts[-1])
|
raise ValueError
|
||||||
|
|
||||||
channel = interaction.guild.get_channel(channel_id)
|
|
||||||
message = await channel.fetch_message(message_id)
|
|
||||||
|
|
||||||
await message.add_reaction("😂")
|
|
||||||
await message.add_reaction("🎉")
|
|
||||||
|
|
||||||
# Store in database
|
|
||||||
record_id = db.add_funny_moment(
|
|
||||||
message_link=message_link,
|
|
||||||
author_id=message.author.id,
|
|
||||||
description=description
|
|
||||||
)
|
|
||||||
|
|
||||||
await interaction.response.send_message(
|
|
||||||
f"✅ Funny moment registered! (ID: {record_id})",
|
|
||||||
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 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
|
|
||||||
):
|
|
||||||
try:
|
|
||||||
messages = [
|
messages = [
|
||||||
msg async for msg in interaction.channel.history(limit=message_count)
|
msg async for msg in interaction.channel.history(limit=count)
|
||||||
][::-1]
|
][::-1]
|
||||||
|
|
||||||
# Prepare messages for storage
|
|
||||||
formatted_messages = [{
|
formatted_messages = [{
|
||||||
"id": msg.id,
|
"id": msg.id,
|
||||||
"author_id": msg.author.id,
|
"author_id": msg.author.id,
|
||||||
@ -86,13 +42,11 @@ async def setup(client):
|
|||||||
"timestamp": msg.created_at
|
"timestamp": msg.created_at
|
||||||
} for msg in messages]
|
} for msg in messages]
|
||||||
|
|
||||||
# Generate incident ID
|
|
||||||
incident_id = f"incident_{int(datetime.now().timestamp())}"
|
incident_id = f"incident_{int(datetime.now().timestamp())}"
|
||||||
|
|
||||||
# Store in database
|
|
||||||
success = db.add_incident(
|
success = db.add_incident(
|
||||||
incident_id=incident_id,
|
incident_id=incident_id,
|
||||||
reason=reason,
|
reason=self.reason.value,
|
||||||
moderator_id=interaction.user.id,
|
moderator_id=interaction.user.id,
|
||||||
messages=formatted_messages
|
messages=formatted_messages
|
||||||
)
|
)
|
||||||
@ -100,21 +54,81 @@ async def setup(client):
|
|||||||
if not success:
|
if not success:
|
||||||
raise Exception("Database storage failed")
|
raise Exception("Database storage failed")
|
||||||
|
|
||||||
|
embed = discord.Embed(
|
||||||
|
title="Incident Logged",
|
||||||
|
description=f"**ID:** `{incident_id}`\n**Reason:** {self.reason.value}",
|
||||||
|
color=0xff0000
|
||||||
|
)
|
||||||
|
preview = messages[0].content[:50] + "..." if messages else "No messages"
|
||||||
|
embed.add_field(name="First Message", value=preview, inline=False)
|
||||||
|
|
||||||
await interaction.response.send_message(
|
await interaction.response.send_message(
|
||||||
f"✅ Incident logged with ID `{incident_id}`\n"
|
embed=embed,
|
||||||
f"**Reason:** {reason}\n"
|
ephemeral=True
|
||||||
f"Captured {len(messages)} messages",
|
)
|
||||||
|
|
||||||
|
except ValueError:
|
||||||
|
await interaction.response.send_message(
|
||||||
|
"❌ Please enter a number between 1-50",
|
||||||
|
ephemeral=True
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def setup(client):
|
||||||
|
# Context menu command
|
||||||
|
@client.tree.context_menu(name="Mark as Funny Moment")
|
||||||
|
async def mark_funny(interaction: discord.Interaction, message: discord.Message):
|
||||||
|
try:
|
||||||
|
await message.add_reaction("😂")
|
||||||
|
await message.add_reaction("🎉")
|
||||||
|
|
||||||
|
message_link = f"https://discord.com/channels/{interaction.guild_id}/{message.channel.id}/{message.id}"
|
||||||
|
record_id = db.add_funny_moment(
|
||||||
|
message_link=message_link,
|
||||||
|
author_id=message.author.id
|
||||||
|
)
|
||||||
|
|
||||||
|
embed = discord.Embed(
|
||||||
|
title="😂 Funny Moment Saved",
|
||||||
|
description=f"[Jump to Message]({message_link})",
|
||||||
|
color=0x00ff00
|
||||||
|
).set_author(
|
||||||
|
name=message.author.display_name,
|
||||||
|
icon_url=message.author.avatar.url
|
||||||
|
).add_field(
|
||||||
|
name="Message Preview",
|
||||||
|
value=message.content[:100] + "..." if len(message.content) > 100 else message.content,
|
||||||
|
inline=False
|
||||||
|
)
|
||||||
|
|
||||||
|
await interaction.response.send_message(
|
||||||
|
embed=embed,
|
||||||
ephemeral=True
|
ephemeral=True
|
||||||
)
|
)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.error(f"Incident log error: {e}")
|
logging.error(f"Context menu error: {e}")
|
||||||
await interaction.response.send_message(
|
await interaction.response.send_message(
|
||||||
"❌ Failed to log incident. Check permissions and message count!",
|
"❌ Couldn't mark this message. Is it too old?",
|
||||||
ephemeral=True
|
ephemeral=True
|
||||||
)
|
)
|
||||||
|
|
||||||
# Add review command
|
# Command group
|
||||||
|
moments_group = app_commands.Group(
|
||||||
|
name="moments",
|
||||||
|
description="Manage memorable moments"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Incident command
|
||||||
|
@moments_group.command(
|
||||||
|
name="incident",
|
||||||
|
description="Log an incident with recent messages"
|
||||||
|
)
|
||||||
|
@app_commands.checks.has_permissions(manage_messages=True)
|
||||||
|
async def incident_log(interaction: discord.Interaction):
|
||||||
|
await interaction.response.send_modal(IncidentModal())
|
||||||
|
|
||||||
|
# Review command
|
||||||
@moments_group.command(
|
@moments_group.command(
|
||||||
name="review",
|
name="review",
|
||||||
description="Review a logged incident"
|
description="Review a logged incident"
|
||||||
@ -136,7 +150,6 @@ async def setup(client):
|
|||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
# Format response
|
|
||||||
messages = "\n\n".join(
|
messages = "\n\n".join(
|
||||||
f"**{msg['timestamp']}** <@{msg['author_id']}>:\n"
|
f"**{msg['timestamp']}** <@{msg['author_id']}>:\n"
|
||||||
f"{msg['content']}"
|
f"{msg['content']}"
|
||||||
@ -166,4 +179,16 @@ async def setup(client):
|
|||||||
ephemeral=True
|
ephemeral=True
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Autocomplete
|
||||||
|
@review_incident.autocomplete("incident_id")
|
||||||
|
async def incident_autocomplete(
|
||||||
|
interaction: discord.Interaction,
|
||||||
|
current: str
|
||||||
|
) -> List[app_commands.Choice[str]]:
|
||||||
|
incidents = db.get_recent_incidents(interaction.user.id, 25)
|
||||||
|
return [
|
||||||
|
app_commands.Choice(name=inc["id"], value=inc["id"])
|
||||||
|
for inc in incidents if current.lower() in inc["id"].lower()
|
||||||
|
][:25]
|
||||||
|
|
||||||
client.tree.add_command(moments_group)
|
client.tree.add_command(moments_group)
|
||||||
|
24
main.py
24
main.py
@ -6,6 +6,7 @@ from dotenv import load_dotenv
|
|||||||
from pydantic_settings import BaseSettings
|
from pydantic_settings import BaseSettings
|
||||||
|
|
||||||
|
|
||||||
|
# Configuration
|
||||||
class Settings(BaseSettings):
|
class Settings(BaseSettings):
|
||||||
DISCORD_TOKEN: str
|
DISCORD_TOKEN: str
|
||||||
DISCORD_GUILD: int
|
DISCORD_GUILD: int
|
||||||
@ -17,7 +18,8 @@ class Settings(BaseSettings):
|
|||||||
|
|
||||||
config = Settings()
|
config = Settings()
|
||||||
|
|
||||||
# Initialize logging
|
|
||||||
|
# Logging setup
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
level=logging.INFO,
|
level=logging.INFO,
|
||||||
format="%(asctime)s | %(levelname)s | %(message)s",
|
format="%(asctime)s | %(levelname)s | %(message)s",
|
||||||
@ -35,23 +37,35 @@ class EmeraldClient(discord.Client):
|
|||||||
self.tree = app_commands.CommandTree(self)
|
self.tree = app_commands.CommandTree(self)
|
||||||
|
|
||||||
async def setup_hook(self):
|
async def setup_hook(self):
|
||||||
"""Sync commands with test guild"""
|
# Load commands
|
||||||
from commands.moments import setup as moments_setup
|
from commands.moments import setup as moments_setup
|
||||||
await moments_setup(self)
|
await moments_setup(self)
|
||||||
|
|
||||||
|
# Sync commands
|
||||||
guild = discord.Object(id=config.DISCORD_GUILD)
|
guild = discord.Object(id=config.DISCORD_GUILD)
|
||||||
self.tree.copy_global_to(guild=guild)
|
self.tree.copy_global_to(guild=guild)
|
||||||
await self.tree.sync(guild=guild)
|
await self.tree.sync(guild=guild)
|
||||||
logging.info("Commands synced to guild")
|
logging.info("Commands synced")
|
||||||
|
|
||||||
async def on_ready(self):
|
async def on_ready(self):
|
||||||
logging.info(f"Logged in as {self.user} (ID: {self.user.id})")
|
logging.info(f"Logged in as {self.user}")
|
||||||
await self.change_presence(activity=discord.Activity(
|
await self.change_presence(activity=discord.Activity(
|
||||||
type=discord.ActivityType.listening,
|
type=discord.ActivityType.listening,
|
||||||
name="your requests"
|
name="/moments"
|
||||||
))
|
))
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
client = EmeraldClient()
|
client = EmeraldClient()
|
||||||
|
|
||||||
|
# Global error handler
|
||||||
|
@client.tree.error
|
||||||
|
async def on_error(interaction: discord.Interaction, error):
|
||||||
|
logging.error(f"Error: {error}")
|
||||||
|
await interaction.response.send_message(
|
||||||
|
"⚠️ Something went wrong. Please try again.",
|
||||||
|
ephemeral=True
|
||||||
|
)
|
||||||
|
|
||||||
client.run(config.DISCORD_TOKEN)
|
client.run(config.DISCORD_TOKEN)
|
@ -119,3 +119,15 @@ class Database:
|
|||||||
"details": dict(incident),
|
"details": dict(incident),
|
||||||
"messages": [dict(msg) for msg in messages]
|
"messages": [dict(msg) for msg in messages]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
def get_recent_incidents(self, moderator_id: int, limit: int = 25):
|
||||||
|
with self._get_connection() as conn:
|
||||||
|
conn.row_factory = sqlite3.Row
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT id FROM incidents
|
||||||
|
WHERE moderator_id = ?
|
||||||
|
ORDER BY timestamp DESC
|
||||||
|
LIMIT ?
|
||||||
|
""", (moderator_id, limit))
|
||||||
|
return [dict(row) for row in cursor.fetchall()]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user