Claude Opus 5
+342
@@ -0,0 +1,342 @@
|
||||
package ai_text_to_novel_ser
|
||||
|
||||
import (
|
||||
"91porn-server/common/db"
|
||||
"91porn-server/common/log"
|
||||
"91porn-server/common/stderr"
|
||||
"91porn-server/models/cache/aitexttonoveldata"
|
||||
"91porn-server/models/commod"
|
||||
"91porn-server/models/v/aitexttonovelmod"
|
||||
"91porn-server/models/v/txnmod"
|
||||
"91porn-server/models/v/walletmod"
|
||||
"91porn-server/web/webg"
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
)
|
||||
|
||||
type WebListReq struct {
|
||||
ID *string `json:"id" form:"id"` // ID
|
||||
UID *uint64 `json:"uid" form:"uid"` // 用户ID
|
||||
Status *int `json:"status" form:"status"` // 状态
|
||||
commod.Page
|
||||
}
|
||||
|
||||
func (q *WebListReq) Filter() primitive.M {
|
||||
filter := bson.M{}
|
||||
if q.ID != nil {
|
||||
id, _ := primitive.ObjectIDFromHex(*q.ID)
|
||||
filter["_id"] = id
|
||||
}
|
||||
if q.Status != nil {
|
||||
filter["status"] = q.Status
|
||||
}
|
||||
if q.UID != nil {
|
||||
filter["uid"] = q.UID
|
||||
}
|
||||
return filter
|
||||
}
|
||||
|
||||
func (q *WebListReq) Sort() primitive.D {
|
||||
return bson.D{{Key: "createdAt", Value: -1}}
|
||||
}
|
||||
|
||||
type WebListRes struct {
|
||||
Total int64 `json:"total"`
|
||||
HasNext bool `json:"hasNext"`
|
||||
List []aitexttonovelmod.AiTextToNovel `json:"list"`
|
||||
}
|
||||
|
||||
// GetList 获取列表
|
||||
func (q *WebListReq) GetList() (res WebListRes, err error) {
|
||||
res.List, res.Total, res.HasNext, err = aitexttonovelmod.GetList(q.Filter(), int64(q.Skip()), int64(q.Limit()), q.Sort())
|
||||
return
|
||||
}
|
||||
|
||||
type WebUpdateReq struct {
|
||||
ID primitive.ObjectID `json:"id" binding:"required"`
|
||||
UID *uint64 `json:"uid"` // 用户ID
|
||||
Background *string `json:"background"` // 小说人物设定/故事背景等
|
||||
Scene *string `json:"scene"` // 地点场景
|
||||
Text *string `json:"text"` // 故事情节描述
|
||||
Other *string `json:"other"` // 小说其他要求
|
||||
ModelType *string `json:"modelType"` // AI小说模型 1，2
|
||||
Content *string `json:"content"` // AI生成小说正文
|
||||
Status *int `json:"status"` // 状态
|
||||
Coin *int64 `json:"coin"` // AI小说金币个数
|
||||
DebitAmountCoin *int64 `json:"debitAmountCoin"` // 此次AI小说扣除金币个数
|
||||
DebitIncomeCoin *int64 `json:"debitIncomeCoin"` // 此次AI小说扣除收益金币个数
|
||||
IsFreeTimes *bool `json:"isFreeTimes"` // 是否使用免费次数
|
||||
IsHide *bool `json:"isHide"` // 是否被用户隐藏
|
||||
Remark *string `json:"remark"` // 备注
|
||||
IsDelete *bool `json:"isDelete"` // 是否删除
|
||||
}
|
||||
|
||||
// Update 更新数据
|
||||
func (p *WebUpdateReq) Update() error {
|
||||
info, err := aitexttonovelmod.GetInfo(p.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if info.ID.IsZero() {
|
||||
return errors.New("data is null")
|
||||
}
|
||||
|
||||
data := make(map[string]interface{})
|
||||
if p.UID != nil {
|
||||
data["uid"] = *p.UID
|
||||
}
|
||||
if p.Background != nil {
|
||||
data["background"] = *p.Background
|
||||
}
|
||||
if p.Scene != nil {
|
||||
data["scene"] = *p.Scene
|
||||
}
|
||||
if p.Text != nil {
|
||||
data["text"] = *p.Text
|
||||
}
|
||||
if p.Other != nil {
|
||||
data["other"] = *p.Other
|
||||
}
|
||||
if p.ModelType != nil {
|
||||
data["modelType"] = *p.ModelType
|
||||
}
|
||||
if p.Content != nil {
|
||||
data["content"] = *p.Content
|
||||
}
|
||||
if p.Status != nil {
|
||||
data["status"] = *p.Status
|
||||
}
|
||||
if p.Coin != nil {
|
||||
data["coin"] = *p.Coin
|
||||
}
|
||||
if p.DebitAmountCoin != nil {
|
||||
data["debitAmountCoin"] = *p.DebitAmountCoin
|
||||
}
|
||||
if p.DebitIncomeCoin != nil {
|
||||
data["debitIncomeCoin"] = *p.DebitIncomeCoin
|
||||
}
|
||||
if p.IsFreeTimes != nil {
|
||||
data["isFreeTimes"] = *p.IsFreeTimes
|
||||
}
|
||||
if p.IsHide != nil {
|
||||
data["isHide"] = *p.IsHide
|
||||
}
|
||||
if p.Remark != nil {
|
||||
data["remark"] = *p.Remark
|
||||
}
|
||||
if p.IsDelete != nil {
|
||||
data["isDelete"] = *p.IsDelete
|
||||
}
|
||||
data["updatedAt"] = time.Now()
|
||||
if p.Status != nil && *p.Status == 5 {
|
||||
if info.Status == int(aitexttonovelmod.StatusRefunded) || info.Status == int(aitexttonovelmod.StatusGenerationFailed) {
|
||||
return errors.New("Please do not refund repeatedly.")
|
||||
}
|
||||
|
||||
if err = webg.VideoDB.Trans(func(t *db.MongoTool) error {
|
||||
if _, err = aitexttonovelmod.UpdateByID(t, info.ID, data); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 退款
|
||||
if info.DebitAmountCoin == 0 && info.DebitIncomeCoin == 0 { // 总金额为0则不需要退款
|
||||
return nil
|
||||
}
|
||||
|
||||
// 退款进入钱包余额
|
||||
wallet, err := walletmod.ReturnAmountAndIncome(t, -info.DebitAmountCoin, -info.DebitIncomeCoin, info.UID)
|
||||
if err != nil {
|
||||
log.Error(fmt.Sprintf("Handle text to novel Edit walletmod.DebitAmount error:%+v:", err), log.Any("uid", info.UID))
|
||||
return err
|
||||
}
|
||||
var tl []txnmod.TransactionLog
|
||||
if info.DebitAmountCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: info.UID,
|
||||
Amount: info.DebitAmountCoin,
|
||||
ActualAmount: float64(info.DebitAmountCoin),
|
||||
TranType: txnmod.AiTextToNovelDebitGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToNovelDebitGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI小说金币-%d", info.DebitAmountCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
if info.DebitIncomeCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: info.UID,
|
||||
Amount: info.DebitIncomeCoin,
|
||||
ActualAmount: float64(info.DebitIncomeCoin),
|
||||
TranType: txnmod.AiTextToNovelDebitIncomeGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToNovelDebitIncomeGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI小说收益金币-%d", info.DebitIncomeCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
|
||||
if len(tl) > 0 {
|
||||
txnErr := txnmod.InsertManyTransactionLog(t, tl)
|
||||
if txnErr != nil {
|
||||
log.Error(fmt.Sprintf("Handle ai text to novel txnmod.InsertTransactionLog error:%+v:", txnErr), log.Any("uid", info.UID))
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}); err != nil {
|
||||
if e, ok := err.(stderr.Code); ok {
|
||||
return e
|
||||
}
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
return nil
|
||||
} else {
|
||||
if _, err = aitexttonoveldata.UpdateData(nil, p.ID.Hex(), data); err != nil {
|
||||
}
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
type WebDeleteReq struct {
|
||||
ID primitive.ObjectID `json:"id" binding:"required"`
|
||||
}
|
||||
|
||||
// Delete 删除数据
|
||||
func (p *WebDeleteReq) Delete() error {
|
||||
data, err := aitexttonovelmod.GetInfo(p.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if err = validateDeleteStatus(data.Status); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := aitexttonoveldata.DeleteData(nil, p.ID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func validateDeleteStatus(status int) error {
|
||||
if status == int(aitexttonovelmod.StatusOrderSuccess) ||
|
||||
status == int(aitexttonovelmod.StatusSubmitted) {
|
||||
return stderr.AiGenningDelForbidden
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// CallbackReq 换脸订单回调请求
|
||||
type CallbackReq struct {
|
||||
Content string `json:"content"` // AI小说内容
|
||||
AppOrderNum string `json:"appOrderNum"` // app订单号
|
||||
Status int `json:"status"` // 订单状态 1:成功，2:失败
|
||||
Msg string `json:"msg"` // 消息
|
||||
}
|
||||
|
||||
func (p *CallbackReq) Callback() error {
|
||||
id, _ := primitive.ObjectIDFromHex(p.AppOrderNum)
|
||||
// 查询该笔订单是否存在
|
||||
data, err := aitexttonovelmod.GetInfo(id)
|
||||
if err != nil {
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
|
||||
if data.ID.IsZero() {
|
||||
return errors.New("ai text to novel data is null")
|
||||
}
|
||||
|
||||
if data.Status != int(aitexttonovelmod.StatusSubmitted) {
|
||||
log.Error("AI小说-重复回调或者状态不正确", log.Any("id", data.ID.Hex()), log.Any("status", data.Status))
|
||||
return nil
|
||||
}
|
||||
|
||||
var status = aitexttonovelmod.StatusGenerationSuccess
|
||||
|
||||
// 处理金币退款以及后续操作
|
||||
set := bson.M{}
|
||||
if p.Content != "" {
|
||||
set["content"] = strings.TrimSpace(p.Content)
|
||||
}
|
||||
if p.Status != 1 {
|
||||
set["remark"] = p.Msg
|
||||
set["content"] = ""
|
||||
status = aitexttonovelmod.StatusGenerationFailed
|
||||
}
|
||||
set["status"] = status
|
||||
set["updatedAt"] = time.Now()
|
||||
|
||||
switch status {
|
||||
case aitexttonovelmod.StatusGenerationSuccess:
|
||||
if _, err = aitexttonovelmod.UpdateByID(nil, id, set); err != nil {
|
||||
return stderr.ErrDbUpdateError
|
||||
}
|
||||
return nil
|
||||
case aitexttonovelmod.StatusGenerationFailed: // 退款
|
||||
if err = webg.VideoDB.Trans(func(t *db.MongoTool) error {
|
||||
if _, err = aitexttonovelmod.UpdateByID(t, id, set); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 退款
|
||||
if data.DebitAmountCoin == 0 && data.DebitIncomeCoin == 0 { // 总金额为0则不需要退款
|
||||
return nil
|
||||
}
|
||||
|
||||
// 退款进入钱包余额
|
||||
wallet, err := walletmod.ReturnAmountAndIncome(t, -data.DebitAmountCoin, -data.DebitIncomeCoin, data.UID)
|
||||
if err != nil {
|
||||
log.Error(fmt.Sprintf("Handle Callback Edit walletmod.DebitAmount error:%+v:", err), log.Any("uid", data.UID))
|
||||
return err
|
||||
}
|
||||
var tl []txnmod.TransactionLog
|
||||
if data.DebitAmountCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: data.UID,
|
||||
Amount: data.DebitAmountCoin,
|
||||
ActualAmount: float64(data.DebitAmountCoin),
|
||||
TranType: txnmod.AiTextToNovelDebitGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToNovelDebitGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI小说金币-%d", data.DebitAmountCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
if data.DebitIncomeCoin > 0 {
|
||||
tl = append(tl, txnmod.TransactionLog{
|
||||
TransNo: primitive.NewObjectID(),
|
||||
UID: data.UID,
|
||||
Amount: data.DebitIncomeCoin,
|
||||
ActualAmount: float64(data.DebitIncomeCoin),
|
||||
TranType: txnmod.AiTextToNovelDebitIncomeGoldReturn.Key(),
|
||||
TranTypeInt: int64(txnmod.AiTextToNovelDebitIncomeGoldReturn),
|
||||
Desc: fmt.Sprintf("退还AI小说收益金币-%d", data.DebitIncomeCoin),
|
||||
RealAmount: walletmod.GetRealAmount(wallet),
|
||||
})
|
||||
}
|
||||
|
||||
if len(tl) > 0 {
|
||||
txnErr := txnmod.InsertManyTransactionLog(t, tl)
|
||||
if txnErr != nil {
|
||||
log.Error(fmt.Sprintf("Handle ai text to image Callback txnmod.InsertTransactionLog error:%+v:", txnErr), log.Any("uid", data.UID))
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}); err != nil {
|
||||
if e, ok := err.(stderr.Code); ok {
|
||||
return e
|
||||
}
|
||||
return stderr.ErrDbQueryError
|
||||
}
|
||||
return nil
|
||||
default:
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user