transpub/transWhat/group.py

import Spectrum2

class Group():

    def __init__(self, id, owner, subject, subjectOwner, backend, user):
        self.id = id
        self.subject = subject
        self.subjectOwner = subjectOwner
        self.owner = owner
        self.joined = False
        self.backend = backend
        self.user = user

        self.nick = "me"
        # Participants is a number -> nickname dict
        self.participants = {}

    def addParticipants(self, participants, buddies, yourNumber):
        """
        Adds participants to the group.

        Args:
            - participants: (Iterable) phone numbers of participants
            - buddies: (dict) Used to get the nicknames of the participants
            - yourNumber: The number you are using
        """
        for jid in participants:
            number = jid.split('@')[0]
            try:
                nick = buddies[number].nick
            except KeyError:
                nick = number
            if number == yourNumber:
                nick = self.nick
            if nick == "":
                nick = number
            self.participants[number] = nick

    def sendParticipantsToSpectrum(self, yourNumber):
        for number, nick in self.participants.iteritems():
            if number == self.owner:
                flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_MODERATOR
            else:
                flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_NONE
            if number == yourNumber:
                flags = flags | Spectrum2.protocol_pb2.PARTICIPANT_FLAG_ME
            
            try:
                self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_ONLINE, 
                    self.backend.sessions[self.user].buddies[number].image_hash)
            except KeyError:
                self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_ONLINE)

    def removeParticipants(self, participants):
        for jid in participants:
            number = jid.split('@')[0]
            nick = self.participants[number]
            flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_NONE
            self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_NONE)
            del self.participants[number]

    def leaveRoom(self):
        for number in self.participants:
            nick = self.participants[number]
            flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_ROOM_NOT_FOUND
            self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_NONE)

    def changeNick(self, number, new_nick):
        if self.participants[number] == new_nick:
            return
        if number == self.owner:
            flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_MODERATOR
        else:
            flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_NONE
        self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_ONLINE, new_nick)
        self.participants[number] = new_nick

    def _updateParticipant(self, number, flags, status, imageHash = "", newNick = ""):
        nick = self.participants[number]
        # Notice the status message is the buddy's number
        if self.joined:
            self.backend.handleParticipantChanged(
                    self.user, nick, self.id, flags,
                    status, number, newname = newNick, iconHash = imageHash)