diff options
Diffstat (limited to 'sugar/chat/chat.py')
| -rwxr-xr-x | sugar/chat/chat.py | 405 |
1 files changed, 405 insertions, 0 deletions
diff --git a/sugar/chat/chat.py b/sugar/chat/chat.py new file mode 100755 index 0000000..2bc7230 --- /dev/null +++ b/sugar/chat/chat.py @@ -0,0 +1,405 @@ +#!/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 + +from sugar.shell import activity +from sugar.p2p.Group import * +from sugar.p2p.StreamReader import * +from sugar.p2p.StreamWriter import * +import sugar.env + +import richtext + +CHAT_SERVICE_TYPE = "_olpc_chat._tcp" +CHAT_SERVICE_PORT = 6100 + +GROUP_CHAT_SERVICE_TYPE = "_olpc_group_chat._udp" +GROUP_CHAT_SERVICE_ADDRESS = "224.0.0.221" +GROUP_CHAT_SERVICE_PORT = 6200 + +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) + self._chat_view.set_cursor_visible(False) + sw.add(self._chat_view) + self._chat_view.show() + chat_vbox.pack_start(sw) + sw.show() + + 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() + 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, self._editor.get_buffer() + + 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.get_nick_name(), 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: + owner = self._controller.get_group().get_owner() + self._insert_rich_message(owner.get_nick_name(), text) + +class BuddyChat(Chat): + def __init__(self, controller, buddy): + self._buddy = buddy + self._act_name = "Chat: %s" % buddy.get_nick_name() + Chat.__init__(self, controller) + + def _start(self): + group = self._controller.get_group() + buddy_name = self._buddy.get_service_name() + service = group.get_service(buddy_name, CHAT_SERVICE_TYPE) + self._stream_writer = StreamWriter(group, service) + + def activity_on_connected_to_shell(self): + Chat.activity_on_connected_to_shell(self) + self.activity_set_can_close(True) + self.activity_set_tab_icon_name("im") + self.activity_show_icon(True) + self._start() + + def recv_message(self, sender, msg): + Chat.recv_message(self, self._buddy, msg) + + def send_message(self, text): + if len(text) > 0: + self._stream_writer.write(text) + self._local_message(True, text) + + def activity_on_close_from_user(self): + Chat.activity_on_close_from_user(self) + del self._chats[self._buddy] + +class GroupChat(Chat): + + _MODEL_COL_NICK = 0 + _MODEL_COL_ICON = 1 + _MODEL_COL_BUDDY = 2 + + def __init__(self): + self._act_name = "Chat" + self._chats = {} + + 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 get_group(self): + return self._group + + def _start(self): + self._group = LocalGroup() + self._group.add_presence_listener(self._on_group_event) + self._group.join() + + name = self._group.get_owner().get_service_name() + service = Service(name, CHAT_SERVICE_TYPE, '', CHAT_SERVICE_PORT) + self._buddy_reader = StreamReader(self._group, service) + self._buddy_reader.set_listener(self._buddy_recv_message) + service.register(self._group) + + service = Service(name, GROUP_CHAT_SERVICE_TYPE, + GROUP_CHAT_SERVICE_ADDRESS, + GROUP_CHAT_SERVICE_PORT, True) + self._group.add_service(service) + + self._buddy_reader = StreamReader(self._group, service) + self._buddy_reader.set_listener(self.recv_message) + + self._stream_writer = StreamWriter(self._group, service) + + 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) + + image_path = sugar.env.get_data_file('bubbleOutline.png') + self._pixbuf_active_chat = gtk.gdk.pixbuf_new_from_file(image_path) + + image_path = sugar.env.get_data_file('bubble.png') + self._pixbuf_new_message = gtk.gdk.pixbuf_new_from_file(image_path) + + 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 activity_on_connected_to_shell(self): + Chat.activity_on_connected_to_shell(self) + + self.activity_set_tab_icon_name("stock_help-chat") + self.activity_show_icon(True) + + 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 _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 self._chats.has_key(buddy): + chat = BuddyChat(self, buddy) + self._chats[buddy] = chat + chat.activity_connect_to_shell() + + def _on_group_event(self, action, buddy): + if buddy.get_nick_name() == self._group.get_owner().get_nick_name(): + # Do not show ourself in the buddy list + pass + elif action == BUDDY_JOIN: + aniter = self._buddy_list_model.append(None) + self._buddy_list_model.set(aniter, self._MODEL_COL_NICK, buddy.get_nick_name(), + self._MODEL_COL_ICON, None, self._MODEL_COL_BUDDY, buddy) + elif action == BUDDY_LEAVE: + 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 send_message(self, text): + if len(text) > 0: + self._stream_writer.write(text) + self._local_message(True, text) + + def recv_message(self, buddy, msg): + if buddy: + self._insert_rich_message(buddy.get_nick_name(), msg) + self._controller.notify_new_message(self, None) + + def _buddy_recv_message(self, sender, msg): + if not self._chats.has_key(sender): + chat = BuddyChat(self, sender) + self._chats[sender] = chat + chat.activity_connect_to_shell() + else: + chat = self._chats[sender] + chat.recv_message(sender, msg) + +class ChatShell(dbus.service.Object): + instance = None + + def get_instance(): + if not ChatShell.instance: + ChatShell.instance = ChatShell() + return ChatShell.instance + + get_instance = staticmethod(get_instance) + + def __init__(self): + 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) + + def open_group_chat(self): + group_chat = GroupChat() + group_chat.activity_connect_to_shell() + + @dbus.service.method('com.redhat.Sugar.ChatShell') + def send_message(self, message): + pass + +def main(): + ChatShell.get_instance().open_group_chat() + try: + gtk.main() + except KeyboardInterrupt: + pass + +if __name__ == "__main__": + main() |