package aichangefaceimgmod

import (
	"91porn-server/models/commod"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo/options"
	"time"
)

// RechargeUpdateReq 修改参数
type RechargeUpdateReq struct {
	StatusDesc *string   `json:"statusDesc" bson:"statusDesc"` // 状态描述
	UpdatedAt  time.Time `json:"updatedAt" bson:"updatedAt"`   // 更新时间
}

type RchgQueryReq struct {
	ID     *string `form:"id" json:"_id,omitempty" bson:"_id"`           // 流水id
	UID    *uint64 `form:"uid"  json:"uid,omitempty" bson:"uid"`         // 用户id
	Status *int    `form:"status" json:"status,omitempty" bson:"status"` // 1、进行中 2、生成成功 3、生成失败 4、已经退款
}

type WebListRequest struct {
	Status *int    `form:"status" json:"status" bson:"status"`   // 1、进行中 2、生成成功 3、生成失败 4、已经退款
	UID    *uint64 `form:"uid"  json:"uid,omitempty" bson:"uid"` // 用户id
	ID     *string `form:"id" json:"_id,omitempty" bson:"_id"`   // 流水id
	MId    *string `form:"mId" json:"mId,omitempty" bson:"_id"`  // AI模版id
	commod.Page
}

func (receiver *WebListRequest) Filter() primitive.M {
	filter := bson.M{}
	if receiver.Status != nil {
		filter["status"] = receiver.Status
	}
	if receiver.UID != nil {
		filter["uid"] = receiver.UID
	}
	if receiver.MId != nil {
		filter["mId"] = receiver.MId
	}
	if receiver.ID != nil {
		id, _ := primitive.ObjectIDFromHex(*receiver.ID)
		filter["_id"] = id
	}
	return filter
}

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

type EditCond struct {
	ID     string    `json:"id" binding:"required"` // 文档
	NewPic *[]string `json:"newPic" bson:"newPic"`  // AI图片换脸后新图
	Status *int      `json:"status" bson:"status"`  // 状态  1、进行中 2、生成成功 3、生成失败
	Remark *string   `json:"remark" bson:"remark"`  // 拒绝理由
}

func (receiver *EditCond) Filter() bson.M {
	objID, _ := primitive.ObjectIDFromHex(receiver.ID)
	return bson.M{"_id": objID}
}

func (receiver *EditCond) Update(updateAct string) bson.M {
	var update = bson.M{}
	if receiver.NewPic != nil {
		update["newPic"] = receiver.NewPic
	}
	if receiver.Status != nil {
		update["status"] = receiver.Status
	}
	if receiver.Remark != nil {
		update["remark"] = receiver.Remark
	}
	update["updateAct"] = updateAct
	update["updatedAt"] = time.Now()
	return bson.M{"$set": update}
}