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,115 @@
+package aichangefacemod
+
+import (
+	"91porn-server/common/db"
+	"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"
+)
+
+type AiChangeFaceWebResp struct {
+	ID         primitive.ObjectID `json:"id"`
+	Uid        uint64             `json:"uid"`        // 用户
+	Pic        []string           `json:"picture"`    // 换脸照片
+	ModId      primitive.ObjectID `json:"modId"`      // 视频模版id
+	ModUrl     string             `json:"modUrl"`     // 视频模版播放地址url
+	ModTitle   string             `json:"modTitle"`   // 模版标题
+	Coin       int64              `json:"coin"`       // 普通金币数
+	IncomeCoin int64              `json:"incomeCoin"` // 可提现金币数
+	Status     AiChangeFaceStatus `json:"status"`     // 0 未完成; 1 已完成; -1 已退款
+	Cover      string             `json:"cover"`      // 换脸后封面大图
+	Url        string             `json:"url"`        // 换脸后视频地址
+	Remark     string             `json:"remark"`     // 拒绝理由
+	Operator   string             `json:"operator"`   // 操作员
+	CreatedAt  time.Time          `json:"createdAt"`  // 创建时间
+	UpdatedAt  time.Time          `json:"updatedAt"`  // 刷新时间
+}
+
+type ListRequest struct {
+	ID     *string             `form:"id" json:"id"`
+	Uid    *uint64             `form:"uid" json:"uid"`
+	Status *AiChangeFaceStatus `form:"status" json:"status"`
+	VidID  *string             `form:"vidId" json:"vidId"`
+	commod.Page
+}
+
+type EditAiChangeFace struct {
+	ID     primitive.ObjectID  `json:"id" binding:"required"` // 被编辑的ai换脸id
+	Status *AiChangeFaceStatus `json:"status"`                // 0 未完成; 1 已完成; -1 已退款 2 已经提交
+	Cover  *string             `json:"cover"`                 // 换脸后封面大图
+	Url    *string             `json:"url"`                   // 换脸后视频地址
+	Remark *string             `json:"remark"`                // 拒绝理由
+	Title  *string             `json:"title"`                 // 标题
+}
+
+func (receiver *EditAiChangeFace) Update(updateAct string) bson.M {
+	var update = bson.M{}
+	if receiver.Url != nil {
+		update["url"] = receiver.Url
+	}
+	if receiver.Cover != nil {
+		update["cover"] = receiver.Cover
+	}
+	if receiver.Status != nil {
+		update["status"] = receiver.Status
+	}
+	if receiver.Remark != nil {
+		update["remark"] = receiver.Remark
+	}
+	update["operator"] = updateAct
+	update["updatedAt"] = time.Now()
+	return bson.M{"$set": update}
+}
+
+func WebList(id *primitive.ObjectID, uid *uint64, status *AiChangeFaceStatus, vidId *string, skip, limit int) ([]AiChangeFace, int64, error) {
+	cond := bson.M{}
+	if id != nil {
+		cond["_id"] = *id
+	}
+	if uid != nil {
+		cond["uid"] = *uid
+	}
+	if status != nil {
+		cond["status"] = *status
+	}
+	if vidId != nil {
+		vidObjId, _ := primitive.ObjectIDFromHex(*vidId)
+		cond["vidId"] = vidObjId
+	}
+	count, err := coll(nil).Count(cond)
+	if err != nil {
+		return nil, 0, err
+	}
+	var acfs []AiChangeFace
+	opts := options.Find().SetSort(bson.D{{Key: "createdAt", Value: -1}})
+	if skip > 0 {
+		opts.SetSkip(int64(skip))
+	}
+	if limit > 0 {
+		opts.SetLimit(int64(limit))
+	}
+	return acfs, count, coll(nil).Find(&acfs, cond, opts)
+}
+
+func Update(t *db.MongoTool, id primitive.ObjectID, set bson.M) error {
+	_, err := coll(t).UpdateOne(bson.M{"_id": id}, set)
+	return err
+}
+
+// TransitionStatus atomically changes an order from the expected status. The
+// returned boolean is false when another worker or callback handled it first.
+func TransitionStatus(t *db.MongoTool, id primitive.ObjectID, from, to AiChangeFaceStatus, set bson.M) (bool, error) {
+	if set == nil {
+		set = bson.M{}
+	}
+	set["status"] = to
+	set["updatedAt"] = time.Now()
+	result, err := coll(t).UpdateOne(bson.M{"_id": id, "status": from}, bson.M{"$set": set})
+	if err != nil {
+		return false, err
+	}
+	return result.ModifiedCount == 1, nil
+}