python/src/server/services/report_service.py

"""
Report Service (Phase 5.1.17: Code Governance Refactoring)
Handles business data context gathering, Map-Reduce workflows, and periodic summaries.
"""

from datetime import UTC, datetime, timedelta
from typing import Any

from src.server.config.logfire_config import get_logger
from src.server.repositories.base_repository import BaseRepository
from src.server.services.shared_constants import AgentUUIDs

logger = get_logger(__name__)


class ReportService(BaseRepository):
    def __init__(self, supabase_client=None):
        super().__init__(supabase_client)
    async def gather_report_context(self, days: int) -> str:
        """Gathers database metrics and events from the last N days to ground periodic summaries."""
        logger.info(f"📊 Gathering report context for the past {days} day(s)...")
        try:
            supabase = self.supabase_client
            cutoff_date = (datetime.now(UTC) - timedelta(days=days)).isoformat()

            # 1. Leads
            success, leads_res = self.execute_query(
                lambda: supabase.table("leads").select("company_name, job_title, status").gt("created_at", cutoff_date).execute(),
                "Failed to get leads"
            )
            leads = leads_res.get("data", []) if success else []
            leads_summary = f"Total New Leads: {len(leads)}\n"
            if leads:
                lead_status_counts: dict[str, int] = {}
                for lead in leads:
                    lead_status_counts[lead["status"]] = lead_status_counts.get(lead["status"], 0) + 1
                leads_summary += "Status Breakdown: " + ", ".join(f"{k}: {v}" for k, v in lead_status_counts.items()) + "\n"
                leads_summary += "Recent Leads Sample:\n" + "\n".join(
                    f"- {lead['company_name']} ({lead['job_title']}) -> {lead['status']}" for lead in leads[:10]
                )
                if len(leads) > 10:
                    leads_summary += f"\n... and {len(leads) - 10} more leads."
            else:
                leads_summary += "No new leads found."

            # 2. Token Usage
            success, token_res = self.execute_query(
                lambda: supabase.table("token_usage").select("input_tokens, output_tokens, cost_usd").gt("created_at", cutoff_date).execute(),
                "Failed to get token usage"
            )
            token_data = token_res.get("data", []) if success else []
            total_input = sum(row.get("input_tokens", 0) or 0 for row in token_data)
            total_output = sum(row.get("output_tokens", 0) or 0 for row in token_data)
            total_cost = sum(float(row.get("cost_usd", 0.0) or 0.0) for row in token_data)
            token_summary = (
                f"API Request Count: {len(token_data)}\n"
                f"Input Tokens: {total_input}\n"
                f"Output Tokens: {total_output}\n"
                f"Total Cost: ${total_cost:.4f} USD"
            )

            # 3. System Alerts / Errors
            success, logs_res = self.execute_query(
                lambda: supabase.table("archon_logs").select("level, message, source, created_at").gt("created_at", cutoff_date).in_("level", ["ALERT", "ERROR"]).execute(),
                "Failed to get alerts"
            )
            logs = logs_res.get("data", []) if success else []
            logs_summary = f"Total Alerts/Errors: {len(logs)}\n"
            if logs:
                logs_summary += "Recent Alerts/Errors:\n" + "\n".join(
                    f"- [{log['level']}] {log['source']}: {log['message']} ({log['created_at']})" for log in logs[:10]
                )
                if len(logs) > 10:
                    logs_summary += f"\n... and {len(logs) - 10} more alerts."
            else:
                logs_summary += "No critical alerts/errors."

            # 4. Tasks Updated
            success, tasks_res = self.execute_query(
                lambda: supabase.table("archon_tasks").select("title, status, assignee").gt("updated_at", cutoff_date).execute(),
                "Failed to get tasks updated"
            )
            tasks = tasks_res.get("data", []) if success else []
            tasks_summary = f"Total Tasks Active/Updated: {len(tasks)}\n"
            if tasks:
                task_status_counts: dict[str, int] = {}
                for t in tasks:
                    task_status_counts[t["status"]] = task_status_counts.get(t["status"], 0) + 1
                status_desc = ", ".join(f"{k}: {v}" for k, v in task_status_counts.items())
                tasks_summary += f"Status Breakdown: {status_desc}\n"
                tasks_summary += "Tasks List:\n" + "\n".join(
                    f"- {t['title']} ({t['assignee']}) -> status: {t['status']}" for t in tasks[:10]
                )
                if len(tasks) > 10:
                    tasks_summary += f"\n... and {len(tasks) - 10} more tasks."
            else:
                tasks_summary += "No task status updates."

            context_md = f"""### 系統運行上下文數據 (過去 {days} 天)

#### 1. 商業開發線告 (Leads)
{leads_summary}

#### 2. AI 運算用量與成本 (Token Usage)
{token_summary}

#### 3. 系統警示與異常紀錄 (Archon Logs)
{logs_summary}

#### 4. 專案任務狀態異動 (Archon Tasks)
{tasks_summary}
"""
            return context_md
        except Exception as e:
            logger.error(f"Failed to gather report context: {e}", exc_info=True)
            return "無法取得系統運行數據，請以無上下文模式進行總結。"

    async def generate_daily_executive_summary(self) -> None:
        """Triggers Star-topology Group Chat for Daily Executive Summary."""
        logger.info("📊 ReportService: Triggering Daily Executive Summary (Group Chat)...")
        try:
            import asyncio

            from src.server.services.agent_service import agent_service
            from src.server.services.projects.task_service import task_service

            # 1. Gather 1-day context
            context_md = await self.gather_report_context(1)

            # 2. Save to DB as Task assigned to Supervisor
            supabase = self.supabase_client
            success, p_res = self.execute_query(
                lambda: supabase.table("archon_projects").select("id").limit(1).execute(),
                "Failed to get projects"
            )
            if not success or not p_res.get("data"):
                logger.warning("ReportService: No projects found to attach summary task.")
                return

            task_title = f"[Daily Report] Executive Summary ({datetime.now().strftime('%Y-%m-%d')})"
            task_desc = (
                f"昨日系統運行數據如下：\n\n{context_md}\n\n"
                "請啟動星環群聊，協調 Alice, Bob, DevBot 進行討論，最後由 Supervisor (Charlie) 彙整並提供每日執行摘要報告。"
            )

            # Get Charlie's ID for assignment
            success, charlie_res = self.execute_query(
                lambda: supabase.table("profiles").select("id").eq("email", "charlie@archon.com").execute(),
                "Failed to get charlie's id"
            )
            charlie_data = charlie_res.get("data", []) if success else []
            try:
                assignee_id = charlie_data[0]["id"]
            except (IndexError, KeyError, TypeError):
                assignee_id = AgentUUIDs.SUPERVISOR

            p_data = p_res.get("data", [])
            try:
                p_id = p_data[0]["id"]
            except (IndexError, KeyError, TypeError):
                p_id = ""

            success, tr = await task_service.create_task(
                project_id=p_id, title=task_title, description=task_desc, assignee_id=assignee_id
            )
            if success:
                logger.info(
                    f"✅ ReportService: Created Daily Executive Summary task {tr['task']['id']}. Dispatching Star-topology Group Chat..."
                )
                # 3. Asynchronously trigger the Star-topology Group Chat workflow
                asyncio.create_task(agent_service.run_agent_task(task_id=tr["task"]["id"], agent_id=AgentUUIDs.SUPERVISOR))

                # Record initial log
                self.execute_query(
                    lambda: supabase.table("archon_logs").insert(
                        {
                            "source": "report-service",
                            "level": "INFO",
                            "message": f"Daily Executive Summary group chat dispatched. Task created: {tr['task']['id']}",
                            "details": {
                                "type": "daily_executive_summary",
                                "status": "dispatched",
                            },
                        }
                    ).execute(),
                    "Failed to record daily summary log"
                )

        except Exception as e:
            logger.error(f"💥 ReportService: Daily Executive Summary generation failed: {e}", exc_info=True)

    async def generate_weekly_executive_summary(self) -> None:
        """Triggers Map-Reduce for Weekly Executive Summary."""
        logger.info("📊 ReportService: Triggering Weekly Executive Summary (Map-Reduce)...")
        try:
            from src.agents.workflow.engine_beta_graph import BetaState, beta_graph
            from src.agents.workflow.state import SharedState
            from src.server.services.projects.task_service import task_service
            from src.server.services.prompt_service import prompt_service

            # 1. Gather 7-day context
            context_md = await self.gather_report_context(7)

            # 2. Initialize State with context as first message
            state = BetaState(shared=SharedState())

            default_weekly_prompt = "這是過去 7 天的系統運行上下文數據：\n\n{context_md}\n\n請對每個專屬領域（Sales, Marketing, System, **Engineering/DevBot**）進行分析提煉，最後由 Supervisor 彙整並提供高知識品質、具體行動建議的執行摘要。"
            prompt_template = prompt_service.get_prompt("WEEKLY_EXECUTIVE_SUMMARY", default=default_weekly_prompt)
            prompt_content = prompt_template.replace("{context_md}", context_md)

            state.shared.messages = [
                {
                    "role": "user",
                    "content": prompt_content,
                }
            ]

            logger.info("📊 ReportService: Executing beta_graph Map-Reduce for Weekly summary...")
            run_result: Any = await beta_graph.run(deps=None, state=state)
            output = run_result.output if hasattr(run_result, "output") else run_result

            # 3. Save to DB as a completed Task
            # Get internal project ID
            supabase = self.supabase_client
            success, p_res = self.execute_query(
                lambda: supabase.table("projects").select("id").limit(1).execute(),
                "Failed to get internal project",
                require_data=True
            )
            if not success or not p_res.get("data"):
                logger.warning("ReportService: No internal project found to attach weekly summary task.")
                return

            task_title = f"[Weekly Report] Executive Summary ({datetime.now().strftime('%Y-%m-%d')})"
            task_desc = str(output)

            # Get Charlie's ID for assignment
            success, charlie_res = self.execute_query(
                lambda: supabase.table("profiles").select("id").eq("email", "charlie@archon.com").execute(),
                "Failed to get charlie's id"
            )
            charlie_data = charlie_res.get("data", []) if success else []
            try:
                assignee_id = charlie_data[0]["id"]
            except (IndexError, KeyError, TypeError):
                assignee_id = AgentUUIDs.SUPERVISOR

            p_data = p_res.get("data", [])
            try:
                p_id = p_data[0]["id"]
            except (IndexError, KeyError, TypeError):
                p_id = ""

            success, tr = await task_service.create_task(
                project_id=p_id, title=task_title, description=task_desc, assignee_id=assignee_id
            )
            if success:
                await task_service.update_task(tr["task"]["id"], {"status": "done"})
                logger.info(f"✅ ReportService: Created and completed Weekly Executive Summary task {tr['task']['id']}.")

                # Record ROI info to logs
                self.execute_query(
                    lambda: supabase.table("archon_logs").insert(
                        {
                            "source": "report-service",
                            "level": "INFO",
                            "message": f"Weekly Executive Summary completed. Task created: {tr['task']['id']}",
                            "details": {
                                "type": "weekly_executive_summary",
                                "input_tokens": state.shared.input_tokens,
                                "output_tokens": state.shared.output_tokens,
                                "model": state.shared.model_used,
                            },
                        }
                    ).execute(),
                    "Failed to record weekly summary log"
                )
        except Exception as e:
            logger.error(f"💥 ReportService: Weekly Executive Summary generation failed: {e}", exc_info=True)

    async def generate_monthly_executive_summary(self) -> None:
        """Triggers Map-Reduce for Monthly Executive Summary."""
        logger.info("📊 ReportService: Triggering Monthly Executive Summary (Map-Reduce)...")
        try:
            from src.agents.workflow.engine_beta_graph import BetaState, beta_graph
            from src.agents.workflow.state import SharedState
            from src.server.services.projects.task_service import task_service

            # 1. Gather 30-day context
            context_md = await self.gather_report_context(30)

            # 2. Initialize State with context as first message
            state = BetaState(shared=SharedState())
            state.shared.messages = [
                {
                    "role": "user",
                    "content": f"這是過去 30 天的系統運行上下文數據：\n\n{context_md}\n\n請對每個專屬領域（Sales, Marketing, System, **Engineering/DevBot**）進行分析提煉，最後由 Supervisor 彙整並提供高知識品質、具體行動建議的執行摘要。",
                }
            ]

            logger.info("📊 ReportService: Executing beta_graph Map-Reduce for Monthly summary...")
            run_result: Any = await beta_graph.run(deps=None, state=state)
            output = run_result.output if hasattr(run_result, "output") else run_result

            # 3. Save to DB as a completed Task
            supabase = self.supabase_client
            success, p_res = self.execute_query(
                lambda: supabase.table("archon_projects").select("id").limit(1).execute(),
                "Failed to get projects"
            )
            if not success or not p_res.get("data"):
                logger.warning("ReportService: No projects found to attach monthly summary task.")
                return

            task_title = f"[Monthly Report] Executive Summary ({datetime.now().strftime('%Y-%m-%d')})"
            task_desc = str(output)

            # Get Charlie's ID for assignment
            success, charlie_res = self.execute_query(
                lambda: supabase.table("profiles").select("id").eq("email", "charlie@archon.com").execute(),
                "Failed to get charlie's id"
            )
            charlie_data = charlie_res.get("data", []) if success else []
            try:
                assignee_id = charlie_data[0]["id"]
            except (IndexError, KeyError, TypeError):
                assignee_id = AgentUUIDs.SUPERVISOR

            p_data = p_res.get("data", [])
            try:
                p_id = p_data[0]["id"]
            except (IndexError, KeyError, TypeError):
                p_id = ""

            success, tr = await task_service.create_task(
                project_id=p_id, title=task_title, description=task_desc, assignee_id=assignee_id
            )
            if success:
                await task_service.update_task(tr["task"]["id"], {"status": "done"})
                logger.info(f"✅ ReportService: Created and completed Monthly Executive Summary task {tr['task']['id']}.")

                # Record ROI info to logs
                self.execute_query(
                    lambda: supabase.table("archon_logs").insert(
                        {
                            "source": "report-service",
                            "level": "INFO",
                            "message": f"Monthly Executive Summary completed. Task created: {tr['task']['id']}",
                            "details": {
                                "type": "monthly_executive_summary",
                                "input_tokens": state.shared.input_tokens,
                                "output_tokens": state.shared.output_tokens,
                                "model": state.shared.model_used,
                            },
                        }
                    ).execute(),
                    "Failed to record monthly summary log"
                )
        except Exception as e:
            logger.error(f"💥 ReportService: Monthly Executive Summary generation failed: {e}", exc_info=True)


report_service = ReportService()