migrate

type ApiKey struct {
	ID          MongoUUID                   `gorm:"primaryKey;type:uuid" json:"id"`
	Key         string                      `gorm:"type:text;not null;uniqueIndex" json:"key"`
	Quota       int                         `gorm:"not null;default:0" json:"quota"`
	Used        int                         `gorm:"not null;default:0" json:"used"`
	Permissions datatypes.JSONSlice[string] `gorm:"type:json" json:"permissions"`
	Description string                      `gorm:"type:text" json:"description"`
}

type Artist struct {
	ID       MongoUUID  `gorm:"primaryKey;type:uuid" json:"id"`
	Name     string     `gorm:"type:text;not null;index" json:"name"`
	Type     SourceType `gorm:"type:text;not null;index" json:"type"`
	UID      string     `gorm:"type:text;not null;index" json:"uid"`
	Username string     `gorm:"type:text;not null;index" json:"username"`

	// reverse relation
	Artworks []*Artwork `gorm:"foreignKey:ArtistID" json:"artworks"`
}

type Artwork struct {
	// keep ObjectID as 24-hex string
	ID          MongoUUID  `gorm:"primaryKey;type:uuid" json:"id"`
	Title       string     `gorm:"type:text;not null;index:idx_artwork_title,sort:asc" json:"title"`
	Description string     `gorm:"type:text" json:"description"`
	R18         bool       `gorm:"not null;default:false" json:"r18"`
	CreatedAt   time.Time  `gorm:"not null;autoCreateTime" json:"created_at"`
	UpdatedAt   time.Time  `gorm:"not null;autoUpdateTime" json:"updated_at"`
	SourceType  SourceType `gorm:"type:text;not null" json:"source_type"`
	SourceURL   string     `gorm:"type:text;not null;uniqueIndex" json:"source_url"`
	LikeCount   uint       `gorm:"not null;default:0" json:"like_count"`

	ArtistID MongoUUID `gorm:"type:uuid;index" json:"artist_id"`
	Artist   *Artist   `gorm:"foreignKey:ArtistID;references:ID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL" json:"artist"`

	// many2many relationship with tags
	Tags []*Tag `gorm:"many2many:artwork_tags;constraint:OnDelete:CASCADE" json:"tags"`

	// one-to-many pictures
	Pictures []*Picture `gorm:"foreignKey:ArtworkID;constraint:OnDelete:CASCADE" json:"pictures"`
}

type ArtworkStatus string

type CachedArtist struct {
	ID       string     `json:"id"`
	Name     string     `json:"name"`
	Type     SourceType `json:"type"`
	UID      string     `json:"uid"`
	Username string     `json:"username"`
}

type CachedArtwork struct {
	ID        MongoUUID                              `gorm:"primaryKey;type:uuid" json:"id"`
	SourceURL string                                 `gorm:"type:text;uniqueIndex" json:"source_url"`
	CreatedAt time.Time                              `gorm:"autoCreateTime" json:"created_at"`
	Artwork   datatypes.JSONType[*CachedArtworkData] `gorm:"type:json" json:"artwork"`
	Status    ArtworkStatus                          `gorm:"type:text;index" json:"status"`
}

type CachedArtworkData struct {
	ID          string     `json:"id"`
	Title       string     `json:"title"`
	Description string     `json:"description"`
	R18         bool       `json:"r18"`
	SourceType  SourceType `json:"source_type"`
	SourceURL   string     `json:"source_url"`

	Artist      *CachedArtist       `json:"artist"`
	Tags        []string            `json:"tags"`
	Pictures    []*CachedPicture    `json:"pictures"`
	UgoiraMetas []*CachedUgoiraMeta `json:"ugoira_metas,omitempty"`

	Version int `json:"version"` // for future schema changes
}

type CachedPicture struct {
	ID         string `json:"id"`
	ArtworkID  string `json:"artwork_id"`
	OrderIndex uint   `json:"index"`
	Thumbnail  string `json:"thumbnail"`
	Original   string `json:"original"`
	Hidden     bool   `json:"hidden"` // don't post when true

	Width     uint   `json:"width"`
	Height    uint   `json:"height"`
	Phash     string `json:"phash"`      // phash
	ThumbHash string `json:"thumb_hash"` // thumbhash

	StorageInfo  StorageInfo  `json:"storage_info"`
	TelegramInfo TelegramInfo `json:"telegram_info"`
}

type CachedUgoiraMeta struct {
	ID              string         `json:"id"`
	ArtworkID       string         `json:"artwork_id"`
	OrderIndex      uint           `json:"index"`
	MetaData        UgoiraMetaData `json:"data"`
	OriginalStorage StorageDetail  `json:"original_storage"`
	TelegramInfo    TelegramInfo   `json:"telegram_info"`
}

type DeletedRecord struct {
	ID        MongoUUID `gorm:"primaryKey;type:uuid" json:"id"`
	ArtworkID MongoUUID `gorm:"type:uuid;uniqueIndex" json:"artwork_id"`
	SourceURL string    `gorm:"type:text;not null;uniqueIndex" json:"source_url"`
	DeletedAt time.Time `gorm:"not null;autoCreateTime" json:"deleted_at"`
}

type MongoUUID uuid.UUID

type Option struct {
	MongoClient *mongo.Client
	GormDB      *gorm.DB
	Cfg         config.Config
}

