import discord
from discord.ext import commands
from config import ALLOWED_CHANNEL_ID, HIT_CHECK_INTERVAL, REASSIGN_DELAY
from cogs.assignments import Assignments
from cogs.commands import HitCommands

intents = discord.Intents.default()
intents.message_content = True

# Global state
enrolled_attackers = []
enemy_queue = []
active_assignments = {}
round_robin_index = 0

class HitDispatchBot(commands.Bot):
    async def setup_hook(self):
        # Load cogs with injected state
        await self.add_cog(
            Assignments(
                self,
                enemy_queue=enemy_queue,
                active_assignments=active_assignments,
                enrolled_attackers=enrolled_attackers,
                hit_check=HIT_CHECK_INTERVAL,
                reassign_delay=REASSIGN_DELAY
            )
        )
        await self.add_cog(
            HitCommands(
                self,
                enrolled_attackers=enrolled_attackers,
                enemy_queue=enemy_queue
            )
        )

    async def cog_check(self, ctx):
        return ctx.channel.id == ALLOWED_CHANNEL_ID

bot = HitDispatchBot(command_prefix="!", intents=intents)

@bot.event
async def on_ready():
    print(f"Logged in as {bot.user.name}")

bot.run("MTQ0Mjg3NjU3NTUzMDg3NzAxMQ.GNuHPr.UreuYD1B7YYjfsbfRcEbhFyjyqvhQDepRCN4kk")