types

type AdapterOption struct {
	LoadTag          bool
	LoadArtist       bool
	LoadPicture      bool
	OnlyIndexPicture bool
}

type AdminModel struct {
	UserID      int64        `bson:"user_id" json:"user_id"`
	Permissions []Permission `bson:"permissions" json:"permissions"`
	GrantBy     int64        `bson:"grant_by" json:"grant_by"`
	SuperAdmin  bool         `bson:"super_admin" json:"super_admin"`
}

type ApiKeyModel struct {
	ID          primitive.ObjectID `bson:"_id,omitempty"`
	Key         string             `bson:"key"`
	Quota       int                `bson:"quota"`
	Used        int                `bson:"used"`
	Permissions []ApiKeyPermission `bson:"permissions"`
	Description string             `bson:"description"`
}

type ApiKeyPermission string

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

type ArtistModel struct {
	ID       primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
	Name     string             `bson:"name" json:"name"`
	Type     SourceType         `bson:"type" json:"type"`
	UID      string             `bson:"uid" json:"uid"`
	Username string             `bson:"username" json:"username"`
}

type Artwork struct {
	ID          string     `json:"id" bson:"_id"`
	Title       string     `json:"title" bson:"title"`
	Description string     `json:"description" bson:"description"`
	R18         bool       `json:"r18" bson:"r18"`
	CreatedAt   time.Time  `json:"created_at" bson:"created_at"`
	SourceType  SourceType `json:"source_type" bson:"source_type"`
	SourceURL   string     `json:"source_url" bson:"source_url"`
	LikeCount   uint       `json:"like_count" bson:"like_count"`

	Artist   *Artist    `json:"artist" bson:"artist"`
	Tags     []string   `json:"tags" bson:"tags"`
	Pictures []*Picture `json:"pictures" bson:"pictures"`
}

type ArtworkModel struct {
	ID          primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
	Title       string             `bson:"title" json:"title"`
	Description string             `bson:"description" json:"description"`
	R18         bool               `bson:"r18" json:"r18"`
	CreatedAt   primitive.DateTime `bson:"created_at" json:"created_at"`
	SourceType  SourceType         `bson:"source_type" json:"source_type"`
	SourceURL   string             `bson:"source_url" json:"source_url"`
	LikeCount   uint               `bson:"like_count" json:"like_count"`

	ArtistID primitive.ObjectID   `bson:"artist_id" json:"artist_id"`
	Tags     []primitive.ObjectID `bson:"tags" json:"tags"`
	Pictures []primitive.ObjectID `bson:"pictures" json:"pictures"`
}

type ArtworkSearchDocument struct {
	ID          string `json:"id"`
	Title       string `json:"title"`
	Artist      string `json:"artist"`
	Description string `json:"description"`
	Tags        string `json:"tags"`
}

type ArtworkStatus string

type AuthMethod string

type CachedArtworksModel struct {
	ID        primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
	SourceURL string             `bson:"source_url" json:"source_url"`
	CreatedAt primitive.DateTime `bson:"created_at" json:"created_at"`
	Artwork   *Artwork           `bson:"artwork" json:"artwork"`
	Status    ArtworkStatus      `bson:"status" json:"status"`
}

type CallbackDataModel struct {
	ID        primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
	Data      string             `bson:"data" json:"data"`
	CreatedAt primitive.DateTime `bson:"created_at" json:"created_at"`
}

type DeletedModel struct {
	ID        primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
	ArtworkID primitive.ObjectID `bson:"artwork_id" json:"artwork_id"`
	SourceURL string             `bson:"source_url" json:"source_url"`
	DeletedAt primitive.DateTime `bson:"deleted_at" json:"deleted_at"`
}

type FavoriteModel struct {
	ID        primitive.ObjectID `bson:"_id,omitempty"`
	ArtworkID primitive.ObjectID `bson:"artwork_id"`
	UserID    primitive.ObjectID `bson:"user_id"`
}

