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

# ==========================================
# 1. Environment Setup
# ==========================================
# Add src to path so we can import our modules
# Assuming we are in scripts/
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 before importing our modules
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

from src.domain.models.data_models import (  # noqa: E402
    ActivityVisualization,
    EmojiStatistics,
    GoldenQuote,
    GroupStatistics,
    QualityDimension,
    QualityReview,
    SummaryTopic,
    TokenUsage,
    UserTitle,
)
from src.infrastructure.reporting.generators import ReportGenerator  # noqa: E402


class MockConfigManager:
    def __init__(
        self,
        template_name: str = "scrapbook",
        profile_mode: str = "mbti",
        profile_image_opacity: float = 0.20,
        profile_mapping_config: str = "",
    ) -> None:
        self.template_name = template_name
        self.profile_mode = profile_mode
        self.profile_image_opacity = profile_image_opacity
        self.profile_mapping_config = profile_mapping_config

    def get_report_template(self) -> str:
        return self.template_name

    def get_max_topics(self) -> int:
        return 8

    def get_max_user_titles(self) -> int:
        return 16

    def get_max_golden_quotes(self) -> int:
        return 8

    def get_html_output_dir(self) -> str:
        return "data/html"

    def get_html_filename_format(self) -> str:
        return "report_{group_id}_{date}.html"

    def get_enable_user_card(self) -> bool:
        return True

    @property
    def playwright_available(self) -> bool:
        return True

    def get_browser_path(self) -> str:
        return ""

    def get_t2i_max_concurrent(self) -> int:
        return 4

    def get_llm_max_concurrent(self) -> int:
        return 2

    def get_profile_display_mode(self) -> str:
        return self.profile_mode

    def get_profile_image_opacity(self) -> float:
        return self.profile_image_opacity

    def get_profile_image_size_mode(self) -> str:
        return "contain"

    def get_profile_mapping_config(self) -> str:
        return self.profile_mapping_config

    def get_html_base_url(self) -> str:
        return ""

    def get_t2i_atri_font_mirror(self) -> str:
        return "https://tc.ciallo.ccwu.cc"

    def get_t2i_google_fonts_mirror(self) -> str:
        return "https://fonts.googleapis.com"

    def get_t2i_gstatic_mirror(self) -> str:
        return "https://fonts.gstatic.com"

    def get_t2i_font_source(self) -> str:
        return "Overseas"

    def get_t2i_rendering_strategies(self) -> list:
        return []


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


