diff options
Diffstat (limited to 'telegram/client.go')
| -rw-r--r-- | telegram/client.go | 29 |
1 files changed, 19 insertions, 10 deletions
diff --git a/telegram/client.go b/telegram/client.go index 5e9dd9f..133f1e6 100644 --- a/telegram/client.go +++ b/telegram/client.go @@ -5,6 +5,7 @@ import ( "path/filepath" "strconv" "sync" + "time" "dev.narayana.im/narayana/telegabber/config" "dev.narayana.im/narayana/telegabber/persistence" @@ -39,12 +40,12 @@ func stringToLogConstant(c string) int32 { // Client stores the metadata for lazily invoked TDlib instance type Client struct { - client *client.Client - authorizer *clientAuthorizer - parameters *client.TdlibParameters - logVerbosity client.Option - me *client.User - listener *client.Listener + client *client.Client + authorizer *clientAuthorizer + parameters *client.TdlibParameters + options []client.Option + me *client.User + listener *client.Listener xmpp *xmpp.Component jid string @@ -61,9 +62,17 @@ type clientLocks struct { // NewClient instantiates a Telegram App func NewClient(conf config.TelegramConfig, jid string, component *xmpp.Component, session *persistence.Session) (*Client, error) { - logVerbosity := client.WithLogVerbosity(&client.SetLogVerbosityLevelRequest{ + var options []client.Option + + options = append(options, client.WithLogVerbosity(&client.SetLogVerbosityLevelRequest{ NewVerbosityLevel: stringToLogConstant(conf.Loglevel), - }) + })) + + if conf.Tdlib.Client.CatchTimeout != 0 { + options = append(options, client.WithCatchTimeout( + time.Duration(conf.Tdlib.Client.CatchTimeout)*time.Second, + )) + } apiID, err := strconv.Atoi(conf.Tdlib.Client.APIID) if err != nil { @@ -103,7 +112,7 @@ func NewClient(conf config.TelegramConfig, jid string, component *xmpp.Component chats: map[int64]*client.Chat{}, users: map[int32]*client.User{}, }, - logVerbosity: logVerbosity, - locks: clientLocks{}, + options: options, + locks: clientLocks{}, }, nil } |