Factor out a very simple p2p messaging system from chat
This commit is contained in:
parent
14d572e72a
commit
91670df8b2
@ -115,6 +115,16 @@ class NavigationToolbar(gtk.Toolbar):
|
|||||||
self.insert(self.reload, -1)
|
self.insert(self.reload, -1)
|
||||||
self.reload.show()
|
self.reload.show()
|
||||||
|
|
||||||
|
separator = gtk.SeparatorToolItem()
|
||||||
|
self.insert(separator, -1)
|
||||||
|
separator.show()
|
||||||
|
|
||||||
|
item = gtk.MenuToolButton(None, "Share")
|
||||||
|
item.set_menu(gtk.Menu())
|
||||||
|
item.connect("show-menu", self.__show_share_menu_cb)
|
||||||
|
toolbar.insert(item, -1)
|
||||||
|
item.show()
|
||||||
|
|
||||||
separator = gtk.SeparatorToolItem()
|
separator = gtk.SeparatorToolItem()
|
||||||
self.insert(separator, -1)
|
self.insert(separator, -1)
|
||||||
separator.show()
|
separator.show()
|
||||||
@ -145,7 +155,20 @@ class NavigationToolbar(gtk.Toolbar):
|
|||||||
|
|
||||||
def __open_address_cb(self, address):
|
def __open_address_cb(self, address):
|
||||||
self.embed.load_address(address)
|
self.embed.load_address(address)
|
||||||
|
|
||||||
|
def __show_share_menu_cb(self, button):
|
||||||
|
menu = gtk.Menu()
|
||||||
|
|
||||||
|
item = gtk.MenuItem("Group", False)
|
||||||
|
item.connect("activate", self.__share_group_activate_cb)
|
||||||
|
menu.append(item)
|
||||||
|
item.show()
|
||||||
|
|
||||||
|
button.set_menu(menu)
|
||||||
|
|
||||||
|
def __share_group_activate_cb(self, item, link):
|
||||||
|
pass
|
||||||
|
|
||||||
class BrowserActivity(activity.Activity):
|
class BrowserActivity(activity.Activity):
|
||||||
def __init__(self, uri):
|
def __init__(self, uri):
|
||||||
activity.Activity.__init__(self)
|
activity.Activity.__init__(self)
|
||||||
|
107
chat/chat.py
107
chat/chat.py
@ -10,8 +10,6 @@ pygtk.require('2.0')
|
|||||||
import gtk, gobject
|
import gtk, gobject
|
||||||
|
|
||||||
import sys
|
import sys
|
||||||
import os
|
|
||||||
import pwd
|
|
||||||
import socket
|
import socket
|
||||||
|
|
||||||
try:
|
try:
|
||||||
@ -21,12 +19,9 @@ except ImportError:
|
|||||||
from sugar import activity
|
from sugar import activity
|
||||||
from sugar.sugar_globals import *
|
from sugar.sugar_globals import *
|
||||||
|
|
||||||
import presence
|
|
||||||
import BuddyList
|
|
||||||
import network
|
|
||||||
import richtext
|
import richtext
|
||||||
import xmlrpclib
|
import xmlrpclib
|
||||||
|
import p2p
|
||||||
|
|
||||||
class Chat(activity.Activity):
|
class Chat(activity.Activity):
|
||||||
def __init__(self, controller):
|
def __init__(self, controller):
|
||||||
@ -166,7 +161,7 @@ class Chat(activity.Activity):
|
|||||||
aniter = buffer.get_end_iter()
|
aniter = buffer.get_end_iter()
|
||||||
buffer.insert(aniter, message)
|
buffer.insert(aniter, message)
|
||||||
else:
|
else:
|
||||||
(nick, realname) = self._controller.local_name()
|
nick = p2p.Owner.get_instance().get_nick()
|
||||||
self._insert_rich_message(nick, text)
|
self._insert_rich_message(nick, text)
|
||||||
|
|
||||||
class BuddyChat(Chat):
|
class BuddyChat(Chat):
|
||||||
@ -175,72 +170,55 @@ class BuddyChat(Chat):
|
|||||||
self._act_name = "Chat: %s" % buddy.nick()
|
self._act_name = "Chat: %s" % buddy.nick()
|
||||||
Chat.__init__(self, controller)
|
Chat.__init__(self, controller)
|
||||||
|
|
||||||
|
def _start(self):
|
||||||
|
group = p2p.Group.get_instance()
|
||||||
|
input_pipe = InputPipe(group, "buddy-chat")
|
||||||
|
input_pipe.listen(self.recv_message)
|
||||||
|
self._output_pipe = OutputPipe(group, "buddy-chat")
|
||||||
|
|
||||||
def activity_on_connected_to_shell(self):
|
def activity_on_connected_to_shell(self):
|
||||||
Chat.activity_on_connected_to_shell(self)
|
Chat.activity_on_connected_to_shell(self)
|
||||||
self.activity_set_can_close(True)
|
self.activity_set_can_close(True)
|
||||||
self.activity_set_tab_icon_name("im")
|
self.activity_set_tab_icon_name("im")
|
||||||
self.activity_show_icon(True)
|
self.activity_show_icon(True)
|
||||||
|
|
||||||
def recv_message(self, msg):
|
def recv_message(self, sender, msg):
|
||||||
Chat.recv_message(self, self._buddy, msg)
|
Chat.recv_message(self, self._buddy, msg)
|
||||||
|
|
||||||
def send_message(self, text):
|
def send_message(self, text):
|
||||||
if len(text) <= 0:
|
if len(text) > 0:
|
||||||
return
|
self._output_pipe.send(self._buddy, text)
|
||||||
addr = "http://%s:%d" % (self._buddy.address(), self._buddy.port())
|
self._local_message(success, msg)
|
||||||
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)
|
|
||||||
|
|
||||||
def activity_on_close_from_user(self):
|
def activity_on_close_from_user(self):
|
||||||
Chat.activity_on_close_from_user(self)
|
Chat.activity_on_close_from_user(self)
|
||||||
self._buddy.set_chat(None)
|
self._buddy.set_chat(None)
|
||||||
|
|
||||||
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):
|
class GroupChat(Chat):
|
||||||
|
|
||||||
_MODEL_COL_NICK = 0
|
_MODEL_COL_NICK = 0
|
||||||
_MODEL_COL_ICON = 1
|
_MODEL_COL_ICON = 1
|
||||||
_MODEL_COL_BUDDY = 2
|
_MODEL_COL_BUDDY = 2
|
||||||
|
|
||||||
_CHAT_SERVER_PORT = 6666
|
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self._act_name = "Chat"
|
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()
|
bus = dbus.SessionBus()
|
||||||
proxy_obj = bus.get_object('com.redhat.Sugar.Browser', '/com/redhat/Sugar/Browser')
|
proxy_obj = bus.get_object('com.redhat.Sugar.Browser', '/com/redhat/Sugar/Browser')
|
||||||
self._browser_shell = dbus.Interface(proxy_obj, 'com.redhat.Sugar.BrowserShell')
|
self._browser_shell = dbus.Interface(proxy_obj, 'com.redhat.Sugar.BrowserShell')
|
||||||
|
|
||||||
Chat.__init__(self, self)
|
Chat.__init__(self, self)
|
||||||
|
|
||||||
|
def _start(self):
|
||||||
|
group = p2p.Group.get_instance()
|
||||||
|
|
||||||
|
self._buddy_list = group.get_buddy_list()
|
||||||
|
self._buddy_list.add_buddy_listener(self._on_buddy_presence_event)
|
||||||
|
|
||||||
|
input_pipe = p2p.InputPipe(group, "group-chat")
|
||||||
|
input_pipe.listen(self.recv_message)
|
||||||
|
self._output_pipe = p2p.BroadcastOutputPipe(group, "group-chat")
|
||||||
|
|
||||||
def _create_sidebar(self):
|
def _create_sidebar(self):
|
||||||
vbox = gtk.VBox(False, 6)
|
vbox = gtk.VBox(False, 6)
|
||||||
|
|
||||||
@ -293,19 +271,6 @@ class GroupChat(Chat):
|
|||||||
sidebar.show()
|
sidebar.show()
|
||||||
self._plug.show_all()
|
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):
|
def activity_on_connected_to_shell(self):
|
||||||
Chat.activity_on_connected_to_shell(self)
|
Chat.activity_on_connected_to_shell(self)
|
||||||
|
|
||||||
@ -321,16 +286,6 @@ class GroupChat(Chat):
|
|||||||
Chat.activity_on_disconnected_from_shell(self)
|
Chat.activity_on_disconnected_from_shell(self)
|
||||||
gtk.main_quit()
|
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):
|
def _on_buddyList_buddy_selected(self, widget, *args):
|
||||||
(model, aniter) = widget.get_selection().get_selected()
|
(model, aniter) = widget.get_selection().get_selected()
|
||||||
name = self._buddy_list_model.get(aniter, self._MODEL_COL_NICK)
|
name = self._buddy_list_model.get(aniter, self._MODEL_COL_NICK)
|
||||||
@ -372,25 +327,15 @@ class GroupChat(Chat):
|
|||||||
aniter = self._get_iter_for_buddy(buddy)
|
aniter = self._get_iter_for_buddy(buddy)
|
||||||
self._buddy_list_model.set(aniter, self._MODEL_COL_ICON, self._pixbuf_active_chat)
|
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):
|
def send_message(self, text):
|
||||||
if len(text) > 0:
|
if len(text) > 0:
|
||||||
self._gc_controller.send_msg(text)
|
self._output_pipe.send(text)
|
||||||
self._local_message(True, text)
|
self._local_message(True, text)
|
||||||
|
|
||||||
def recv_message(self, buddy, msg):
|
def recv_message(self, buddy, msg):
|
||||||
self._insert_rich_message(buddy.nick(), msg)
|
|
||||||
self._controller.notify_new_message(self, None)
|
|
||||||
|
|
||||||
def _recv_group_message(self, msg):
|
|
||||||
buddy = self.find_buddy_by_address(msg['addr'])
|
|
||||||
if buddy:
|
if buddy:
|
||||||
self.recv_message(buddy, msg['data'])
|
self._insert_rich_message(buddy.nick(), msg)
|
||||||
|
self._controller.notify_new_message(self, None)
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
try:
|
try:
|
||||||
|
158
chat/p2p.py
Normal file
158
chat/p2p.py
Normal file
@ -0,0 +1,158 @@
|
|||||||
|
import os
|
||||||
|
import pwd
|
||||||
|
|
||||||
|
import presence
|
||||||
|
import BuddyList
|
||||||
|
import network
|
||||||
|
|
||||||
|
class GroupRequestHandler(object):
|
||||||
|
def __init__(self, group):
|
||||||
|
self._group = group
|
||||||
|
|
||||||
|
def message(self, message):
|
||||||
|
address = network.get_authinfo()
|
||||||
|
self._group.recv(address, message)
|
||||||
|
|
||||||
|
class Owner:
|
||||||
|
instance = None
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
ent = pwd.getpwuid(os.getuid())
|
||||||
|
self._nick = ent[0]
|
||||||
|
if not self._nick or not len(self._nick):
|
||||||
|
self._nick = "n00b"
|
||||||
|
self._realname = ent[4]
|
||||||
|
if not self._realname or not len(self._realname):
|
||||||
|
self._realname = "Some Clueless User"
|
||||||
|
|
||||||
|
def get_realname(self):
|
||||||
|
return self._realname
|
||||||
|
|
||||||
|
def get_nick(self):
|
||||||
|
return self._nick
|
||||||
|
|
||||||
|
def get_instance():
|
||||||
|
if not Owner.instance:
|
||||||
|
Owner.instance = Owner()
|
||||||
|
return Owner.instance
|
||||||
|
|
||||||
|
get_instance = staticmethod(get_instance)
|
||||||
|
|
||||||
|
class Group:
|
||||||
|
instance = None
|
||||||
|
|
||||||
|
_SERVER_PORT = 6666
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self._pipes = {}
|
||||||
|
|
||||||
|
def get_instance():
|
||||||
|
if not Group.instance:
|
||||||
|
Group.instance = Group()
|
||||||
|
Group.instance.join()
|
||||||
|
return Group.instance
|
||||||
|
|
||||||
|
get_instance = staticmethod(get_instance)
|
||||||
|
|
||||||
|
def join(self):
|
||||||
|
self._pannounce = presence.PresenceAnnounce()
|
||||||
|
|
||||||
|
realname = Owner.get_instance().get_realname()
|
||||||
|
nick = Owner.get_instance().get_nick()
|
||||||
|
|
||||||
|
self._buddy_list = BuddyList.BuddyList(realname)
|
||||||
|
self._buddy_list.start()
|
||||||
|
|
||||||
|
self._pannounce.register_service(realname, self._SERVER_PORT, presence.OLPC_CHAT_SERVICE,
|
||||||
|
name = nick, realname = realname)
|
||||||
|
|
||||||
|
self._p2p_req_handler = GroupRequestHandler(self)
|
||||||
|
self._p2p_server = network.GlibXMLRPCServer(("", self._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 get_buddy_list(self):
|
||||||
|
return self._buddy_list
|
||||||
|
|
||||||
|
def _serialize_msg(self, pipe_id, msg):
|
||||||
|
return pipe_id + "|" + msg
|
||||||
|
|
||||||
|
def _deserialize_msg(self, msg):
|
||||||
|
sep_index = msg.find("|")
|
||||||
|
pipe_id = msg[0 : sep_index]
|
||||||
|
message = msg[sep_index + 1 :]
|
||||||
|
return [pipe_id, message]
|
||||||
|
|
||||||
|
def send(self, buddy, pipe_id, msg):
|
||||||
|
addr = "http://%s:%d" % (buddy.address(), buddy.port())
|
||||||
|
peer = xmlrpclib.ServerProxy(addr)
|
||||||
|
msg = text
|
||||||
|
success = True
|
||||||
|
try:
|
||||||
|
peer.message(self._serialize_msg(pipe_id, msg))
|
||||||
|
except (socket.error, xmlrpclib.Fault), e:
|
||||||
|
msg = str(e)
|
||||||
|
success = False
|
||||||
|
return success
|
||||||
|
|
||||||
|
def broadcast(self, pipe_id, msg):
|
||||||
|
self._gc_controller.send_msg(self._serialize_msg(pipe_id, msg))
|
||||||
|
|
||||||
|
def register_pipe(self, input_pipe):
|
||||||
|
self._pipes[input_pipe.get_id()] = input_pipe
|
||||||
|
|
||||||
|
def _recv_group_message(self, msg):
|
||||||
|
self.recv(msg['addr'], msg['data'])
|
||||||
|
|
||||||
|
def recv(self, address, message):
|
||||||
|
sender = self._buddy_list.find_buddy_by_address(address)
|
||||||
|
[pipe_id, msg] = self._deserialize_msg(message)
|
||||||
|
pipe = self._pipes[pipe_id]
|
||||||
|
if pipe:
|
||||||
|
pipe.recv(sender, msg)
|
||||||
|
|
||||||
|
class AbstractPipe:
|
||||||
|
def __init__(self, group, pipe_id=None):
|
||||||
|
self._group = group
|
||||||
|
self._pipe_id = pipe_id
|
||||||
|
|
||||||
|
def get_id(self):
|
||||||
|
return self._pipe_id
|
||||||
|
|
||||||
|
def send(self, msg):
|
||||||
|
pass
|
||||||
|
|
||||||
|
class AbstractOutputPipe(AbstractPipe):
|
||||||
|
def __init__(self, group, pipe_id=None):
|
||||||
|
AbstractPipe.__init__(self, group, pipe_id)
|
||||||
|
|
||||||
|
def send(self, msg):
|
||||||
|
pass
|
||||||
|
|
||||||
|
class OutputPipe(AbstractOutputPipe):
|
||||||
|
def __init__(self, group, buddy, pipe_id=None):
|
||||||
|
AbstractOutputPipe.__init__(self, group, pipe_id)
|
||||||
|
self._buddy = buddy
|
||||||
|
|
||||||
|
def send(self, msg):
|
||||||
|
self._group.send(self._buddy, self._pipe_id, msg)
|
||||||
|
|
||||||
|
class BroadcastOutputPipe(AbstractOutputPipe):
|
||||||
|
def __init__(self, group, pipe_id=None):
|
||||||
|
AbstractOutputPipe.__init__(self, group, pipe_id)
|
||||||
|
|
||||||
|
def send(self, msg):
|
||||||
|
self._group.broadcast(self._pipe_id, msg)
|
||||||
|
|
||||||
|
class InputPipe(AbstractPipe):
|
||||||
|
def __init__(self, group, pipe_id=None):
|
||||||
|
AbstractPipe.__init__(self, group, pipe_id)
|
||||||
|
group.register_pipe(self)
|
||||||
|
|
||||||
|
def listen(self, callback):
|
||||||
|
self.__callback = callback
|
||||||
|
|
||||||
|
def recv(self, sender, msg):
|
||||||
|
self.__callback(sender, msg)
|
Loading…
Reference in New Issue
Block a user