iartamonov/mautrix-telegram
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

msgconv: start experimenting with direct download URL

Browse Source 
Signed-off-by: Sumner Evans <sumner.evans@automattic.com>
This commit is contained in:
Sumner Evans
2024-05-15 16:53:57 -06:00
parent 43212ad8db
commit 72fc81b239
1 changed files with 22 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
msgconv/from-tg.go
+22 -23
View File
@@ -1,14 +1,13 @@
package msgconv package msgconv
import ( import (
"bytes"
"context" "context"
"encoding/base64"
"fmt" "fmt"
"os"
"github.com/gotd/td/telegram/downloader"
"github.com/gotd/td/tg" "github.com/gotd/td/tg"
"maunium.net/go/mautrix/event" "maunium.net/go/mautrix/event"
"maunium.net/go/mautrix/id"
) )
type ConvertedMessage struct { type ConvertedMessage struct {
@@ -46,12 +45,6 @@ func getLargestPhotoSize(sizes []tg.PhotoSizeClass) (largest tg.PhotoSizeClass)
return return
} }
func (mc *MessageConverter) downloadFile(ctx context.Context, file tg.InputFileLocationClass) ([]byte, error) {
var buf bytes.Buffer
_, err := downloader.NewDownloader().Download(mc.Client.API(), file).Stream(ctx, &buf)
return buf.Bytes(), err
}
func (mc *MessageConverter) ToMatrix(ctx context.Context, msg tg.MessageClass) *ConvertedMessage { func (mc *MessageConverter) ToMatrix(ctx context.Context, msg tg.MessageClass) *ConvertedMessage {
log := mc.getLogger(ctx).With().Str("action", "to_matrix").Logger() log := mc.getLogger(ctx).With().Str("action", "to_matrix").Logger()
cm := &ConvertedMessage{ cm := &ConvertedMessage{
@@ -85,21 +78,27 @@ func (mc *MessageConverter) ToMatrix(ctx context.Context, msg tg.MessageClass) *
fmt.Printf("photo: %v\n", photo) fmt.Printf("photo: %v\n", photo)
largest := getLargestPhotoSize(photo.GetSizes()) largest := getLargestPhotoSize(photo.GetSizes())
file := tg.InputPhotoFileLocation{ // file := tg.InputPhotoFileLocation{
ID: photo.GetID(), // ID: photo.GetID(),
AccessHash: photo.GetAccessHash(), // AccessHash: photo.GetAccessHash(),
FileReference: photo.GetFileReference(), // FileReference: photo.GetFileReference(),
ThumbSize: largest.GetType(), // ThumbSize: largest.GetType(),
} // }
data, err := mc.downloadFile(ctx, &file) mxc := id.ContentURIString(
if err != nil { fmt.Sprintf("mxc://telegram.sumner.user.beeper.com/p.i%d.a%d.f%s.t%s", photo.GetID(), photo.GetAccessHash(), base64.RawURLEncoding.EncodeToString(photo.GetFileReference()), largest.GetType()),
panic(err) )
}
err = os.WriteFile("/home/sumner/tmp/test.jpg", data, 0644) fmt.Printf("%s\n", mxc)
if err != nil {
panic(err) // data, err := mc.downloadFile(ctx, &file)
} // if err != nil {
// panic(err)
// }
// err = os.WriteFile("/home/sumner/tmp/test.jpg", data, 0644)
// if err != nil {
// panic(err)
// }
default: default:
log.Error().Type("msg", msg).Msg("Unhandled photo type") log.Error().Type("msg", msg).Msg("Unhandled photo type")
} }