feat(feeds): Enhance search functionality with additional filter options

- Added support for sorting, note type, time range, search scope, and location distance in the search feeds functionality.
- Updated SearchFeedsArgs struct to include new parameters for filtering.
- Modified handleSearchFeeds method to process and apply filters during feed search.
- Improved logging to include the number of applied filters.
This commit is contained in:
chekayo
2025-10-28 02:01:42 +08:00
18 changed files with 751 additions and 142 deletions

View File

@@ -3,6 +3,8 @@ package main
import (
"context"
"encoding/base64"
"fmt"
"runtime/debug"
"github.com/modelcontextprotocol/go-sdk/mcp"
"github.com/sirupsen/logrus"
@@ -28,7 +30,17 @@ type PublishVideoArgs struct {
// SearchFeedsArgs 搜索内容的参数
type SearchFeedsArgs struct {
Keyword string `json:"keyword" jsonschema:"搜索关键词"`
Keyword string `json:"keyword" jsonschema:"搜索关键词"`
Filters FilterOption `json:"filters,omitempty" jsonschema:"筛选选项"`
}
// FilterOption 筛选选项结构体
type FilterOption struct {
SortBy string `json:"sort_by,omitempty" jsonschema:"排序依据: 综合|最新|最多点赞|最多评论|最多收藏,默认为'综合'"`
NoteType string `json:"note_type,omitempty" jsonschema:"笔记类型: 不限|视频|图文,默认为'不限'"`
PublishTime string `json:"publish_time,omitempty" jsonschema:"发布时间: 不限|一天内|一周内|半年内,默认为'不限'"`
SearchScope string `json:"search_scope,omitempty" jsonschema:"搜索范围: 不限|已看过|未看过|已关注,默认为'不限'"`
Location string `json:"location,omitempty" jsonschema:"位置距离: 不限|同城|附近,默认为'不限'"`
}
// FeedDetailArgs 获取Feed详情的参数
@@ -68,8 +80,8 @@ type LikeFeedArgs struct {
// FavoriteFeedArgs 收藏参数
type FavoriteFeedArgs struct {
FeedID string `json:"feed_id" jsonschema:"小红书笔记ID从Feed列表获取"`
XsecToken string `json:"xsec_token" jsonschema:"访问令牌从Feed列表的xsecToken字段获取"`
FeedID string `json:"feed_id" jsonschema:"小红书笔记ID从Feed列表获取"`
XsecToken string `json:"xsec_token" jsonschema:"访问令牌从Feed列表的xsecToken字段获取"`
Unfavorite bool `json:"unfavorite,omitempty" jsonschema:"是否取消收藏true为取消收藏false或未设置则为收藏"`
}
@@ -92,6 +104,38 @@ func InitMCPServer(appServer *AppServer) *mcp.Server {
return server
}
func withPanicRecovery[T any](
toolName string,
handler func(context.Context, *mcp.CallToolRequest, T) (*mcp.CallToolResult, any, error),
) func(context.Context, *mcp.CallToolRequest, T) (*mcp.CallToolResult, any, error) {
return func(ctx context.Context, req *mcp.CallToolRequest, args T) (result *mcp.CallToolResult, resp any, err error) {
defer func() {
if r := recover(); r != nil {
logrus.WithFields(logrus.Fields{
"tool": toolName,
"panic": r,
}).Error("Tool handler panicked")
logrus.Errorf("Stack trace:\n%s", debug.Stack())
result = &mcp.CallToolResult{
Content: []mcp.Content{
&mcp.TextContent{
Text: fmt.Sprintf("工具 %s 执行时发生内部错误: %v\n\n请查看服务端日志获取详细信息。", toolName, r),
},
},
IsError: true,
}
resp = nil
err = nil
}
}()
return handler(ctx, req, args)
}
}
// registerTools 注册所有 MCP 工具
func registerTools(server *mcp.Server, appServer *AppServer) {
// 工具 1: 检查登录状态
@@ -100,10 +144,10 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "check_login_status",
Description: "检查小红书登录状态",
},
func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) {
withPanicRecovery("check_login_status", func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) {
result := appServer.handleCheckLoginStatus(ctx)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 2: 获取登录二维码
@@ -112,10 +156,10 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "get_login_qrcode",
Description: "获取登录二维码(返回 Base64 图片和超时时间)",
},
func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) {
withPanicRecovery("get_login_qrcode", func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) {
result := appServer.handleGetLoginQrcode(ctx)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 3: 发布内容
@@ -124,7 +168,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "publish_content",
Description: "发布小红书图文内容",
},
func(ctx context.Context, req *mcp.CallToolRequest, args PublishContentArgs) (*mcp.CallToolResult, any, error) {
withPanicRecovery("publish_content", func(ctx context.Context, req *mcp.CallToolRequest, args PublishContentArgs) (*mcp.CallToolResult, any, error) {
// 转换参数格式到现有的 handler
argsMap := map[string]interface{}{
"title": args.Title,
@@ -134,19 +178,19 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
}
result := appServer.handlePublishContent(ctx, argsMap)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 4: 获取Feed列表
mcp.AddTool(server,
&mcp.Tool{
Name: "list_feeds",
Description: "获取用户发布的内容列表",
Description: "获取首页 Feeds 列表",
},
func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) {
withPanicRecovery("list_feeds", func(ctx context.Context, req *mcp.CallToolRequest, _ any) (*mcp.CallToolResult, any, error) {
result := appServer.handleListFeeds(ctx)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 5: 搜索内容
@@ -155,13 +199,10 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "search_feeds",
Description: "搜索小红书内容(需要已登录)",
},
func(ctx context.Context, req *mcp.CallToolRequest, args SearchFeedsArgs) (*mcp.CallToolResult, any, error) {
argsMap := map[string]interface{}{
"keyword": args.Keyword,
}
result := appServer.handleSearchFeeds(ctx, argsMap)
withPanicRecovery("search_feeds", func(ctx context.Context, req *mcp.CallToolRequest, args SearchFeedsArgs) (*mcp.CallToolResult, any, error) {
result := appServer.handleSearchFeeds(ctx, args)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 6: 获取Feed详情
@@ -170,30 +211,30 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "get_feed_detail",
Description: "获取小红书笔记详情,返回笔记内容、图片、作者信息、互动数据(点赞/收藏/分享数)及评论列表",
},
func(ctx context.Context, req *mcp.CallToolRequest, args FeedDetailArgs) (*mcp.CallToolResult, any, error) {
withPanicRecovery("get_feed_detail", func(ctx context.Context, req *mcp.CallToolRequest, args FeedDetailArgs) (*mcp.CallToolResult, any, error) {
argsMap := map[string]interface{}{
"feed_id": args.FeedID,
"xsec_token": args.XsecToken,
}
result := appServer.handleGetFeedDetail(ctx, argsMap)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 7: 获取用户主页
mcp.AddTool(server,
&mcp.Tool{
Name: "user_profile",
Description: "获取小红书用户主页,返回用户基本信息,关注、粉丝、获赞量及其笔记内容",
Description: "获取指定的小红书用户主页,返回用户基本信息,关注、粉丝、获赞量及其笔记内容",
},
func(ctx context.Context, req *mcp.CallToolRequest, args UserProfileArgs) (*mcp.CallToolResult, any, error) {
withPanicRecovery("user_profile", func(ctx context.Context, req *mcp.CallToolRequest, args UserProfileArgs) (*mcp.CallToolResult, any, error) {
argsMap := map[string]interface{}{
"user_id": args.UserID,
"xsec_token": args.XsecToken,
}
result := appServer.handleUserProfile(ctx, argsMap)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 8: 发表评论
@@ -202,7 +243,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "post_comment_to_feed",
Description: "发表评论到小红书笔记",
},
func(ctx context.Context, req *mcp.CallToolRequest, args PostCommentArgs) (*mcp.CallToolResult, any, error) {
withPanicRecovery("post_comment_to_feed", func(ctx context.Context, req *mcp.CallToolRequest, args PostCommentArgs) (*mcp.CallToolResult, any, error) {
argsMap := map[string]interface{}{
"feed_id": args.FeedID,
"xsec_token": args.XsecToken,
@@ -210,7 +251,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
}
result := appServer.handlePostComment(ctx, argsMap)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 9: 回复评论
@@ -245,7 +286,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "publish_with_video",
Description: "发布小红书视频内容(仅支持本地单个视频文件)",
},
func(ctx context.Context, req *mcp.CallToolRequest, args PublishVideoArgs) (*mcp.CallToolResult, any, error) {
withPanicRecovery("publish_with_video", func(ctx context.Context, req *mcp.CallToolRequest, args PublishVideoArgs) (*mcp.CallToolResult, any, error) {
argsMap := map[string]interface{}{
"title": args.Title,
"content": args.Content,
@@ -254,7 +295,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
}
result := appServer.handlePublishVideo(ctx, argsMap)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 11: 点赞笔记
@@ -263,7 +304,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "like_feed",
Description: "为指定笔记点赞或取消点赞(如已点赞将跳过点赞,如未点赞将跳过取消点赞)",
},
func(ctx context.Context, req *mcp.CallToolRequest, args LikeFeedArgs) (*mcp.CallToolResult, any, error) {
withPanicRecovery("like_feed", func(ctx context.Context, req *mcp.CallToolRequest, args LikeFeedArgs) (*mcp.CallToolResult, any, error) {
argsMap := map[string]interface{}{
"feed_id": args.FeedID,
"xsec_token": args.XsecToken,
@@ -271,7 +312,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
}
result := appServer.handleLikeFeed(ctx, argsMap)
return convertToMCPResult(result), nil, nil
},
}),
)
// 工具 12: 收藏笔记
@@ -280,7 +321,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
Name: "favorite_feed",
Description: "收藏指定笔记或取消收藏(如已收藏将跳过收藏,如未收藏将跳过取消收藏)",
},
func(ctx context.Context, req *mcp.CallToolRequest, args FavoriteFeedArgs) (*mcp.CallToolResult, any, error) {
withPanicRecovery("favorite_feed", func(ctx context.Context, req *mcp.CallToolRequest, args FavoriteFeedArgs) (*mcp.CallToolResult, any, error) {
argsMap := map[string]interface{}{
"feed_id": args.FeedID,
"xsec_token": args.XsecToken,
@@ -288,7 +329,7 @@ func registerTools(server *mcp.Server, appServer *AppServer) {
}
result := appServer.handleFavoriteFeed(ctx, argsMap)
return convertToMCPResult(result), nil, nil
},
}),
)
logrus.Infof("Registered %d MCP tools", 11)