#!/usr/bin/python -t # -*- tab-width: 4; indent-tabs-mode: t -*- import dbus import dbus.service import dbus.glib import pygtk pygtk.require('2.0') import gtk, gobject import sys import os import pwd import socket import activity import presence import BuddyList import network import richtext import xmlrpclib from sugar_globals import * class Chat(activity.Activity): def __init__(self, controller): self._controller = controller activity.Activity.__init__(self) def activity_on_connected_to_shell(self): self.activity_set_tab_text(self._act_name) self._plug = self.activity_get_gtk_plug() self._ui_setup(self._plug) self._plug.show_all() def _create_chat(self): chat_vbox = gtk.VBox() chat_vbox.set_spacing(6) sw = gtk.ScrolledWindow() sw.set_shadow_type(gtk.SHADOW_IN) sw.set_policy(gtk.POLICY_NEVER, gtk.POLICY_ALWAYS) self._chat_view = richtext.RichTextView() self._chat_view.connect("link-clicked", self.__link_clicked_cb) self._chat_view.set_editable(False) sw.add(self._chat_view) self._chat_view.show() chat_vbox.pack_start(sw) sw.show() rich_buf = richtext.RichTextBuffer() chat_view_sw = gtk.ScrolledWindow() chat_view_sw.set_shadow_type(gtk.SHADOW_IN) chat_view_sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC) self._editor = richtext.RichTextView(rich_buf) self._editor.connect("key-press-event", self.__key_press_event_cb) self._editor.set_size_request(-1, 50) chat_view_sw.add(self._editor) self._editor.show() chat_vbox.pack_start(chat_view_sw, False) chat_view_sw.show() return chat_vbox, rich_buf def _ui_setup(self, base): vbox = gtk.VBox(False, 6) self._hbox = gtk.HBox(False, 12) self._hbox.set_border_width(12) [chat_vbox, buffer] = self._create_chat() self._hbox.pack_start(chat_vbox) chat_vbox.show() vbox.pack_start(self._hbox) self._hbox.show() toolbar = self._create_toolbar(buffer) vbox.pack_start(toolbar, False) toolbar.show() base.add(vbox) vbox.show() def __link_clicked_cb(self, view, address): self._browser_shell.open_browser(address) def __key_press_event_cb(self, text_view, event): if event.keyval == gtk.keysyms.Return: buf = text_view.get_buffer() serializer = richtext.RichTextSerializer() text = serializer.serialize(buf) self.send_message(text) buf.set_text("") buf.place_cursor(buf.get_start_iter()) return True def _create_toolbar(self, rich_buf): toolbar = richtext.RichTextToolbar(rich_buf) item = gtk.MenuToolButton(None, "Links") item.set_menu(gtk.Menu()) item.connect("show-menu", self.__show_link_menu_cb) toolbar.insert(item, -1) item.show() return toolbar def __link_activate_cb(self, item, link): buf = self._editor.get_buffer() buf.append_link(link['title'], link['address']) def __show_link_menu_cb(self, button): menu = gtk.Menu() links = self._browser_shell.get_links() for link in links: item = gtk.MenuItem(link['title'], False) item.connect("activate", self.__link_activate_cb, link) menu.append(item) item.show() button.set_menu(menu) def activity_on_close_from_user(self): print "act %d: in activity_on_close_from_user"%self.activity_get_id() self.activity_shutdown() def activity_on_lost_focus(self): print "act %d: in activity_on_lost_focus"%self.activity_get_id() def activity_on_got_focus(self): print "act %d: in activity_on_got_focus"%self.activity_get_id() self._controller.notify_activate(self) def recv_message(self, buddy, msg): self._insert_rich_message(buddy.nick(), msg) self._controller.notify_new_message(self, buddy) def _insert_rich_message(self, nick, msg): buffer = self._chat_view.get_buffer() aniter = buffer.get_end_iter() buffer.insert(aniter, nick + ": ") serializer = richtext.RichTextSerializer() serializer.deserialize(msg, buffer) aniter = buffer.get_end_iter() buffer.insert(aniter, "\n") def _local_message(self, success, text): if not success: message = "Error: %s\n" % text buffer = self._chat_view.get_buffer() aniter = buffer.get_end_iter() buffer.insert(aniter, message) else: (nick, realname) = self._controller.local_name() self._insert_rich_message(nick, text) class BuddyChat(Chat): def __init__(self, controller, buddy): self._buddy = buddy self._act_name = "Chat: %s" % buddy.nick() Chat.__init__(self, controller) def recv_message(self, msg): Chat.recv_message(self, self._buddy, msg) def send_message(self, text): if len(text) <= 0: return addr = "http://%s:%d" % (self._buddy.address(), self._buddy.port()) peer = xmlrpclib.ServerProxy(addr) msg = text success = True try: peer.message(text) except (socket.error, xmlrpclib.Fault), e: msg = str(e) success = False self._local_message(success, msg) class ChatRequestHandler(object): def __init__(self, parent): self._parent = parent def message(self, message): client_address = network.get_authinfo() buddy = self._parent.find_buddy_by_address(client_address[0]) if buddy: chat = buddy.chat() if not chat: chat = BuddyChat(self._parent, buddy) buddy.set_chat(chat) chat.activity_connect_to_shell() chat.recv_message(message) return True class GroupChat(Chat): _MODEL_COL_NICK = 0 _MODEL_COL_ICON = 1 _MODEL_COL_BUDDY = 2 _CHAT_SERVER_PORT = 6666 def __init__(self): self._act_name = "Chat" self._pannounce = presence.PresenceAnnounce() (self._nick, self._realname) = self._get_name() self._buddy_list = BuddyList.BuddyList(self._realname) self._buddy_list.add_buddy_listener(self._on_buddy_presence_event) bus = dbus.SessionBus() proxy_obj = bus.get_object('com.redhat.Sugar.Browser', '/com/redhat/Sugar/Browser') self._browser_shell = dbus.Interface(proxy_obj, 'com.redhat.Sugar.BrowserShell') Chat.__init__(self, self) def _create_sidebar(self): vbox = gtk.VBox(False, 6) label = gtk.Label("Who's around:") label.set_alignment(0.0, 0.5) vbox.pack_start(label, False) label.show() self._buddy_list_model = gtk.ListStore(gobject.TYPE_STRING, gtk.gdk.Pixbuf, gobject.TYPE_PYOBJECT) self._pixbuf_active_chat = gtk.gdk.pixbuf_new_from_file(data_dir + "/bubbleOutline.png") self._pixbuf_new_message = gtk.gdk.pixbuf_new_from_file(data_dir + "/bubble.png") sw = gtk.ScrolledWindow() sw.set_shadow_type(gtk.SHADOW_IN) sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC) self._buddy_list_view = gtk.TreeView(self._buddy_list_model) self._buddy_list_view.set_headers_visible(False) self._buddy_list_view.connect("cursor-changed", self._on_buddyList_buddy_selected) self._buddy_list_view.connect("row-activated", self._on_buddyList_buddy_double_clicked) sw.set_size_request(120, -1) sw.add(self._buddy_list_view) self._buddy_list_view.show() renderer = gtk.CellRendererPixbuf() column = gtk.TreeViewColumn("", renderer, pixbuf=self._MODEL_COL_ICON) column.set_resizable(False) column.set_expand(False); self._buddy_list_view.append_column(column) renderer = gtk.CellRendererText() column = gtk.TreeViewColumn("", renderer, text=self._MODEL_COL_NICK) column.set_resizable(True) column.set_sizing("GTK_TREE_VIEW_COLUMN_GROW_ONLY"); column.set_expand(True); self._buddy_list_view.append_column(column) vbox.pack_start(sw) sw.show() return vbox def _ui_setup(self, base): Chat._ui_setup(self, base) sidebar = self._create_sidebar() self._hbox.pack_start(sidebar, False) sidebar.show() self._plug.show_all() def _start(self): self._buddy_list.start() self._pannounce.register_service(self._realname, self._CHAT_SERVER_PORT, presence.OLPC_CHAT_SERVICE, name = self._nick, realname = self._realname) # Create the P2P chat XMLRPC server self._p2p_req_handler = ChatRequestHandler(self) self._p2p_server = network.GlibXMLRPCServer(("", self._CHAT_SERVER_PORT)) self._p2p_server.register_instance(self._p2p_req_handler) self._gc_controller = network.GroupChatController('224.0.0.221', 6666, self._recv_group_message) self._gc_controller.start() def activity_on_connected_to_shell(self): Chat.activity_on_connected_to_shell(self) aniter = self._buddy_list_model.append(None) self._buddy_list_model.set(aniter, self._MODEL_COL_NICK, "Group", self._MODEL_COL_ICON, self._pixbuf_active_chat, self._MODEL_COL_BUDDY, None) self._start() def activity_on_disconnected_from_shell(self): Chat.activity_on_disconnected_from_shell(self) gtk.main_quit() def _get_name(self): ent = pwd.getpwuid(os.getuid()) nick = ent[0] if not nick or not len(nick): nick = "n00b" realname = ent[4] if not realname or not len(realname): realname = "Some Clueless User" return (nick, realname) def _on_buddyList_buddy_selected(self, widget, *args): (model, aniter) = widget.get_selection().get_selected() name = self._buddy_list_model.get(aniter, self._MODEL_COL_NICK) print "Selected %s" % name def _on_buddyList_buddy_double_clicked(self, widget, *args): """ Select the chat for this buddy or group """ (model, aniter) = widget.get_selection().get_selected() chat = None buddy = self._buddy_list_model.get_value(aniter, self._MODEL_COL_BUDDY) if buddy and not buddy.chat(): chat = BuddyChat(self, buddy) buddy.set_chat(chat) chat.activity_connect_to_shell() def _on_buddy_presence_event(self, action, buddy): if action == BuddyList.ACTION_BUDDY_ADDED: aniter = self._buddy_list_model.append(None) self._buddy_list_model.set(aniter, self._MODEL_COL_NICK, buddy.nick(), self._MODEL_COL_ICON, None, self._MODEL_COL_BUDDY, buddy) elif action == BuddyList.ACTION_BUDDY_REMOVED: aniter = self._get_iter_for_buddy(buddy) if aniter: self._buddy_list_model.remove(aniter) def _get_iter_for_buddy(self, buddy): aniter = self._buddy_list_model.get_iter_first() while aniter: list_buddy = self._buddy_list_model.get_value(aniter, self._MODEL_COL_BUDDY) if buddy == list_buddy: return aniter aniter = self._buddy_list_model.iter_next(aniter) def notify_new_message(self, chat, buddy): aniter = self._get_iter_for_buddy(buddy) self._buddy_list_model.set(aniter, self._MODEL_COL_ICON, self._pixbuf_new_message) def notify_activate(self, chat): aniter = self._get_iter_for_buddy(buddy) self._buddy_list_model.set(aniter, self._MODEL_COL_ICON, self._pixbuf_active_chat) def find_buddy_by_address(self, address): return self._buddy_list.find_buddy_by_address(address) def local_name(self): return (self._nick, self._realname) def send_message(self, text): if len(text) > 0: self._gc_controller.send_msg(text) self._local_message(True, text) def recv_message(self, buddy, msg): self._insert_rich_message(buddy.nick(), msg) self._parent.notify_new_message(self, None) def _recv_group_message(self, msg): buddy = self.find_buddy_by_address(msg['addr']) if buddy: self.recv_message(buddy, msg['data']) def run(self): try: gtk.main() except KeyboardInterrupt: pass def main(): app = GroupChat() app.activity_connect_to_shell() app.run() if __name__ == "__main__": main()