async def debug_render(
    template_name: str,
    output_file: str = "debug_output.html",
    profile_mode: str = "mbti",
) -> None:
    # 1. Setup Mock Data
    config_manager = MockConfigManager(
        template_name=template_name,
        profile_mode=profile_mode,
    )

    # 2. Mock Analysis Result using Data Models
    stats = GroupStatistics(
        message_count=1250,
        total_characters=45000,
        participant_count=42,
        most_active_period="20:00 - 22:00",
        golden_quotes=[],  # Will be filled later
        emoji_count=156,
        emoji_statistics=EmojiStatistics(face_count=100, mface_count=56),
        activity_visualization=ActivityVisualization(
            hourly_activity={
                i: (10 + i * 5 if i < 12 else 100 - i * 2) for i in range(24)
            }
        ),
        token_usage=TokenUsage(
            prompt_tokens=1500, completion_tokens=800, total_tokens=2300
        ),
        chat_quality_review=QualityReview(
            title="互联网难民收容所",
            subtitle="只要不工作，我们就是最好的朋友",
            dimensions=[
                QualityDimension(
                    "水群闲聊",
                    44.0,
                    "这里的群友不生产代码，只生产各种表情包和废话，建议送去加个班。",
                    "#607d8b",
                ),
                QualityDimension(
                    "技术探讨",
                    25.5,
                    "偶尔冒出的技术术语像是在荒漠里发现绿洲，虽然很快就被废话淹没了。",
                    "#2196f3",
                ),
                QualityDimension(
                    "深夜发情",
                    15.0,
                    "凌晨三点的群聊内容需要打上 R18 标签，建议各位群友早点休息。",
                    "#f44336",
                ),
                QualityDimension(
                    "就业焦虑",
                    10.5,
                    "谈到工作时群里笼罩着一股淡淡的忧伤，大家都在比谁的工位更像牢房。",
                    "#ff9800",
                ),
            ],
            summary="今天也是充满活力（或者说充满废话）的一天，继续保持这份不求上进的快乐吧。",
        ),
    )

    topics = [
        SummaryTopic(
            topic="关于AstrBot插件开发的讨论",
            contributors=["张三", "李四", "王五"],
            detail="大家深入探讨了专家 [123456789] 提到的如何利用Jinja2模板渲染出精美的分析报告，[987654321] 也分享了调试技巧。",
        ),
        SummaryTopic(
            topic="午餐吃什么的终极哲学问题",
            contributors=["赵六", "孙七"],
            detail="[112233445] 提议去吃黄焖鸡，但群友对螺蛳粉的优劣进行了长达一小时的辩论，最终未能达成共识。",
        ),
        SummaryTopic(
            topic="新出的3A大作测评",
            contributors=["周八", "吴九"],
            detail="[200000000] 开始尝试新的游戏，发现 INTJ 类型的玩家在策略游戏中非常吃香。",
        ),
    ]

    mbti_types = [
        "INTJ",
        "INTP",
        "ENTJ",
        "ENTP",
        "INFJ",
        "INFP",
        "ENFJ",
        "ENFP",
        "ISTJ",
        "ISFJ",
        "ESTJ",
        "ESTP",
        "ISTP",
        "ISFP",
        "ESFJ",
        "ESFP",
    ]
    user_titles = []
    user_analysis = {
        "123456789": {"nickname": "张三"},
        "987654321": {"nickname": "李四"},
        "112233445": {"nickname": "潜水员"},
    }

    for i, mbti in enumerate(mbti_types):
        uid = str(200000000 + i)
        uname = f"用户_{mbti}"
        user_titles.append(
            UserTitle(
                name=uname,
                user_id=uid,
                title=f"{mbti} 王者",
                mbti=mbti,
                reason=f"在日常交流中表现出极强的 {mbti} 特征，获得了大家的认可。",
            )
        )
        user_analysis[uid] = {"nickname": uname}

    golden_quotes = [
        GoldenQuote(
            content="代码写得好，下班走得早。",
            sender="张三",
            reason="深刻揭示了程序员的生存法则",
            user_id="123456789",
        ),
        GoldenQuote(
            content="这个Bug我不修，它就是个Feature。",
            sender="李四",
            reason="经典的开发辩解",
            user_id="987654321",
        ),
        GoldenQuote(
            content="PHP是世界上最好的语言！",
            sender="王五",
            reason="[200000001] 表示强烈赞同，并引发了后续的长篇大论。",
            user_id="112233445",
        ),
    ]

    stats.golden_quotes = golden_quotes
    # token_usage already set in constructor

    analysis_result = {
        "statistics": stats,
        "topics": topics,
        "user_titles": user_titles,
        "user_analysis": user_analysis,
        "chat_quality_review": stats.chat_quality_review,
        "analysis_date": "2026年02月11日",
        "group_id": "123456",
        "group_name": "插件逻辑调试群",
    }

    # 3. Initialize Generator
    data_dir = Path("data/debug_data")
    data_dir.mkdir(parents=True, exist_ok=True)
    generator = ReportGenerator(config_manager, data_dir)

    # Mock avatar cache to avoid KeyError in debug mode
    class MockCache(dict):
        def __getitem__(self, key):
            return self.get(key, "")

        def set(self, key, value, expire=None):
            self[key] = value

    generator._avatar_cache = MockCache()

    # 4. Prepare Render Data
    # Note: _prepare_render_data handles converting Entities to template-friendly dicts
    render_payload = await generator._prepare_render_data(
        analysis_result, avatar_url_getter=mock_get_user_avatar
    )

    # Use Jinja2 renderer
    final_html = generator.html_templates.render_template(
        "image_template.html", **render_payload
    )

    # 复用最终 HTML 中所有内联头像资源，并注入复用样式
    final_html = generator._reuse_avatars_in_final_html(
        final_html,
        render_payload.get("avatar_reuse_registry", {}),
        render_payload.get("avatar_reuse_aliases", {}),
    )

    # 6. Save to file
    output_path = Path(output_file)
    output_path.write_text(final_html, encoding="utf-8")

    # 7. Close generator
    await generator.close()

    print(
        f"Successfully rendered template '{template_name}' in mode '{profile_mode}' to {output_path.absolute()}"
    )
    print("You can now open this file with your browser to debug your HTML/CSS.")


def main() -> None:
    parser = argparse.ArgumentParser(
        description="Debug render tool for astrbot_plugin_qq_group_daily_analysis report templates."
    )
    parser.add_argument(
        "-t",
        "--template",
        type=str,
        default="scrapbook",
        help="Template name to render (default: scrapbook)",
    )
    parser.add_argument(
        "-o",
        "--output",
        type=str,
        default="debug_output.html",
        help="Output HTML file path (default: debug_output.html)",
    )
    parser.add_argument(
        "-m",
        "--mode",
        type=str,
        default="mbti",
        choices=["mbti", "sbti", "acgti"],
        help="Profile display mode to render (default: mbti)",
    )
    args = parser.parse_args()

    asyncio.run(debug_render(args.template, args.output, args.mode))


if __name__ == "__main__":
    main()