feat: 支持返回登录二维码与 Docker 部署 (#155)

* feat: 支持返回登录二维码与 Docker 部署

* feat: 完善扫码登录功能

* fix: 修复当存在已经登录的情况,上层还会启动 goroutine的问题，并把 mcp 的返回增加为图片格式

Author: lmxdawn

.dockerignore

@@ -0,0 +1,16 @@
+.git
+.idea
+.vscode
+.claude
+.cursor
+.github
+**/*.log
+bin
+dist
+vendor
+Dockerfile
+docker-compose.yml
+docker
+.DS_Store
+
+cookies.json

.gitignore

@@ -8,6 +8,8 @@
 *.so
 *.dylib
 
+.idea
+
 # Test binary, built with `go test -c`
 *.test
 

Dockerfile

@@ -0,0 +1,49 @@
+# ---- build stage ----
+FROM golang:1.24 AS builder
+
+WORKDIR /src
+COPY go.mod go.sum ./
+RUN go mod download
+
+COPY . .
+RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o /out/app .
+
+# ---- run stage ----
+FROM debian:bookworm-slim
+
+WORKDIR /app
+
+# 1. 装 Chromium + 依赖（无头模式运行 rod）
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+      chromium \
+      ca-certificates \
+      fonts-liberation \
+      fonts-noto-cjk \
+      libasound2 \
+      libatk1.0-0 \
+      libatk-bridge2.0-0 \
+      libcups2 \
+      libdrm2 \
+      libxkbcommon0 \
+      libxcomposite1 \
+      libxdamage1 \
+      libxfixes3 \
+      libxrandr2 \
+      libgbm1 \
+      libpango-1.0-0 \
+      libnss3 \
+      libxshmfence1 \
+      wget \
+      tzdata \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY --from=builder /out/app .
+
+# 2. 设置默认 Chromium 路径（rod 会用）
+ENV ROD_BROWSER_BIN=/usr/bin/chromium
+
+EXPOSE 18060
+
+CMD ["./app"]
+

cookies/cookies.go

@@ -56,6 +56,11 @@ func GetCookiesFilePath() string {
 		return oldPath
 	}
 
+	path := os.Getenv("COOKIES_PATH") // 判断环境变量
+	if path == "" {
+		path = "cookies.json" // fallback，本地调试时用当前目录
+	}
+
 	// 文件不存在，使用新路径（当前目录）
-	return "cookies.json"
+	return path
 }

docker/README.md

@@ -0,0 +1,34 @@
+# Docker 使用说明
+<!-- TOC depthFrom:2 -->
+
+- [1. 自己构建镜像](#1-自己构建镜像)
+- [2. 手动 Docker Compose](#2-手动 Docker Compose)
+
+## 1. 自己构建镜像
+
+可以使用源码自己构建镜像，如下：
+
+在有项目的Dockerfile的目录运行
+
+`docker build -t xpzouying/xiaohongshu-mcp .`
+
+`xpzouying/xiaohongshu-mcp`为镜像名称和版本，可以自己起个名字
+
+## 2. 手动 Docker Compose
+
+```
+# 启动 docker-compose
+docker compose up -d
+
+# 停止 docker-compose
+docker compose stop
+
+# 查看实时日志
+docker logs -f xpzouying/xiaohongshu-mcp
+
+# 进入容器
+docker exec -it xpzouying/xiaohongshu-mcp /bin/bash
+
+# 手动更新容器
+docker compose pull && docker compose up -d
+```

docker/sample/docker-compose.yml

@@ -0,0 +1,13 @@
+services:
+  xiaohongshu-mcp:
+    image: xpzouying/xiaohongshu-mcp
+    container_name: xiaohongshu-mcp
+    restart: unless-stopped
+    tty: true
+    volumes:
+      - ./data:/app/data
+    environment:
+      - ROD_BROWSER_BIN=/usr/bin/chromium  # ← 无头浏览器
+      - COOKIES_PATH=/app/data/cookies.json  # ← 程序读取/写入这个路径
+    ports:
+      - "18060:18060"

handlers_api.go

@@ -48,6 +48,19 @@ func (s *AppServer) checkLoginStatusHandler(c *gin.Context) {
 	respondSuccess(c, status, "检查登录状态成功")
 }
 
+// getLoginQrcodeHandler 处理 [GET /api/login/qrcode] 请求。
+// 用于生成并返回登录二维码（Base64 图片 + 超时时间），供前端展示给用户扫码登录。
+func (s *AppServer) getLoginQrcodeHandler(c *gin.Context) {
+	result, err := s.xiaohongshuService.GetLoginQrcode(c.Request.Context())
+	if err != nil {
+		respondError(c, http.StatusInternalServerError, "STATUS_CHECK_FAILED",
+			"获取登录二维码失败", err.Error())
+		return
+	}
+
+	respondSuccess(c, result, "获取登录二维码成功")
+}
+
 // publishHandler 发布内容
 func (s *AppServer) publishHandler(c *gin.Context) {
 	var req PublishRequest

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"os"
 
 	"github.com/sirupsen/logrus"
 	"github.com/xpzouying/xiaohongshu-mcp/configs"
@@ -16,6 +17,10 @@ func main() {
 	flag.StringVar(&binPath, "bin", "", "浏览器二进制文件路径")
 	flag.Parse()
 
+	if len(binPath) == 0 {
+		binPath = os.Getenv("ROD_BROWSER_BIN")
+	}
+
 	configs.InitHeadless(headless)
 	configs.SetBinPath(binPath)
 

mcp_handlers.go

@@ -4,8 +4,9 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-
 	"github.com/sirupsen/logrus"
+	"strings"
+	"time"
 )
 
 // MCP 工具处理函数
@@ -34,6 +35,46 @@ func (s *AppServer) handleCheckLoginStatus(ctx context.Context) *MCPToolResult {
 	}
 }
 
+// handleGetLoginQrcode 处理获取登录二维码请求。
+// 返回二维码图片的 Base64 编码和超时时间，供前端展示扫码登录。
+func (s *AppServer) handleGetLoginQrcode(ctx context.Context) *MCPToolResult {
+	logrus.Info("MCP: 获取登录扫码图片")
+
+	result, err := s.xiaohongshuService.GetLoginQrcode(ctx)
+	if err != nil {
+		return &MCPToolResult{
+			Content: []MCPContent{{Type: "text", Text: "获取登录扫码图片失败: " + err.Error()}},
+			IsError: true,
+		}
+	}
+
+	if result.IsLoggedIn {
+		return &MCPToolResult{
+			Content: []MCPContent{{Type: "text", Text: "你当前已处于登录状态"}},
+		}
+	}
+
+	now := time.Now()
+	deadline := func() string {
+		d, err := time.ParseDuration(result.Timeout)
+		if err != nil {
+			return now.Format("2006-01-02 15:04:05")
+		}
+		return now.Add(d).Format("2006-01-02 15:04:05")
+	}()
+
+	// 已登录：文本 + 图片
+	contents := []MCPContent{
+		{Type: "text", Text: "请用小红书 App 在 " + deadline + " 前扫码登录 👇"},
+		{
+			Type:     "image",
+			MimeType: "image/png",
+			Data:     strings.TrimPrefix(result.Img, "data:image/png;base64,"),
+		},
+	}
+	return &MCPToolResult{Content: contents}
+}
+
 // handlePublishContent 处理发布内容
 func (s *AppServer) handlePublishContent(ctx context.Context, args map[string]interface{}) *MCPToolResult {
 	logrus.Info("MCP: 发布内容")

routes.go

@@ -29,6 +29,7 @@ func setupRoutes(appServer *AppServer) *gin.Engine {
 	api := router.Group("/api/v1")
 	{
 		api.GET("/login/status", appServer.checkLoginStatusHandler)
+		api.GET("/login/qrcode", appServer.getLoginQrcodeHandler)
 		api.POST("/publish", appServer.publishHandler)
 		api.GET("/feeds/list", appServer.listFeedsHandler)
 		api.GET("/feeds/search", appServer.searchFeedsHandler)