package useractmod

import (
	"fmt"
	"time"

	"91porn-server/common/db"
	"91porn-server/common/log"
	"91porn-server/models"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var mdb *db.MongoDB

const table = models.UserAct

func coll(t *db.MongoTool) *db.MongoTool {
	if t == nil {
		return mdb.Coll(table)
	}
	return t.Coll(table)
}

// initIndex 索引设置
func initIndex() {
	many := []mongo.IndexModel{ //batch set indexes   //value is the type 1 or -1
		{
			Keys: bson.D{{Key: "uid", Value: 1}, {Key: "vid", Value: 1}},
		},
		{
			Keys: bson.D{{Key: "uid", Value: 1}, {Key: "dailyDate", Value: 1}},
		},
		{
			Keys: bson.D{{Key: "vid", Value: 1}},
		},
		{
			Keys: bson.D{{Key: "createdAt", Value: -1}},
		},
		{
			Keys: bson.D{{Key: "addPoint", Value: 1}},
		},
		{
			Keys: bson.D{{Key: "consumeKey", Value: 1}},
			Options: options.Index().
				SetName("uniq_daily_free_watch_consume").
				SetUnique(true).
				SetPartialFilterExpression(bson.M{"consumeKey": bson.M{"$gt": ""}}),
		},
	}
	if _, err := coll(nil).CreateIndex(many); err != nil {
		panic(fmt.Sprintf("user model set index err ==>[%+v]", err))
	}
}

// UserActUpsert 更新用户观看次数
func UserActUpsert(uid uint64, duration time.Time, watchCount int64) error {
	filter := bson.M{"uid": uid, "dailyDate": duration}
	update := bson.M{"$inc": bson.M{"addWatchCount": watchCount}, "$set": bson.M{"addUpdatedAt": time.Now()}}
	if _, err := coll(nil).UpsertOne(filter, update); err != nil {
		log.ZapLog.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UserActUpsert", table, "UpsertOne", err))
		return err
	}
	return nil
}

// UserActInsert 生成用户观看记录
func UserActInsert(act UserAct) error {
	return UserActInsertTrans(nil, act)
}

// UserActInsertTrans 生成用户观看记录，支持复用Mongo事务。
func UserActInsertTrans(t *db.MongoTool, act UserAct) error {
	if _, err := coll(t).InsertOne(act); err != nil {
		log.ZapLog.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UserActInsert", table, "InsertOne", err))
		return err
	}
	return nil
}

// ActCount 查看用户观看次数 只统计通过非vip观看的次数
func ActCount(uid uint64, duration time.Time) int64 {
	filter := bson.M{"uid": uid, "dailyDate": duration, "playWay": IsNoVip}
	cnt, err := coll(nil).Count(filter)
	if err != nil {
		log.ZapLog.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "ActCount", table, "Count", err))
		return 0
	}
	return cnt
}

// IsViewTodayByNoVip
func IsViewTodayByNoVip(uid uint64, vid primitive.ObjectID, duration time.Time) bool {
	result, err := IsViewTodayByNoVipTrans(nil, uid, vid, duration)
	if err != nil {
		log.ZapLog.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "IsViewTodayByNoVip", table, "Exists", err))
		return false
	}
	return result
}

// IsViewTodayByNoVipTrans 查询当天是否已通过非VIP次数观看，支持复用Mongo事务。
func IsViewTodayByNoVipTrans(t *db.MongoTool, uid uint64, vid primitive.ObjectID, duration time.Time) (bool, error) {
	filter := bson.M{"uid": uid, "dailyDate": duration, "vid": vid, "playWay": IsNoVip}
	return coll(t).Exists(filter)
}

// UserActFindByUid
func UserActFindByUid(uid uint64, duration time.Time) (u *UserAct, err error) {
	filter := bson.M{"uid": uid, "dailyDate": duration}
	if err = coll(nil).FindOne(&u, filter); err != nil {
		log.ZapLog.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UserActUpsert", table, "UpsertOne", err))
		return
	}
	return
}

// UserActFindByUidDesc
func UserActFindByUidDesc(uid uint64, duration time.Time) (u []*UserAct, err error) {
	filter := bson.M{"uid": uid, "dailyDate": duration, "addPoint": Begin}
	opt := options.Find().SetSort(bson.D{{Key: "createdAt", Value: -1}}).SetLimit(1)
	if err = coll(nil).Find(&u, filter, opt); err != nil {
		log.ZapLog.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UserActFindByUidDesc", table, "FindOne", err))
		return
	}
	return
}

func DeleteBeforeDailyDate(t *db.MongoTool, tm time.Time) error {
	_, err := coll(t).DeleteMany(bson.M{"dailyDate": bson.M{"$lt": tm}})
	return err
}