From c6390bf0148ec28457f740ef484c27f61626f062 Mon Sep 17 00:00:00 2001 From: chekayo <9827969+chekayo@user.noreply.gitee.com> Date: Mon, 6 Oct 2025 03:26:52 +0800 Subject: [PATCH] feat: add like and favorite functionality for feeds - Implemented handleLikeFeed and handleFavoriteFeed methods in mcp_handlers.go to manage liking and favoriting feeds. - Added LikeFavoriteArgs struct in mcp_server.go for handling parameters. - Registered new MCP tools for liking and favoriting feeds in registerTools function. - Introduced LikeFeed and FavoriteFeed methods in XiaohongshuService to interact with the respective actions. - Created LikeFavoriteAction in a new file to encapsulate the logic for liking and favoriting feeds on the Xiaohongshu platform. --- mcp_handlers.go | 34 ++++ mcp_server.go | 40 ++++- service.go | 49 ++++-- xiaohongshu/like_favorite.go | 332 +++++++++++++++++++++++++++++++++++ 4 files changed, 443 insertions(+), 12 deletions(-) create mode 100644 xiaohongshu/like_favorite.go diff --git a/mcp_handlers.go b/mcp_handlers.go index 6c7fd1e..aebbe25 100644 --- a/mcp_handlers.go +++ b/mcp_handlers.go @@ -391,6 +391,40 @@ func (s *AppServer) handleUserProfile(ctx context.Context, args map[string]any) } } +// handleLikeFeed 处理点赞 +func (s *AppServer) handleLikeFeed(ctx context.Context, args map[string]interface{}) *MCPToolResult { + feedID, ok := args["feed_id"].(string) + if !ok || feedID == "" { + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: "点赞失败: 缺少feed_id参数"}}, IsError: true} + } + xsecToken, ok := args["xsec_token"].(string) + if !ok || xsecToken == "" { + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: "点赞失败: 缺少xsec_token参数"}}, IsError: true} + } + res, err := s.xiaohongshuService.LikeFeed(ctx, feedID, xsecToken) + if err != nil { + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: "点赞失败: " + err.Error()}}, IsError: true} + } + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: fmt.Sprintf("点赞成功 - Feed ID: %s", res.FeedID)}}} +} + +// handleFavoriteFeed 处理收藏 +func (s *AppServer) handleFavoriteFeed(ctx context.Context, args map[string]interface{}) *MCPToolResult { + feedID, ok := args["feed_id"].(string) + if !ok || feedID == "" { + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: "收藏失败: 缺少feed_id参数"}}, IsError: true} + } + xsecToken, ok := args["xsec_token"].(string) + if !ok || xsecToken == "" { + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: "收藏失败: 缺少xsec_token参数"}}, IsError: true} + } + res, err := s.xiaohongshuService.FavoriteFeed(ctx, feedID, xsecToken) + if err != nil { + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: "收藏失败: " + err.Error()}}, IsError: true} + } + return &MCPToolResult{Content: []MCPContent{{Type: "text", Text: fmt.Sprintf("收藏成功 - Feed ID: %s", res.FeedID)}}} +} + // handlePostComment 处理发表评论到Feed func (s *AppServer) handlePostComment(ctx context.Context, args map[string]interface{}) *MCPToolResult { logrus.Info("MCP: 发表评论到Feed") diff --git a/mcp_server.go b/mcp_server.go index 9d6895b..2c7e4b8 100644 --- a/mcp_server.go +++ b/mcp_server.go @@ -50,6 +50,12 @@ type PostCommentArgs struct { Content string `json:"content" jsonschema:"评论内容"` } +// LikeFavoriteArgs 点赞/收藏参数 +type LikeFavoriteArgs struct { + FeedID string `json:"feed_id" jsonschema:"小红书笔记ID，从Feed列表获取"` + XsecToken string `json:"xsec_token" jsonschema:"访问令牌，从Feed列表的xsecToken字段获取"` +} + // InitMCPServer 初始化 MCP Server func InitMCPServer(appServer *AppServer) *mcp.Server { // 创建 MCP Server @@ -208,7 +214,39 @@ func registerTools(server *mcp.Server, appServer *AppServer) { }, ) - logrus.Infof("Registered %d MCP tools", 9) + // 工具 10: 点赞笔记 + mcp.AddTool(server, + &mcp.Tool{ + Name: "like_feed", + Description: "为指定笔记点赞（如已点赞将跳过）", + }, + func(ctx context.Context, req *mcp.CallToolRequest, args LikeFavoriteArgs) (*mcp.CallToolResult, any, error) { + argsMap := map[string]interface{}{ + "feed_id": args.FeedID, + "xsec_token": args.XsecToken, + } + result := appServer.handleLikeFeed(ctx, argsMap) + return convertToMCPResult(result), nil, nil + }, + ) + + // 工具 11: 收藏笔记 + mcp.AddTool(server, + &mcp.Tool{ + Name: "favorite_feed", + Description: "收藏指定笔记（如已收藏将跳过）", + }, + func(ctx context.Context, req *mcp.CallToolRequest, args LikeFavoriteArgs) (*mcp.CallToolResult, any, error) { + argsMap := map[string]interface{}{ + "feed_id": args.FeedID, + "xsec_token": args.XsecToken, + } + result := appServer.handleFavoriteFeed(ctx, argsMap) + return convertToMCPResult(result), nil, nil + }, + ) + + logrus.Infof("Registered %d MCP tools", 11) } // convertToMCPResult 将自定义的 MCPToolResult 转换为官方 SDK 的格式 diff --git a/service.go b/service.go index ebc1534..ab909d2 100644 --- a/service.go +++ b/service.go @@ -20,6 +20,13 @@ import ( // XiaohongshuService 小红书业务服务 type XiaohongshuService struct{} +// ActionResult 通用动作响应（点赞/收藏等） +type ActionResult struct { + FeedID string `json:"feed_id"` + Success bool `json:"success"` + Message string `json:"message"` +} + // NewXiaohongshuService 创建小红书服务实例 func NewXiaohongshuService() *XiaohongshuService { return &XiaohongshuService{} @@ -368,29 +375,49 @@ func (s *XiaohongshuService) UserProfile(ctx context.Context, userID, xsecToken // PostCommentToFeed 发表评论到Feed func (s *XiaohongshuService) PostCommentToFeed(ctx context.Context, feedID, xsecToken, content string) (*PostCommentResponse, error) { - // 使用非无头模式以便查看操作过程 b := newBrowser() defer b.Close() page := b.NewPage() defer page.Close() - // 创建 Feed 评论 action action := xiaohongshu.NewCommentFeedAction(page) - // 发表评论 - err := action.PostComment(ctx, feedID, xsecToken, content) - if err != nil { + if err := action.PostComment(ctx, feedID, xsecToken, content); err != nil { return nil, err } - response := &PostCommentResponse{ - FeedID: feedID, - Success: true, - Message: "评论发表成功", - } + return &PostCommentResponse{FeedID: feedID, Success: true, Message: "评论发表成功"}, nil +} - return response, nil +// LikeFeed 点赞笔记 +func (s *XiaohongshuService) LikeFeed(ctx context.Context, feedID, xsecToken string) (*ActionResult, error) { + b := newBrowser() + defer b.Close() + + page := b.NewPage() + defer page.Close() + + action := xiaohongshu.NewLikeFavoriteAction(page) + if err := action.Like(ctx, feedID, xsecToken); err != nil { + return nil, err + } + return &ActionResult{FeedID: feedID, Success: true, Message: "点赞成功或已点赞"}, nil +} + +// FavoriteFeed 收藏笔记 +func (s *XiaohongshuService) FavoriteFeed(ctx context.Context, feedID, xsecToken string) (*ActionResult, error) { + b := newBrowser() + defer b.Close() + + page := b.NewPage() + defer page.Close() + + action := xiaohongshu.NewLikeFavoriteAction(page) + if err := action.Favorite(ctx, feedID, xsecToken); err != nil { + return nil, err + } + return &ActionResult{FeedID: feedID, Success: true, Message: "收藏成功或已收藏"}, nil } func newBrowser() *headless_browser.Browser { diff --git a/xiaohongshu/like_favorite.go b/xiaohongshu/like_favorite.go new file mode 100644 index 0000000..19b85ee --- /dev/null +++ b/xiaohongshu/like_favorite.go @@ -0,0 +1,332 @@ +package xiaohongshu + +import ( + "context" + "encoding/json" + "fmt" + "strings" + "time" + + "github.com/go-rod/rod" + "github.com/go-rod/rod/lib/proto" + "github.com/pkg/errors" + "github.com/sirupsen/logrus" +) + +// LikeFavoriteAction 点赞/收藏 动作 +// 提供在笔记详情页执行点赞和收藏的能力，并在可能的情况下避免重复点击 +// 通过读取 window.__INITIAL_STATE__ 判断当前状态 +// 并尽量采用多种选择器/文案做回退，避免因页面样式变更导致失败 +// 注意：该实现依赖页面 DOM，可能随页面升级而变化 + +type LikeFavoriteAction struct { + page *rod.Page +} + +func NewLikeFavoriteAction(page *rod.Page) *LikeFavoriteAction { + return &LikeFavoriteAction{page: page} +} + +// Like 点赞指定笔记，如果已点赞则直接返回 +func (a *LikeFavoriteAction) Like(ctx context.Context, feedID, xsecToken string) error { + page := a.page.Context(ctx).Timeout(60 * time.Second) + url := makeFeedDetailURL(feedID, xsecToken) + logrus.Infof("Opening feed detail page for like: %s", url) + + page.MustNavigate(url) + page.MustWaitDOMStable() + time.Sleep(1 * time.Second) + + liked, _, err := a.getInteractState(page, feedID) + if err != nil { + logrus.Warnf("failed to read interact state: %v (continue to try clicking)", err) + } else if liked { + logrus.Infof("feed %s already liked, skip clicking", feedID) + return nil + } + + // 依次尝试多种选择器或按文案匹配 + selectors := []string{ + "span.like-lottie", // 页面提供的喜欢图标容器 (根据您提供的HTML) + ".like-lottie", // 页面提供的喜欢图标容器 + "button.like", // 常见按钮类名 + "div.interaction-bar .like", // 交互区域 like + "div.footer .like", // 底部工具栏 + ".side-action .like", // 侧边操作栏 + ".like-wrapper", // 包裹元素 + ".interactions .like", // 通用交互区 + } + // 同时尝试 SVG use 的 like 图标 + selectors = append(selectors, + "svg.like-icon", "use[href='#like']", "use[xlink\\:href='#like']", + ) + textCandidates := []string{"点赞", "赞", "喜欢"} + if err := clickFirstMatch(page, selectors, textCandidates); err != nil { + return errors.Wrap(err, "点击点赞按钮失败") + } + + time.Sleep(3 * time.Second) // 增加等待时间，确保状态更新 + + // 验证点赞是否成功 + newLiked, _, err := a.getInteractState(page, feedID) + if err == nil && newLiked { + logrus.Infof("feed %s 点赞成功", feedID) + return nil + } + + if err != nil { + logrus.Warnf("验证点赞状态失败: %v", err) + } else { + logrus.Warnf("feed %s 点赞可能未成功，状态未变化，尝试再次点击", feedID) + // 如果第一次点击失败，尝试再次点击 + if err := clickFirstMatch(page, selectors, textCandidates); err != nil { + logrus.Warnf("第二次点击点赞按钮也失败: %v", err) + } else { + time.Sleep(2 * time.Second) + newLiked2, _, err2 := a.getInteractState(page, feedID) + if err2 == nil && newLiked2 { + logrus.Infof("feed %s 第二次点击点赞成功", feedID) + return nil + } else if err2 == nil && !newLiked2 { + logrus.Warnf("feed %s 第二次点击后取消了点赞，这是正常行为", feedID) + return nil + } + } + } + + return nil +} + +// Favorite 收藏指定笔记，如果已收藏则直接返回 +func (a *LikeFavoriteAction) Favorite(ctx context.Context, feedID, xsecToken string) error { + page := a.page.Context(ctx).Timeout(60 * time.Second) + url := makeFeedDetailURL(feedID, xsecToken) + logrus.Infof("Opening feed detail page for favorite: %s", url) + + page.MustNavigate(url) + page.MustWaitDOMStable() + time.Sleep(1 * time.Second) + + _, collected, err := a.getInteractState(page, feedID) + if err != nil { + logrus.Warnf("failed to read interact state: %v (continue to try clicking)", err) + } else if collected { + logrus.Infof("feed %s already favorited, skip clicking", feedID) + return nil + } + + selectors := []string{ + "#note-page-collect-board-guide", // 直接通过ID点击收藏按钮容器 + ".collect-wrapper", // 收藏按钮的包裹容器 + ".collect-wrapper svg", // 容器内的SVG + ".collect-wrapper .reds-icon.collect-icon", // 容器内的收藏图标 + ".collect-wrapper use", // 容器内的use元素 + "use[xlink:href='#collect']", // 直接点击SVG内部的use元素 + "use[href='#collect']", // 备用use选择器 + "svg.reds-icon.collect-icon use", // SVG内部的use元素 + "svg.reds-icon.collect-icon", // SVG容器（可能需要点击父容器） + ".reds-icon.collect-icon use", // 类组合的use元素 + ".reds-icon.collect-icon", // 类组合的容器 + "svg.collect-icon use", // 通用SVG收藏图标内部的use + "svg.collect-icon", // 通用SVG收藏图标 + ".collect-icon", // 通用收藏图标类 + "button.collect", // 常见按钮类名（收藏/收藏夹） + "button.favorite", + "div.interaction-bar .collect", + "div.footer .collect", + ".side-action .collect", + ".interactions .collect", + } + textCandidates := []string{"收藏", "收藏夹", "喜欢"} + if err := clickFirstMatch(page, selectors, textCandidates); err != nil { + return errors.Wrap(err, "点击收藏按钮失败") + } + + time.Sleep(3 * time.Second) // 增加等待时间，确保状态更新 + + // 验证收藏是否成功 + _, newCollected, err := a.getInteractState(page, feedID) + if err == nil && newCollected { + logrus.Infof("feed %s 收藏成功", feedID) + return nil + } + + if err != nil { + logrus.Warnf("验证收藏状态失败: %v", err) + } else { + logrus.Warnf("feed %s 收藏可能未成功，状态未变化，尝试再次点击", feedID) + // 如果第一次点击失败，尝试再次点击 + if err := clickFirstMatch(page, selectors, textCandidates); err != nil { + logrus.Warnf("第二次点击收藏按钮也失败: %v", err) + } else { + time.Sleep(2 * time.Second) + _, newCollected2, err2 := a.getInteractState(page, feedID) + if err2 == nil && newCollected2 { + logrus.Infof("feed %s 第二次点击收藏成功", feedID) + return nil + } + } + } + + return nil +} + +// getInteractState 从 __INITIAL_STATE__ 读取笔记的点赞/收藏状态 +func (a *LikeFavoriteAction) getInteractState(page *rod.Page, feedID string) (liked bool, collected bool, err error) { + result := page.MustEval(`() => { + if (window.__INITIAL_STATE__) { + return JSON.stringify(window.__INITIAL_STATE__); + } + return ""; + }`).String() + if result == "" { + return false, false, fmt.Errorf("__INITIAL_STATE__ not found") + } + + var state struct { + Note struct { + NoteDetailMap map[string]struct { + Note struct { + InteractInfo struct { + Liked bool `json:"liked"` + Collected bool `json:"collected"` + } `json:"interactInfo"` + } `json:"note"` + } `json:"noteDetailMap"` + } `json:"note"` + } + if err := json.Unmarshal([]byte(result), &state); err != nil { + return false, false, errors.Wrap(err, "unmarshal __INITIAL_STATE__ failed") + } + + detail, ok := state.Note.NoteDetailMap[feedID] + if !ok { + return false, false, fmt.Errorf("feed %s not in noteDetailMap", feedID) + } + return detail.Note.InteractInfo.Liked, detail.Note.InteractInfo.Collected, nil +} + +// clickFirstMatch 依次尝试选择器点击；若失败，尝试按按钮/链接文本模糊匹配 +func clickFirstMatch(page *rod.Page, selectors []string, textCandidates []string) error { + // 1) 尝试按选择器查找多个元素并点击（优先点击最后一个，即笔记的点赞按钮） + for _, sel := range selectors { + if els, err := page.Elements(sel); err == nil && len(els) > 0 { + // 从最后一个元素开始尝试（笔记的点赞按钮通常在评论区之前） + for i := len(els) - 1; i >= 0; i-- { + if tryClickChain(els[i]) { + return nil + } + } + } + // 单个元素回退 + if el, err := page.Element(sel); err == nil && el != nil { + if tryClickChain(el) { + return nil + } + } + } + // 2) 文案匹配：在按钮/链接/容器中查找包含文案的元素 + for _, txt := range textCandidates { + if els, err := page.Elements("button, a, div, span, svg, use"); err == nil && len(els) > 0 { + // 从最后一个元素开始尝试匹配文本 + for i := len(els) - 1; i >= 0; i-- { + text, _ := els[i].Text() + if strings.Contains(strings.ToLower(text), strings.ToLower(txt)) { + if tryClickChain(els[i]) { + return nil + } + } + } + } + // 单个元素回退 + if el, err := page.ElementR("button, a, div, span, svg, use", fmt.Sprintf("(?i)%s", regexpEscape(txt))); err == nil && el != nil { + if tryClickChain(el) { + return nil + } + } + } + return errors.New("no clickable element matched for selectors/text") +} + +// tryClickChain 对元素自身及其若干父级尝试点击（scrollIntoView + js click + rod click） +func tryClickChain(el *rod.Element) bool { + current := el + for i := 0; i < 6 && current != nil; i++ { + if clickElement(current) { + return true + } + parent, _ := current.Parent() + current = parent + } + return false +} + +func clickElement(el *rod.Element) bool { + defer func() { _ = recover() }() + // 滚动到可见区域 + _, _ = el.Eval(`() => { try { this.scrollIntoView({block: "center", inline: "center", behavior: "instant"}); } catch (e) {} return true }`) + + // 检查元素类型，对SVG元素使用特殊处理 - 简化处理，直接尝试所有方法 + // 不检查元素类型，直接尝试多种点击方式 + + // 1. 尝试触发MouseEvent（对SVG元素特别有效） + _, jsErr := el.Eval(`() => { + try { + const event = new MouseEvent('click', { + view: window, + bubbles: true, + cancelable: true + }); + this.dispatchEvent(event); + return true; + } catch (e) { + console.error('MouseEvent click error:', e); + return false; + } + }`) + if jsErr == nil { + return true + } + + // 优先尝试标准 JS click + _, jsErr2 := el.Eval(`() => { + try { + this.click(); + return true; + } catch (e) { + console.error('JS click error:', e); + return false; + } + }`) + if jsErr2 == nil { + return true + } + + // 再尝试 rod 的 Click + if err := el.Click(proto.InputMouseButtonLeft, 1); err != nil { + return false + } + + return true +} + +// regexpEscape 对用户文案做正则转义，避免特殊字符 +func regexpEscape(s string) string { + replacer := strings.NewReplacer( + "\\", "\\\\", + ".", "\\.", + "+", "\\+", + "*", "\\*", + "?", "\\?", + "(", "\\(", + ")", "\\)", + "[", "\\[", + "]", "\\]", + "{", "\\{", + "}", "\\}", + "^", "\\^", + "$", "\\$", + "|", "\\|", + ) + return replacer.Replace(s) +}