VisionApp/tickets/cog.py

from __future__ import annotations

import logging

import discord
from discord import app_commands
from discord.ext import commands

from tickets import storage
from tickets.data import TicketCategory
from tickets.ui import PanelCreateButton

logger = logging.getLogger("vision")


class TicketSystem(commands.Cog):
    def __init__(self, bot: commands.Bot):
        self.bot = bot
        self.bot.add_view(PanelView())

    @commands.Cog.listener()
    async def on_ready(self):
        logger.info("TicketSystem cog charge")


class PanelView(discord.ui.LayoutView):
    def __init__(self):
        super().__init__(timeout=None)
        container = discord.ui.Container()
        container.add_item(
            discord.ui.TextDisplay(
                "## Systeme de tickets\n"
                "Besoin d'aide ? Creez un ticket et notre equipe vous repondra.\n"
                "Cliquez sur le bouton ci-dessous pour commencer."
            )
        )
        container.add_item(discord.ui.Separator())
        container.add_item(discord.ui.ActionRow(PanelCreateButton()))
        self.add_item(container)


ticket_group = app_commands.Group(name="ticket", description="Systeme de tickets")


@ticket_group.command(name="setup", description="Envoyer le panneau de tickets")
@app_commands.describe(channel="Canal pour le panneau")
@app_commands.checks.has_permissions(manage_guild=True)
async def ticket_setup(interaction: discord.Interaction, channel: discord.TextChannel):
    config = storage.get_config(interaction.guild_id)
    config.enabled = True
    config.panel_channel_id = channel.id
    storage.save_config(config)

    view = PanelView()
    msg = await channel.send(view=view)
    config.panel_message_id = msg.id
    storage.save_config(config)

    container = discord.ui.Container()
    container.add_item(discord.ui.TextDisplay(f"## Panneau envoye\nCanal : {channel.mention}\nStatut : Actif"))
    resp_view = discord.ui.LayoutView()
    resp_view.add_item(container)
    await interaction.response.send_message(view=resp_view, ephemeral=True)


@ticket_group.command(name="config", description="Configuration du systeme de tickets")
@app_commands.checks.has_permissions(manage_guild=True)
async def ticket_config(interaction: discord.Interaction):
    config = storage.get_config(interaction.guild_id)
    cats = "\n".join(f"- {c.emoji} **{c.name}**" for c in config.categories) or "*Aucune*"

    container = discord.ui.Container()
    container.add_item(
        discord.ui.TextDisplay(
            f"## Configuration Tickets\n"
            f"**Statut :** {'🟢 Actif' if config.enabled else '🔴 Inactif'}\n"
            f"**Role staff :** <@&{config.staff_role_id}>\n"
            if config.staff_role_id
            else f"**Canal de logs :** <#{config.log_channel_id}>\n"
            if config.log_channel_id
            else f"### Categories\n{cats}"
        )
    )
    container.add_item(discord.ui.Separator())
    container.add_item(
        discord.ui.ActionRow(
            ConfigToggleButton(config.enabled),
            ConfigAddCatButton(),
            ConfigSetStaffButton(),
            ConfigSetLogButton(),
        )
    )
    view = discord.ui.LayoutView()
    view.add_item(container)
    await interaction.response.send_message(view=view, ephemeral=True)


class ConfigToggleButton(discord.ui.Button):
    def __init__(self, enabled: bool):
        super().__init__(
            label="Desactiver" if enabled else "Activer",
            style=discord.ButtonStyle.danger if enabled else discord.ButtonStyle.success,
            custom_id="ticket_cfg_toggle",
        )

    async def callback(self, interaction: discord.Interaction):
        config = storage.get_config(interaction.guild_id)
        config.enabled = not config.enabled
        storage.save_config(config)
        await interaction.response.edit_message(view=ConfigView())


class ConfigAddCatButton(discord.ui.Button):
    def __init__(self):
        super().__init__(
            label="Ajouter categorie",
            style=discord.ButtonStyle.primary,
            custom_id="ticket_cfg_addcat",
        )

    async def callback(self, interaction: discord.Interaction):
        await interaction.response.send_modal(AddCategoryModal())


class ConfigSetStaffButton(discord.ui.Button):
    def __init__(self):
        super().__init__(
            label="Role staff",
            style=discord.ButtonStyle.secondary,
            custom_id="ticket_cfg_staff",
        )

    async def callback(self, interaction: discord.Interaction):
        await interaction.response.send_message(
            "Selectionnez le role staff :",
            view=StaffRoleSelectView(),
            ephemeral=True,
        )


class ConfigSetLogButton(discord.ui.Button):
    def __init__(self):
        super().__init__(
            label="Canal logs",
            style=discord.ButtonStyle.secondary,
            custom_id="ticket_cfg_log",
        )

    async def callback(self, interaction: discord.Interaction):
        await interaction.response.send_message(
            "Selectionnez le canal de logs :",
            view=LogChannelSelectView(),
            ephemeral=True,
        )


class ConfigView(discord.ui.LayoutView):
    def __init__(self):
        super().__init__(timeout=None)
        container = discord.ui.Container()
        container.add_item(discord.ui.TextDisplay("## Configuration Tickets\nModifications enregistrees."))
        self.add_item(container)


class AddCategoryModal(discord.ui.Modal, title="Ajouter une categorie"):
    name = discord.ui.TextInput(label="Nom", max_length=50, required=True)
    emoji = discord.ui.TextInput(label="Emoji", max_length=10, required=False)
    description = discord.ui.TextInput(
        label="Description",
        max_length=100,
        required=False,
        style=discord.TextStyle.paragraph,
    )

    async def on_submit(self, interaction: discord.Interaction):
        config = storage.get_config(interaction.guild_id)
        cat_id = self.name.value.lower().replace(" ", "-")[:20]
        cat = TicketCategory(
            id=cat_id,
            name=self.name.value,
            emoji=self.emoji.value or "",
            description=self.description.value or "",
        )
        config.categories.append(cat)
        storage.save_config(config)
        await interaction.response.send_message(f"Categorie **{cat.name}** ajoutee.", ephemeral=True)


class StaffRoleSelect(discord.ui.RoleSelect):
    def __init__(self):
        super().__init__(
            placeholder="Role staff...",
            custom_id="ticket_staff_role_select",
            min_values=1,
            max_values=1,
        )

    async def callback(self, interaction: discord.Interaction):
        config = storage.get_config(interaction.guild_id)
        config.staff_role_id = self.values[0].id
        storage.save_config(config)
        await interaction.response.edit_message(
            content=f"Role staff defini : {self.values[0].mention}",
            view=None,
        )


class StaffRoleSelectView(discord.ui.View):
    def __init__(self):
        super().__init__(timeout=300)
        self.add_item(StaffRoleSelect())


class LogChannelSelect(discord.ui.ChannelSelect):
    def __init__(self):
        super().__init__(
            placeholder="Canal de logs...",
            custom_id="ticket_log_channel_select",
            channel_types=[discord.ChannelType.text, discord.ChannelType.public_thread],
            min_values=1,
            max_values=1,
        )

    async def callback(self, interaction: discord.Interaction):
        config = storage.get_config(interaction.guild_id)
        config.log_channel_id = self.values[0].id
        storage.save_config(config)
        await interaction.response.edit_message(
            content=f"Canal de logs defini : {self.values[0].mention}",
            view=None,
        )


class LogChannelSelectView(discord.ui.View):
    def __init__(self):
        super().__init__(timeout=300)
        self.add_item(LogChannelSelect())


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