sugar-toolkit-gtk3/activities/chat/ChatActivity.py
2006-08-07 11:08:10 +02:00

90 lines
2.5 KiB
Python

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',
in_signature="o", out_signature="")
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.get(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()