Kuby/commandes/bug_report.py

import disnake
from disnake.ext import commands
from utils.gitlab_client import gitlab_client
from aiohttp import web
import logging
import json
import os
import asyncio

kuby_logger = logging.getLogger("KubyBot")
# Utilisation de chemins absolus pour la persistance après reboot
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
REPORTS_FILE = os.path.join(BASE_DIR, "data", "gitlab_reports.json")

def save_report(issue_iid, user_id):
    os.makedirs(os.path.dirname(REPORTS_FILE), exist_ok=True)
    try:
        data = {}
        if os.path.exists(REPORTS_FILE):
            try:
                with open(REPORTS_FILE, "r", encoding="utf-8") as f:
                    data = json.load(f)
            except json.JSONDecodeError:
                kuby_logger.warning("REPORTS_FILE corrupted, re-initializing to avoid tracking loss.")
        
        data[str(issue_iid)] = user_id
        
        temp_file = f"{REPORTS_FILE}.tmp"
        with open(temp_file, "w", encoding="utf-8") as f:
            json.dump(data, f, indent=4)
        os.replace(temp_file, REPORTS_FILE)
    except Exception as e:
        kuby_logger.error(f"Error saving report to JSON: {e}")

class BugReportModal(disnake.ui.Modal):
    def __init__(self, priority_choice):
        super().__init__(title="Signaler un Bug", components=[self.bug_title, self.description])
        self.priority_choice = priority_choice

    bug_title = disnake.ui.TextInput(
        label="Titre du bug",
        placeholder="Décrivez brièvement le problème...",
        required=True,
        max_length=100
    )
    
    description = disnake.ui.TextInput(
        label="Description détaillée",
        style=disnake.TextInputStyle.paragraph,
        placeholder="Que s'est-il passé ? Comment reproduire le bug ?",
        required=True,
        max_length=1000
    )

    async def callback(self, interaction: disnake.Interaction):
        await interaction.response.send_message("Envoi de votre rapport de bug à GitLab...", ephemeral=True)
        
        description_text = f"**Rapporté par:** {interaction.user} ({interaction.user.id})\n"
        description_text += f"**Priorité:** {self.priority_choice.name}\n\n"
        description_text += f"**Description:**\n{self.description.value}"
        
        # Mapping des priorités vers des labels GitLab stylisés (Scoped Labels)
        labels = ["bug", "manual-report", f"Priority::{self.priority_choice.value}"]
        
        result = await gitlab_client.create_issue(
            title=f"[MANUAL] {self.bug_title.value}",
            description=description_text,
            labels=labels
        )
        
        if result:
            issue_iid = result.get("iid")
            save_report(issue_iid, interaction.user.id)
            await interaction.edit_original_response(content=f"✅ Votre bug a été signalé avec succès ! Gameur a bien été averti et le bug sera réglé dans la prochaine mise à jour. [Voir l'issue]({result.get('web_url')})")
        else:
            await interaction.edit_original_response(content="❌ Une erreur est survenue lors de l'envoi du rapport à GitLab. Veuillez contacter un administrateur.")