type LikeModel struct {
	ID        primitive.ObjectID `bson:"_id,omitempty"`
	ArtworkID primitive.ObjectID `bson:"artwork_id"`
	UserID    primitive.ObjectID `bson:"user_id"`
	CreatedAt primitive.DateTime `bson:"created_at"`
}

type Permission string

type Picture struct {
	ID        string `json:"id" bson:"_id"`
	ArtworkID string `json:"artwork_id" bson:"artwork_id"`
	Index     uint   `json:"index" bson:"index"`
	Thumbnail string `json:"thumbnail" bson:"thumbnail"`
	Original  string `json:"original" bson:"original"`

	Width  uint   `json:"width" bson:"width"`
	Height uint   `json:"height" bson:"height"`
	Hash   string `json:"hash" bson:"hash"`

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

type PictureModel struct {
	ID        primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
	ArtworkID primitive.ObjectID `bson:"artwork_id" json:"artwork_id"`
	Index     uint               `bson:"index" json:"index"`
	Thumbnail string             `bson:"thumbnail" json:"thumbnail"`
	Original  string             `bson:"original" json:"original"`
	Width     uint               `bson:"width" json:"width"`
	Height    uint               `bson:"height" json:"height"`
	Hash      string             `bson:"hash" json:"hash"`
	BlurScore float64            `bson:"blur_score" json:"blur_score"`

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

type R18Type int

type Service interface {
	GetArtworkByURL(ctx context.Context, url string, opts ...*AdapterOption) (*Artwork, error)
}

type Source interface {
	Init(service Service)
	FetchNewArtworksWithCh(artworkCh chan *Artwork, limit int) error
	FetchNewArtworks(limit int) ([]*Artwork, error)
	GetArtworkInfo(sourceURL string) (*Artwork, error)
	// GetPictureInfo(sourceURL string, index uint) (*Picture, error)
	GetSourceURLRegexp() *regexp.Regexp
	// CommonSourceURl should has prefix "https://"
	GetCommonSourceURL(url string) string
	// FileName 返回图片的用于存储和传输的文件名
	GetFileName(artwork *Artwork, picture *Picture) string
	Config() *config.SourceCommonConfig
}

type SourceType string

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

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 TagModel struct {
	// example:
	/*
		{
			id: objectId,
			name: "碧蓝档案",
			alias: ["碧蓝档案","蔚蓝档案","blue archive"]
		}
	*/
	ID    primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"`
	Name  string             `bson:"name" json:"name"`
	Alias []string           `bson:"alias" json:"alias"`
}

type TelegramInfo struct {
	PhotoFileID    string `json:"photo_file_id" bson:"photo_file_id"`
	DocumentFileID string `json:"document_file_id" bson:"document_file_id"`
	MessageID      int    `json:"message_id" bson:"message_id"`
	MediaGroupID   string `json:"media_group_id" bson:"media_group_id"`
}

type UnauthUserModel struct {
	ID         primitive.ObjectID `bson:"_id,omitempty"`
	Code       string             `bson:"code"` // 注册验证码
	Username   string             `bson:"username"`
	TelegramID int64              `bson:"telegram_id"`
	Email      string             `bson:"email"`
	AuthMethod AuthMethod         `bson:"auth_method"`
	CreatedAt  primitive.DateTime `bson:"created_at"`
}

type UserModel struct {
	ID         primitive.ObjectID `bson:"_id,omitempty"`
	Username   string             `bson:"username"`
	Password   string             `bson:"password"`
	Email      string             `bson:"email"`
	TelegramID int64              `bson:"telegram_id"`
	Blocked    bool               `bson:"blocked"`
	UpdatedAt  primitive.DateTime `bson:"updated_at"`
	DeletedAt  primitive.DateTime `bson:"deleted_at,omitempty"`

	// Settings
	Settings *UserSettings `bson:"settings" json:"settings"`
}

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