mirror of
https://github.com/krau/SaveAny-Bot.git
synced 2026-08-28 19:56:41 +08:00
krau
fa101f4f49
Photos pass size 0 (unknown, not available in InputPhotoFileLocation); the guard then answered the first request at offset 0 with an empty chunk, silently saving every photo as a 0-byte file. Fixes #238
36 lines
1.2 KiB
Go
36 lines
1.2 KiB
Go
package tdler
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gotd/td/telegram/downloader"
|
|
"github.com/gotd/td/tg"
|
|
"github.com/krau/SaveAny-Bot/common/utils/dlutil"
|
|
"github.com/krau/SaveAny-Bot/config"
|
|
"github.com/krau/SaveAny-Bot/pkg/consts/tglimit"
|
|
"github.com/krau/SaveAny-Bot/pkg/tfile"
|
|
)
|
|
|
|
func NewDownloader(file tfile.TGFile) *downloader.Builder {
|
|
return downloader.NewDownloader().WithPartSize(tglimit.MaxPartSize).
|
|
Download(eofAwareClient{Client: file.Dler(), size: file.Size()}, file.Location()).
|
|
WithThreads(dlutil.BestThreads(file.Size(), config.C().Threads))
|
|
}
|
|
|
|
// eofAwareClient answers upload.getFile requests at or past the end of the
|
|
// file with an empty chunk. gotd's downloader is size-unaware: for files
|
|
// whose size is an exact multiple of the part size it issues one final
|
|
// request at offset == size and expects an empty chunk, but Telegram rejects
|
|
// it with 400 OFFSET_INVALID and the whole download fails.
|
|
type eofAwareClient struct {
|
|
downloader.Client
|
|
size int64
|
|
}
|
|
|
|
func (c eofAwareClient) UploadGetFile(ctx context.Context, req *tg.UploadGetFileRequest) (tg.UploadFileClass, error) {
|
|
if c.size > 0 && req.Offset >= c.size {
|
|
return &tg.UploadFile{}, nil
|
|
}
|
|
return c.Client.UploadGetFile(ctx, req)
|
|
}
|