Handle unsubscribe presence too, improve logging

This commit is contained in:
bodqhrohro 2019-11-20 23:45:30 +02:00
parent f0c0d0ba94
commit c0c21a35a4
2 changed files with 9 additions and 2 deletions

View file

@ -3,6 +3,7 @@ package telegram
import (
"github.com/pkg/errors"
log "github.com/sirupsen/logrus"
"github.com/zelenin/go-tdlib/client"
)
@ -12,6 +13,8 @@ func (c *Client) Connect() error {
return nil
}
log.Warn("Connecting to Telegram network...")
authorizer := client.ClientAuthorizer()
go func() {
for {
@ -48,6 +51,10 @@ func (c *Client) Disconnect() {
return
}
log.Warn("Disconnecting from Telegram network...")
// TODO: send unavailable presence to cached chats
c.client.Stop()
c.online = false
}

View file

@ -96,7 +96,7 @@ func handlePresence(s xmpp.Sender, p stanza.Presence) {
}
switch p.Type {
case "unsubscribed":
case "unsubscribed", "unsubscribe":
session.Disconnect()
delete(sessions, bareFromJid)
case "unavailable", "error":
@ -105,7 +105,7 @@ func handlePresence(s xmpp.Sender, p stanza.Presence) {
// due to the weird implentation of go-tdlib wrapper, it won't
// return the client instance until successful authorization
go func() {
session.Connect()
err = session.Connect()
if err != nil {
log.Error(errors.Wrap(err, "TDlib connection failure"))
}