100 lines
3.4 KiB
Go
100 lines
3.4 KiB
Go
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}
|
|
}
|