# -*- coding: utf-8 -*- """XiaoHongShu -- check if mcporter + xiaohongshu MCP is available.""" 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 _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" ) try: r = subprocess.run( [mcporter, "config", "get", "xiaohongshu", "--json"], capture_output=True, encoding="utf-8", errors="replace", timeout=5, ) 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 连接异常" try: r = subprocess.run( [mcporter, "list", "xiaohongshu", "--json"], capture_output=True, encoding="utf-8", errors="replace", timeout=10, ) out = r.stdout.lower() if r.returncode == 0 and '"status": "ok"' in out: return "ok", "MCP 已连接(阅读、搜索、发帖、评论、点赞)" return "warn", "MCP 已配置,但连接异常;请检查 xiaohongshu-mcp 服务状态" except subprocess.TimeoutExpired: return "warn", "MCP 已配置,但健康检查超时;请检查 xiaohongshu-mcp 服务状态" except Exception: return "warn", "MCP 已配置,但连接异常;请检查 xiaohongshu-mcp 服务状态"