base backend for gc

toxygen/profile.py

@@ -16,6 +16,7 @@ import basecontact
 import items_factory
 import cv2
 import threading
+from group_chat import *
 
 
 class Profile(basecontact.BaseContact, Singleton):
@@ -177,7 +178,7 @@ class Profile(basecontact.BaseContact, Singleton):
     # -----------------------------------------------------------------------------------------------------------------
 
     def get_friend_by_number(self, num):
-        return list(filter(lambda x: x.number == num, self._contacts))[0]
+        return list(filter(lambda x: x.number == num and type(x) is Friend, self._contacts))[0]
 
     def get_friend(self, num):
         if num < 0 or num >= len(self._contacts):
@@ -636,6 +637,16 @@ class Profile(basecontact.BaseContact, Singleton):
         return self._factory.message_item(text, time, name, owner != MESSAGE_OWNER['NOT_SENT'],
                                           message_type, append, pixmap)
 
+    def create_gc_message_item(self, text, time, owner, name, message_type, append=True):
+        pixmap = None
+        if self._show_avatars:
+            if owner == MESSAGE_OWNER['FRIEND']:
+                pixmap = self.get_curr_friend().get_pixmap()
+            else:
+                pixmap = self.get_pixmap()
+        return self._factory.message_item(text, time, name, True,
+                                          message_type, append, pixmap)
+
     def create_file_transfer_item(self, tm, append=True):
         data = list(tm.get_data())
         data[3] = self.get_friend_by_number(data[4]).name if data[3] else self._name
@@ -1265,6 +1276,60 @@ class Profile(basecontact.BaseContact, Singleton):
             self.create_message_item(text, time.time(), '', MESSAGE_TYPE['INFO_MESSAGE'])
             self._messages.scrollToBottom()
 
+    # -----------------------------------------------------------------------------------------------------------------
+    # GC support
+    # -----------------------------------------------------------------------------------------------------------------
+
+    def is_active_a_friend(self):
+        return type(self.get_curr_friend()) is Friend
+
+    def get_group_by_number(self, number):
+        groups = filter(lambda x: type(x) is GroupChat and x.number == number, self._contacts)
+        return list(groups)[0]
+
+    def add_gc(self, number):
+        widget = self.create_friend_item()
+        gc = GroupChat('', '', widget, self._tox, number)
+        self._contacts.append(gc)
+
+    def create_group_chat(self):
+        number = self._tox.add_av_groupchat()
+        self.add_gc(number)
+
+    def group_invite(self, friend_number, gc_type, data):
+        text = QtWidgets.QApplication.translate('MainWindow', 'User {} invites you to group chat. Accept?')
+        title = QtWidgets.QApplication.translate('MainWindow', 'Group chat invite')
+        reply = QtWidgets.QMessageBox.question(None, title, text, QtWidgets.QMessageBox.Yes, QtWidgets.QMessageBox.No)
+        if reply == QtWidgets.QMessageBox.Yes:  # accepted
+            if gc_type == TOX_GROUPCHAT_TYPE['TEXT']:
+                number = self._tox.join_groupchat(friend_number, data)
+            else:
+                number = self._tox.join_av_groupchat(friend_number, data)
+            self.add_gc(number)
+
+    def new_gc_message(self, group_number, peer_number, message_type, message):
+        name = self._tox.group_peername(group_number, peer_number)
+        if group_number == self.get_active_number() and not self.is_active_a_friend():  # add message to list
+            t = time.time()
+            self.create_gc_message_item(message, t, MESSAGE_OWNER['FRIEND'], name, message_type)
+            self._messages.scrollToBottom()
+            self.get_curr_friend().append_message(
+                GroupChatMessage(message, MESSAGE_OWNER['FRIEND'], t, message_type. name))
+        else:
+            gc = self.get_group_by_number(group_number)
+            gc.inc_messages()
+            gc.append_message(
+                GroupChatMessage(message, MESSAGE_OWNER['FRIEND'], time.time(), message_type, name))
+            if not gc.visibility:
+                self.update_filtration()
+
+    def new_gc_title(self, group_number, title):
+        gc = self.get_group_by_number(group_number)
+        gc.new_title(title)
+
+    def update_gc(self, group_number):
+        pass
+
 
 def tox_factory(data=None, settings=None):
     """