From 8f702f2cf547abc3e719346f113a14423f048ca9 Mon Sep 17 00:00:00 2001 From: Marco Pesenti Gritti Date: Wed, 26 Apr 2006 16:35:19 -0400 Subject: [PATCH] Fix up browser life cycle --- browser/browser.py | 4 + chat/chat.py | 474 +++++++++++++++++++++++++-------------------- chat/richtext.py | 7 +- sugar_globals.py | 2 +- 4 files changed, 269 insertions(+), 218 deletions(-) diff --git a/browser/browser.py b/browser/browser.py index f193241e..1b19c792 100755 --- a/browser/browser.py +++ b/browser/browser.py @@ -213,6 +213,10 @@ class WebActivity(activity.Activity): BrowserShell.get_instance().open_browser(uri) return True + def activity_on_disconnected_from_shell(self): + gtk.main_quit() + gc.collect() + class BrowserShell(dbus.service.Object): instance = None diff --git a/chat/chat.py b/chat/chat.py index 8c5a10a9..bb66d8f8 100755 --- a/chat/chat.py +++ b/chat/chat.py @@ -12,6 +12,7 @@ import gtk, gobject import sys import os import pwd +import gc import socket import activity @@ -24,28 +25,144 @@ import xmlrpclib from sugar_globals import * +class Chat(object): + def __init__(self, parent, view, label): + self._parent = parent + self._buffer = richtext.RichTextBuffer() + self._view = view + self._label = label -class Chat(activity.Activity): - def __init__(self, controller): - self._controller = controller - activity.Activity.__init__(self) + def activate(self, label): + self._view.set_buffer(self._buffer) + self._label.set_text(label) - 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 recv_message(self, buddy, msg): + self._insert_rich_message(buddy.nick(), msg) + self._parent.notify_new_message(self, buddy) + + def _insert_rich_message(self, nick, msg): + aniter = self._buffer.get_end_iter() + self._buffer.insert(aniter, nick + ": ") + serializer = richtext.RichTextSerializer() + serializer.deserialize(msg, self._buffer) + + aniter = self._buffer.get_end_iter() + self._buffer.insert(aniter, "\n") + + def _local_message(self, success, text): + if not success: + message = "Error: %s\n" % text + aniter = self._buffer.get_end_iter() + self._buffer.insert(aniter, message) + else: + (nick, realname) = self._parent.local_name() + self._insert_rich_message(nick, text) + +class BuddyChat(Chat): + def __init__(self, parent, buddy, view, label): + self._buddy = buddy + Chat.__init__(self, parent, view, label) + + def activate(self): + Chat.activate(self, self._buddy.nick()) + + 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 GroupChat(Chat): + def __init__(self, parent, view, label): + Chat.__init__(self, parent, view, label) + self._gc_controller = network.GroupChatController('224.0.0.221', 6666, self._recv_group_message) + self._gc_controller.start() + + def activate(self): + Chat.activate(self, "Group Chat") + + 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._parent.find_buddy_by_address(msg['addr']) + if buddy: + self.recv_message(buddy, msg['data']) + + +class ChatRequestHandler(object): + def __init__(self, parent, chat_view, chat_label): + self._parent = parent + self._chat_view = chat_view + self._chat_label = chat_label + + 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, self._chat_view, self._chat_label) + buddy.set_chat(chat) + chat.recv_message(message) + return True + +class ChatActivity(activity.Activity): + + _MODEL_COL_NICK = 0 + _MODEL_COL_ICON = 1 + _MODEL_COL_BUDDY = 2 + + def __init__(self): + activity.Activity.__init__(self) + self._act_name = "Chat" + self._active_chat_buddy = None + 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') + + def __link_clicked_cb(self, view, address): + self.browser_shell.open_browser(address) + def _create_chat(self): chat_vbox = gtk.VBox() chat_vbox.set_spacing(6) + self._chat_label = gtk.Label() + chat_vbox.pack_start(self._chat_label, False) + # Do we actually need this label? + # self._chat_label.show() + 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) @@ -66,168 +183,6 @@ class Chat(activity.Activity): 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) @@ -235,7 +190,7 @@ class GroupChat(Chat): 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") @@ -270,39 +225,123 @@ class GroupChat(Chat): vbox.pack_start(sw) sw.show() + button_box = gtk.VButtonBox() + button_box.set_border_width(18) + + talk_alone_button = gtk.Button("Talk alone") + button_box.pack_start(talk_alone_button) + talk_alone_button.show() + + vbox.pack_start(button_box, False) + button_box.show() + return vbox - def _ui_setup(self, base): - Chat._ui_setup(self, base) + 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 _ui_setup(self, plug): + vbox = gtk.VBox(False, 6) + + hbox = gtk.HBox(False, 12) + hbox.set_border_width(12) + + [chat, rich_buf] = self._create_chat() + hbox.pack_start(chat) + chat.show() + sidebar = self._create_sidebar() - self._hbox.pack_start(sidebar, False) + 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) + vbox.pack_start(hbox) + hbox.show() - # 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) + toolbar = self._create_toolbar(rich_buf) + vbox.pack_start(toolbar, False); + toolbar.show() + + self._group_chat = GroupChat(self, self._chat_view, self._chat_label) 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._activate_chat_for_buddy(None) + + plug.add(vbox) + vbox.show() + + def __key_press_event_cb(self, text_view, event): + if event.keyval == gtk.keysyms.Return: + buf = text_view.get_buffer() + chat = self._get_active_chat() + + serializer = richtext.RichTextSerializer() + text = serializer.serialize(buf) + chat.send_message(text) + + buf.set_text("") + buf.place_cursor(buf.get_start_iter()) + + return True + + def _start(self): + self._buddy_list.start() + self._pannounce.register_service(self._realname, 6666, presence.OLPC_CHAT_SERVICE, + name = self._nick, realname = self._realname) + + # Create the P2P chat XMLRPC server + self._p2p_req_handler = ChatRequestHandler(self, self._chat_view, self._chat_label) + self._p2p_server = network.GlibXMLRPCServer(("", 6666)) + self._p2p_server.register_instance(self._p2p_req_handler) + + def activity_on_connected_to_shell(self): + print "act %d: in activity_on_connected_to_shell" % self.activity_get_id() + self.activity_set_tab_text(self._act_name) + self._plug = self.activity_get_gtk_plug() + self._ui_setup(self._plug) + self._plug.show() self._start() def activity_on_disconnected_from_shell(self): - Chat.activity_on_disconnected_from_shell(self) + print "act %d: in activity_on_disconnected_from_shell"%self.activity_get_id() + print "act %d: Shell disappeared..."%self.activity_get_id() gtk.main_quit() + gc.collect() + + 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() def _get_name(self): ent = pwd.getpwuid(os.getuid()) @@ -325,9 +364,9 @@ class GroupChat(Chat): chat = None buddy = self._buddy_list_model.get_value(aniter, self._MODEL_COL_BUDDY) if buddy and not buddy.chat(): - chat = BuddyChat(self, buddy) + chat = BuddyChat(self, buddy, self._chat_view, self._chat_label) buddy.set_chat(chat) - chat.activity_connect_to_shell() + self._activate_chat_for_buddy(buddy) def _on_buddy_presence_event(self, action, buddy): if action == BuddyList.ACTION_BUDDY_ADDED: @@ -348,12 +387,9 @@ class GroupChat(Chat): 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) + if chat != self._get_active_chat(): + aniter = self._get_iter_for_buddy(buddy) + self._buddy_list_model.set(aniter, self._MODEL_COL_ICON, self._pixbuf_new_message) def find_buddy_by_address(self, address): return self._buddy_list.find_buddy_by_address(address) @@ -361,19 +397,33 @@ class GroupChat(Chat): 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 _activate_chat_for_buddy(self, buddy): + self._active_chat_buddy = buddy - def recv_message(self, buddy, msg): - self._insert_rich_message(buddy.nick(), msg) - self._parent.notify_new_message(self, None) + # Clear the "new message" icon when the user activates the chat + aniter = self._get_iter_for_buddy(buddy) + # Select the row in the list + if aniter: + selection = self._buddy_list_view.get_selection() + selection.select_iter(aniter) + icon = self._buddy_list_model.get_value(aniter, self._MODEL_COL_ICON) + if icon == self._pixbuf_new_message: + self._buddy_list_model.set_value(aniter, self._MODEL_COL_ICON, self._pixbuf_active_chat) - def _recv_group_message(self, msg): - buddy = self.find_buddy_by_address(msg['addr']) - if buddy: - self.recv_message(buddy, msg['data']) + # Actually activate the chat + chat = self._group_chat + if self._active_chat_buddy: + chat = buddy.chat() + chat.activate() + + def _on_main_window_delete(self, widget, *args): + self.quit() + + def _get_active_chat(self): + chat = self._group_chat + if self._active_chat_buddy: + chat = self._active_chat_buddy.chat() + return chat def run(self): try: @@ -382,7 +432,7 @@ class GroupChat(Chat): pass def main(): - app = GroupChat() + app = ChatActivity() app.activity_connect_to_shell() app.run() diff --git a/chat/richtext.py b/chat/richtext.py index 88d40357..2951d16e 100644 --- a/chat/richtext.py +++ b/chat/richtext.py @@ -35,11 +35,8 @@ class RichTextView(gtk.TextView): gtk.gdk.flush() def __iter_is_link(self, it): - item = self.get_buffer().get_tag_table().lookup("link") - if item: - return it.has_tag(item) - return False - + return it.has_tag(self.get_buffer().get_tag_table().lookup("link")) + def __get_event_iter(self, event): return self.get_iter_at_location(int(event.x), int(event.y)) diff --git a/sugar_globals.py b/sugar_globals.py index 859d6788..2f9c4553 100644 --- a/sugar_globals.py +++ b/sugar_globals.py @@ -1 +1 @@ -data_dir = "/usr/share/sugar" +data_dir = "/home/marco/sugar/share/sugar"