move gotd fork into repo. (#111)

- update to latest telegram layer
- remove some references to fields in tg.Entities that don't exist in
the schema
- originally added here:
https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e
  - referenced here
-
https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3
-
https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
This commit is contained in:
Adam Van Ymeren
2025-06-27 20:03:37 -07:00
committed by GitHub
parent 0952df0244
commit 7a04f298d2
19264 changed files with 1539697 additions and 84 deletions
+2
View File
@@ -0,0 +1,2 @@
// Package unpack contains some updates result unpacking helpers.
package unpack
@@ -0,0 +1,80 @@
package unpack
import (
"github.com/go-faster/errors"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/internal/upconv"
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
)
func extractClass(u tg.UpdateClass) (tg.MessageClass, bool) {
switch v := u.(type) {
case *tg.UpdateNewMessage:
return v.Message, true
case *tg.UpdateNewChannelMessage:
return v.Message, true
default:
return nil, false
}
}
// MessageClass tries to unpack sent message and returns it as MessageClass.
func MessageClass(u tg.UpdatesClass, err error) (tg.MessageClass, error) {
if err != nil {
return nil, err
}
var updates []tg.UpdateClass
switch v := u.(type) {
case *tg.UpdateShortMessage:
short := upconv.ShortMessage(v)
updates = []tg.UpdateClass{short.Update}
case *tg.UpdateShortChatMessage:
short := upconv.ShortChatMessage(v)
updates = []tg.UpdateClass{short.Update}
case *tg.UpdateShortSentMessage:
short := upconv.ShortSentMessage(v)
updates = []tg.UpdateClass{short.Update}
case *tg.UpdateShort:
updates = []tg.UpdateClass{v.Update}
case *tg.UpdatesCombined:
updates = v.GetUpdates()
case *tg.Updates:
updates = v.GetUpdates()
default:
return nil, errors.Errorf("unexpected type %T", u)
}
for _, update := range updates {
if msg, ok := extractClass(update); ok {
return msg, nil
}
}
return nil, errors.Errorf("bad updates result %+v", updates)
}
// Message tries to unpack sent message and returns it as Message.
func Message(u tg.UpdatesClass, err error) (*tg.Message, error) {
msg, err := MessageClass(u, err)
if err != nil {
return nil, err
}
m, ok := msg.(*tg.Message)
if !ok {
return nil, errors.Errorf("unexpected type %T", msg)
}
return m, nil
}
// MessageID tries to unpack sent message and returns message id.
func MessageID(u tg.UpdatesClass, err error) (int, error) {
msg, err := MessageClass(u, err)
if err != nil {
return 0, err
}
return msg.GetID(), nil
}
@@ -0,0 +1,44 @@
package unpack_test
import (
"context"
"fmt"
"os"
"os/signal"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/unpack"
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
)
func unpackMessage(ctx context.Context) error {
client, err := telegram.ClientFromEnvironment(telegram.Options{})
if err != nil {
return err
}
return client.Run(ctx, func(ctx context.Context) error {
sender := message.NewSender(tg.NewClient(client))
msg, err := unpack.Message(sender.Resolve("@durovschat").Dice(ctx))
// Sends dice "🎲" to the @durovschat.
if err != nil {
return err
}
fmt.Println("Sent message ID:", msg.ID)
return nil
})
}
func ExampleMessage() {
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
defer cancel()
if err := unpackMessage(ctx); err != nil {
_, _ = fmt.Fprintf(os.Stderr, "%+v\n", err)
os.Exit(2)
}
}
@@ -0,0 +1,111 @@
package unpack
import (
"testing"
"github.com/go-faster/errors"
"github.com/stretchr/testify/require"
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
)
func TestMessage(t *testing.T) {
testMsg := &tg.Message{
ID: 10,
Message: "Golang is always going to do some approximation of the right thing.",
}
testErr := errors.New("женой накормили толпу")
type args struct {
u tg.UpdatesClass
err error
}
tests := []struct {
name string
input args
output *tg.Message
isErr bool
}{
{"Good", args{
u: &tg.Updates{
Updates: []tg.UpdateClass{
&tg.UpdateNewMessage{
Message: testMsg,
},
},
},
err: nil,
}, testMsg, false},
{"GoodChannel", args{
u: &tg.Updates{
Updates: []tg.UpdateClass{
&tg.UpdateNewChannelMessage{
Message: testMsg,
},
},
},
err: nil,
}, testMsg, false},
{"Short", args{
u: &tg.UpdateShort{
Update: &tg.UpdateNewMessage{
Message: testMsg,
},
},
err: nil,
}, testMsg, false},
{"ShortSent", args{
u: &tg.UpdateShortSentMessage{
Out: testMsg.Out,
ID: testMsg.ID,
Date: testMsg.Date,
Media: testMsg.Media,
Entities: testMsg.Entities,
TTLPeriod: testMsg.TTLPeriod,
},
err: nil,
}, testMsg, false},
{"ShortChat", args{
u: &tg.UpdateShortChatMessage{
Out: testMsg.Out,
Mentioned: testMsg.Mentioned,
MediaUnread: testMsg.MediaUnread,
Silent: testMsg.Silent,
ID: testMsg.ID,
Message: testMsg.Message,
Date: testMsg.Date,
FwdFrom: testMsg.FwdFrom,
ViaBotID: testMsg.ViaBotID,
ReplyTo: testMsg.ReplyTo,
Entities: testMsg.Entities,
TTLPeriod: testMsg.TTLPeriod,
},
err: nil,
}, testMsg, false},
{"ExternalError", args{
u: nil,
err: testErr,
}, nil, true},
{"BadUnpack", args{
u: &tg.UpdateShort{
Update: &tg.UpdateConfig{},
},
err: nil,
}, nil, true},
}
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
a := require.New(t)
r, err := Message(test.input.u, test.input.err)
if test.isErr {
a.Error(err)
return
}
a.NoError(err)
a.Equal(testMsg.ID, r.ID)
})
}
}