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:
@@ -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
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user