import asyncio
import os
import sys
import types
from pathlib import Path

# ==========================================
# 1. Environment Setup
# ==========================================
# Add project root to sys.path so we can import 'astrbot' and plugin modules
# In docker environment:
# /AstrBot
# /AstrBot/data/plugins/astrbot_plugin_qq_group_daily_analysis
current_dir = os.path.dirname(os.path.abspath(__file__))
plugin_root = os.path.abspath(os.path.join(current_dir, ".."))
sys.path.insert(0, plugin_root)

# Mock astrbot.api
astrbot_api = types.ModuleType("astrbot.api")


class MockLogger:
    def info(self, msg, *args, **kwargs):
        print(f"[INFO] {msg}")

    def error(self, msg, *args, **kwargs):
        print(f"[ERROR] {msg}")

    def warning(self, msg, *args, **kwargs):
        print(f"[WARN] {msg}")

    def debug(self, msg, *args, **kwargs):
        print(f"[DEBUG] {msg}")

    def log(self, level, msg, *args, **kwargs):
        print(f"[LOG {level}] {msg}")

    def isEnabledFor(self, level):
        return True


astrbot_api.logger = MockLogger()
astrbot_api.AstrBotConfig = dict
sys.modules["astrbot.api"] = astrbot_api

# Mock astrbot.core.utils.astrbot_path
astrbot_core_utils = types.ModuleType("astrbot.core.utils")
astrbot_path = types.ModuleType("astrbot.core.utils.astrbot_path")
astrbot_path.get_astrbot_data_path = lambda: Path(".")
sys.modules["astrbot.core.utils"] = astrbot_core_utils
sys.modules["astrbot.core.utils.astrbot_path"] = astrbot_path

# Now import plugin modules
from src.domain.entities.analysis_result import (  # noqa: E402
    ActivityVisualization,
    EmojiStatistics,
    GoldenQuote,
    GroupStatistics,
    SummaryTopic,
    TokenUsage,
    UserTitle,
)
from src.infrastructure.reporting.generators import ReportGenerator  # noqa: E402


# ==========================================
# 2. Mocks
# ==========================================
class MockConfigManager:
    def __init__(self):
        pass

    def get_pdf_output_dir(self):
        # Output to the scripts directory for easy access
        return os.path.join(current_dir, "output")

    def get_pdf_filename_format(self):
        return "mock_report_{group_id}_{date}.pdf"

    def get_max_topics(self):
        return 5

    def get_max_user_titles(self):
        return 8

    def get_max_golden_quotes(self):
        return 5

    def get_report_template(self):
        return "scrapbook"

    def get_enable_user_card(self):
        return True

    def get_t2i_max_concurrent(self):
        return 2

    @property
    def playwright_available(self):
        return True

    def get_browser_path(self):
        return ""


async def mock_get_user_avatar(avatar_id: str, avatar_url_getter=None) -> str:
    # Return a known avatar URL for testing
    return f"https://q4.qlogo.cn/headimg_dl?dst_uin={avatar_id}&spec=640"


