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,128 @@
+package dailytaskmod
+
+import (
+	"errors"
+	"fmt"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"time"
+
+	"91porn-server/common/db"
+	"91porn-server/models"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+)
+
+var mdb *db.MongoDB
+
+const table = models.DailyTask
+
+// InitIndex 设置index
+func initIndex() {
+	many := []mongo.IndexModel{
+		{
+			Keys: bson.D{{Key: "type", Value: 1}},
+		},
+	}
+	if _, err := coll(nil).CreateIndex(many); err != nil {
+		panic(fmt.Sprintf("%s model set index err ==>[%+v]", table, err))
+	}
+}
+
+func coll(t *db.MongoTool) *db.MongoTool {
+	if t == nil {
+		return mdb.Coll(table)
+	}
+	return t.Coll(table)
+}
+
+func Init() {
+	mdb = db.Init(table)
+	initIndex()
+}
+
+type DailyTaskTypeEnum int64
+
+const (
+	DailyTaskTypeAdsClick   DailyTaskTypeEnum = 0 // 每日广告点击
+	DailyTaskTypeUserInvite DailyTaskTypeEnum = 1 // 每日邀请
+	DailyUserLogin          DailyTaskTypeEnum = 2 // 每日登录
+	DailyComment            DailyTaskTypeEnum = 3 // 每日评论
+	DailyPublish            DailyTaskTypeEnum = 4 // 每日发布
+	CommentSetGod           DailyTaskTypeEnum = 5 // 评论被设置成神评
+	DailyDoCollect          DailyTaskTypeEnum = 6 // 每日收藏
+	DailyUnlockByGold       DailyTaskTypeEnum = 7 // 每日解锁(video/acg)
+)
+
+type DailyTask struct {
+	ID        primitive.ObjectID `json:"_id"       bson:"_id,omitempty"`
+	Title     string             `json:"title"     bson:"title"`   // 任务标题
+	Img       string             `json:"img"       bson:"img"`     // 任务图片
+	Type      DailyTaskTypeEnum  `json:"type"      bson:"type"`    // 0 每日广告点击; 1 每日邀请
+	Desc      string             `json:"desc"      bson:"desc"`    // 任务说明
+	Detail    []TaskDetail       `json:"detail"    bson:"detail"`  // 任务详情
+	Status    bool               `json:"status"    bson:"status"`  // 启用/禁用
+	Link      string             `json:"link"      bson:"link"`    // 跳转链接
+	SortNum   int                `json:"sortNum"   bson:"sortNum"` // 排序
+	CreatedAt time.Time          `json:"createdAt" bson:"createdAt"`
+	UpdatedAt time.Time          `json:"updatedAt" bson:"updatedAt"`
+}
+
+type TaskDetail struct {
+	Prizes          []primitive.ObjectID `json:"prizes"          bson:"prizes"`          // 奖品列表
+	FinishCondition uint64               `json:"finishCondition" bson:"finishCondition"` // 达成条件
+}
+
+func GetUserDailyTasks(t *db.MongoTool) ([]DailyTask, error) {
+	var ds []DailyTask
+	op := options.Find().SetSort(bson.D{{"sortNum", 1}})
+	return ds, coll(t).Find(&ds, bson.M{"status": true}, op)
+}
+func GetTasks(filter bson.M, opt *options.FindOptions) ([]DailyTask, bool, error) {
+	var ds []DailyTask
+	if opt == nil {
+		opt = options.Find()
+	}
+	if opt.Limit == nil || *opt.Limit == 0 {
+		opt.SetLimit(101)
+	}
+
+	err := coll(nil).Find(&ds, filter, opt)
+	if err != nil {
+		return nil, false, err
+	}
+	next := len(ds) == int(*opt.Limit)
+	if next {
+		ds = ds[:*opt.Limit-1]
+	}
+	return ds, true, nil
+}
+
+func GetUserDailyTaskByType(t *db.MongoTool, taskType DailyTaskTypeEnum) (*DailyTask, error) {
+	var dt DailyTask
+	if err := coll(t).FindOne(&dt, bson.M{"type": taskType, "status": true}); err != nil {
+		return nil, err
+	}
+	if dt.ID.IsZero() {
+		return nil, errors.New("not found")
+	}
+	if !dt.Status {
+		return nil, errors.New("task not active")
+	}
+	return &dt, nil
+}
+
+func GetUserDailyTaskByID(t *db.MongoTool, id primitive.ObjectID) (*DailyTask, error) {
+	var dt DailyTask
+	if err := coll(t).FindOne(&dt, bson.M{"_id": id}); err != nil {
+		return nil, err
+	}
+	if dt.ID.IsZero() {
+		return nil, errors.New("not found")
+	}
+	if !dt.Status {
+		return nil, errors.New("task not active")
+	}
+	return &dt, nil
+}