import discord
from discord import app_commands
from discord.ext import commands
from typing import Optional
from app.internal.c2_client import c2
from app.internal.logger import logger


class RadioCommands(commands.Cog):
    def __init__(self, bot: commands.Bot):
        self.bot = bot

    # ------------------------------------------------------------------
    # Autocomplete — system names from C2
    # ------------------------------------------------------------------

    async def system_autocomplete(
        self, interaction: discord.Interaction, current: str
    ) -> list[app_commands.Choice[str]]:
        systems = await c2.get_systems()
        return [
            app_commands.Choice(name=s["name"], value=s["system_id"])
            for s in systems
            if current.lower() in s["name"].lower()
        ][:25]

    # ------------------------------------------------------------------
    # /join
    # ------------------------------------------------------------------

    @app_commands.command(name="join", description="Stream a radio system to your voice channel.")
    @app_commands.describe(system="The radio system to listen to.")
    @app_commands.autocomplete(system=system_autocomplete)
    async def join(self, interaction: discord.Interaction, system: str):
        await interaction.response.defer(ephemeral=True)

        if not interaction.user.voice or not interaction.user.voice.channel:
            await interaction.followup.send("You need to be in a voice channel first.")
            return

        channel = interaction.user.voice.channel
        guild_id = str(interaction.guild_id)
        channel_id = str(channel.id)

        node = await c2.find_node_for_system(system)
        if not node:
            await interaction.followup.send(
                "No online node is assigned to that system. Check `/status` for availability."
            )
            return

        ok = await c2.send_command(node["node_id"], {
            "action": "discord_join",
            "guild_id": guild_id,
            "channel_id": channel_id,
        })

        if ok:
            systems = await c2.get_systems()
            system_name = next((s["name"] for s in systems if s["system_id"] == system), system)
            await interaction.followup.send(
                f"Streaming **{system_name}** from node `{node['node_id']}` to {channel.mention}."
            )
        else:
            await interaction.followup.send("Failed to contact the node. It may be offline.")

    # ------------------------------------------------------------------
    # /leave
    # ------------------------------------------------------------------

    @app_commands.command(name="leave", description="Stop streaming radio in this server.")
    async def leave(self, interaction: discord.Interaction):
        await interaction.response.defer(ephemeral=True)

        # Find any node currently streaming to this guild
        nodes = await c2.get_nodes()
        streaming_nodes = [
            n for n in nodes if n.get("status") in ("online", "recording")
        ]

        if not streaming_nodes:
            await interaction.followup.send("No nodes appear to be active right now.")
            return

        # Send leave to all online nodes in case more than one joined
        for node in streaming_nodes:
            await c2.send_command(node["node_id"], {"action": "discord_leave"})

        await interaction.followup.send("Disconnected.")

    # ------------------------------------------------------------------
    # /status
    # ------------------------------------------------------------------

    @app_commands.command(name="status", description="Show all node and system status.")
    async def status(self, interaction: discord.Interaction):
        await interaction.response.defer()

        nodes = await c2.get_nodes()
        systems = await c2.get_systems()

        system_map = {s["system_id"]: s["name"] for s in systems}

        status_emoji = {
            "online": "🟢",
            "recording": "🔴",
            "offline": "⚫",
            "unconfigured": "🟡",
        }

        embed = discord.Embed(title="DRB Node Status", color=0x2b2d31)

        if not nodes:
            embed.description = "No nodes registered."
        else:
            for node in sorted(nodes, key=lambda n: n.get("name", "")):
                s = node.get("status", "offline")
                emoji = status_emoji.get(s, "⚪")
                system_name = system_map.get(node.get("assigned_system_id", ""), "Unassigned")
                embed.add_field(
                    name=f"{emoji} {node.get('name', node['node_id'])}",
                    value=f"`{s}` — {system_name}",
                    inline=True,
                )

        await interaction.followup.send(embed=embed)


async def setup(bot: commands.Bot):
    await bot.add_cog(RadioCommands(bot))