class FeatureSuggestionModal(disnake.ui.Modal):
    def __init__(self):
        super().__init__(title="Suggérer une fonctionnalité", components=[self.suggestion_title, self.description])

    suggestion_title = disnake.ui.TextInput(
        label="Titre de la suggestion",
        placeholder="Que voulez-vous ajouter ?",
        required=True,
        max_length=100
    )
    
    description = disnake.ui.TextInput(
        label="Détails de la fonctionnalité",
        style=disnake.TextInputStyle.paragraph,
        placeholder="Décrivez comment cela devrait fonctionner...",
        required=True,
        max_length=1000
    )

    async def callback(self, interaction: disnake.Interaction):
        await interaction.response.send_message("Envoi de votre suggestion à GitLab...", ephemeral=True)
        
        description_text = f"**Suggéré par:** {interaction.user} ({interaction.user.id})\n\n"
        description_text += f"**Description:**\n{self.description.value}"
        
        labels = ["enhancement", "manual-suggestion"]
        
        result = await gitlab_client.create_issue(
            title=f"[SUGGESTION] {self.suggestion_title.value}",
            description=description_text,
            labels=labels
        )
        
        if result:
            issue_iid = result.get("iid")
            save_report(issue_iid, interaction.user.id)
            await interaction.edit_original_response(content=f"✅ Votre suggestion a été envoyée avec succès ! Merci de contribuer à l'amélioration du bot. [Voir l'issue]({result.get('web_url')})")
        else:
            await interaction.edit_original_response(content="❌ Une erreur est survenue lors de l'envoi de la suggestion à GitLab.")

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

    async def cog_load(self):
        self.web_app = web.Application()
        self.web_app.router.add_post('/bot_event', self.handle_bot_event)
        self.runner = web.AppRunner(self.web_app)
        await self.runner.setup()
        
        max_retries = 5
        retry_delay = 2
        
        for attempt in range(1, max_retries + 1):
            try:
                self.site = web.TCPSite(self.runner, '127.0.0.1', 5001)
                await self.site.start()
                kuby_logger.info(f"Internal Bot Webhook Server started on 127.0.0.1:5001 (Attempt {attempt})")
                return
            except OSError as e:
                if e.errno == 98:  # Address already in use
                    if attempt < max_retries:
                        kuby_logger.warning(f"Port 5001 already in use, retrying in {retry_delay}s... ({attempt}/{max_retries})")
                        await asyncio.sleep(retry_delay)
                    else:
                        kuby_logger.error(f"Failed to bind to port 5001 after {max_retries} attempts.")
                        raise e
                else:
                    raise e

    def cog_unload(self):
        if hasattr(self, 'runner'):
            # On utilise create_task car cog_unload est synchrone
            async def cleanup():
                try:
                    if hasattr(self, 'site'):
                        await self.site.stop()
                    await self.runner.cleanup()
                    kuby_logger.info("Internal Bot Webhook Server stopped and cleaned up.")
                except Exception as e:
                    kuby_logger.error(f"Error during webhook server cleanup: {e}")
            
            self.bot.loop.create_task(cleanup())

    async def handle_bot_event(self, request):
        # Sécurité interne désactivée comme demandé (le port 5001 n'est accessible que via localhost)
        provided_token = request.headers.get("X-Gitlab-Token")
        # On ignore la vérification du token pour faciliter le relais local

        try:
            # On attend que le bot soit prêt si on vient de rebooter
            if not self.bot.is_ready():
                kuby_logger.info("Webhook received but bot is not ready yet. Waiting up to 10s...")
                try:
                    await asyncio.wait_for(self.bot.wait_until_ready(), timeout=10.0)
                except asyncio.TimeoutError:
                    kuby_logger.warning("Bot still not ready after 10s. Proceeding anyway (best effort).")

            payload = await request.json()
            
            event_type = payload.get("object_kind")
            if event_type not in ["issue", "note"]:
                return web.json_response({"status": "ignored"}, status=200)

            if event_type == "issue":
                issue_iid = payload.get("object_attributes", {}).get("iid")
            else:
                issue_iid = payload.get("issue", {}).get("iid")

            if not issue_iid:
                return web.json_response({"status": "no issue id"}, status=200)

            if not os.path.exists(REPORTS_FILE):
                return web.json_response({"status": "no reports file"}, status=200)

            try:
                with open(REPORTS_FILE, "r", encoding="utf-8") as f:
                    reports = json.load(f)
            except json.JSONDecodeError:
                kuby_logger.error("REPORTS_FILE corrupted during read!")
                return web.json_response({"status": "corrupted file"}, status=500)

            user_id = reports.get(str(issue_iid))
            if not user_id:
                return web.json_response({"status": "untracked issue"}, status=200)

            user = self.bot.get_user(user_id)
            if not user:
                try:
                    user = await self.bot.fetch_user(user_id)
                except:
                    pass

            if not user:
                return web.json_response({"status": "user not found"}, status=200)

            app_info = await self.bot.application_info()
            dev = app_info.owner
            issue_title = payload.get("object_attributes", {}).get("title", f"#{issue_iid}")
            if event_type == "note":
                issue_title = payload.get("issue", {}).get("title", f"#{issue_iid}")

            if event_type == "issue":
                action = payload.get("object_attributes", {}).get("action")
                changes = payload.get("changes", {})
                
                # Check for label changes
                if "labels" in changes:
                    curr_labels = [l.get("title") for l in changes["labels"].get("current", []) if l.get("title")]
                    display_labels = [l for l in curr_labels if not l.startswith("Priority::") and l not in ["bug", "manual-report", "manual-suggestion"]]
                    
                    if display_labels:
                        current_labels_str = ", ".join(display_labels)
                    else:
                        current_labels_str = "Mis à jour"

                    try:
                        # Création de l'embed pour la mise à jour de statut
                        embed = disnake.Embed(
                            title="🛠️ Mise à jour de votre signalement !",
                            description=f"Le statut de votre rapport **{issue_title}** a été mis à jour par l'équipe.",
                            color=disnake.Color.blue()
                        )
                        embed.add_field(name="Nouveaux Labels / Statuts", value=current_labels_str)
                        
                        try:
                            await user.send(embed=embed)
                            # On logue systématiquement pour que le staff sache si ça a marché (utile en prod)
                            kuby_logger.info(f"Notification de label envoyée à {user} ({user.id}) pour l'issue #{issue_iid}")
                            if dev and dev.id != user.id:
                                await dev.send(f"✅ [SUIVI] L'utilisateur {user} ({user.id}) a bien reçu la notification de statut **{current_labels_str}** pour l'issue #{issue_iid} ({issue_title}).")
                        except disnake.Forbidden:
                            kuby_logger.warning(f"Impossible d'envoyer un MP à {user} ({user.id}) - MPs désactivés.")
                            if dev and dev.id != user.id:
                                await dev.send(f"⚠️ [DM BLOQUÉ] {user} ({user.id}) a ses MPs fermés. Impossible de notifier pour l'issue #{issue_iid}.")
                        except Exception as e:
                            kuby_logger.error(f"Erreur lors de l'envoi du MP de statut à {user}: {e}")
                    except Exception as e:
                        kuby_logger.error(f"Erreur inattendue dans la notification de label : {e}")

                # Check if it was closed
                state = payload.get("object_attributes", {}).get("state")
                is_closing_now = action == "close" or (state == "closed" and "state_id" in changes)
                
                if is_closing_now:
                    try:
                        embed = disnake.Embed(
                            title="🛠️ Bug / Suggestion Terminé(e) !",
                            description=f"Bonne nouvelle ! Votre signalement **{issue_title}** a été marqué comme terminé ou résolu.",
                            color=disnake.Color.green()
                        )
                        embed.add_field(name="Prochaine étape", value="La mise à jour arrive prochainement (ou est déjà là) !")
                        embed.set_footer(text="Merci pour votre aide !")
                        
                        try:
                            await user.send(embed=embed)
                            kuby_logger.info(f"Notification de clôture envoyée à {user} ({user.id}) pour l'issue #{issue_iid}")
                            if dev and dev.id != user.id:
                                await dev.send(f"✅ [SUIVI] L'utilisateur {user} ({user.id}) a bien reçu la notification de CLÔTURE pour l'issue #{issue_iid}.")
                        except disnake.Forbidden:
                            kuby_logger.warning(f"Impossible d'envoyer le MP de clôture à {user} ({user.id}).")
                            if dev and dev.id != user.id:
                                await dev.send(f"⚠️ [DM BLOQUÉ] Impossible d'annoncer la clôture à {user} ({user.id}) pour l'issue #{issue_iid} (DMs désactivés).")
                        except Exception as e:
                            kuby_logger.error(f"Erreur lors de l'envoi du MP de clôture à {user}: {e}")
                    except Exception as e:
                        kuby_logger.error(f"Erreur inattendue dans la notification de clôture : {e}")

            elif event_type == "note":
                note_attr = payload.get("object_attributes", {})
                is_system = note_attr.get("system", False)
                
                if not is_system:
                    note_body = note_attr.get("note", "")
                    author_name = payload.get("user", {}).get("name", "Développeur")
                    
                    try:
                        embed = disnake.Embed(
                            title="💬 Nouveau commentaire sur votre signalement",
                            description=f"Le développeur a répondu à votre rapport **{issue_title}** :\n\n>>> {note_body}",
                            color=disnake.Color.orange()
                        )
                        embed.set_footer(text=f"Par {author_name}")
                        
                        try:
                            await user.send(embed=embed)
                            kuby_logger.info(f"Commentaire GitLab envoyé à {user} ({user.id}) pour l'issue #{issue_iid}")
                            if dev and dev.id != user.id:
                                await dev.send(f"✅ [SUIVI] L'utilisateur {user} ({user.id}) a bien reçu votre commentaire pour l'issue #{issue_iid}.")
                        except disnake.Forbidden:
                            kuby_logger.warning(f"Impossible d'envoyer le commentaire en MP à {user} ({user.id}).")
                            if dev and dev.id != user.id:
                                await dev.send(f"⚠️ [DM BLOQUÉ] Impossible d'envoyer votre commentaire à {user} ({user.id}) pour l'issue #{issue_iid} (DMs désactivés).")
                        except Exception as e:
                            kuby_logger.error(f"Erreur lors de l'envoi du MP de commentaire à {user}: {e}")
                    except Exception as e:
                        kuby_logger.error(f"Erreur inattendue dans la notification de commentaire : {e}")

            return web.json_response({"status": "success"}, status=200)

        except Exception as e:
            kuby_logger.error(f"Error handling bot event: {e}")
            return web.json_response({"status": "error", "message": str(e)}, status=500)

    @commands.slash_command(name="signaler_bug", description="Signaler un bug aux développeurs")
    async def signaler_bug(self, interaction: disnake.ApplicationCommandInteraction, 
                           priority: str = commands.Param(description="Priorité du bug", choices={"Basse": "Low", "Normale": "Normal", "Haute": "High", "Urgente": "Urgent"})):
        # Simulate app_commands.Choice behavior for existing code
        from collections import namedtuple
        Choice = namedtuple('Choice', ['name', 'value'])
        # find the name from the value
        p_name = [k for k, v in {"Basse": "Low", "Normale": "Normal", "Haute": "High", "Urgente": "Urgent"}.items() if v == priority][0]
        choice = Choice(name=p_name, value=priority)
        await interaction.response.send_modal(BugReportModal(choice))

    @commands.slash_command(name="suggerer_fonctionnalite", description="Proposer une nouvelle fonctionnalité pour le bot")
    async def suggerer_fonctionnalite(self, interaction: disnake.ApplicationCommandInteraction):
        await interaction.response.send_modal(FeatureSuggestionModal())

def setup(bot):
    bot.add_cog(BugReport(bot))