type Picture struct {
	ID        MongoUUID `gorm:"primaryKey;type:uuid" json:"id"`
	ArtworkID MongoUUID `gorm:"type:uuid;index" json:"artwork_id"`
	Artwork   *Artwork  `gorm:"foreignKey:ArtworkID;references:ID;constraint:OnDelete:CASCADE" json:"-"`

	OrderIndex uint   `gorm:"column:order_index;not null;default:0;index:idx_picture_artwork_index,priority:1" json:"index"`
	Thumbnail  string `gorm:"type:text" json:"thumbnail"`
	Original   string `gorm:"type:text;index" json:"original"`
	Width      uint   `json:"width"`
	Height     uint   `json:"height"`
	Phash      string `gorm:"type:text;index" json:"phash"` // phash
	ThumbHash  string `gorm:"type:text" json:"thumb_hash"`  // thumbhash

	TelegramInfo datatypes.JSONType[TelegramInfo] `gorm:"type:json" json:"telegram_info"` // original TelegramInfo struct as JSON
	StorageInfo  datatypes.JSONType[StorageInfo]  `gorm:"type:json" json:"storage_info"`  // StorageInfo as JSON

	CreatedAt time.Time `gorm:"autoCreateTime" json:"created_at"`
	UpdatedAt time.Time `gorm:"autoUpdateTime" json:"updated_at"`
}

type SourceType string

type StorageDetail struct {
	Type StorageType `json:"type" bson:"type"`
	Path string      `json:"path" bson:"path"`
	Mime string      `json:"mime" bson:"mime"`
}

type StorageInfo struct {
	Original *StorageDetail `json:"original" bson:"original"`
	Regular  *StorageDetail `json:"regular" bson:"regular"`
	Thumb    *StorageDetail `json:"thumb" bson:"thumb"`
}

type StorageType string

type Tag struct {
	ID    MongoUUID   `gorm:"primaryKey;type:uuid" json:"id"`
	Name  string      `gorm:"type:text;not null;uniqueIndex" json:"name"`
	Alias []*TagAlias `gorm:"foreignKey:TagID;constraint:OnDelete:CASCADE" json:"alias"`

	// reverse relation via many2many
	Artworks []*Artwork `gorm:"many2many:artwork_tags" json:"artworks"`
}

type TagAlias struct {
	ID    MongoUUID `gorm:"primaryKey;type:uuid" json:"id"`
	TagID MongoUUID `gorm:"type:uuid;index" json:"tag_id"`
	Alias string    `gorm:"type:text;not null;index" json:"alias"`
}

type TelegramInfo struct {

	// {bot_id:{"photo": file_id, "document": file_id}}
	FileIDs  map[int64]map[TelegramMediaType]string `json:"file_ids"` // (file_id is bot specific)
	Messages map[int64]TelegramMessage              `json:"messages"` // key: chat id
}

type TelegramMediaType string

type TelegramMessage struct {
	MessageID    int    `json:"message_id"`
	MediaGroupID string `json:"media_group_id,omitempty"`
}

type UgoiraFrame struct {
	File  string `json:"file"`
	Delay int    `json:"delay"`
}

type UgoiraMeta struct {
	ID        MongoUUID `gorm:"primaryKey;type:uuid" json:"id"`
	ArtworkID MongoUUID `gorm:"type:uuid;index" json:"artwork_id"`
	Artwork   *Artwork  `gorm:"foreignKey:ArtworkID;references:ID;constraint:OnDelete:CASCADE" json:"-"`

	OrderIndex uint                               `gorm:"column:order_index;not null;default:0;index:idx_ugoira_artwork_index,priority:1" json:"index"`
	Data       datatypes.JSONType[UgoiraMetaData] `json:"data"`

	CreatedAt time.Time `gorm:"autoCreateTime" json:"created_at"`
	UpdatedAt time.Time `gorm:"autoUpdateTime" json:"updated_at"`

	OriginalStorage datatypes.JSONType[StorageDetail] `json:"original_storage"`
	TelegramInfo    datatypes.JSONType[TelegramInfo]  `json:"telegram_info"`
}

type UgoiraMetaData struct {
	PosterOriginal string        `json:"poster_original"`
	PosterThumb    string        `json:"poster_thumb"`
	OriginalZip    string        `json:"original_zip"`
	ThumbZip       string        `json:"thumb_zip"`
	MimeType       string        `json:"mime_type"`
	Width          int           `json:"width"`
	Height         int           `json:"height"`
	Frames         []UgoiraFrame `json:"frames"`
}

type User struct {
	ID         MongoUUID      `gorm:"primaryKey;type:uuid" json:"id"`
	Username   string         `gorm:"type:text;uniqueIndex" json:"username"`
	Password   string         `gorm:"type:text;not null" json:"password"`
	Email      *string        `gorm:"type:text;uniqueIndex" json:"email"`
	TelegramID *int64         `gorm:"type:bigint;uniqueIndex" json:"telegram_id"`
	Blocked    bool           `gorm:"not null;default:false;index" json:"blocked"`
	UpdatedAt  time.Time      `gorm:"autoUpdateTime" json:"updated_at"`
	DeletedAt  gorm.DeletedAt `gorm:"index" json:"deleted_at"`

	Favorites []*Artwork `gorm:"many2many:user_favorites;constraint:OnDelete:CASCADE" json:"favorites"`

	Settings datatypes.JSONType[*UserSettings] `gorm:"type:json" json:"settings"`
}

type UserSettings struct {
	Language string `json:"language"`
	Theme    string `json:"theme"`
	R18      bool   `json:"r18"`
}