// 库存预警DAO层
// 职责：预警查询(无Insert/Update/Delete，由系统自动生成)
// 紧密耦合：service.InventoryWarning
// 注意：预警记录由定时任务或库存变动触发，非用户手动创建
package dao

import (
	"assets/consts/public"
	dto "assets/model/dto/stock"
	entity "assets/model/entity/stock"
	"context"

	"gitea.redpowerfuture.com/red-future/common/db/mongo"
	"github.com/gogf/gf/v2/frame/g"
	"go.mongodb.org/mongo-driver/v2/bson"
)

var InventoryWarning = new(inventoryWarning)

type inventoryWarning struct{}

func (d *inventoryWarning) GetOne(ctx context.Context, req *dto.GetInventoryWarningReq) (res *entity.InventoryWarning, err error) {
	filter := bson.M{"_id": req.Id}
	err = mongo.DB().FindOne(ctx, filter, &res, public.InventoryWarningCollection)
	return
}

func (d *inventoryWarning) List(ctx context.Context, req *dto.ListInventoryWarningReq) (res []entity.InventoryWarning, total int64, err error) {
	filter, err := d.buildListFilter(ctx, req)
	if err != nil {
		return
	}
	total, err = mongo.DB().Find(ctx, filter, &res, public.InventoryWarningCollection, req.Page, req.OrderBy)
	return
}

func (d *inventoryWarning) buildListFilter(ctx context.Context, req *dto.ListInventoryWarningReq) (filter bson.M, err error) {
	_ = ctx
	filter = bson.M{}
	if !g.IsEmpty(req.WarningType) {
		filter["warningType"] = req.WarningType
	}
	// 兼容单值和数组参数（优先使用数组）
	if len(req.BatchIDs) > 0 {
		filter["batchId"] = bson.M{"$in": req.BatchIDs}
	} else if !g.IsEmpty(req.BatchID) {
		filter["batchId"] = req.BatchID
	}
	if len(req.AssetIDs) > 0 {
		filter["assetId"] = bson.M{"$in": req.AssetIDs}
	} else if !g.IsEmpty(req.AssetID) {
		filter["assetId"] = req.AssetID
	}
	if len(req.AssetSkuIDs) > 0 {
		filter["assetSkuId"] = bson.M{"$in": req.AssetSkuIDs}
	} else if !g.IsEmpty(req.AssetSkuID) {
		filter["assetSkuId"] = req.AssetSkuID
	}
	if len(req.SupplierIDs) > 0 {
		filter["supplierId"] = bson.M{"$in": req.SupplierIDs}
	} else if !g.IsEmpty(req.SupplierID) {
		filter["supplierId"] = req.SupplierID
	}
	if !g.IsEmpty(req.Status) {
		filter["status"] = req.Status
	}
	if !g.IsEmpty(req.Keyword) {
		filter["batchNo"] = bson.M{"$regex": req.Keyword, "$options": "i"}
	}
	return
}