# ==========================================
# 3. Main Execution
# ==========================================
async def main():
    print("Initializing ReportGenerator...")
    config_manager = MockConfigManager()
    data_dir = Path(current_dir) / "data"
    generator = ReportGenerator(config_manager, data_dir)

    # Override avatar fetching to avoid real network calls during testing
    generator._get_user_avatar = mock_get_user_avatar

    # 1. Mock Analysis Result using Entities
    stats = GroupStatistics(
        message_count=1280,
        total_characters=8500,
        participant_count=42,
        most_active_period="20:00-22:00",
        emoji_count=156,
        emoji_statistics=EmojiStatistics(face_count=100, mface_count=56),
        activity_visualization=ActivityVisualization(
            hourly_activity={i: (i * 5) % 65 for i in range(24)},
            daily_activity={"2026-02-11": 1280},
        ),
    )

    topics = [
        SummaryTopic(
            topic="AstrBot新功能",
            detail="大家对 [10001] 提到的PDF生成功能的讨论非常热烈，[10002] 提出了很多优化建议。",
            contributors=["开发者", "测试员"],
        ),
        SummaryTopic(
            topic="周末计划",
            detail="[10003] 提议去爬山，也有人想在家打游戏。",
            contributors=["旅行家", "宅男"],
        ),
        SummaryTopic(
            topic="代码调试",
            detail="关于Python异步编程的深入探讨，[10001]分享了一些心得。",
            contributors=["小白", "大神"],
        ),
        SummaryTopic(
            topic="美食分享",
            detail="深夜放毒，[10004] 发了很多火锅和烧烤的照片。",
            contributors=["吃货A", "吃货B"],
        ),
        SummaryTopic(
            topic="模组推荐",
            detail="[10005] 推荐了一些好用的Minecraft模组。",
            contributors=["MC玩家"],
        ),
    ]

    user_titles = [
        UserTitle(
            name="极客",
            user_id="10001",
            title="代码魔术师",
            mbti="INTJ",
            reason="总是能用一行代码解决复杂问题。",
        ),
        UserTitle(
            name="社牛",
            user_id="10002",
            title="气氛组组长",
            mbti="ENFP",
            reason="群里冷场时总能第一时间活跃气氛。",
        ),
        UserTitle(
            name="百科",
            user_id="10003",
            title="移动维基",
            mbti="ISTJ",
            reason="不管问什么问题，他都知道答案。",
        ),
        UserTitle(
            name="潜水",
            user_id="10004",
            title="深海幽灵",
            mbti="INTP",
            reason="虽然很少说话，但每次发言都直击要害。",
        ),
        UserTitle(
            name="欧皇",
            user_id="10005",
            title="天选之子",
            mbti="ESFJ",
            reason="抽卡次次出金，让人羡慕嫉妒恨。",
        ),
    ]

    stats.golden_quotes = [
        GoldenQuote(
            sender="大佬",
            content="这代码能跑就行，别动它！",
            reason="至理名言，动了就崩。",
            user_id="20001",
        ),
        GoldenQuote(
            sender="萌新",
            content="为什么我的报错和你不一？",
            reason="经典的灵魂发问。",
            user_id="20002",
        ),
        GoldenQuote(
            sender="群主",
            content="再发涩图全部禁言！",
            reason="来自管理层的威慑。",
            user_id="888888",
        ),
    ]
    stats.token_usage = TokenUsage(
        prompt_tokens=2000, completion_tokens=1000, total_tokens=3000
    )

    analysis_result = {
        "statistics": stats,
        "topics": topics,
        "user_titles": user_titles,
        "user_analysis": {
            "10001": {"nickname": "极客"},
            "10002": {"nickname": "社牛"},
            "10003": {"nickname": "百科"},
            "10004": {"nickname": "潜水"},
            "10005": {"nickname": "欧皇"},
        },
        "token_usage": TokenUsage(
            prompt_tokens=2000, completion_tokens=1000, total_tokens=3000
        ),
    }

    class MockDimension:
        def __init__(self, name, percentage, comment, color):
            self.name = name
            self.percentage = percentage
            self.comment = comment
            self.color = color

    class MockChatQualityReview:
        def __init__(self):
            self.title = "今日群聊质量分析"
            self.subtitle = "基于AI的深度神经网络评估"
            self.dimensions = [
                MockDimension("讨论活跃度", 85, "大家讨论非常热烈", "#FF5722"),
                MockDimension("知识共享度", 70, "分享了很多有用的技术信息", "#4CAF50"),
                MockDimension("情感温度", 92, "群内氛围非常融洽", "#E91E63")
            ]
            self.summary = "整体聊天的质量很高，技术讨论和日常闲聊达到了完美的平衡。"

    analysis_result["chat_quality_review"] = MockChatQualityReview()

    print("Generating PDF Report...")
    group_id = "test_group_mock"

    # Direct generation
    pdf_path = await generator.generate_pdf_report(analysis_result, group_id=group_id)

    if pdf_path:
        print(f"\n[SUCCESS] PDF Generated Successfully: {pdf_path}")
        if os.path.exists(pdf_path):
            print(f"File Size: {os.path.getsize(pdf_path) / 1024:.2f} KB")
        else:
            print(
                "[WARN] Generated path returned but file not found on local disk (might be in container)."
            )
    else:
        print("\n[FAILURE] PDF Generation Failed.")


if __name__ == "__main__":
    asyncio.run(main())