341 lines
13 KiB
Go
341 lines
13 KiB
Go
package domain
|
||
|
||
// 本文件定义媒体相关的业务值对象（文档、照片、贴纸集、可用 reaction、消息媒体）。
|
||
// 这些类型完全不依赖 tg.*；rpc 层负责 domain↔tg 转换。
|
||
//
|
||
// 字段带 json tag 是为了 store 层可直接 json.Marshal 落 JSONB（消息 media 快照、
|
||
// 文档/照片元数据）。它们是协议无关的纯数据，不是 tg 生成类型。
|
||
|
||
// MediaBackend 标识 blob 字节实际存放后端。第一阶段只有本地磁盘。
|
||
type MediaBackend string
|
||
|
||
const (
|
||
// MediaBackendLocalFS 表示 blob 字节存在本地磁盘（object_key 为相对路径）。
|
||
MediaBackendLocalFS MediaBackend = "localfs"
|
||
)
|
||
|
||
// FileBlob 是一个可下载的二进制对象的索引项：location_key → 后端/对象键/大小/mime。
|
||
// 真正的字节由 blob backend 按 ObjectKey 读写；本结构只描述定位与元数据。
|
||
type FileBlob struct {
|
||
// LocationKey 是稳定的逻辑定位键，由 getFile 的 InputFileLocation 推导：
|
||
// doc:<id> 文档主体
|
||
// doc:<id>:<type> 文档缩略图（PhotoSize type）
|
||
// photo:<id>:<type> 照片某尺寸
|
||
LocationKey string `json:"location_key"`
|
||
Backend MediaBackend `json:"backend"`
|
||
ObjectKey string `json:"object_key"`
|
||
Size int64 `json:"size"`
|
||
SHA256 []byte `json:"sha256,omitempty"`
|
||
MimeType string `json:"mime_type,omitempty"`
|
||
}
|
||
|
||
// UploadPart 是 upload.saveFilePart/saveBigFilePart 累积的一个分片（落 PG，组装后清理）。
|
||
type UploadPart struct {
|
||
OwnerUserID int64
|
||
FileID int64
|
||
Part int
|
||
TotalParts int // big file 已知总数；small file 为 0
|
||
Big bool
|
||
Bytes []byte
|
||
}
|
||
|
||
// UploadedFileRef 引用一个客户端已通过 upload.saveFilePart(Big) 上传完毕的文件。
|
||
// rpc 层从 tg.InputFile/InputFileBig 转换得到；files 服务据此组装 blob。
|
||
type UploadedFileRef struct {
|
||
OwnerUserID int64
|
||
FileID int64
|
||
Parts int
|
||
Name string
|
||
Big bool
|
||
MD5 string // small file 客户端 md5_checksum（hex），可校验；big file 为空
|
||
}
|
||
|
||
// DocumentSpec 描述从上传文件创建 Document 的元数据（来自 InputMediaUploadedDocument）。
|
||
type DocumentSpec struct {
|
||
MimeType string
|
||
Attributes []DocumentAttribute
|
||
Thumb *UploadedFileRef // 可选缩略图上传，生成 doc:<id>:m
|
||
ForceFile bool
|
||
}
|
||
|
||
// FileDownloadRequest 是 upload.getFile 解析后的下载请求；
|
||
// LocationKey 由 rpc 层从 tg.InputFileLocation 推导（doc:<id> / photo:<id>:<type> 等）。
|
||
type FileDownloadRequest struct {
|
||
LocationKey string
|
||
Offset int64
|
||
Limit int
|
||
}
|
||
|
||
// FileChunk 是 upload.getFile 返回的一段内容。
|
||
type FileChunk struct {
|
||
Bytes []byte
|
||
MimeType string
|
||
Total int64
|
||
}
|
||
|
||
// PhotoSizeKind 标识 PhotoSize 的 TL 变体。
|
||
type PhotoSizeKind string
|
||
|
||
const (
|
||
// PhotoSizeKindDefault → photoSize（可下载，type/w/h/size）。
|
||
PhotoSizeKindDefault PhotoSizeKind = "size"
|
||
// PhotoSizeKindStripped → photoStrippedSize（内联字节，秒开模糊图）。
|
||
PhotoSizeKindStripped PhotoSizeKind = "stripped"
|
||
// PhotoSizeKindCached → photoCachedSize（内联字节 + w/h）。
|
||
PhotoSizeKindCached PhotoSizeKind = "cached"
|
||
// PhotoSizeKindPath → photoPathSize（内联 svg path 字节，矢量占位）。
|
||
PhotoSizeKindPath PhotoSizeKind = "path"
|
||
// PhotoSizeKindProgressive → photoSizeProgressive（渐进式 jpeg 多段大小）。
|
||
PhotoSizeKindProgressive PhotoSizeKind = "progressive"
|
||
)
|
||
|
||
// PhotoSize 描述照片/缩略图的一种渲染尺寸。
|
||
type PhotoSize struct {
|
||
Kind PhotoSizeKind `json:"kind"`
|
||
Type string `json:"type"`
|
||
W int `json:"w,omitempty"`
|
||
H int `json:"h,omitempty"`
|
||
Size int `json:"size,omitempty"`
|
||
Bytes []byte `json:"bytes,omitempty"` // stripped/cached/path 内联内容
|
||
Sizes []int `json:"sizes,omitempty"` // progressive
|
||
}
|
||
|
||
// Downloadable 表示该尺寸需要客户端通过 upload.getFile 拉取（而非内联字节）。
|
||
func (s PhotoSize) Downloadable() bool {
|
||
return s.Kind == PhotoSizeKindDefault || s.Kind == PhotoSizeKindProgressive
|
||
}
|
||
|
||
// DocumentAttributeKind 标识 TL DocumentAttribute 变体。
|
||
type DocumentAttributeKind string
|
||
|
||
const (
|
||
DocAttrImageSize DocumentAttributeKind = "image_size"
|
||
DocAttrAnimated DocumentAttributeKind = "animated"
|
||
DocAttrSticker DocumentAttributeKind = "sticker"
|
||
DocAttrVideo DocumentAttributeKind = "video"
|
||
DocAttrAudio DocumentAttributeKind = "audio"
|
||
DocAttrFilename DocumentAttributeKind = "filename"
|
||
DocAttrCustomEmoji DocumentAttributeKind = "custom_emoji"
|
||
)
|
||
|
||
// DocumentAttribute 是主路径用到的 TL DocumentAttribute 变体的并集。
|
||
type DocumentAttribute struct {
|
||
Kind DocumentAttributeKind `json:"kind"`
|
||
|
||
// image_size / video / sticker box
|
||
W int `json:"w,omitempty"`
|
||
H int `json:"h,omitempty"`
|
||
|
||
// sticker / custom_emoji
|
||
Alt string `json:"alt,omitempty"`
|
||
Mask bool `json:"mask,omitempty"`
|
||
StickerSetID int64 `json:"sticker_set_id,omitempty"`
|
||
StickerSetAccessHash int64 `json:"sticker_set_access_hash,omitempty"`
|
||
Free bool `json:"free,omitempty"` // custom_emoji
|
||
TextColor bool `json:"text_color,omitempty"` // custom_emoji
|
||
|
||
// video
|
||
Duration float64 `json:"duration,omitempty"`
|
||
RoundMessage bool `json:"round_message,omitempty"`
|
||
SupportsStreaming bool `json:"supports_streaming,omitempty"`
|
||
|
||
// audio
|
||
AudioDuration int `json:"audio_duration,omitempty"`
|
||
Voice bool `json:"voice,omitempty"`
|
||
Title string `json:"title,omitempty"`
|
||
Performer string `json:"performer,omitempty"`
|
||
Waveform []byte `json:"waveform,omitempty"`
|
||
|
||
// filename
|
||
FileName string `json:"file_name,omitempty"`
|
||
}
|
||
|
||
// Document 是已存储的 Telegram 文档（贴纸、gif、文件、视频、音频、自定义 emoji……）。
|
||
type Document struct {
|
||
ID int64 `json:"id"`
|
||
AccessHash int64 `json:"access_hash"`
|
||
FileReference []byte `json:"file_reference,omitempty"`
|
||
Date int `json:"date,omitempty"`
|
||
MimeType string `json:"mime_type,omitempty"`
|
||
Size int64 `json:"size,omitempty"`
|
||
DCID int `json:"dc_id,omitempty"`
|
||
Attributes []DocumentAttribute `json:"attributes,omitempty"`
|
||
Thumbs []PhotoSize `json:"thumbs,omitempty"`
|
||
}
|
||
|
||
// StickerSetRef 返回该文档归属的贴纸集引用（若有 sticker/custom_emoji 属性）。
|
||
func (d Document) StickerSetRef() (id, accessHash int64, ok bool) {
|
||
for _, attr := range d.Attributes {
|
||
if attr.Kind == DocAttrSticker || attr.Kind == DocAttrCustomEmoji {
|
||
if attr.StickerSetID != 0 {
|
||
return attr.StickerSetID, attr.StickerSetAccessHash, true
|
||
}
|
||
}
|
||
}
|
||
return 0, 0, false
|
||
}
|
||
|
||
// Photo 是已存储的 Telegram 照片（头像或图片消息）。
|
||
type Photo struct {
|
||
ID int64 `json:"id"`
|
||
AccessHash int64 `json:"access_hash"`
|
||
FileReference []byte `json:"file_reference,omitempty"`
|
||
Date int `json:"date,omitempty"`
|
||
DCID int `json:"dc_id,omitempty"`
|
||
HasStickers bool `json:"has_stickers,omitempty"`
|
||
Sizes []PhotoSize `json:"sizes,omitempty"`
|
||
}
|
||
|
||
// MessageMediaKind 枚举消息可挂载的媒体载荷。
|
||
type MessageMediaKind string
|
||
|
||
const (
|
||
MessageMediaKindNone MessageMediaKind = ""
|
||
MessageMediaKindPhoto MessageMediaKind = "photo"
|
||
MessageMediaKindDocument MessageMediaKind = "document"
|
||
)
|
||
|
||
// MessageMedia 是一条消息媒体载荷的业务表示（落库为消息行上的 JSONB 快照）。
|
||
type MessageMedia struct {
|
||
Kind MessageMediaKind `json:"kind"`
|
||
Photo *Photo `json:"photo,omitempty"`
|
||
Document *Document `json:"document,omitempty"`
|
||
Spoiler bool `json:"spoiler,omitempty"`
|
||
TTLSeconds int `json:"ttl_seconds,omitempty"`
|
||
Nopremium bool `json:"nopremium,omitempty"`
|
||
Voice bool `json:"voice,omitempty"`
|
||
Round bool `json:"round,omitempty"`
|
||
Video bool `json:"video,omitempty"`
|
||
}
|
||
|
||
// IsZero 表示无媒体（用于落库时跳过空快照、转换时回退 MessageMediaEmpty）。
|
||
func (m *MessageMedia) IsZero() bool {
|
||
return m == nil || m.Kind == MessageMediaKindNone
|
||
}
|
||
|
||
// StickerPack 是 emoji→文档 id 的映射条目（messages.stickerSet.packs）。
|
||
type StickerPack struct {
|
||
Emoticon string `json:"emoticon"`
|
||
DocumentIDs []int64 `json:"document_ids"`
|
||
}
|
||
|
||
// StickerSetKind 区分贴纸集用途（影响 getAllStickers / getEmojiStickers 归类）。
|
||
type StickerSetKind string
|
||
|
||
const (
|
||
StickerSetKindStickers StickerSetKind = "stickers"
|
||
StickerSetKindEmoji StickerSetKind = "emoji"
|
||
StickerSetKindMasks StickerSetKind = "masks"
|
||
// StickerSetKindSystem 是 TDesktop 通过 InputStickerSetDice/AnimatedEmoji 等系统集请求的内置集。
|
||
StickerSetKindSystem StickerSetKind = "system"
|
||
)
|
||
|
||
// StickerSet 是贴纸/自定义 emoji 集的元数据 + 有序文档 id。
|
||
type StickerSet struct {
|
||
ID int64 `json:"id"`
|
||
AccessHash int64 `json:"access_hash"`
|
||
ShortName string `json:"short_name"`
|
||
Title string `json:"title"`
|
||
Count int `json:"count"`
|
||
Hash int `json:"hash"`
|
||
Kind StickerSetKind `json:"set_kind"`
|
||
Official bool `json:"official,omitempty"`
|
||
Animated bool `json:"animated,omitempty"`
|
||
Videos bool `json:"videos,omitempty"`
|
||
Emojis bool `json:"emojis,omitempty"`
|
||
Masks bool `json:"masks,omitempty"`
|
||
Installed bool `json:"installed,omitempty"`
|
||
Archived bool `json:"archived,omitempty"`
|
||
InstalledDate int `json:"installed_date,omitempty"`
|
||
ThumbDocumentID int64 `json:"thumb_document_id,omitempty"`
|
||
Thumbs []PhotoSize `json:"thumbs,omitempty"`
|
||
ThumbDCID int `json:"thumb_dc_id,omitempty"`
|
||
ThumbVersion int `json:"thumb_version,omitempty"`
|
||
DocumentIDs []int64 `json:"document_ids,omitempty"`
|
||
Packs []StickerPack `json:"packs,omitempty"`
|
||
SortOrder int `json:"sort_order,omitempty"`
|
||
// SystemKey 是 TDesktop 系统集的稳定标识（如 "animated_emoji"、"dice:🎲"），用于 InputStickerSet* 路由。
|
||
SystemKey string `json:"system_key,omitempty"`
|
||
}
|
||
|
||
// ProfilePhotoKind distinguishes a user's real profile photo from the fallback
|
||
// public photo shown when privacy hides the real one.
|
||
type ProfilePhotoKind string
|
||
|
||
const (
|
||
ProfilePhotoKindProfile ProfilePhotoKind = "profile"
|
||
ProfilePhotoKindFallback ProfilePhotoKind = "fallback"
|
||
)
|
||
|
||
// ProfilePhotoRef 是渲染头像所需的最小信息（当前 profile/fallback/personal photo）。
|
||
type ProfilePhotoRef struct {
|
||
PhotoID int64
|
||
DCID int
|
||
Stripped []byte // photoStrippedSize 内联缩略图，可空
|
||
Personal bool // true 表示 viewer 私有联系人头像
|
||
}
|
||
|
||
// StrippedFromSizes 从照片尺寸列表里取出 stripped 缩略图字节（用于 UserProfilePhoto/ChatPhoto 占位）。
|
||
func StrippedFromSizes(sizes []PhotoSize) []byte {
|
||
for _, s := range sizes {
|
||
if s.Kind == PhotoSizeKindStripped {
|
||
return s.Bytes
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// StickerSetRefKind 标识 InputStickerSet 的解析方式。
|
||
type StickerSetRefKind string
|
||
|
||
const (
|
||
StickerSetRefByID StickerSetRefKind = "id"
|
||
StickerSetRefByShortName StickerSetRefKind = "short_name"
|
||
StickerSetRefBySystem StickerSetRefKind = "system"
|
||
)
|
||
|
||
// StickerSetRef 是 rpc 层从 tg.InputStickerSet 转换得到的贴纸集引用。
|
||
type StickerSetRef struct {
|
||
Kind StickerSetRefKind
|
||
ID int64
|
||
AccessHash int64
|
||
ShortName string
|
||
SystemKey string
|
||
}
|
||
|
||
// AvailableReaction 描述 messages.getAvailableReactions 的一项（真实资源由文档 id 引用）。
|
||
type AvailableReaction struct {
|
||
Reaction string `json:"reaction"`
|
||
Title string `json:"title"`
|
||
Inactive bool `json:"inactive,omitempty"`
|
||
Premium bool `json:"premium,omitempty"`
|
||
StaticIconID int64 `json:"static_icon_id,omitempty"`
|
||
AppearAnimationID int64 `json:"appear_animation_id,omitempty"`
|
||
SelectAnimationID int64 `json:"select_animation_id,omitempty"`
|
||
ActivateAnimationID int64 `json:"activate_animation_id,omitempty"`
|
||
EffectAnimationID int64 `json:"effect_animation_id,omitempty"`
|
||
AroundAnimationID int64 `json:"around_animation_id,omitempty"`
|
||
CenterIconID int64 `json:"center_icon_id,omitempty"`
|
||
Order int `json:"order,omitempty"`
|
||
}
|
||
|
||
// DocumentIDs 收集该 reaction 引用的全部文档 id（去零去重，便于批量加载）。
|
||
func (r AvailableReaction) DocumentIDs() []int64 {
|
||
raw := []int64{
|
||
r.StaticIconID, r.AppearAnimationID, r.SelectAnimationID,
|
||
r.ActivateAnimationID, r.EffectAnimationID, r.AroundAnimationID, r.CenterIconID,
|
||
}
|
||
out := make([]int64, 0, len(raw))
|
||
seen := make(map[int64]struct{}, len(raw))
|
||
for _, id := range raw {
|
||
if id == 0 {
|
||
continue
|
||
}
|
||
if _, ok := seen[id]; ok {
|
||
continue
|
||
}
|
||
seen[id] = struct{}{}
|
||
out = append(out, id)
|
||
}
|
||
return out
|
||
}
|