|
|
@@ -0,0 +1,354 @@
|
|
|
+package service
|
|
|
+
|
|
|
+import (
|
|
|
+ "context"
|
|
|
+ "encoding/json"
|
|
|
+ "errors"
|
|
|
+ "fmt"
|
|
|
+ "io/ioutil"
|
|
|
+ "net/http"
|
|
|
+ "strconv"
|
|
|
+ "strings"
|
|
|
+)
|
|
|
+
|
|
|
+// RedBookNoteResponse 小红书链接接收结构体
|
|
|
+type RedBookNoteResponse struct {
|
|
|
+ Code int `json:"code"`
|
|
|
+ Data struct {
|
|
|
+ ResponseBody struct {
|
|
|
+ Data struct {
|
|
|
+ Items []struct {
|
|
|
+ NoteCard struct {
|
|
|
+ InteractInfo struct {
|
|
|
+ LikedCount string `json:"liked_count"` // 点赞数
|
|
|
+ CommentCount string `json:"comment_count"` // 评论数
|
|
|
+ CollectedCount string `json:"collected_count"` // 收藏数
|
|
|
+ ShareCount string `json:"share_count"` // 分享数
|
|
|
+ } `json:"interact_info"`
|
|
|
+ } `json:"note_card"`
|
|
|
+ } `json:"items"`
|
|
|
+ } `json:"data"`
|
|
|
+ } `json:"response_body"`
|
|
|
+ } `json:"data"`
|
|
|
+}
|
|
|
+
|
|
|
+// WeiboPostResponse 微博链接接收结构体
|
|
|
+type WeiboPostResponse struct {
|
|
|
+ Code int `json:"code"`
|
|
|
+ Msg string `json:"msg"`
|
|
|
+ Data struct {
|
|
|
+ AttitudesCount int `json:"attitudes_count"` // 点赞数
|
|
|
+ CommentsCount int `json:"comments_count"` // 评论数
|
|
|
+ RepostsCount int `json:"reposts_count"` // 转发数
|
|
|
+ ReadsCount int `json:"reads_count"` // 浏览数
|
|
|
+ } `json:"data"`
|
|
|
+}
|
|
|
+
|
|
|
+// parseNoteStats 小红书接收方法
|
|
|
+func parseNoteStats(jsonData []byte) (likes, comments, collects, shares string, err error) {
|
|
|
+ var resp RedBookNoteResponse
|
|
|
+ if err := json.Unmarshal(jsonData, &resp); err != nil {
|
|
|
+ return "", "", "", "", fmt.Errorf("JSON解析失败: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ // 检查数据是否存在
|
|
|
+ if len(resp.Data.ResponseBody.Data.Items) == 0 {
|
|
|
+ return "", "", "", "", errors.New("未找到笔记数据")
|
|
|
+ }
|
|
|
+
|
|
|
+ stats := resp.Data.ResponseBody.Data.Items[0].NoteCard.InteractInfo
|
|
|
+ return stats.LikedCount, stats.CommentCount, stats.CollectedCount, stats.ShareCount, nil
|
|
|
+}
|
|
|
+
|
|
|
+// parseWeiboStats 微博接收方法
|
|
|
+func parseWeiboStats(jsonData []byte) (likes, comments, reposts, reads string, err error) {
|
|
|
+ var resp WeiboPostResponse
|
|
|
+
|
|
|
+ // 解析JSON
|
|
|
+ if err := json.Unmarshal(jsonData, &resp); err != nil {
|
|
|
+ return "", "", "", "", fmt.Errorf("JSON解析失败: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ // 检查状态码（微博通常用 code=200 表示成功）
|
|
|
+ if resp.Code != 200 {
|
|
|
+ return "", "", "", "", fmt.Errorf("API返回错误: %s", resp.Msg)
|
|
|
+ }
|
|
|
+
|
|
|
+ // 返回数值（转为字符串，保持与原函数一致）
|
|
|
+ return strconv.Itoa(resp.Data.AttitudesCount),
|
|
|
+ strconv.Itoa(resp.Data.CommentsCount),
|
|
|
+ strconv.Itoa(resp.Data.RepostsCount),
|
|
|
+ strconv.Itoa(resp.Data.ReadsCount),
|
|
|
+ nil
|
|
|
+}
|
|
|
+
|
|
|
+// GetRedBookLinkDetail 获取小红书链接数据详情
|
|
|
+func GetRedBookLinkDetail(ctx context.Context, shareText string) (int, int, int, int, error) {
|
|
|
+
|
|
|
+ url := "http://120.46.92.62:6888/api/xhs/note_detail"
|
|
|
+ method := "POST"
|
|
|
+
|
|
|
+ // 构造请求体结构
|
|
|
+ requestBody := struct {
|
|
|
+ ShareText string `json:"share_text"`
|
|
|
+ Proxy string `json:"proxy"`
|
|
|
+ }{
|
|
|
+ ShareText: shareText,
|
|
|
+ Proxy: "",
|
|
|
+ }
|
|
|
+
|
|
|
+ jsonData, err := json.Marshal(requestBody)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("JSON编码失败:", err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+
|
|
|
+ payload := strings.NewReader(string(jsonData))
|
|
|
+
|
|
|
+ client := &http.Client{}
|
|
|
+ req, err := http.NewRequest(method, url, payload)
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ // fmt.Println(err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+ req.Header.Add("Cookie", "abRequestId=87d8ec6f-314f-5914-af1d-21296bf34c5f; a1=197a0b8a06djmry91j6taqhe0kcutwojdcu15nypd50000386246; webId=ae01b40eef7780a778b6624a3b295f4f; gid=yjW08S8Y2yffyjW08DY08SIvKfV6FvjyVK907UfAxd8TSy2879ElV2888qYKJ4K82qfKy82D; webBuild=4.68.0; acw_tc=0a4a2d6e17510033969926232e7b39d55058a43fb37e1ddc19745e6642f746; websectiga=7750c37de43b7be9de8ed9ff8ea0e576519e8cd2157322eb972ecb429a7735d4; sec_poison_id=b4fae566-5d78-4269-85c6-28fc28ec20d3; xsecappid=xhs-pc-web; web_session=040069b22467fb3084e74f796b3a4b11424ee0; loadts=1751003476156; unread={%22ub%22:%22685b56850000000020019eef%22%2C%22ue%22:%22685df405000000002201fd00%22%2C%22uc%22:14}")
|
|
|
+ req.Header.Add("Content-Type", "application/json")
|
|
|
+
|
|
|
+ res, err := client.Do(req)
|
|
|
+ if err != nil {
|
|
|
+ // fmt.Println(err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+ defer res.Body.Close()
|
|
|
+
|
|
|
+ body, err := ioutil.ReadAll(res.Body)
|
|
|
+ if err != nil {
|
|
|
+ // fmt.Println(err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+ fmt.Println(string(body))
|
|
|
+
|
|
|
+ // 提取互动数据
|
|
|
+ likes, comments, collects, shares, err := parseNoteStats(body)
|
|
|
+ if err != nil {
|
|
|
+ //fmt.Println("解析互动数据失败:", err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+
|
|
|
+ like, likeErr := strconv.Atoi(likes)
|
|
|
+ if likeErr != nil {
|
|
|
+ }
|
|
|
+ comment, commentErr := strconv.Atoi(comments)
|
|
|
+ if commentErr != nil {
|
|
|
+ }
|
|
|
+ collect, collectErr := strconv.Atoi(collects)
|
|
|
+ if collectErr != nil {
|
|
|
+ }
|
|
|
+ share, shareErr := strconv.Atoi(shares)
|
|
|
+ if shareErr != nil {
|
|
|
+ }
|
|
|
+ return like, comment, collect, share, nil
|
|
|
+}
|
|
|
+
|
|
|
+// GetWeiBoLinkDetail 获取微博链接数据详情
|
|
|
+func GetWeiBoLinkDetail(ctx context.Context, shareText string) (int, int, int, int, error) {
|
|
|
+ url := "http://120.46.92.62:6888/api/weibo/post_detail"
|
|
|
+ method := "POST"
|
|
|
+
|
|
|
+ fmt.Println(shareText)
|
|
|
+
|
|
|
+ requestBody := struct {
|
|
|
+ Id string `json:"id"`
|
|
|
+ ShareText string `json:"share_text"`
|
|
|
+ Proxy string `json:"proxy"`
|
|
|
+ }{
|
|
|
+ Id: "",
|
|
|
+ ShareText: shareText,
|
|
|
+ Proxy: "",
|
|
|
+ }
|
|
|
+
|
|
|
+ jsonData, err := json.Marshal(requestBody)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("JSON编码失败:", err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+
|
|
|
+ payload := strings.NewReader(string(jsonData))
|
|
|
+
|
|
|
+ client := &http.Client{}
|
|
|
+ req, err := http.NewRequest(method, url, payload)
|
|
|
+
|
|
|
+ fmt.Println(req)
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ //fmt.Println(err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+ req.Header.Add("Content-Type", "application/json")
|
|
|
+
|
|
|
+ res, err := client.Do(req)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println(err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+ defer res.Body.Close()
|
|
|
+
|
|
|
+ body, err := ioutil.ReadAll(res.Body)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println(err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+ // fmt.Println(string(body))
|
|
|
+
|
|
|
+ likes, comments, collects, shares, err := parseWeiboStats(body)
|
|
|
+ if err != nil {
|
|
|
+ //fmt.Println("解析互动数据失败:", err)
|
|
|
+ return 0, 0, 0, 0, err
|
|
|
+ }
|
|
|
+
|
|
|
+ like, likeErr := strconv.Atoi(likes)
|
|
|
+ if likeErr != nil {
|
|
|
+ }
|
|
|
+ comment, commentErr := strconv.Atoi(comments)
|
|
|
+ if commentErr != nil {
|
|
|
+ }
|
|
|
+ collect, collectErr := strconv.Atoi(collects)
|
|
|
+ if collectErr != nil {
|
|
|
+ }
|
|
|
+ share, shareErr := strconv.Atoi(shares)
|
|
|
+ if shareErr != nil {
|
|
|
+ }
|
|
|
+ return like, comment, collect, share, nil
|
|
|
+}
|
|
|
+
|
|
|
+// WeiboPostsResponse 微博多条数据接收结构体
|
|
|
+type WeiboPostsResponse struct {
|
|
|
+ Code int `json:"code"`
|
|
|
+ Msg string `json:"msg"`
|
|
|
+ Data struct {
|
|
|
+ Data struct {
|
|
|
+ Cards []struct {
|
|
|
+ Mblog struct {
|
|
|
+ Id string `json:"id"`
|
|
|
+ Text string `json:"text"`
|
|
|
+ CreatedAt string `json:"created_at"`
|
|
|
+ AttitudesCount int `json:"attitudes_count"` // 点赞数
|
|
|
+ CommentsCount int `json:"comments_count"` // 评论数
|
|
|
+ RepostsCount int `json:"reposts_count"` // 转发数
|
|
|
+ PendingApproval int `json:"pending_approval"` // 待审数
|
|
|
+ PicIds []string `json:"pic_ids"` // 图片ID列表
|
|
|
+ PageInfo struct {
|
|
|
+ Type string `json:"type"` // 内容类型(video/article)
|
|
|
+ PagePic struct {
|
|
|
+ Url string `json:"url"` // 封面图URL
|
|
|
+ } `json:"page_pic"`
|
|
|
+ MediaInfo struct {
|
|
|
+ StreamUrl string `json:"stream_url"` // 视频流地址
|
|
|
+ } `json:"media_info"`
|
|
|
+ } `json:"page_info"`
|
|
|
+ } `json:"mblog"`
|
|
|
+ } `json:"cards"`
|
|
|
+ } `json:"data"`
|
|
|
+ } `json:"data"`
|
|
|
+}
|
|
|
+
|
|
|
+// WeiboPostStats 单条微博的互动数据
|
|
|
+type WeiboPostStats struct {
|
|
|
+ Id string `json:"id"`
|
|
|
+ CreatedAt string `json:"created_at"`
|
|
|
+ Likes int `json:"likes"` // 点赞数
|
|
|
+ Comments int `json:"comments"` // 评论数
|
|
|
+ Reposts int `json:"reposts"` // 转发数
|
|
|
+ Pending int `json:"pending"` // 待审数
|
|
|
+ HasImages bool `json:"has_images"` // 是否有图片
|
|
|
+ HasVideo bool `json:"has_video"` // 是否有视频
|
|
|
+ CoverImageUrl string `json:"cover_image_url"` // 封面图URL(视频/文章)
|
|
|
+ VideoUrl string `json:"video_url"` // 视频地址(如果有)
|
|
|
+}
|
|
|
+
|
|
|
+// parseWeiboPosts 解析微博多条数据
|
|
|
+func parseWeiboPosts(body []byte) ([]WeiboPostStats, error) {
|
|
|
+ var resp WeiboPostsResponse
|
|
|
+ if err := json.Unmarshal(body, &resp); err != nil {
|
|
|
+ return nil, fmt.Errorf("JSON解析失败: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ if resp.Code != 200 {
|
|
|
+ return nil, fmt.Errorf("API返回错误: %s", resp.Msg)
|
|
|
+ }
|
|
|
+
|
|
|
+ var posts []WeiboPostStats
|
|
|
+ for i, card := range resp.Data.Data.Cards {
|
|
|
+ if i >= 5 { // 只取前5条
|
|
|
+ break
|
|
|
+ }
|
|
|
+
|
|
|
+ mblog := card.Mblog
|
|
|
+ hasVideo := mblog.PageInfo.Type == "video"
|
|
|
+
|
|
|
+ posts = append(posts, WeiboPostStats{
|
|
|
+ Id: mblog.Id,
|
|
|
+ CreatedAt: mblog.CreatedAt,
|
|
|
+ Likes: mblog.AttitudesCount,
|
|
|
+ Comments: mblog.CommentsCount,
|
|
|
+ Reposts: mblog.RepostsCount,
|
|
|
+ Pending: mblog.PendingApproval,
|
|
|
+ HasImages: len(mblog.PicIds) > 0,
|
|
|
+ HasVideo: hasVideo,
|
|
|
+ CoverImageUrl: mblog.PageInfo.PagePic.Url,
|
|
|
+ VideoUrl: mblog.PageInfo.MediaInfo.StreamUrl,
|
|
|
+ })
|
|
|
+ }
|
|
|
+ return posts, nil
|
|
|
+}
|
|
|
+
|
|
|
+func GetWeiBoHistoryList(ctx context.Context, homePageUrl string) ([]WeiboPostStats, error) {
|
|
|
+ url := "http://120.46.92.62:6888/api/weibo/user_post" // 替换为实际API地址
|
|
|
+ method := "POST"
|
|
|
+
|
|
|
+ requestBody := struct {
|
|
|
+ Uid string `json:"uid"`
|
|
|
+ ShareText string `json:"share_text"`
|
|
|
+ SinceId string `json:"since_id"`
|
|
|
+ Proxy string `json:"proxy"`
|
|
|
+ }{
|
|
|
+ ShareText: homePageUrl,
|
|
|
+ Proxy: "",
|
|
|
+ }
|
|
|
+
|
|
|
+ client := &http.Client{}
|
|
|
+
|
|
|
+ jsonData, err := json.Marshal(requestBody)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println("JSON编码失败:", err)
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+ payload := strings.NewReader(string(jsonData))
|
|
|
+ req, err := http.NewRequest(method, url, payload)
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println(err)
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ req.Header.Add("Content-Type", "application/json")
|
|
|
+
|
|
|
+ res, err := client.Do(req)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println(err)
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ defer res.Body.Close()
|
|
|
+
|
|
|
+ body, err := ioutil.ReadAll(res.Body)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Println(err)
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ hisData, hisErr := parseWeiboPosts(body)
|
|
|
+ if hisErr != nil {
|
|
|
+ fmt.Println(hisErr)
|
|
|
+ }
|
|
|
+ return hisData, nil
|
|
|
+}
|
