package aichangefaceimgmod

import (
	"91porn-server/models/commod"
	"time"

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

// AiChangFaceImgAppRes AI记录 App返回给前端
type AiChangFaceImgAppRes struct {
	UID       uint64    `json:"uid" bson:"uid"`             // 用户id
	OriginPic string    `json:"originPic" bson:"originPic"` // 脱衣原图
	NewPic    []string  `json:"newPic" bson:"newPic"`       // 脱衣后新图
	Status    int       `json:"status" bson:"status"`       // 1、进行中 2、生成成功 3、生成失败
	Remark    string    `json:"remark" bson:"remark"`       // 备注
	CreatedAt time.Time `json:"createdAt" bson:"createdAt"` // 创建时间
	UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt"` // 刷新时间
}

type ListRequest struct {
	Status *int `form:"status" json:"status" bson:"status"` // 1、进行中 2、生成成功 3、生成失败
	commod.Page
}

func (receiver *ListRequest) Filter(uid uint64) primitive.M {
	filter := bson.M{}
	if receiver.Status != nil {
		filter["status"] = receiver.Status
		if *receiver.Status == 1 {
			filter["status"] = bson.M{"$in": []int{Processing, StatusSubmit}}
		}
	}
	filter["uid"] = uid
	filter["isHide"] = bson.M{"$in": []any{nil, false}}
	return filter
}

func (receiver *ListRequest) Options() *options.FindOptions {
	return options.Find().SetSkip(int64(receiver.Skip())).SetLimit(int64(receiver.Limit())).SetSort(bson.M{"fakeLikeCount": -1})
}

type GenerateRequest struct {
	OriginPic   string `json:"originPic" bson:"originPic" binding:"required"` // 换脸原图
	MId         string `json:"mId" bson:"mId" binding:"required"`             // 模版ID
	IsFreeTimes bool   `json:"isFreeTimes" bson:"isFreeTimes"`                // 是否使用免费次数
	Coin        int64  `json:"coin" bson:"coin"`                              // 此次脱衣金币个数
	ShareTitle  string `json:"shareTitle" bson:"shareTitle"`                  // 分享标题
	ShareStatus int    `json:"shareStatus" bson:"shareStatus"`                // 是否分享 0-不分享 1-分享

}

func (receiver *GenerateRequest) Generate(orderId primitive.ObjectID, createdAt time.Time, uid uint64, modTitle, modPic string, privilegeCount, debitFreeCount, debitCoins, debitIncomeCoins int64) AiChangeFaceImg {
	isFreeTimes := false
	if debitFreeCount > 0 {
		isFreeTimes = true
	}
	if privilegeCount > 0 {
		isFreeTimes = true
	}
	return AiChangeFaceImg{
		ID:              orderId,
		Coin:            debitIncomeCoins + debitCoins,
		DebitAmountCoin: debitCoins,
		DebitIncomeCoin: debitIncomeCoins,
		IsFreeTimes:     isFreeTimes,
		Count:           debitFreeCount + privilegeCount,
		OriginPic:       receiver.OriginPic,
		MId:             receiver.MId,
		ModTitle:        modTitle,
		ModPic:          modPic,
		UID:             uid,
		Status:          Processing,
		IsHide:          false,
		ShareTitle:      receiver.ShareTitle,
		ShareStatus:     receiver.ShareStatus,
		UpdatedAt:       createdAt,
		CreatedAt:       createdAt,
	}
}

type DelRequest struct {
	ID primitive.ObjectID `json:"id" bson:"id"` // AI订单ID
}

func (receiver *DelRequest) Filter() primitive.M {
	filter := bson.M{}
	filter["_id"] = receiver.ID
	return filter
}

func (receiver *DelRequest) Update() primitive.M {
	update := bson.M{}
	update["isHide"] = true
	update["updatedAt"] = time.Now()
	return bson.M{"$set": update}
}