Files
mautrix-telegram/pkg/gotd/telegram/peers/manager_example_test.go
T
2025-06-27 20:03:37 -07:00

62 lines
1.4 KiB
Go

package peers_test
import (
"context"
"github.com/go-faster/errors"
"go.uber.org/zap"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/peers"
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/updates"
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
)
func ExampleManager() {
logger := zap.NewExample()
var (
dispatcher = tg.NewUpdateDispatcher()
h telegram.UpdateHandler
)
client, err := telegram.ClientFromEnvironment(telegram.Options{
Logger: logger.Named("client"),
UpdateHandler: telegram.UpdateHandlerFunc(func(ctx context.Context, u tg.UpdatesClass) error {
return h.Handle(ctx, u)
}),
})
if err != nil {
panic(err)
}
peerManager := peers.Options{
Logger: logger,
}.Build(client.API())
gaps := updates.New(updates.Config{
Handler: dispatcher,
AccessHasher: peerManager,
Logger: logger.Named("gaps"),
})
h = peerManager.UpdateHook(gaps)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
if err := client.Run(ctx, func(ctx context.Context) error {
if err := peerManager.Init(ctx); err != nil {
return err
}
u, err := peerManager.Self(ctx)
if err != nil {
return err
}
_, isBot := u.ToBot()
if err := gaps.Run(ctx, client.API(), u.ID(), updates.AuthOptions{IsBot: isBot}); err != nil {
return errors.Wrap(err, "gaps")
}
return nil
}); err != nil {
panic(err)
}
}