feat:获取用户主页功能 (#122)

增加读取用户的个人主页的信息

xiaohongshu/user_profile.go

@@ -0,0 +1,71 @@
+package xiaohongshu
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"github.com/go-rod/rod"
+)
+
+type UserProfileAction struct {
+	page *rod.Page
+}
+
+func NewUserProfileAction(page *rod.Page) *UserProfileAction {
+	pp := page.Timeout(60 * time.Second)
+	return &UserProfileAction{page: pp}
+}
+
+// UserProfile 获取用户基本信息及帖子
+func (u *UserProfileAction) UserProfile(ctx context.Context, userID, xsecToken string) (*UserProfileResponse, error) {
+	page := u.page.Context(ctx)
+
+	searchURL := makeUserProfileURL(userID, xsecToken)
+	page.MustNavigate(searchURL)
+	page.MustWaitStable()
+
+	page.MustWait(`() => window.__INITIAL_STATE__ !== undefined`)
+
+	// 获取 window.__INITIAL_STATE__ 并转换为 JSON 字符串
+	result := page.MustEval(`() => {
+			if (window.__INITIAL_STATE__) {
+				return JSON.stringify(window.__INITIAL_STATE__);
+			}
+			return "";
+		}`).String()
+
+	if result == "" {
+		return nil, fmt.Errorf("__INITIAL_STATE__ not found")
+	}
+	// 定义响应结构并直接反序列化
+	var initialState = struct {
+		User struct {
+			UserPageData UserPageData `json:"userPageData"`
+			Notes        struct {
+				Feeds [][]Feed `json:"_rawValue"` // 帖子为双重数组
+			} `json:"notes"`
+		} `json:"user"`
+	}{}
+	if err := json.Unmarshal([]byte(result), &initialState); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal __INITIAL_STATE__: %w", err)
+	}
+	response := &UserProfileResponse{
+		UserBasicInfo: initialState.User.UserPageData.RawValue.BasicInfo,
+		Interactions:  initialState.User.UserPageData.RawValue.Interactions,
+	}
+	// 添加用户贴子
+	for _, feeds := range initialState.User.Notes.Feeds {
+		if len(feeds) != 0 {
+			response.Feeds = append(response.Feeds, feeds...)
+		}
+	}
+
+	return response, nil
+
+}
+
+func makeUserProfileURL(userID, xsecToken string) string {
+	return fmt.Sprintf("https://www.xiaohongshu.com/user/profile/%s?xsec_token=%s&xsec_source=pc_note", userID, xsecToken)
+}