package main

import (
	"91porn-server/generate/common"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"time"
)

type AiPlaza struct {
	ID            primitive.ObjectID `bson:"_id,omitempty" json:"id" comment:"文档id"`
	Type          int                `json:"type" bson:"type"`                          // 1-ai图片换脸 2-ai视频换脸 3-ai脱衣 4-ai图生视频 5-ai绘画
	ModuleId      string             `json:"moduleId" bson:"moduleId" comment:"关联模版id"` // 关联模版id
	Uid           uint64             `json:"uid" bson:"uid"`
	Title         string             `json:"title" bson:"title" comment:"标题"`
	OriginalImage string             `json:"originalImage" bson:"originalImage" comment:"原图"`       // 原图
	OriginalVideo string             `json:"originalVideo" bson:"originalVideo" comment:"原视频"`      // 原视频
	OriginContent string             `json:"originContent" bson:"originContent" comment:"输入的原文"`    // 输入的原文
	GenerateImage string             `json:"generateImage" bson:"generateImage" comment:"生成的图"`     // 生成的图
	GenerateVideo string             `json:"generateVideo" bson:"generateVideo" comment:"生成的视频"`    // 生成的视频
	Status        int                `json:"status" bson:"status" comment:"0-待审核 1-审核通过 2-拒绝 3-下架"` // 0-待审核 1-审核通过 2-拒绝 3-下架
	Reason        string             `json:"reason" bson:"reason" comment:"原因"`
	SortCode      int                `json:"sortCode" bson:"sortCode" comment:"置顶排序号，大于0就是置顶"`
	ReviewAt      time.Time          `json:"reviewAt" bson:"reviewAt" comment:"审核时间"`     // 审核时间
	CreatedAt     time.Time          `json:"createdAt" bson:"createdAt" comment:"文档创建时间"` // 文档创建时间
	UpdatedAt     time.Time          `json:"updatedAt" bson:"updatedAt" comment:"文档更新时间"` // 文档更新时间
}

func main() {
	g := common.NewGen("91porn-server", "ai广场帖子")
	g.AppGenerate = true
	g.Generate(AiPlaza{})
}