package announInfoMod

import (
	"context"
	"fmt"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"stat/common/log"
	"stat/common/mongos"
	"stat/common/utils"
	"time"

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

func Init() {
	// 获取数据库表
	db = mongos.New(mongos.MgoManager, CollAnnounInfo)

	// 初始化索引
	indexes := []mongo.IndexModel{
		{
			Keys: bson.D{{Key: "appId", Value: 1}},
		},
	}

	if _, err := db.CreateIndex(context.Background(), indexes); err != nil {
		panic(fmt.Sprintf("Model %s InitIndex error %+v", CollAnnounInfo, err))
	}
}

// 获取公告列表
func FindAnnounceList(ctx context.Context, appId *int32, skip int64, limit int64) (count int64, data []AnnounInfo, err error) {
	filter := bson.M{}

	if appId != nil {
		filter["appId"] = appId
	}

	opts := options.Find().SetSkip(skip).SetLimit(limit)

	if err = db.Find(ctx, &data, filter, opts); err != nil {
		log.Error("announce FindAnnounceList Find err", log.E(err))
		return
	}
	if count, err = db.Count(ctx, filter); err != nil {
		log.Error("announce FindAnnounceList Count err", log.E(err))
		return
	}

	return
}

// 新增管理员
func AddAnnounce(ctx context.Context, doc *AnnounInfo) (err error) {
	if _, err = db.InsertOne(ctx, doc); err != nil {
		log.Error("announce AddAnnounce InsertOne err", log.E(err))
		return
	}

	return
}

// 更新管理员
func UpdateAnnounce(ctx context.Context, id string, doc UpdateDoc) (err error) {
	if id == "" {
		return
	}

	objId, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		log.Error("announce UpdateAnnounce ObjectIDFromHex err", log.E(err))
		return
	}

	filter := bson.M{"_id": objId}
	update, _ := utils.ToBsonM(doc)

	if _, err = db.UpdateOne(ctx, filter, bson.M{"$set": update}); err != nil {
		log.Error("announce UpdateAdmin UpdateOne err", log.E(err))
		return
	}

	return
}

// 删除管理员
func DelAnnounce(ctx context.Context, id string) (err error) {
	if id == "" {
		return
	}

	objId, err := primitive.ObjectIDFromHex(id)
	if err != nil {
		log.Error("announce DelAnnounce ObjectIDFromHex err", log.E(err))
		return
	}

	filter := bson.M{"_id": objId}

	if _, err = db.DeleteOne(ctx, filter); err != nil {
		log.Error("announce DelAnnounce DeleteOne err", log.E(err))
		return
	}
	return
}

// 根据appId获取公告
func FindAnnounceByAppId(ctx context.Context, appId int32) (data []AnnounInfo, err error) {
	now := time.Now()

	filer := bson.M{"appId": appId, "startTime": bson.M{"$lt": now}, "endTime": bson.M{"$gt": now}}

	if err = db.Find(ctx, &data, filer); err != nil {
		log.Error("announce DelAnnounce DeleteOne err", log.E(err))
		return
	}

	return
}