Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,99 @@
+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}
+}