Fix various roster count issues

disconnections not clearing the number of connected contacts, leading to
an ever-increasing count, and roster count not always being setup

1 files changed, 2 insertions, 0 deletions

diff --git a/src/core/handlers.py b/src/core/handlers.py
index 6572bca4..e17476a4 100644
--- a/src/core/handlers.py
+++ b/src/core/handlers.py
@@ -748,6 +748,7 @@ def on_roster_update(self, iq):
                 del roster[jid]
             else:
                 roster.update_contact_groups(jid)
+    roster.update_size()
     if isinstance(self.current_tab(), tabs.RosterInfoTab):
         self.refresh_window()
 
@@ -936,6 +937,7 @@ def on_disconnected(self, event):
     """
     When we are disconnected from remote server
     """
+    roster.connected = 0
     # Stop the ping plugin. It would try to send stanza on regular basis
     self.xmpp.plugin['xep_0199'].disable_keepalive()
     roster.modified()