comment_handler.py

#!/usr/bin/env python3
"""
Comment Handler - Creates formatted PR comments and summaries.
"""
from typing import Dict
from github import Repository
from github.PullRequest import PullRequest


class CommentHandler:
    """Handles creation of PR comments and summaries."""

    def __init__(self, repo: Repository):
        """
        Initialize comment handler.

        Args:
            repo: GitHub repository object
        """
        self.repo = repo

    def create_pr_summary(self, pr: PullRequest, status: Dict, all_passed: bool = False, details: Dict = None) -> str:
        """
        Create a summary comment for a PR.

        Args:
            pr: Pull request object
            status: Workflow status dictionary
            all_passed: Whether all checks truly passed (including branch workflows)
            details: Additional details about check status

        Returns:
            Formatted markdown comment
        """
        lines = [
            "## 🤖 faneX-ID Bot - PR Summary\n",
            f"**PR:** #{pr.number} - {pr.title}\n",
            f"**Author:** @{pr.user.login}\n",
            f"**Branch:** `{pr.head.ref}` → `{pr.base.ref}`\n",
        ]

        # Add overall status
        if details:
            if details.get("truly_all_passed"):
                lines.append("\n✅ **All checks passed successfully!**\n\n")
            else:
                lines.append("\n⚠️ **Some checks are still running or have failed.**\n\n")

                if details.get("running_workflows"):
                    lines.append(f"⏳ **{len(details['running_workflows'])} workflow(s) still running:**\n")
                    for wf in details["running_workflows"][:10]:
                        lines.append(f"- {wf['name']}: {wf['status']}\n")
                    if len(details["running_workflows"]) > 10:
                        lines.append(f"- ... and {len(details['running_workflows']) - 10} more\n")
                    lines.append("\n")

                if details.get("failed_workflows"):
                    lines.append(f"❌ **{len(details['failed_workflows'])} workflow(s) failed:**\n")
                    for wf in details["failed_workflows"][:10]:
                        lines.append(f"- {wf['name']}: {wf.get('conclusion', 'failure')}\n")
                    if len(details["failed_workflows"]) > 10:
                        lines.append(f"- ... and {len(details['failed_workflows']) - 10} more\n")
                    lines.append("\n")

                if details.get("pending_checks"):
                    lines.append(f"⏳ **{len(details['pending_checks'])} check(s) still running**\n\n")

                if details.get("failed_checks"):
                    lines.append(f"❌ **{len(details['failed_checks'])} check(s) failed**\n\n")

        # Add CI/CD status
        if status.get("workflows"):
            lines.append("\n### 📊 CI/CD Status\n\n")
            lines.append("| Workflow | Status |\n")
            lines.append("|----------|--------|\n")

            for workflow in status["workflows"]:
                name = workflow["name"]
                state = workflow["status"]
                # Fix: Handle None values properly
                conclusion = workflow.get("conclusion") or "unknown"

                # Emoji based on status
                if conclusion == "success":
                    emoji = "✅"
                elif conclusion == "failure":
                    emoji = "❌"
                elif conclusion == "cancelled":
                    emoji = "⚠️"
                elif state == "in_progress":
                    emoji = "🔄"
                else:
                    emoji = "⏳"

                # Fix: Handle None values properly
                if conclusion and conclusion != "unknown":
                    status_text = f"{emoji} {conclusion.upper()}"
                else:
                    status_text = f"{emoji} {state.upper() if state else 'UNKNOWN'}"
                workflow_url = workflow.get("url", "")
                if workflow_url:
                    lines.append(f"| [`{name}`]({workflow_url}) | {status_text} |\n")
                else:
                    lines.append(f"| `{name}` | {status_text} |\n")

            # Summary
            total = len(status["workflows"])
            success = sum(1 for w in status["workflows"] if w.get("conclusion") == "success")
            failed = sum(1 for w in status["workflows"] if w.get("conclusion") == "failure")
            in_progress = sum(1 for w in status["workflows"] if w.get("status") == "in_progress")

            lines.append(f"\n**Summary:** {success}/{total} passed, {failed} failed, {in_progress} in progress\n")

            # Add helpful commands if there are failures
            if failed > 0 or (details and (details.get("failed_workflows") or details.get("failed_checks"))):
                lines.append("\n### 🔧 Quick Actions\n\n")
                lines.append("If workflows failed, you can retry them:\n\n")
                lines.append("- `/retry` - Retry all failed workflows\n")
                lines.append("- `/retry <workflow-name>` - Retry a specific workflow\n")
                lines.append("- `/status` - Check current status\n")
        else:
            lines.append("\n⏳ CI/CD checks are running...\n")

        # Add helpful links
        lines.append("\n### 📚 Useful Commands\n\n")
        lines.append("- `/help` - Show all available commands\n")
        lines.append("- `/status` - Get current CI/CD status\n")
        lines.append("- `/retry` - Retry failed workflows\n")

        return "".join(lines)

    def create_error_comment(self, error_message: str, context: str = "") -> str:
        """
        Create an error comment.

        Args:
            error_message: Error message
            context: Additional context

        Returns:
            Formatted error comment
        """
        lines = [
            "## ❌ Error\n\n",
            f"**Message:** {error_message}\n",
        ]

        if context:
            lines.append(f"\n**Context:** {context}\n")

        lines.append("\nPlease check the logs and try again.\n")

        return "".join(lines)

    def create_success_comment(self, message: str) -> str:
        """
        Create a success comment.

        Args:
            message: Success message

        Returns:
            Formatted success comment
        """
        return f"## ✅ Success\n\n{message}\n"