From e6bf9921ea08b37dc05800fa4a03fbe6c8557a37 Mon Sep 17 00:00:00 2001
From: chekayo <9827969+chekayo@user.noreply.gitee.com>
Date: Thu, 9 Oct 2025 21:38:52 +0800
Subject: [PATCH] refactor: improve comment posting logic with enhanced error
 handling and stability checks

- Updated the PostComment method to include error handling for navigation and element interactions.
- Replaced sleep calls with more reliable wait mechanisms to ensure page stability.
- Added checks for the presence of input elements and improved logging for better debugging.
---
 xiaohongshu/comment_feed.go | 70 ++++++++++++++++++++++++++++---------
 xiaohongshu/publish.go      |  7 ++++
 2 files changed, 61 insertions(+), 16 deletions(-)

diff --git a/xiaohongshu/comment_feed.go b/xiaohongshu/comment_feed.go
index 5aee894..d2ad332 100644
--- a/xiaohongshu/comment_feed.go
+++ b/xiaohongshu/comment_feed.go
@@ -23,28 +23,66 @@ func NewCommentFeedAction(page *rod.Page) *CommentFeedAction {
 // PostComment 发表评论到 Feed
 func (f *CommentFeedAction) PostComment(ctx context.Context, feedID, xsecToken, content string) error {
 	page := f.page.Context(ctx).Timeout(60 * time.Second)
+
 	// 构建详情页 URL
 	url := makeFeedDetailURL(feedID, xsecToken)
+
 	logrus.Infof("Opening feed detail page: %s", url)
+
 	// 导航到详情页
-	page.MustNavigate(url)
-	page.MustWaitDOMStable()
-	time.Sleep(3 * time.Second) // 增加等待时间确保页面完全加载
-	
-	// 等待评论容器加载
-	waitForCommentsContainer(page)
-	
-	elem := page.MustElement("div.input-box div.content-edit span")
-	elem.MustClick()
-	elem2 := page.MustElement("div.input-box div.content-edit p.content-input")
-	elem2.MustInput(content)
-	time.Sleep(2 * time.Second) // 增加等待时间
-	submitButton := page.MustElement("div.bottom button.submit")
-	submitButton.MustClick()
-	time.Sleep(2 * time.Second) // 增加等待时间确保提交完成
+	if err := page.Navigate(url); err != nil {
+		logrus.Warnf("Failed to navigate to feed detail page: %v", err)
+		return fmt.Errorf("无法打开帖子详情页，该帖子可能在网页端不可访问: %w", err)
+	}
+
+	if err := page.WaitStable(2 * time.Second); err != nil {
+		logrus.Warnf("Failed to wait for page stable: %v", err)
+		return fmt.Errorf("页面加载超时，该帖子可能在网页端不可访问: %w", err)
+	}
+
+	time.Sleep(1 * time.Second)
+
+	// 查找评论输入框
+	elem, err := page.Element("div.input-box div.content-edit span")
+	if err != nil {
+		logrus.Warnf("Failed to find comment input box: %v", err)
+		return fmt.Errorf("未找到评论输入框，该帖子可能不支持评论或网页端不可访问: %w", err)
+	}
+
+	if err := elem.Click(proto.InputMouseButtonLeft, 1); err != nil {
+		logrus.Warnf("Failed to click comment input box: %v", err)
+		return fmt.Errorf("无法点击评论输入框: %w", err)
+	}
+
+	elem2, err := page.Element("div.input-box div.content-edit p.content-input")
+	if err != nil {
+		logrus.Warnf("Failed to find comment input field: %v", err)
+		return fmt.Errorf("未找到评论输入区域: %w", err)
+	}
+
+	if err := elem2.Input(content); err != nil {
+		logrus.Warnf("Failed to input comment content: %v", err)
+		return fmt.Errorf("无法输入评论内容: %w", err)
+	}
+
+	time.Sleep(1 * time.Second)
+
+	submitButton, err := page.Element("div.bottom button.submit")
+	if err != nil {
+		logrus.Warnf("Failed to find submit button: %v", err)
+		return fmt.Errorf("未找到提交按钮: %w", err)
+	}
+
+	if err := submitButton.Click(proto.InputMouseButtonLeft, 1); err != nil {
+		logrus.Warnf("Failed to click submit button: %v", err)
+		return fmt.Errorf("无法点击提交按钮: %w", err)
+	}
+
+	time.Sleep(1 * time.Second)
+
+	logrus.Infof("Comment posted successfully to feed: %s", feedID)
 	return nil
 }
-
 // ReplyToComment 回复指定评论
 func (f *CommentFeedAction) ReplyToComment(ctx context.Context, feedID, xsecToken, commentID, userID, content string) error {
 	page := f.page.Context(ctx).Timeout(60 * time.Second)
diff --git a/xiaohongshu/publish.go b/xiaohongshu/publish.go
index a8d1439..d9844e8 100644
--- a/xiaohongshu/publish.go
+++ b/xiaohongshu/publish.go
@@ -7,6 +7,8 @@ import (
 	"os"
 	"strings"
 	"time"
+	"unicode"
+	"unicode/utf8"
 
 	"github.com/go-rod/rod"
 	"github.com/go-rod/rod/lib/input"
@@ -55,6 +57,11 @@ func (p *PublishAction) Publish(ctx context.Context, content PublishImageContent
 		return errors.New("图片不能为空")
 	}
 
+	trimmedContent := strings.TrimRightFunc(content.Content, unicode.IsSpace)
+	if utf8.RuneCountInString(trimmedContent) > 1000 {
+		return errors.New("正文内容不能超过1000个字符")
+	}
+
 	page := p.page.Context(ctx)
 
 	if err := uploadImages(page, content.ImagePaths); err != nil {