1#!/bin/bash
2# 启动小红书 MCP 服务
3 
4XHS_MCP="$HOME/.local/bin/xiaohongshu-mcp"
5PID_FILE="$HOME/.xiaohongshu/mcp.pid"
6LOG_FILE="$HOME/.xiaohongshu/mcp.log"
7XVFB_PID_FILE="$HOME/.xiaohongshu/xvfb.pid"
8XVFB_DISPLAY_FILE="$HOME/.xiaohongshu/xvfb.display"
9 
10# Cookies 路径（可通过环境变量覆盖）
11# XHS_COOKIES_SRC: 源 cookies 文件（用于远程服务器场景）
12# 默认检查 ~/cookies.json 和 ~/.xiaohongshu/cookies.json
13COOKIES_DST="/tmp/cookies.json"
14 
15mkdir -p "$HOME/.xiaohongshu"
16 
17# 检测是否有显示器（桌面环境）
18has_display() {
19    [ -n "$DISPLAY" ] && xdpyinfo >/dev/null 2>&1
20}
21 
22# 在无桌面环境下自动启动 Xvfb
23ensure_display() {
24    if has_display; then
25        return 0
26    fi
27 
28    # 已有 Xvfb 在运行
29    if [ -f "$XVFB_PID_FILE" ]; then
30        local pid
31        pid=$(cat "$XVFB_PID_FILE")
32        if kill -0 "$pid" 2>/dev/null; then
33            export DISPLAY=$(cat "$XVFB_DISPLAY_FILE" 2>/dev/null || echo ":99")
34            echo "复用已有 Xvfb (PID: $pid, DISPLAY=$DISPLAY)"
35            return 0
36        fi
37    fi
38 
39    # 检查 Xvfb 是否安装
40    if ! command -v Xvfb >/dev/null 2>&1; then
41        echo "⚠ 未检测到桌面环境，且未安装 Xvfb。"
42        echo "  请安装：sudo apt-get install -y xvfb"
43        echo "  安装后重新运行本脚本即可自动配置。"
44        exit 1
45    fi
46 
47    echo "未检测到桌面环境，自动启动 Xvfb 虚拟显示..."
48 
49    # 自动选择可用的 display 号（99-109）
50    local display_num=""
51    local d
52    for d in $(seq 99 109); do
53        if [ ! -e "/tmp/.X${d}-lock" ]; then
54            display_num=$d
55            break
56        fi
57        # 锁文件存在但进程已死，尝试清理后使用
58        local lock_pid
59        lock_pid=$(cat "/tmp/.X${d}-lock" 2>/dev/null | tr -d ' ')
60        if [ -n "$lock_pid" ] && ! kill -0 "$lock_pid" 2>/dev/null; then
61            rm -f "/tmp/.X${d}-lock" "/tmp/.X11-unix/X${d}" 2>/dev/null
62            if [ ! -e "/tmp/.X${d}-lock" ]; then
63                display_num=$d
64                break
65            fi
66        fi
67    done
68 
69    if [ -z "$display_num" ]; then
70        echo "✗ 无法找到可用的 display 号（:99-:109 均被占用）"
71        exit 1
72    fi
73 
74    # -ac: 关闭访问控制，允许 chromium 连接虚拟显示（仅用于 headless 自动化）
75    Xvfb ":${display_num}" -screen 0 1024x768x24 -ac >/dev/null 2>&1 &
76    echo $! > "$XVFB_PID_FILE"
77    echo ":${display_num}" > "$XVFB_DISPLAY_FILE"
78    export DISPLAY=":${display_num}"
79    sleep 1
80 
81    if kill -0 "$(cat "$XVFB_PID_FILE")" 2>/dev/null; then
82        echo "✓ Xvfb 已启动 (DISPLAY=:${display_num})"
83    else
84        echo "✗ Xvfb 启动失败"
85        exit 1
86    fi
87}
88 
89# 同步 cookies（支持多个可能的来源）
90sync_cookies() {
91    local src=""
92 
93    # 优先使用环境变量指定的路径
94    if [ -n "$XHS_COOKIES_SRC" ] && [ -f "$XHS_COOKIES_SRC" ]; then
95        src="$XHS_COOKIES_SRC"
96    elif [ -f "$HOME/cookies.json" ]; then
97        src="$HOME/cookies.json"
98    elif [ -f "$HOME/.xiaohongshu/cookies.json" ]; then
99        src="$HOME/.xiaohongshu/cookies.json"
100    fi
101 
102    if [ -n "$src" ]; then
103        if [ ! -f "$COOKIES_DST" ] || [ "$src" -nt "$COOKIES_DST" ]; then
104            install -m 600 "$src" "$COOKIES_DST"
105            echo "已同步 cookies: $src -> $COOKIES_DST"
106        fi
107    else
108        # 确保已有的 cookies 文件权限正确
109        [ -f "$COOKIES_DST" ] && chmod 600 "$COOKIES_DST"
110    fi
111}
112 
113sync_cookies
114ensure_display
115 
116# 检查是否已在运行
117if [ -f "$PID_FILE" ]; then
118    PID=$(cat "$PID_FILE")
119    if kill -0 "$PID" 2>/dev/null; then
120        echo "MCP 服务已在运行 (PID: $PID)"
121        echo "如需重启，请先运行 stop-mcp.sh"
122        exit 0
123    fi
124fi
125 
126# 解析参数
127HEADLESS="true"
128PORT="${XHS_MCP_PORT:-18060}"
129for arg in "$@"; do
130    case $arg in
131        --headless=false)
132            HEADLESS="false"
133            ;;
134        --port=*)
135            PORT="${arg#*=}"
136            ;;
137    esac
138done
139 
140# 校验端口号
141if [[ ! "$PORT" =~ ^[0-9]+$ ]]; then
142    echo "错误: 无效端口号: $PORT"
143    exit 1
144fi
145 
146# 启动服务
147echo "启动小红书 MCP 服务..."
148if [ "$HEADLESS" = "false" ]; then
149    nohup "$XHS_MCP" -port ":${PORT}" -headless=false > "$LOG_FILE" 2>&1 &
150else
151    nohup "$XHS_MCP" -port ":${PORT}" > "$LOG_FILE" 2>&1 &
152fi
153 
154echo $! > "$PID_FILE"
155sleep 2
156 
157# 验证启动
158if kill -0 $(cat "$PID_FILE") 2>/dev/null; then
159    echo "✓ MCP 服务已启动 (PID: $(cat $PID_FILE))"
160    echo "  端点: http://localhost:${PORT}/mcp"
161    echo "  日志: $LOG_FILE"
162else
163    echo "✗ 启动失败，查看日志: $LOG_FILE"
164    cat "$LOG_FILE"
165    exit 1
166fi
167