package uservisitlog

import (
	"context"
	"fmt"
	"time"
	"videoweb/pub/common/log"
	"videoweb/pub/common/srvdef"
	"videoweb/pub/mongos"

	"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"
)

// Init 初始化访问日志表
func Init(ctx context.Context) {
	visitColl = mongos.Shop.Coll(TN.UserVisitLog)
	initIndex(ctx)
}

func initIndex(ctx context.Context) {
	indexes := []mongo.IndexModel{
		{
			Keys: bson.D{
				{Key: FN.UserVisitLog.UserId, Value: -1},
			},
		},
		{
			Keys: bson.D{
				{Key: FN.UserVisitLog.VisitAt, Value: -1},
			},
			Options: options.Index().SetExpireAfterSeconds(5 * 24 * 60 * 60),
		},
	}
	if _, err := visitColl.CreateIndex(ctx, indexes); err != nil {
		panic(fmt.Sprintf("(uservisitlog:initIndex) Create uservisitlog Index failed! err:%+v:", err))
	}
}

// InsertOneVisitLog 插入一条访问记录
func InsertOneVisitLog(ctx context.Context, data *UserVisitLog) *srvdef.SrvErr {
	if _, err := visitColl.InsertOne(ctx, data); err != nil {
		log.Info("visitColl:InsertOneVisitLog db.InsertOne failed",
			log.E(err), log.Any("data", data))
		return srvdef.New(srvdef.ErrDbInsertError, "", "")
	}
	return nil
}

// FindManyVisitLogs 查找一批访问记录
func FindManyVisitLogs(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (VisitLogList, *srvdef.SrvErr) {
	data := make(VisitLogList, 0)
	if err := visitColl.Find(ctx, &data, filter, opts...); err != nil {
		log.Warn("visitColl:FindManyVisitLogs db.Find failed",
			log.E(err), log.Any("filter", filter))
		return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
	}
	return data, nil
}

// AccessSyncById 按 ObjectId 增量拉取访问数据（统计中心同步）
func AccessSyncById(ctx context.Context, id string, size int64) (data VisitLogList, err error) {
	oid, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		return
	}
	endTime := time.Now().Add(-time.Minute * 1)
	visitFilter := bson.M{
		FN.UserVisitLog.Id:      bson.M{"$gt": oid},
		FN.UserVisitLog.VisitAt: bson.M{"$lt": endTime},
	}
	visitOpt := (&options.FindOptions{}).SetLimit(size).SetSort(bson.M{FN.UserVisitLog.Id: 1})
	err = visitColl.Find(ctx, &data, visitFilter, visitOpt)
	return data, err
}