Private messages! fixed #1077

6 files changed, 162 insertions, 26 deletions

diff --git a/src/connection.py b/src/connection.py
index 5c94b3fa..e20aaab5 100644
--- a/src/connection.py
+++ b/src/connection.py
@@ -173,7 +173,10 @@ class Connection(threading.Thread):
         if message.getType() == 'error':
             self.error_message(message)
             return
-        self.handler.emit('room-message', stanza=message)
+        if message.getType() == 'groupchat':
+            self.handler.emit('room-message', stanza=message)
+        else:
+            self.handler.emit('private-message', stanza=message)
         raise xmpp.protocol.NodeProcessed
 
     def process(self, timeout=10):
@@ -213,3 +216,10 @@ def jid_get_domain(jid):
     if isinstance(jid, basestring):
         jid = xmpp.JID(jid)
     return jid.getDomain()
+
+def is_jid_the_same(a, b):
+    if isinstance(a, basestring):
+        a = xmpp.JID(a)
+    if isinstance(b, basestring):
+        b = xmpp.JID(b)
+    return a.bareMatch(b)
diff --git a/src/gui.py b/src/gui.py
index 685bcb97..b127abb5 100644
--- a/src/gui.py
+++ b/src/gui.py
@@ -40,6 +40,8 @@ from user import User
 from room import Room
 from message import Message
 
+from connection import is_jid_the_same
+
 from common import debug
 def doupdate():
     debug("doupdate")
@@ -83,6 +85,8 @@ class Gui(object):
             'set': (self.command_set, _("Usage: /set <option> [value]\nSet: Sets the value to the option in your configuration file. You can, for example, change your default nickname by doing `/set default_nick toto` or your resource with `/set resource blabla`. You can also set an empty value (nothing) by providing no [value] after <option>.")),
             'kick': (self.command_kick, _("Usage: /kick <nick> [reason]\nKick: Kick the user with the specified nickname. You also can give an optional reason.")),
             'topic': (self.command_topic, _("Usage: /topic <subject> \nTopic: Change the subject of the room")),
+            'query': (self.command_query, _('Usage: /query <nick>\nQuery: Open a private conversation with <nick>. This nick has to be present in the room you\'re currently in.')),
+
             'nick': (self.command_nick, _("Usage: /nick <nickname> \nNick: Change your nickname in the current room"))
             }
 
@@ -109,6 +113,7 @@ class Gui(object):
         self.handler.connect('join-room', self.join_room)
         self.handler.connect('room-presence', self.room_presence)
         self.handler.connect('room-message', self.room_message)
+        self.handler.connect('private-message', self.private_message)
         self.handler.connect('error-message', self.room_error)
         self.handler.connect('error', self.information)
 
@@ -195,7 +200,7 @@ class Gui(object):
                          curses.COLOR_RED) # highlight room
         curses.init_pair(14, curses.COLOR_WHITE,
                          curses.COLOR_YELLOW)
