146 lines
3.0 KiB
Go
146 lines
3.0 KiB
Go
package tfile
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/celestix/gotgproto/functions"
|
|
"github.com/gotd/td/telegram/downloader"
|
|
"github.com/gotd/td/tg"
|
|
)
|
|
|
|
type TGFile interface {
|
|
Location() tg.InputFileLocationClass
|
|
Dler() downloader.Client // witch client to use for downloading
|
|
Size() int64
|
|
Name() string
|
|
SetName(name string)
|
|
}
|
|
|
|
type TGFileMessage interface {
|
|
TGFile
|
|
Message() *tg.Message
|
|
}
|
|
|
|
type tgFile struct {
|
|
location tg.InputFileLocationClass
|
|
size int64
|
|
name string
|
|
message *tg.Message
|
|
dler downloader.Client
|
|
}
|
|
|
|
func (f *tgFile) SetName(name string) {
|
|
f.name = name
|
|
}
|
|
|
|
func (f *tgFile) Location() tg.InputFileLocationClass {
|
|
return f.location
|
|
}
|
|
|
|
func (f *tgFile) Size() int64 {
|
|
return f.size
|
|
}
|
|
|
|
func (f *tgFile) Name() string {
|
|
return f.name
|
|
}
|
|
|
|
func (f *tgFile) Message() *tg.Message {
|
|
return f.message
|
|
}
|
|
|
|
func (f *tgFile) Dler() downloader.Client {
|
|
return f.dler
|
|
}
|
|
|
|
func NewTGFile(
|
|
location tg.InputFileLocationClass,
|
|
dler downloader.Client,
|
|
size int64,
|
|
name string,
|
|
opts ...TGFileOption,
|
|
) TGFile {
|
|
f := &tgFile{
|
|
location: location,
|
|
dler: dler,
|
|
size: size,
|
|
name: name,
|
|
}
|
|
for _, opt := range opts {
|
|
opt(f)
|
|
}
|
|
return f
|
|
}
|
|
|
|
func FromMedia(media tg.MessageMediaClass, client downloader.Client, opts ...TGFileOption) (TGFile, error) {
|
|
switch m := media.(type) {
|
|
case *tg.MessageMediaDocument:
|
|
document, ok := m.Document.AsNotEmpty()
|
|
if !ok {
|
|
return nil, errors.New("document is empty")
|
|
}
|
|
fileName := ""
|
|
for _, attribute := range document.Attributes {
|
|
if name, ok := attribute.(*tg.DocumentAttributeFilename); ok {
|
|
fileName = name.GetFileName()
|
|
break
|
|
}
|
|
}
|
|
file := NewTGFile(
|
|
document.AsInputDocumentFileLocation(),
|
|
client,
|
|
document.Size,
|
|
fileName,
|
|
opts...,
|
|
)
|
|
return file, nil
|
|
case *tg.MessageMediaPhoto:
|
|
photo, ok := m.Photo.AsNotEmpty()
|
|
if !ok {
|
|
return nil, errors.New("photo is empty")
|
|
}
|
|
sizes := photo.Sizes
|
|
if len(sizes) == 0 {
|
|
return nil, errors.New("photo sizes are empty")
|
|
}
|
|
photoSize := sizes[len(sizes)-1]
|
|
size, ok := photoSize.AsNotEmpty()
|
|
if !ok {
|
|
return nil, errors.New("photo size is empty")
|
|
}
|
|
location := new(tg.InputPhotoFileLocation)
|
|
location.ID = photo.GetID()
|
|
location.AccessHash = photo.GetAccessHash()
|
|
location.FileReference = photo.GetFileReference()
|
|
location.ThumbSize = size.GetType()
|
|
fileName, err := functions.GetMediaFileName(m)
|
|
if err != nil {
|
|
fileName = fmt.Sprintf("photo_%d.png", photo.GetID())
|
|
}
|
|
file := NewTGFile(
|
|
location,
|
|
client,
|
|
0, // Photo size is not available in InputPhotoFileLocation
|
|
fileName,
|
|
opts...,
|
|
)
|
|
return file, nil
|
|
}
|
|
return nil, fmt.Errorf("unsupported media type: %T", media)
|
|
}
|
|
|
|
func FromMediaMessage(media tg.MessageMediaClass, client downloader.Client, msg *tg.Message, opts ...TGFileOption) (TGFileMessage, error) {
|
|
file, err := FromMedia(media, client, opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &tgFile{
|
|
location: file.Location(),
|
|
dler: file.Dler(),
|
|
size: file.Size(),
|
|
name: file.Name(),
|
|
message: msg,
|
|
}, nil
|
|
}
|