58faab15c4
* refactor: remove redundant validation in service and clean up code - Remove validatePublishRequest function as gin binding already handles validation - Remove unused errors import - Simplify PublishContent method by relying on gin's built-in validation - Add comprehensive image processing support with URL download capability * 添加 MCP 说明文档，完善教程
112 lines
2.8 KiB
Go
112 lines
2.8 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/xpzouying/xiaohongshu-mcp/browser"
|
||
"github.com/xpzouying/xiaohongshu-mcp/configs"
|
||
"github.com/xpzouying/xiaohongshu-mcp/pkg/downloader"
|
||
"github.com/xpzouying/xiaohongshu-mcp/xiaohongshu"
|
||
)
|
||
|
||
// XiaohongshuService 小红书业务服务
|
||
type XiaohongshuService struct{}
|
||
|
||
// NewXiaohongshuService 创建小红书服务实例
|
||
func NewXiaohongshuService() *XiaohongshuService {
|
||
return &XiaohongshuService{}
|
||
}
|
||
|
||
// PublishRequest 发布请求
|
||
type PublishRequest struct {
|
||
Title string `json:"title" binding:"required"`
|
||
Content string `json:"content" binding:"required"`
|
||
Images []string `json:"images" binding:"required,min=1"`
|
||
}
|
||
|
||
// LoginStatusResponse 登录状态响应
|
||
type LoginStatusResponse struct {
|
||
IsLoggedIn bool `json:"is_logged_in"`
|
||
Username string `json:"username,omitempty"`
|
||
}
|
||
|
||
// PublishResponse 发布响应
|
||
type PublishResponse struct {
|
||
Title string `json:"title"`
|
||
Content string `json:"content"`
|
||
Images int `json:"images"`
|
||
Status string `json:"status"`
|
||
PostID string `json:"post_id,omitempty"`
|
||
}
|
||
|
||
// CheckLoginStatus 检查登录状态
|
||
func (s *XiaohongshuService) CheckLoginStatus(ctx context.Context) (*LoginStatusResponse, error) {
|
||
// 使用全局单例浏览器创建新页面
|
||
page := browser.NewPage()
|
||
defer page.Close()
|
||
loginAction := xiaohongshu.NewLogin(page)
|
||
|
||
isLoggedIn, err := loginAction.CheckLoginStatus(ctx)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
response := &LoginStatusResponse{
|
||
IsLoggedIn: isLoggedIn,
|
||
Username: configs.Username,
|
||
}
|
||
|
||
return response, nil
|
||
}
|
||
|
||
// PublishContent 发布内容
|
||
func (s *XiaohongshuService) PublishContent(ctx context.Context, req *PublishRequest) (*PublishResponse, error) {
|
||
// 处理图片：下载URL图片或使用本地路径
|
||
imagePaths, err := s.processImages(req.Images)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 构建发布内容
|
||
content := xiaohongshu.PublishImageContent{
|
||
Title: req.Title,
|
||
Content: req.Content,
|
||
ImagePaths: imagePaths,
|
||
}
|
||
|
||
// 执行发布
|
||
if err := s.publishContent(ctx, content); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
response := &PublishResponse{
|
||
Title: req.Title,
|
||
Content: req.Content,
|
||
Images: len(imagePaths),
|
||
Status: "发布完成",
|
||
}
|
||
|
||
return response, nil
|
||
}
|
||
|
||
// processImages 处理图片列表，支持URL下载和本地路径
|
||
func (s *XiaohongshuService) processImages(images []string) ([]string, error) {
|
||
processor := downloader.NewImageProcessor()
|
||
return processor.ProcessImages(images)
|
||
}
|
||
|
||
// publishContent 执行内容发布
|
||
func (s *XiaohongshuService) publishContent(ctx context.Context, content xiaohongshu.PublishImageContent) error {
|
||
// 使用全局单例浏览器创建新页面
|
||
page := browser.NewPage()
|
||
defer page.Close()
|
||
|
||
action, err := xiaohongshu.NewPublishImageAction(page)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 执行发布
|
||
return action.Publish(ctx, content)
|
||
}
|