45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
import logging
|
|
|
|
from sugar.chat.GroupChat import GroupChat
|
|
|
|
class ActivityChat(GroupChat):
|
|
SERVICE_TYPE = "_olpc_activity_chat._udp"
|
|
|
|
def __init__(self, activity):
|
|
GroupChat.__init__(self)
|
|
self._chat_service = None
|
|
|
|
self._activity = activity
|
|
self._pservice.register_service_type(ActivityChat.SERVICE_TYPE)
|
|
self._pservice.connect('service-appeared', self._service_appeared_cb)
|
|
|
|
# Find an existing activity chat to latch onto
|
|
ps_activity = self._pservice.get_activity(activity.get_id())
|
|
if ps_activity is not None:
|
|
services = ps_activity.get_services_of_type(ActivityChat.SERVICE_TYPE)
|
|
if len(services) > 0:
|
|
self._service_appeared_cb(self._pservice, services[0])
|
|
|
|
def _service_appeared_cb(self, pservice, service):
|
|
if service.get_activity_id() != self._activity.get_id():
|
|
return
|
|
if service.get_type() != ActivityChat.SERVICE_TYPE:
|
|
return
|
|
if self._chat_service:
|
|
return
|
|
|
|
logging.debug('Activity chat service appeared, setup the stream.')
|
|
# Ok, there's an existing chat service that we copy
|
|
# parameters and such from
|
|
addr = service.get_address()
|
|
port = service.get_port()
|
|
self._chat_service = self._pservice.share_activity(self._activity,
|
|
stype=ActivityChat.SERVICE_TYPE, address=addr, port=port)
|
|
self._setup_stream(self._chat_service)
|
|
|
|
def share(self):
|
|
"""Only called when we share the activity this chat is tied to."""
|
|
self._chat_service = self._pservice.share_activity(self._activity,
|
|
stype=ActivityChat.SERVICE_TYPE)
|
|
self._setup_stream(self._chat_service)
|