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
@@ -0,0 +1,98 @@
package e2etest
import (
"context"
"time"
"github.com/go-faster/errors"
"github.com/stretchr/testify/require"
"go.uber.org/zap"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message"
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
"go.mau.fi/mautrix-telegram/pkg/gotd/tgerr"
)
// User is a simple user bot.
type User struct {
suite *Suite
text []string
username string
logger *zap.Logger
message chan string
}
// NewUser creates new User bot.
func NewUser(suite *Suite, text []string, username string) User {
return User{
suite: suite,
text: text,
username: username,
logger: suite.logger.Named("terentyev"),
message: make(chan string, 1),
}
}
func (u User) messageHandler(ctx context.Context, entities tg.Entities, update *tg.UpdateNewMessage) error {
if filterMessage(update) {
return nil
}
if m, ok := update.Message.(interface{ GetMessage() string }); ok {
u.logger.Named("dispatcher").
Info("Got new message update", zap.String("message", m.GetMessage()))
}
msg, ok := update.Message.(*tg.Message)
if !ok {
return errors.Errorf("unexpected type %T", update.Message)
}
select {
case u.message <- msg.Message:
return nil
case <-ctx.Done():
return ctx.Err()
}
}
// Run setups and starts user bot.
func (u User) Run(ctx context.Context) error {
dispatcher := tg.NewUpdateDispatcher()
dispatcher.OnNewMessage(u.messageHandler)
client := u.suite.Client(u.logger, dispatcher)
sender := message.NewSender(tg.NewClient(retryInvoker{prev: client}))
return client.Run(ctx, func(ctx context.Context) error {
if err := u.suite.RetryAuthenticate(ctx, client.Auth()); err != nil {
return errors.Wrap(err, "authenticate")
}
peer, err := sender.Resolve(u.username).AsInputPeer(ctx)
if err != nil {
return errors.Wrapf(err, "resolve bot username %q", u.username)
}
for _, line := range u.text {
time.Sleep(2 * time.Second)
_, err = sender.To(peer).Text(ctx, line)
if flood, err := tgerr.FloodWait(ctx, err); err != nil {
if flood {
continue
}
return err
}
select {
case gotMessage := <-u.message:
require.Equal(u.suite.TB, line, gotMessage)
case <-ctx.Done():
return ctx.Err()
}
}
return nil
})
}