bot.py

import os
import sys
import logging
import discord
from discord.ext import commands

from db.database import Database

logging.basicConfig(level=logging.INFO, stream=sys.stdout)

REQUIRED_ENV = ["DISCORD_TOKEN", "GITHUB_USERNAME", "MOD_LOG_CHANNEL_ID", "WELCOME_CHANNEL_ID", "GITHUB_NOTIFY_CHANNEL_ID"]

COGS = [
    "cogs.github",
    "cogs.qflix",
    "cogs.info",
    "cogs.welcome",
    "cogs.reminders",
    "cogs.polls",
    "cogs.autoresponse",
    "cogs.logging",
    "cogs.announcements",
    "cogs.purge",
    "cogs.trim",
    "cogs.celebrations",
    "cogs.timestamp",
    "cogs.role_panels",
]


def check_env():
    missing = [v for v in REQUIRED_ENV if not os.environ.get(v)]
    if missing:
        print(f"Missing required environment variables: {', '.join(missing)}", file=sys.stderr)
        sys.exit(1)


class QuadstroNot(commands.Bot):
    def __init__(self):
        intents = discord.Intents.default()
        intents.message_content = True
        intents.members = True
        intents.voice_states = True
        intents.presences = True
        super().__init__(command_prefix="!", intents=intents)
        self.database = Database()

    async def setup_hook(self):
        await self.database.connect()
        for cog in COGS:
            try:
                await self.load_extension(cog)
                print(f"Loaded {cog}")
            except Exception as e:
                print(f"Failed to load {cog}: {e}", file=sys.stderr)

        sync_guild_id = os.environ.get("SLASH_SYNC_GUILD_ID")
        if sync_guild_id:
            guild = discord.Object(id=int(sync_guild_id))
            self.tree.copy_global_to(guild=guild)
            try:
                synced = await self.tree.sync(guild=guild)
                print(f"Synced {len(synced)} slash command(s) to guild {sync_guild_id}")
            except discord.HTTPException as e:
                print(f"Slash sync failed: {e}", file=sys.stderr)

    async def close(self):
        await self.database.close()
        await super().close()

    async def on_ready(self):
        print(f"QuadstroNot is online as {self.user} (ID: {self.user.id})")
        print(f"Connected to {len(self.guilds)} guild(s)")
        for guild in self.guilds:
            print(f"  - {guild.name} (ID: {guild.id})")

    async def on_message(self, message: discord.Message):
        if message.author.bot:
            return
        if message.content.startswith("!"):
            print(f"Command received: '{message.content}' from {message.author} in #{message.channel}")
        await self.process_commands(message)

    async def on_command_error(self, ctx: commands.Context, error: commands.CommandError):
        if isinstance(error, commands.CommandNotFound):
            return  # silently ignore unknown commands
        if isinstance(error, commands.MissingPermissions):
            await ctx.send("You don't have permission to use that command.")
        elif isinstance(error, commands.MissingRequiredArgument):
            await ctx.send(f"Missing argument: `{error.param.name}`. Use `!help {ctx.command}` for usage.")
        elif isinstance(error, commands.BadArgument):
            await ctx.send(f"Invalid argument. Use `!help {ctx.command}` for usage.")
        else:
            await ctx.send("Something went wrong running that command.")
            print(f"Unhandled error in {ctx.command}: {error}", file=sys.stderr)


def main():
    check_env()
    bot = QuadstroNot()
    bot.run(os.environ["DISCORD_TOKEN"])


if __name__ == "__main__":
    main()