Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,116 @@
+package mediabookshelfdata
+
+import (
+	"91porn-server/models/v/mediabookshelfmod"
+	"91porn-server/models/v/mediamod"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"time"
+)
+
+// AppData 需要组装返回的数据
+type AppData struct {
+	BaseInfo *mediabookshelfmod.MediaBookshelfInfo `json:"baseInfo"` // 基础信息
+}
+
+// FormatAppDataList 格式化app列表数据
+func FormatAppDataList(origin []mediabookshelfmod.MediaBookshelf) (res []AppData) {
+	res = make([]AppData, len(origin))
+	if len(origin) == 0 {
+		return
+	}
+
+	// 组装返回数据
+	for i, item := range origin {
+		res[i] = FormatAppData(item)
+	}
+
+	return
+}
+
+// FormatAppData 格式化app数据
+func FormatAppData(item mediabookshelfmod.MediaBookshelf) (res AppData) {
+	if item.ID.IsZero() {
+		return
+	}
+
+	// 组装返回数据
+	res.BaseInfo = &mediabookshelfmod.MediaBookshelfInfo{
+		ID:         item.ID,
+		Name:       item.Name,
+		UID:        item.UID,
+		ReadAt:     item.ReadAt,
+		CreatedAt:  item.CreatedAt,
+		UpdateTime: item.UpdateTime,
+	}
+
+	return
+}
+
+// AppSimpleData 仅返回基础数据
+type AppSimpleData *mediabookshelfmod.MediaBookshelfInfo
+
+// FormatAppSimpleDataList 格式化app列表数据
+func FormatAppSimpleDataList(origin []mediabookshelfmod.MediaBookshelf, skip, limit int64, sort bson.D) (res []AppSimpleData) {
+	res = make([]AppSimpleData, len(origin))
+	if len(origin) == 0 {
+		return
+	}
+
+	var mids []primitive.ObjectID
+	for i := range origin {
+		mids = append(mids, origin[i].MID)
+	}
+
+	filter := bson.M{"_id": bson.M{"$in": mids}}
+	opts := options.Find().SetSkip(skip).SetLimit(limit)
+	if len(sort) > 0 {
+		opts = opts.SetSort(sort)
+	}
+	media, err := mediamod.QueryMediaByCond(filter, opts)
+	if err != nil {
+		return
+	}
+	m := make(map[primitive.ObjectID]*mediamod.Media)
+	for i := range media {
+		m[media[i].ID] = media[i]
+	}
+	// 组装返回数据
+	for i, item := range origin {
+		res[i] = FormatAppSimpleData(item, m[item.MID])
+	}
+
+	return
+}
+
+// FormatAppSimpleData 格式化app数据
+func FormatAppSimpleData(item mediabookshelfmod.MediaBookshelf, media *mediamod.Media) (res AppSimpleData) {
+	if item.ID.IsZero() {
+		return
+	}
+
+	// 组装返回数据
+	res = &mediabookshelfmod.MediaBookshelfInfo{
+		ID: item.MID,
+		//MID:            item.MID,
+		Name:           media.Title,
+		VerticalCover:  media.VerticalCover,
+		TotalEpisode:   media.TotalEpisode,
+		CurrentEpisode: media.CurrentEpisode,
+		MediaType:      media.MediaType,
+		MediaSubType:   int(media.MediaSubType),
+		UpdateStatus:   media.UpdateStatus,
+		CountBrowse:    media.CountBrowse,
+		Permission:     media.Permission,
+		Price:          media.Price,
+		UID:            item.UID,
+		ReadAt:         item.ReadAt,
+		CreatedAt:      item.CreatedAt,
+		UpdateTime:     item.UpdateTime,
+	}
+	if time.Now().Add(-time.Hour*24*7).Unix() < media.ContentUpdateTime.Unix() {
+		res.IsNew = true
+	}
+	return
+}