sugar-toolkit-gtk3/sugar/chat/GroupChat.py

import logging

from sugar.chat.Chat import Chat
from sugar.p2p.Stream import Stream
from sugar.presence.PresenceService import PresenceService 

class GroupChat(Chat):
	SERVICE_TYPE = "_olpc_group_chat._udp"
	SERVICE_PORT = 6200

	def __init__(self, activity):
		Chat.__init__(self)
		self._chats = {}
		self._activity = activity

		self._pservice = PresenceService.get_instance()
		self._pservice.start()		
		self._pservice.connect('service-appeared', self._service_appeared_cb)
		self._pservice.track_service_type(GroupChat.SERVICE_TYPE)
		service = self._pservice.get_activity_service(activity, GroupChat.SERVICE_TYPE)
		if service is not None:
			self._service_appeared_cb(self._pservice, None, service)

	def _service_appeared_cb(self, pservice, buddy, service):
		if service.get_activity_uid() == self._activity.get_id():
			if service.get_type() == GroupChat.SERVICE_TYPE:
				logging.debug('Group chat service appeared, setup the stream.')
				self._setup_stream(service)

	def publish(self):
		service = self._pservice.share_activity(self._activity,
				stype = GroupChat.SERVICE_TYPE, port = GroupChat.SERVICE_PORT)

	def _setup_stream(self, service):
		self._group_stream = Stream.new_from_service(service)
		self._group_stream.set_data_listener(self._group_recv_message)
		self._stream_writer = self._group_stream.new_writer()

	def _group_recv_message(self, address, msg):
		pservice = PresenceService.get_instance()
		[nick, msg] = self.deserialize_message(msg)
		buddy = pservice.get_buddy_by_nick_name(nick)
		if buddy:
			self.recv_message(buddy, msg)
		else:
			logging.error('The buddy %s is not present.' % (nick))