Adam Van Ymeren
7a04f298d2
- 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
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package tgtest
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/clock"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/crypto"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/mt"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/mtproto"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/proto"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tmap"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/transport"
|
|
)
|
|
|
|
// ServerOptions of Server.
|
|
type ServerOptions struct {
|
|
// DC ID of this server. Default to 2.
|
|
DC int
|
|
// Random is random source. Defaults to rand.Reader.
|
|
Random io.Reader
|
|
// Logger is instance of zap.Logger. No logs by default.
|
|
Logger *zap.Logger
|
|
// Codec constructor.
|
|
// Defaults to nil (underlying transport server detects protocol automatically).
|
|
Codec func() transport.Codec
|
|
// Clock to use. Defaults to clock.System.
|
|
Clock clock.Clock
|
|
// MessageID generator. Creates a new proto.MessageIDGen by default.
|
|
// Clock will be used for creation.
|
|
MessageID mtproto.MessageIDSource
|
|
// Types map, used in verbose logging of incoming message.
|
|
Types *tmap.Map
|
|
// ReadTimeout is a connection read timeout.
|
|
ReadTimeout time.Duration
|
|
// ReadTimeout is a connection write timeout.
|
|
WriteTimeout time.Duration
|
|
}
|
|
|
|
func (opt *ServerOptions) setDefaults() {
|
|
if opt.DC == 0 {
|
|
opt.DC = 2
|
|
}
|
|
if opt.Random == nil {
|
|
opt.Random = crypto.DefaultRand()
|
|
}
|
|
if opt.Logger == nil {
|
|
opt.Logger = zap.NewNop()
|
|
}
|
|
|
|
// Ignore opt.Codec, will be handled by transport.NewCustomServer.
|
|
|
|
if opt.Clock == nil {
|
|
opt.Clock = clock.System
|
|
}
|
|
if opt.MessageID == nil {
|
|
opt.MessageID = proto.NewMessageIDGen(opt.Clock.Now)
|
|
}
|
|
if opt.Types == nil {
|
|
opt.Types = tmap.New(
|
|
tg.TypesMap(),
|
|
mt.TypesMap(),
|
|
proto.TypesMap(),
|
|
)
|
|
}
|
|
if opt.ReadTimeout == 0 {
|
|
opt.ReadTimeout = 30 * time.Second
|
|
}
|
|
if opt.WriteTimeout == 0 {
|
|
opt.WriteTimeout = 30 * time.Second
|
|
}
|
|
}
|