Add new unread_msgs structure to index

Fixes #496

Author: kaustubh-nair

zulipterminal/helper.py

@@ -37,6 +37,8 @@
     'search': Set[int],  # {message_id, ...}
     # Downloaded message data
     'messages': Dict[int, Message],  # message_id: Message
+    # unread message data
+    'unread_msgs': Set[int]  # message_ids
 })
 
 initial_index = Index(
@@ -51,6 +53,7 @@
     topics=defaultdict(list),
     search=set(),
     messages=defaultdict(dict),
+    unread_msgs=set(),
 )
 
 
@@ -350,6 +353,22 @@ def index_messages(messages: List[Any],
     return index
 
 
+def index_unread_messsages(unread_msgs: Dict[str, Any], index: Index) -> Index:
+
+    for pm in unread_msgs['pms']:
+        message_ids = pm['unread_message_ids']
+        index['unread_msgs'].update(message_ids)
+
+    for stream in unread_msgs['streams']:
+        message_ids = stream['unread_message_ids']
+        index['unread_msgs'].update(message_ids)
+
+    for group_pm in unread_msgs['huddles']:
+        message_ids = group_pm['unread_message_ids']
+        index['unread_msgs'].update(message_ids)
+
+    return index
+
 def classify_unread_counts(model: Any) -> UnreadCounts:
     # TODO: support group pms
     unread_msg_counts = model.initial_data['unread_msgs']

zulipterminal/model.py

@@ -22,6 +22,7 @@
     initial_index,
     Message,
     notify,
+    index_unread_messsages,
 )
 from zulipterminal.ui_tools.utils import create_msg_box_list
 
@@ -446,6 +447,7 @@ def _update_initial_data(self) -> None:
             self.user_email = self.initial_data['email']
             self.user_full_name = self.initial_data['full_name']
             self.server_name = self.initial_data['realm_name']
+            self.index = index_unread_messsages(self.initial_data['unread_msgs'], self.index)
         else:
             failures = [name for name, result in results.items() if not result]
             raise ServerConnectionFailure(", ".join(failures))