sugar-toolkit-gtk3/activities/chat/ChatActivity.py

import dbus
import random
import logging

import pygtk
pygtk.require('2.0')
import gtk
import gobject

from sugar.activity.Activity import Activity
from sugar.LogWriter import LogWriter
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_type._tcp"

class ChatActivity(Activity):
	def __init__(self, service):
		Activity.__init__(self, _CHAT_ACTIVITY_TYPE)
		self._service = service
		self._chat = BuddyChat(self._service)
	
	def on_connected_to_shell(self):
		self.set_tab_text(self._service.get_name())
		self.set_can_close(True)
		self.set_tab_icon(name = "im")
		self.set_show_tab_icon(True)

		plug = self.gtk_plug()		
		plug.add(self._chat)
		self._chat.show()

		plug.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, serialized_service):
		self._parent.open_chat(Service.deserialize(serialized_service))

class ChatShell:
	instance = None

	def get_instance():
		if not ChatShell.instance:
			ChatShell.instance = ChatShell()
		return ChatShell.instance		
	get_instance = staticmethod(get_instance)

	def __init__(self):
		ChatShellDbusService(self)

		self._chats = {}
	
		self._pservice = PresenceService.get_instance()
		self._pservice.start()
		self._pservice.track_service_type(BuddyChat.SERVICE_TYPE)

	def start(self):
		port = random.randint(5000, 65535)
		service = Service.Service(sugar.env.get_nick_name(), BuddyChat.SERVICE_TYPE,
						  		  'local', '', port)
		self._buddy_stream = Stream.new_from_service(service)
		self._buddy_stream.set_data_listener(self._recv_message)
		self._pservice.register_service(service)

	def _recv_message(self, address, message):
		[nick, msg] = Chat.deserialize_message(message)
		buddy = self._pservice.get_buddy_by_nick_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(self, service):
		chat = ChatActivity(service)
		self._chats[service.get_name()] = chat
		gobject.idle_add(self._connect_chat, chat)
		return chat

	def _connect_chat(self, chat):
		chat.connect_to_shell()
		return False

log_writer = LogWriter("Chat")
log_writer.start()

chat_shell = ChatShell.get_instance()
chat_shell.start()

gtk.main()