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, 6 insertions, 1 deletions

diff --git a/src/roster.py b/src/roster.py
index 7866655f..ba7da63e 100644
--- a/src/roster.py
+++ b/src/roster.py
@@ -149,9 +149,14 @@ class Roster(object):
             contact = self.get_and_set(key)
             if key != self.jid and (contact and self.exists(contact)):
                 l.append(key)
-        self.length = len(l)
+        self.update_size(l)
         return l
 
+    def update_size(self, jids=None):
+        if jids is None:
+            jids = self.jids()
+        self.length = len(jids)
+
     def get_contacts(self):
         """
         Return a list of all the contacts