54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from sugar.chat.Chat import Chat
|
|
from sugar.presence.Service import Service
|
|
from sugar.p2p.Stream import Stream
|
|
|
|
GROUP_CHAT_SERVICE_TYPE = "_olpc_group_chat._udp"
|
|
GROUP_CHAT_SERVICE_ADDRESS = "224.0.0.221"
|
|
GROUP_CHAT_SERVICE_PORT = 6200
|
|
|
|
class GroupChat(Chat):
|
|
def __init__(self):
|
|
Chat.__init__(self)
|
|
self._chats = {}
|
|
|
|
def get_group(self):
|
|
return self._group
|
|
|
|
def new_buddy_writer(self, buddy):
|
|
service = buddy.get_service(Chat.SERVICE_TYPE)
|
|
return self._buddy_stream.new_writer(service)
|
|
|
|
def _start(self):
|
|
name = self._group.get_owner().get_nick_name()
|
|
|
|
# Group controls the Stream for incoming messages for
|
|
# specific buddy chats
|
|
buddy_service = Service(name, Chat.SERVICE_TYPE, Chat.SERVICE_PORT)
|
|
self._buddy_stream = Stream.new_from_service(buddy_service, self._group)
|
|
self._buddy_stream.set_data_listener(getattr(self, "_buddy_recv_message"))
|
|
buddy_service.register(self._group)
|
|
|
|
# Group chat Stream
|
|
group_service = Service(name,
|
|
GROUP_CHAT_SERVICE_TYPE,
|
|
GROUP_CHAT_SERVICE_PORT,
|
|
GROUP_CHAT_SERVICE_ADDRESS)
|
|
self._group.add_service(group_service)
|
|
|
|
self._group_stream = Stream.new_from_service(group_service, self._group)
|
|
self._group_stream.set_data_listener(self._group_recv_message)
|
|
self._stream_writer = self._group_stream.new_writer()
|
|
|
|
def _group_recv_message(self, buddy, msg):
|
|
self.recv_message(buddy, msg)
|
|
|
|
def _buddy_recv_message(self, buddy, msg):
|
|
if not self._chats.has_key(buddy):
|
|
# chat = BuddyChat(self, buddy)
|
|
# self._chats[buddy] = chat
|
|
# chat.connect_to_shell()
|
|
pass
|
|
else:
|
|
chat = self._chats[buddy]
|
|
chat.recv_message(buddy, msg)
|