-        curses.init_pair(15, curses.COLOR_WHITE,
+        curses.init_pair(15, curses.COLOR_WHITE, # new message in private room
                          curses.COLOR_GREEN)
 
     def reset_curses(self):
@@ -270,12 +275,55 @@ class Gui(object):
         room = self.get_room_by_name(room)
         code = error.getAttr('code')
         typ = error.getAttr('type')
-        body = error.getTag('text').getData()
+        if error.getTag('text'):
+            body = error.getTag('text').getData()
+        else:
+            body = _('Unknown error')
         self.add_message_to_room(room, _('Error: %(code)s-%(msg)s: %(body)s' %
                                    {'msg':msg, 'code':code, 'body':body}))
         if code == '401':
             room.add(_('To provide a password in order to join the room, type "/join / password" (replace "password" by the real password)'))
 
+    def private_message(self, stanza):
+        """
+        When a private message is received
+        """
+        jid = stanza.getFrom()
+        nick_from = stanza.getFrom().getResource()
+        room_from = stanza.getFrom().getStripped()
+        room = self.get_room_by_name(jid) # get the tab with the private conversation
+        if not room: # It's the first message we receive: create the tab
+            room = self.open_private_window(room_from, nick_from, False)
+        body = stanza.getBody()
+        self.add_message_to_room(room, body, None, nick_from)
+        self.window.input.refresh()
+        doupdate()
+
+    def open_private_window(self, room_name, user_nick, focus=True):
+        complete_jid = room_name+'/'+user_nick
+        for room in self.rooms: # if the room exists, focus it and return
+            if room.jid:
+                if room.jid == complete_jid:
+                    self.command_win(str(room.nb))
+                    return
+        # create the new tab
+        own_nick = self.get_room_by_name(room_name).own_nick
+        r = Room(complete_jid, own_nick, self.window, complete_jid)
+        # insert it in the rooms
+        if self.current_room().nb == 0:
+            self.rooms.append(r)
+        else:
+            for ro in self.rooms:
+                if ro.nb == 0:
+                    self.rooms.insert(self.rooms.index(ro), r)
+                    break
+        if focus:               # focus the room if needed
+            while self.current_room().nb != r.nb:
+                self.rooms.insert(0, self.rooms.pop())
+        # self.window.new_room(r)
+        self.window.refresh(self.rooms)
+        return r
+
     def room_message(self, stanza, date=None):
         """
         Display the message on the room window
@@ -356,8 +404,22 @@ class Gui(object):
                 elif change_nick:
                     if user.nick == room.own_nick:
                         room.own_nick = stanza.getNick().encode('utf-8')
+                        # also change our nick in all private discussion of this room
+                        for _room in self.rooms:
+                            if _room.jid is not None and is_jid_the_same(_room.jid, room.name):
+                                debug(_room.jid)
+                                debug(room.name)
+                                _room.own_nick = stanza.getNick()
                     user.change_nick(stanza.getNick())
                     self.add_message_to_room(room, _('%(old)s is now known as %(new)s') % {'old':from_nick, 'new':stanza.getNick()})
+                    # rename the private tabs if needed
+                    private_room = self.get_room_by_name(stanza.getFrom())
+                    if private_room:
+                        self.add_message_to_room(private_room, _('%(old_nick)s is now known as %(new_nick)s') % {'old_nick':from_nick, 'new_nick':stanza.getNick()})
+                        new_jid = private_room.name.split('/')[0]+'/'+stanza.getNick()
+                        private_room.jid = new_jid
+                        private_room.name = new_jid
+
                 # kick
                 elif kick:
                     room.users.remove(user)
@@ -386,8 +448,22 @@ class Gui(object):
                     hide_exit_join = config.get('hide_exit_join', -1) if config.get('hide_exit_join', -1) >= -1 else -1
                     if hide_exit_join == -1 or user.has_talked_since(hide_exit_join):
                         self.add_message_to_room(room, _('%s has left the room') % (from_nick))
+                    private_room = self.get_room_by_name(stanza.getFrom())
+                    if private_room:
+                        self.add_message_to_room(private_room, _('%s has left the room') % (from_nick))
                 # status change
                 else:
+                    # build the message
+                    msg = _('%s changed his/her status: ')% from_nick
+                    if affiliation != user.affiliation:
+                        msg += _('affiliation: %s,') % affiliation
+                    if role != user.role:
+                        msg += _('role: %s,') % role
+                    if show != user.show:
+                        msg += _('show: %s,') % show
+                    if status != user.status:
+                        msg += _('status: %s,') % status
+                    msg = msg[:-1] # remove the last ","
                     hide_status_change = config.get('hide_status_change', -1) if config.get('hide_status_change', -1) >= -1 else -1
                     if (hide_status_change == -1 or \
                             user.has_talked_since(hide_status_change) or\
@@ -397,23 +473,18 @@ class Gui(object):
                                 role != user.role or\
                                 show != user.show or\
                                 status != user.status):
-                        msg = _('%s changed his/her status: ')% from_nick
-                        if affiliation != user.affiliation:
-                            msg += _('affiliation: %s,') % affiliation
-                        if role != user.role:
-                            msg += _('role: %s,') % role
-                        if show != user.show:
-                            msg += _('show: %s,') % show
-                        if status != user.status:
-                            msg += _('status: %s,') % status
-                        # (a)s, %(b)s, %(c)s, %(d)s') % {'nick':from_nick, 'a':affiliation, 'b':role, 'c':show, 'd':status})
-                        user.update(affiliation, show, status, role)
-                        msg = msg[:-1] # remove the last ","
+                        # display the message in the room
                         self.add_message_to_room(room, msg)
-
+                    private_room = self.get_room_by_name(stanza.getFrom())
+                    # debug('status change: ' + stanza.getFrom()+'\n')
+                    if private_room: # display the message in private
+                        self.add_message_to_room(private_room, msg)
+                    # finally, effectively change the user status
+                    user.update(affiliation, show, status, role)
             if room == self.current_room():
                 self.window.user_win.refresh(room.users)
         self.window.input.refresh()
+        self.window.info_win.refresh(self.rooms, self.current_room())
         doupdate()
 
     def add_message_to_room(self, room, txt, time=None, nickname=None):
@@ -446,7 +517,11 @@ class Gui(object):
             else:
                 self.add_message_to_room(self.current_room(), _("Error: unknown command (%s)") % (command))
         elif self.current_room().name != 'Info':
-            self.muc.send_message(self.current_room().name, line)
+            if self.current_room().jid is not None:
+                self.muc.send_private_message(self.current_room().name, line)
+                self.add_message_to_room(self.current_room(), line.decode('utf-8'), None, self.current_room().own_nick)
+            else:
+                self.muc.send_message(self.current_room().name, line)
         self.window.input.refresh()
         doupdate()
 
@@ -703,6 +778,17 @@ class Gui(object):
         self.rooms.remove(self.current_room())
         self.window.refresh(self.rooms)
 
+    def command_query(self, args):
+        if len(args) != 1:
+            return
+        nick = args[0]
+        room = self.current_room()
+        if room.name == "Info" or room.jid is not None:
+            return
+        for user in room.users:
+            if user.nick == nick:
+                self.open_private_window(room.name, user.nick)
+
     def command_topic(self, args):
         """
         /topic [new topic]
diff --git a/src/handler.py b/src/handler.py
index 69669af0..a8bebe23 100644
--- a/src/handler.py
+++ b/src/handler.py
@@ -46,6 +46,10 @@ class Handler(Singleton):
                 # A message is received
                 # Args: the stanza object
 
+                'private-message': list(),
+                # A message is received
+                # Args: the stanza object
+
                 'room-delayed-message': list(),
                 # A message is received
                 # Args: the stanza object
diff --git a/src/multiuserchat.py b/src/multiuserchat.py
index fec1d0d3..4eaa696c 100644
--- a/src/multiuserchat.py
+++ b/src/multiuserchat.py
@@ -149,6 +149,13 @@ class MultiUserChat(object):
         mes.setType('groupchat')
         self.connection.send(mes)
 
+    def send_private_message(self, user_jid, message):
+        from common import debug
+        mes = Message(to=user_jid)
+        mes.setBody(message)
+        mes.setType('chat')
+        self.connection.send(mes)
+
     def join_room(self, room, nick, password=None):
         """Join a new room"""
         pres = Presence(to='%s/%s' % (room, nick))
diff --git a/src/room.py b/src/room.py
index 28f8cfad..bea68d05 100644
--- a/src/room.py
+++ b/src/room.py
@@ -27,7 +27,8 @@ class Room(object):
     """
     """
     number = 0
-    def __init__(self, name, nick, window):
+    def __init__(self, name, nick, window, jid=None):
+        self.jid = jid          # used for a private chat. None if it's a MUC
         self.name = name
         self.own_nick = nick
         self.color_state = 11   # color used in RoomInfo
@@ -72,7 +73,10 @@ class Room(object):
         time = time if time is not None else datetime.now()
         color = None
         if nickname is not None:
-            self.set_color_state(12)
+            if not self.jid:
+                self.set_color_state(12)
+            else:
+                self.set_color_state(15)
         else:
             color = 8
         if nickname != self.own_nick and self.joined and nickname is not None: # do the highlight
diff --git a/src/window.py b/src/window.py
index 4c6b27bb..879151c3 100644
--- a/src/window.py
+++ b/src/window.py
@@ -17,6 +17,17 @@
 # You should have received a copy of the GNU General Public License
 # along with Poezio.  If not, see <http://www.gnu.org/licenses/>.
 
+
+from gettext import (bindtextdomain, textdomain, bind_textdomain_codeset,
+                     gettext as _)
+
+bindtextdomain('poezio')
+textdomain('poezio')
+bind_textdomain_codeset('poezio', 'utf-8')
+
+import locale
+locale.setlocale(locale.LC_ALL, '')
+
 import curses
 from config import config
 
@@ -108,14 +119,23 @@ class Topic(Win):
     def resize(self, height, width, y, x, stdscr, visible):
         self._resize(height, width, y, x, stdscr)
 
-    def refresh(self, room_name):
+    def refresh(self, topic, jid=None):
         if not self.visible:
             return
         self.win.erase()
-        try:
-            self.win.addnstr(0, 0, room_name + " "*(self.width-len(room_name)), self.width
-                             , curses.color_pair(1))
-        except:pass
+        if not jid:
+            try:
+                self.win.addnstr(0, 0, topic + " "*(self.width-len(topic)), self.width
+                                 , curses.color_pair(1))
+            except:
+                pass
+        elif jid:
+            room = jid.split('/')[0]
+            nick = '/'.join(jid.split('/')[1:])
+            topic = _('%(nick)s from room %(room)s' % {'nick': nick, 'room':room})
+            self.win.addnstr(0, 0, topic.encode('utf-8') + " "*(self.width-len(topic)), self.width-1
+                             , curses.color_pair(15))
+
         self.win.refresh()
 
 class RoomInfo(Win):
@@ -147,9 +167,14 @@ class RoomInfo(Win):
                 break
         (y, x) = self.win.getyx()
         try:
-            self.win.addstr(y, x-1, '] '+ current.name+ (' '*((self.width)-x)), curses.color_pair(1))
+            self.win.addstr(y, x-1, '] '+ current.name.encode('utf-8') , curses.color_pair(1))
         except:
             pass
+        while True:
+            try:
+                self.win.addstr(' ', curses.color_pair(1))
+            except:
+                break
         self.win.refresh()
 
 class TextWin(Win):
@@ -577,7 +602,7 @@ class Window(object):
         # self.text_win.redraw(room)
         self.text_win.refresh(room)
         self.user_win.refresh(room.users)
-        self.topic_win.refresh(room.topic)
+        self.topic_win.refresh(room.topic, room.jid)
         self.info_win.refresh(rooms, room)
         self.input.refresh()