# -*- coding: utf-8 -*-
"""XiaoHongShu -- check if mcporter + xiaohongshu MCP is available."""

import json
import platform
import shutil
import subprocess
from .base import Channel


def _is_arm64() -> bool:
    """Detect ARM64 architecture (e.g. Apple Silicon)."""
    machine = platform.machine().lower()
    return machine in ("arm64", "aarch64")


def _mcporter_status_ok(stdout: str) -> bool:
    """Return True if mcporter JSON output indicates status == 'ok'.

    Uses proper JSON parsing to handle Windows BOM, CRLF line endings, and
    whitespace variations.  Falls back to normalised string matching so the
    check still works if mcporter ever returns non-JSON text.
    """
    text = stdout.strip()
    # Strip UTF-8 BOM that Windows PowerShell sometimes prepends.
    if text.startswith("\ufeff"):
        text = text[1:]
    try:
        data = json.loads(text)
        if isinstance(data, dict):
            return str(data.get("status", "")).lower() == "ok"
    except (json.JSONDecodeError, ValueError):
        pass
    # Fallback: normalise whitespace and CRLF, then do string search.
    normalised = text.lower().replace("\r\n", "\n").replace("\r", "\n").replace(" ", "")
    return '"status":"ok"' in normalised


def _docker_run_hint() -> str:
    """Return the docker run command, with --platform flag for ARM64."""
    if _is_arm64():
        return (
            "  docker run -d --name xiaohongshu-mcp -p 18060:18060 "
            "--platform linux/amd64 xpzouying/xiaohongshu-mcp\n"
            "  # ARM64 also: build from source: "
            "https://github.com/xpzouying/xiaohongshu-mcp"
        )
    return (
        "  docker run -d --name xiaohongshu-mcp -p 18060:18060 "
        "xpzouying/xiaohongshu-mcp"
    )


class XiaoHongShuChannel(Channel):
    name = "xiaohongshu"
    description = "小红书笔记"
    backends = ["xiaohongshu-mcp"]
    tier = 2

    def can_handle(self, url: str) -> bool:
        from urllib.parse import urlparse
        d = urlparse(url).netloc.lower()
        return "xiaohongshu.com" in d or "xhslink.com" in d

    def check(self, config=None):
        mcporter = shutil.which("mcporter")
        if not mcporter:
            return "off", (
                "需要 mcporter + xiaohongshu-mcp。安装步骤：\n"
                "  1. npm install -g mcporter\n"
                "  2. " + _docker_run_hint().strip() + "\n"
                "  3. mcporter config add xiaohongshu http://localhost:18060/mcp\n"
                "  详见 https://github.com/xpzouying/xiaohongshu-mcp"
            )
        is_windows = platform.system() == "Windows"
        config_timeout = 15 if is_windows else 5
        try:
            r = subprocess.run(
                [mcporter, "config", "get", "xiaohongshu", "--json"],
                capture_output=True,
                encoding="utf-8",
                errors="replace",
                timeout=config_timeout,
            )
            if r.returncode != 0 or "xiaohongshu" not in r.stdout.lower():
                return "off", (
                    "mcporter 已装但小红书 MCP 未配置。运行：\n"
                    + _docker_run_hint() + "\n"
                    "  mcporter config add xiaohongshu http://localhost:18060/mcp"
                )
        except Exception:
            return "off", "mcporter 连接异常"

        # Use longer timeouts on Windows where mcporter may be slower to respond.
        list_timeout = 30 if is_windows else 10
        try:
            r = subprocess.run(
                [mcporter, "list", "xiaohongshu", "--json"],
                capture_output=True,
                encoding="utf-8",
                errors="replace",
                timeout=list_timeout,
            )
            if r.returncode == 0 and _mcporter_status_ok(r.stdout):
                return "ok", "MCP 已连接（阅读、搜索、发帖、评论、点赞）"
            return "warn", "MCP 已配置，但连接异常；请检查 xiaohongshu-mcp 服务状态"
        except subprocess.TimeoutExpired:
            return "warn", "MCP 已配置，但健康检查超时；请检查 xiaohongshu-mcp 服务状态"
        except Exception:
            return "warn", "MCP 已配置，但连接异常；请检查 xiaohongshu-mcp 服务状态"