import dbus import random import logging import gtk import gobject from gettext import gettext as _ from sugar.activity.Activity import Activity from sugar.presence import Service from sugar.chat.Chat import Chat from sugar.chat.BuddyChat import BuddyChat from sugar.p2p.Stream import Stream from sugar.presence.PresenceService import PresenceService import sugar.env _CHAT_ACTIVITY_TYPE = "_chat_activity._tcp" class ChatActivity(Activity): def __init__(self, service): Activity.__init__(self) self.set_title(_('Private chat')) self._service = service self._chat = BuddyChat(self._service) self.add(self._chat) self._chat.show() def recv_message(self, message): self._chat.recv_message(message) class ChatShellDbusService(dbus.service.Object): def __init__(self, parent): self._parent = parent session_bus = dbus.SessionBus() bus_name = dbus.service.BusName('com.redhat.Sugar.Chat', bus=session_bus) object_path = '/com/redhat/Sugar/Chat' dbus.service.Object.__init__(self, bus_name, object_path) @dbus.service.method('com.redhat.Sugar.ChatShell') def open_chat(self, service_path): self._parent.open_chat_from_path(service_path) class ChatListener: def __init__(self): ChatShellDbusService(self) self._chats = {} self._pservice = PresenceService() self._pservice.register_service_type(BuddyChat.SERVICE_TYPE) def start(self): self._service = self._pservice.register_service(sugar.env.get_nick_name(), BuddyChat.SERVICE_TYPE) self._buddy_stream = Stream.new_from_service(self._service) self._buddy_stream.set_data_listener(self._recv_message) def _recv_message(self, address, message): [nick, msg] = Chat.deserialize_message(message) buddy = self._pservice.get_buddy_by_name(nick) if buddy: if buddy == self._pservice.get_owner(): return service = buddy.get_service_of_type(BuddyChat.SERVICE_TYPE) name = service.get_name() if service: if not self._chats.has_key(name): self.open_chat(service) self._chats[name].recv_message(message) else: logging.error('The buddy %s does not have a chat service.' % (nick)) else: logging.error('The buddy %s is not present.' % (nick)) return def open_chat_from_path(self, service_path): service = self._pservice._new_object(service_path) self.open_chat(service) def open_chat(self, service): chat = ChatActivity(service) self._chats[service.get_name()] = chat return chat def start(): chat_listener = ChatListener() chat_listener.start()