gui.py 47.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
# -*- coding:utf-8 -*-
#
# Copyright 2010 Le Coz Florent <louizatakk@fedoraproject.org>
#
# This file is part of Poezio.
#
# Poezio is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# Poezio is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Poezio.  If not, see <http://www.gnu.org/licenses/>.

19 20
from gettext import (bindtextdomain, textdomain, bind_textdomain_codeset,
                     gettext as _)
21
from os.path import isfile
22

23 24
from time import sleep

25
import sys
26
import os
27
import re
28
import curses
29 30
import webbrowser

31
from datetime import datetime
32 33

import common
34
import theme
35

36
import multiuserchat as muc
37
from handler import Handler
38
from config import config
39
from window import Window
40 41
from user import User
from room import Room
42
from message import Message
43
from keyboard import read_char
44
from common import is_jid_the_same, jid_get_domain, jid_get_resource, is_jid
45

46 47 48 49 50 51 52 53 54 55 56 57
# http://xmpp.org/extensions/xep-0045.html#errorstatus
ERROR_AND_STATUS_CODES = {
    '401': 'A password is required',
    '403': 'You are banned from the room',
    '404': 'The room does\'nt exist',
    '405': 'Your are not allowed to create a new room',
    '406': 'A reserved nick must be used',
    '407': 'You are not in the member list',
    '409': 'This nickname is already in use or has been reserved',
    '503': 'The maximum number of users has been reached',
    }

58 59 60 61 62 63 64
SHOW_NAME = {
    'dnd': _('busy'),
    'away': _('away'),
    'xa': _('not available'),
    'chat': _('chatty'),
    '': _('available')
    }
65 66
def doupdate():
    curses.doupdate()
67

68 69
class Gui(object):
    """
70
    User interface using ncurses
71
    """
72 73 74 75 76
    def __init__(self, xmpp):
        self.stdscr = curses.initscr()
        self.init_curses(self.stdscr)
        self.xmpp = xmpp
        self.window = Window(self.stdscr)
77
        self.rooms = [Room('Info', '', self.window)]
78
        self.ignores = {}
79

80
        self.commands = {
81
            'help': (self.command_help, u'\_o< KOIN KOIN KOIN'),
82
            'join': (self.command_join, _("Usage: /join [room_name][@server][/nick] [password]\nJoin: Join the specified room. You can specify a nickname after a slash (/). If no nickname is specified, you will use the default_nick in the configuration file. You can omit the room name: you will then join the room you\'re looking at (useful if you were kicked). You can also provide a room_name without specifying a server, the server of the room you're currently in will be used. You can also provide a password to join the room.\nExamples:\n/join room@server.tld\n/join room@server.tld/John\n/join room2\n/join /me_again\n/join\n/join room@server.tld/my_nick password\n/join / password")),
83 84 85 86 87 88 89 90
            'quit': (self.command_quit, _("Usage: /quit\nQuit: Just disconnect from the server and exit poezio.")),
            'exit': (self.command_quit, _("Usage: /exit\nExit: Just disconnect from the server and exit poezio.")),
            'next': (self.rotate_rooms_right, _("Usage: /next\nNext: Go to the next room.")),
            'n': (self.rotate_rooms_right, _("Usage: /n\nN: Go to the next room.")),
            'prev': (self.rotate_rooms_left, _("Usage: /prev\nPrev: Go to the previous room.")),
            'p': (self.rotate_rooms_left, _("Usage: /p\nP: Go to the previous room.")),
            'win': (self.command_win, _("Usage: /win <number>\nWin: Go to the specified room.")),
            'w': (self.command_win, _("Usage: /w <number>\nW: Go to the specified room.")),
91 92
            'ignore': (self.command_ignore, _("Usage: /ignore <nickname> \nIgnore: Ignore a specified nickname.")),
            'unignore': (self.command_unignore, _("Usage: /unignore <nickname>\nUnignore: Remove the specified nickname from the ignore list.")),
93 94 95 96 97 98 99
            'part': (self.command_part, _("Usage: /part [message]\n Part: disconnect from a room. You can specify an optional message.")),
            'show': (self.command_show, _("Usage: /show <availability> [status]\nShow: Change your availability and (optionaly) your status. The <availability> argument is one of \"avail, available, ok, here, chat, away, afk, dnd, busy, xa\" and the optional [message] argument will be your status message")),
            'away': (self.command_away, _("Usage: /away [message]\nAway: Sets your availability to away and (optional) sets your status message. This is equivalent to '/show away [message]'")),
            'busy': (self.command_busy, _("Usage: /busy [message]\nBusy: Sets your availability to busy and (optional) sets your status message. This is equivalent to '/show busy [message]'")),
            'avail': (self.command_avail, _("Usage: /avail [message]\nAvail: Sets your availability to available and (optional) sets your status message. This is equivalent to '/show available [message]'")),
            'available': (self.command_avail, _("Usage: /available [message]\nAvailable: Sets your availability to available and (optional) sets your status message. This is equivalent to '/show available [message]'")),
           'bookmark': (self.command_bookmark, _("Usage: /bookmark [roomname][/nick]\nBookmark: Bookmark the specified room (you will then auto-join it on each poezio start). This commands uses the same syntaxe as /join. Type /help join for syntaxe examples. Note that when typing \"/bookmark\" on its own, the room will be bookmarked with the nickname you\'re currently using in this room (instead of default_nick)")),
100
            'unquery': (self.command_unquery, _("Usage: /unquery\nClose the private conversation window")),
101 102 103
            '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")),
104
            'link': (self.command_link, _("Usage: /link [option] [number]\nLink: Interact with a link in the conversation. Available options are 'open', 'copy'. Open just opens the link in the browser if it's http://, Copy just copy the link in the clipboard. An optional number can be provided, it indicates which link to interact with.")),
105
            'query': (self.command_query, _('Usage: /query <nick> [message]\nQuery: Open a private conversation with <nick>. This nick has to be present in the room you\'re currently in. If you specified a message after the nickname, it will immediately be sent to this user')),
106 107
            'nick': (self.command_nick, _("Usage: /nick <nickname>\nNick: Change your nickname in the current room")),
            'say': (self.command_say, _('Usage: /say <message>\nSay: Just send the message. Useful if you want your message to begin with a "/"')),
108
            'whois': (self.command_whois, _('Usage: /whois <nickname>\nWhois: Request many informations about the user.')),
109
            'theme': (self.command_theme, _('Usage: /theme\nTheme: Reload the theme defined in the config file.')),
110 111
            }

112 113 114 115 116 117 118
        self.key_func = {
            "KEY_LEFT": self.window.input.key_left,
            "KEY_RIGHT": self.window.input.key_right,
            "KEY_UP": self.window.input.key_up,
            "KEY_END": self.window.input.key_end,
            "KEY_HOME": self.window.input.key_home,
            "KEY_DOWN": self.window.input.key_down,
119 120
            "KEY_PPAGE": self.scroll_page_up,
            "KEY_NPAGE": self.scroll_page_down,
121
            "KEY_DC": self.window.input.key_dc,
122
            "KEY_F(5)": self.rotate_rooms_left,
123
            "^P": self.rotate_rooms_left,
124
            "KEY_F(6)": self.rotate_rooms_right,
125
            "^N": self.rotate_rooms_right,
126 127
            "\t": self.completion,
            "^I": self.completion,
128
            "KEY_BTAB": self.last_words_completion,
129
            "KEY_RESIZE": self.resize_window,
130
            "KEY_BACKSPACE": self.window.input.key_backspace,
131
            '^?': self.window.input.key_backspace,
132 133 134 135 136 137 138 139 140 141 142 143
            '^J': self.execute,
            '\n': self.execute,
            '^D': self.window.input.key_dc,
            '^W': self.window.input.delete_word,
            '^K': self.window.input.delete_end_of_line,
            '^U': self.window.input.delete_begining_of_line,
            '^Y': self.window.input.paste_clipboard,
            '^A': self.window.input.key_home,
            '^E': self.window.input.key_end,
            'M-f': self.window.input.jump_word_right,
            '^X': self.go_to_important_room,
            'M-b': self.window.input.jump_word_left
144 145
            }

146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197
        # Add handlers
        self.xmpp.add_event_handler("session_start", self.on_connected)
        self.xmpp.add_event_handler("groupchat_presence", self.on_groupchat_presence)
        self.xmpp.add_event_handler("groupchat_message", self.on_groupchat_message)
        self.xmpp.add_event_handler("message", self.on_message)
        # self.handler = Handler()
        # self.handler.connect('on-connected', self.on_connected)
        # 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)

    def on_connected(self, event):
        """
        Called when we are connected and authenticated
        """
        self.information(_("Welcome on Poezio \o/!"))
        self.information(_("Your JID is %s") % self.xmpp.fulljid)

        rooms = config.get('rooms', '')
        if rooms == '' or not isinstance(rooms, str):
            return
        rooms = rooms.split(':')
        for room in rooms:
            args = room.split('/')
            if args[0] == '':
                return
            roomname = args[0]
            if len(args) == 2:
                nick = args[1]
            else:
                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
                nick = config.get('default_nick', '')
                if nick == '':
                    nick = default
            self.open_new_room(roomname, nick)
            muc.join_groupchat(self.xmpp, roomname, nick)
        # Todo: SEND VCARD
        return
        if config.get('jid', '') == '': # Don't send the vcard if we're not anonymous
            self.vcard_sender.start()   # because the user ALREADY has one on the server

    def on_groupchat_presence(self, presence):
        """
        Triggered whenever a presence stanza is received from a user in a multi-user chat room.
        Display the presence on the room window and update the
        presence information of the concerned user
        """
        from_nick = presence['from'].resource
        from_room = presence['from'].bare
198
        room = self.get_room_by_name(from_room)
199 200 201 202 203
        code = presence.find('{jabber:client}status')
        status_codes = set([s.attrib['code'] for s in presence.findall('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}status')])
        # Check if it's not an error presence.
        if presence['type'] == 'error':
            return self.room_error(presence, from_room)
204
        if not room:
205
            return
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
        msg = None
        affiliation = presence['muc']['affiliation']
        show = presence['show']
        status = presence['status']
        role = presence['muc']['role']
        jid = presence['muc']['jid']
        typ = presence['type']
        if not room.joined:     # user in the room BEFORE us.
            # ignore redondant presence message, see bug #1509
            if from_nick not in [user.nick for user in room.users]:
                new_user = User(from_nick, affiliation, show, status, role)
                room.users.append(new_user)
                if from_nick.encode('utf-8') == room.own_nick:
                    room.joined = True
                    new_user.color = theme.COLOR_OWN_NICK
                    self.add_message_to_room(room, _("Your nickname is %s") % (from_nick))
                    if '170' in status_codes:
                        self.add_message_to_room(room, 'Warning: this room is publicly logged')
224
        else:
225 226 227 228 229 230 231 232
            change_nick = '303' in status_codes
            kick = '307' in status_codes and typ == 'unavailable'
            user = room.get_user_by_name(from_nick)
            # New user
            if not user:
                self.on_user_join(room, from_nick, affiliation, show, status, role, jid)
            # nick change
            elif change_nick:
233
                self.on_user_nick_change(room, presence, user, from_nick, from_room)
234 235
            # kick
            elif kick:
236
                self.on_user_kicked(room, presence, user, from_nick)
237 238
            # user quit
            elif typ == 'unavailable':
239
                self.on_user_leave_groupchat(room, user, jid, status, from_nick, from_room)
240 241
            # status change
            else:
242
                self.on_user_change_status(room, user, from_nick, from_room, affiliation, role, show, status)
243 244
        if room == self.current_room():
            self.window.user_win.refresh(room.users)
245 246 247
        self.window.input.refresh()
        doupdate()

248 249 250 251 252 253 254 255 256 257 258 259 260
    def on_user_join(self, room, from_nick, affiliation, show, status, role, jid):
        """
        When a new user joins a groupchat
        """
        room.users.append(User(from_nick, affiliation,
                               show, status, role))
        hide_exit_join = config.get('hide_exit_join', -1)
        if hide_exit_join != 0:
            if not jid.full:
                self.add_message_to_room(room, _("%(spec)s [%(nick)s] joined the room") % {'nick':from_nick, 'spec':theme.CHAR_JOIN}, colorized=True)
            else:
                self.add_message_to_room(room, _("%(spec)s [%(nick)s] (%(jid)s) joined the room") % {'spec':theme.CHAR_JOIN, 'nick':from_nick, 'jid':jid.full}, colorized=True)

261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277
    def on_user_nick_change(self, room, presence, user, from_nick, from_room):
        new_nick = presence.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item').attrib['nick']
        if user.nick == room.own_nick:
            room.own_nick = new_nick
            # 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):
                    _room.own_nick = new_nick
        user.change_nick(new_nick)
        self.add_message_to_room(room, _('[%(old)s] is now known as [%(new)s]') % {'old':from_nick, 'new':new_nick}, colorized=True)
        # rename the private tabs if needed
        private_room = self.get_room_by_name('%s/%s' % (from_room, from_nick))
        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':new_nick}, colorized=True)
            new_jid = private_room.name.split('/')[0]+'/'+new_nick
            private_room.jid = private_room.name = new_jid

278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304
    def on_user_kicked(self, room, presence, user, from_nick):
        """
        When someone is kicked
        """
        room.users.remove(user)
        by = presence.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item/{http://jabber.org/protocol/muc#user}actor')
        reason = presence.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item/{http://jabber.org/protocol/muc#user}reason')
        by = by.attrib['jid'] if by else ''
        reason = reason.text# if reason else ''
        if from_nick == room.own_nick: # we are kicked
            room.disconnect()
            if by:
                kick_msg = _("%(spec) [You] have been kicked by [%(by)s].") % {'spec': theme.CHAR_KICK, 'by':by}
            else:
                kick_msg = _("%(spec)s [You] have been kicked.") % {'spec':theme.CHAR_KICK}
            # try to auto-rejoin
            if config.get('autorejoin', 'false') == 'true':
                muc.join_groupchat(self.xmpp, room.name, room.own_nick)
        else:
            if by:
                kick_msg = _("%(spec)s [%(nick)s] has been kicked by %(by)s.") % {'spec':theme.CHAR_KICK, 'nick':from_nick, 'by':by}
            else:
                kick_msg = _("%(spec)s [%(nick)s] has been kicked") % {'spec':theme.CHAR_KICK, 'nick':from_nick}
        if reason:
            kick_msg += _(' Reason: %(reason)s') % {'reason': reason}
        self.add_message_to_room(room, kick_msg, colorized=True)

305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325
    def on_user_leave_groupchat(self, room, user, jid, status, from_nick, from_room):
        """
        When an user leaves a groupchat
        """
        room.users.remove(user)
        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):
            if not jid.full:
                leave_msg = _('%(spec)s [%(nick)s] has left the room') % {'nick':from_nick, 'spec':theme.CHAR_QUIT}
            else:
                leave_msg = _('%(spec)s [%(nick)s] (%(jid)s) has left the room') % {'spec':theme.CHAR_QUIT, 'nick':from_nick, 'jid':jid.full}
            if status:
                leave_msg += ' (%s)' % status
            self.add_message_to_room(room, leave_msg, colorized=True)
        private_room = self.get_room_by_name('%s/%s' % (from_room, from_nick))
        if private_room:
            if not status:
                self.add_message_to_room(private_room, _('%(spec)s [%(nick)s] has left the room') % {'nick':from_nick, 'spec':theme.CHAR_QUIT}, colorized=True)
            else:
                self.add_message_to_room(private_room, _('%(spec)s [%(nick)s] has left the room (%(status)s)') % {'nick':from_nick, 'spec':theme.CHAR_QUIT, 'status': status}, colorized=True)

326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357
    def on_user_change_status(self, room, user, from_nick, from_room, affiliation, role, show, status):
        """
        When an user changes her status
        """
        # 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 and show in SHOW_NAME.keys():
            msg += _('show: %s,') % SHOW_NAME[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\
                user.nick == room.own_nick)\
                and\
                (affiliation != user.affiliation or\
                    role != user.role or\
                    show != user.show or\
                    status != user.status):
            # display the message in the room
            self.add_message_to_room(room, msg)
        private_room = self.get_room_by_name('%s/%s' % (from_room, from_nick))
        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)

358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376
    def on_message(self, message):
        """
        When receiving private message from a muc OR a normal message
        (from one of our contacts)
        """
        if message['type'] == 'groupchat':
            return None
        # Differentiate both type of messages, and call the appropriate handler.
        jid_from = message['from']
        for room in self.rooms:
            if room.jid is None and room.name == jid_from.bare: # check all the MUC we are in
                return self.on_groupchat_private_message(message)
        return self.on_normal_message(message)

    def on_groupchat_private_message(self, message):
        """
        We received a Private Message (from someone in a Muc)
        """
        jid = message['from']
377 378
        nick_from = jid.resource
        room_from = jid.bare
379 380 381 382 383 384 385 386 387 388 389 390 391 392 393
        room = self.get_room_by_name(jid.full) # 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.encode('utf-8'), False)
            if not room:
                return
        body = message['body']
        self.add_message_to_room(room, body, None, nick_from)
        self.window.input.refresh()
        doupdate()

    def on_normal_message(self, message):
        """
        When receiving "normal" messages (from someone in our roster)
        """
        return
394

395 396 397 398 399 400 401
    def resize_window(self):
        """
        Resize the whole screen
        """
        self.window.resize(self.stdscr)
        self.window.refresh(self.rooms)

402
    def main_loop(self):
403 404 405
        """
        main loop waiting for the user to press a key
        """
406
        self.refresh_window()
407
        while True:
408
            doupdate()
409
            char=read_char(self.stdscr)
410
            try: # if this is not a valide utf-8 char, discard it
411
                char.decode('utf-8')
412 413
            except UnicodeDecodeError:
                continue
414 415 416
            # search for keyboard shortcut
            if char in self.key_func.keys():
                self.key_func[char]()
417
            else:
418 419
                if len(char.decode('utf-8')) > 1:
                    continue    # ignore non-handled keyboard shortcuts
420
                self.window.do_command(char)
421

422
    def current_room(self):
423 424 425
        """
        returns the current room, the one we are viewing
        """
426
        return self.rooms[0]
427 428

    def get_room_by_name(self, name):
429 430 431 432
        """
        returns the room that has this name
        """
        for room in self.rooms:
433 434 435 436 437 438
            try:
                if room.name.decode('utf-8') == name:
                    return room
            except UnicodeEncodeError:
                if room.name == name:
                    return room
439
        return None
440

441
    def init_curses(self, stdscr):
442 443 444
        """
        ncurses initialization
        """
445
        theme.init_colors()
446
        curses.noecho()
447
        curses.curs_set(0)
448
        stdscr.keypad(True)
449

450
    def reset_curses(self):
451 452 453 454
        """
        Reset terminal capabilities to what they were before ncurses
        init
        """
455
        # TODO remove me?
456
        curses.echo()
457
        curses.nocbreak()
458
        curses.endwin()
459

460 461 462 463
    def refresh_window(self):
        """
        Refresh everything
        """
464
        self.current_room().set_color_state(theme.COLOR_TAB_CURRENT)
465 466
        self.window.refresh(self.rooms)

467
    def open_new_room(self, room, nick, focus=True):
468
        """
469
        Open a new Tab containing a Muc room, using the specified nick
470
        """
471
        r = Room(room, nick, self.window)
472
        self.current_room().set_color_state(theme.COLOR_TAB_NORMAL)
473 474 475 476 477 478 479
        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
480 481
        if focus:
            self.command_win("%s" % r.nb)
482
        self.refresh_window()
483

484
    def completion(self):
485 486 487
        """
        Called when Tab is pressed, complete the nickname in the input
        """
488 489
        def compare_users(a, b):
            """
490
            Used to sort users by their last_talked
491
            """
492
            if not a.last_talked and b.last_talked:
493
                return 0
494 495 496 497 498
            elif not b.last_talked and a.last_talked:
                return 1
            if a.last_talked <  b.last_talked:
                return 1
            else:
499
                return -1
500
        self.window.input.auto_completion([user.nick for user in sorted(self.current_room().users, compare_users)])
501

502 503 504 505 506 507 508
    def last_words_completion(self):
        """
        Complete the input with words recently said
        """
        # build the list of the recent words
        char_we_dont_want = [',', '(', ')', '.']
        words = list()
509
        for msg in self.current_room().messages[:-9:-1]:
510 511
            if not msg:
                continue
512 513 514 515
            for word in msg.txt.split():
                for char in char_we_dont_want: # remove the chars we don't want
                    word = word.replace(char, '')
                if len(word) > 5:
516
                    words.append(word.encode('utf-8'))
517 518
        self.window.input.auto_completion(words)

519 520 521 522 523 524 525 526
    def go_to_important_room(self):
        """
        Go to the next room with activity, in this order:
        - A personal conversation with a new message
        - A Muc with an highlight
        - A Muc with any new message
        """
        for room in self.rooms:
527
            if room.color_state == theme.COLOR_TAB_PRIVATE:
528
                self.command_win('%s' % room.nb)
529 530
                return
        for room in self.rooms:
531
            if room.color_state == theme.COLOR_TAB_HIGHLIGHT:
532
                self.command_win('%s' % room.nb)
533 534
                return
        for room in self.rooms:
535
            if room.color_state == theme.COLOR_TAB_NEW_MESSAGE:
536 537
                self.command_win('%s' % room.nb)
                return
538

539
    def rotate_rooms_right(self, args=None):
540 541 542
        """
        rotate the rooms list to the right
        """
543
        self.current_room().set_color_state(theme.COLOR_TAB_NORMAL)
544
        self.current_room().remove_line_separator()
545
        self.rooms.append(self.rooms.pop(0))
546
        self.current_room().set_color_state(theme.COLOR_TAB_CURRENT)
547
        self.refresh_window()
548

549
    def rotate_rooms_left(self, args=None):
550 551 552
        """
        rotate the rooms list to the right
        """
553
        self.current_room().set_color_state(theme.COLOR_TAB_NORMAL)
554
        self.current_room().remove_line_separator()
555
        self.rooms.insert(0, self.rooms.pop())
556
        self.current_room().set_color_state(theme.COLOR_TAB_CURRENT)
557
        self.refresh_window()
558

559
    def scroll_page_down(self, args=None):
560
        self.current_room().scroll_down(self.window.text_win.height-1)
561
        self.refresh_window()
562 563

    def scroll_page_up(self, args=None):
564
        self.current_room().scroll_up(self.window.text_win.height-1)
565
        self.refresh_window()
566

567
    def room_error(self, error, room_name):
568 569 570
        """
        Display the error on the room window
        """
571
        room = self.get_room_by_name(room_name)
572 573
        if not room:
            room = self.get_room_by_name('Info')
574 575 576 577 578
        msg = error['error']['type']
        condition = error['error']['condition']
        code = error['error']['code']
        body = error['error']['text']
        if not body:
579 580 581
            if code in ERROR_AND_STATUS_CODES.keys():
                body = ERROR_AND_STATUS_CODES[code]
            else:
582 583 584 585 586 587 588
                body = condition or _('Unknown error')
        if code:
            self.add_message_to_room(room, _('Error: %(code)s - %(msg)s: %(body)s' %
                                             {'msg':msg, 'body':body, 'code':code}))
        else:
            self.add_message_to_room(room, _('Error: %(msg)s: %(body)s' %
                                             {'msg':msg, 'body':body}))
589
        if code == '401':
590 591
            self.add_message_to_room(room, _('To provide a password in order to join the room, type "/join / password" (replace "password" by the real password)'))
        if code == '409':
592 593 594
            if config.get('alternative_nickname', '') != '':
                self.command_join('%s/%s'% (room.name, room.own_nick+config.get('alternative_nickname', '')))
            else:
595
                self.add_message_to_room(room, _('You can join the room with an other nick, by typing "/join /other_nick"'))
596
        self.refresh_window()
597

598
    def open_private_window(self, room_name, user_nick, focus=True):
599
        complete_jid = room_name.decode('utf-8')+'/'+user_nick
600 601 602
        for room in self.rooms: # if the room exists, focus it and return
            if room.jid:
                if room.jid == complete_jid:
603
                    self.command_win('%s' % room.nb)
604 605
                    return
        # create the new tab
606
        room = self.get_room_by_name(room_name)
607 608 609
        if not room:
            return None
        own_nick = room.own_nick
610 611 612 613 614 615 616 617 618 619
        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
620
            self.command_win('%s' % (r.nb))
621
        # self.window.new_room(r)
622
        self.refresh_window()
623 624
        return r

625
    def on_groupchat_message(self, message):
626
        """
627
        Triggered whenever a message is received from a multi-user chat room.
628
        """
629 630 631
        # FIXME: not receiving subjects? :/
        delay_tag = message.find('{urn:xmpp:delay}delay')
        if delay_tag is not None:
632
            delayed = True
633
            date = common.datetime_tuple(delay_tag.attrib['stamp'])
634
        else:
635 636
            # We support the OLD and deprecated XEP: http://xmpp.org/extensions/xep-0091.html
            # But it sucks, please, Jabber servers, don't do this :(
637 638
            delay_tag = message.find('{jabber:x:delay}x')
            if delay_tag is not None:
639
                delayed = True
640
                date = common.datetime_tuple(delay_tag.attrib['stamp'])
641 642
            else:
                delayed = False
643
                date = None
644
        nick_from = message['mucnick']
645
        room_from = message.getMucroom()
646 647
        if nick_from == room_from:
            nick_from = None
648
        room = self.get_room_by_name(room_from)
649 650
        if (self.ignores.has_key(room_from)) and (nick_from in self.ignores[room_from]):
            return
651
        room = self.get_room_by_name(room_from)
652 653
        if not room:
            self.information(_("message received for a non-existing room: %s") % (room_from))
654
            return
655 656
        body = message['body']
        subject = message['subject']
657
        if subject:
658
            if nick_from:
659
                self.add_message_to_room(room, _("%(nick)s changed the subject to: %(subject)s") % {'nick':nick_from, 'subject':subject}, time=date)
660
            else:
661
                self.add_message_to_room(room, _("The subject is: %(subject)s") % {'subject':subject}, time=date)
662
            room.topic = subject.encode('utf-8').replace('\n', '|')
663 664
            if room == self.current_room():
                self.window.topic_win.refresh(room.topic)
665
        elif body:
666
            if body.startswith('/me '):
667
                self.add_message_to_room(room, "* "+nick_from + ' ' + body[4:], date)
668
            else:
669 670
                date = date if delayed == True else None
                self.add_message_to_room(room, body, date, nick_from)
671
        self.refresh_window()
672
        doupdate()
673

674

675
    def add_message_to_room(self, room, txt, time=None, nickname=None, colorized=False):
676
        """
677 678
        Add the message to the room and refresh the associated component
        of the interface
679
        """
680 681
        if room != self.current_room():
            room.add_line_separator()
682
        room.add_message(txt, time, nickname, colorized)
683
        if room == self.current_room():
684 685
            self.window.text_win.refresh(room)
        else:
686
            self.window.info_win.refresh(self.rooms, self.current_room())
687
        self.window.input.refresh()
688 689

    def execute(self):
690 691 692
        """
        Execute the /command or just send the line on the current room
        """
693 694
        line = self.window.input.get_text()
        self.window.input.clear_text()
695
        self.window.input.refresh()
696 697
        if line == "":
            return
698
        if line.startswith('/') and not line.startswith('/me '):
699
            command = line.strip()[:].split()[0][1:]
700
            arg = line[2+len(command):] # jump the '/' and the ' '
701
            # example. on "/link 0 open", command = "link" and arg = "0 open"
702
            if command in self.commands.keys():
703
                func = self.commands[command][0]
704
                func(arg)
705
                return
706
            else:
707
                self.add_message_to_room(self.current_room(), _("Error: unknown command (%s)") % (command))
708
        elif self.current_room().name != 'Info':
709
            if self.current_room().jid is not None:
710
                muc.send_private_message(self.xmpp, self.current_room().name, line)
711
                self.add_message_to_room(self.current_room(), line.decode('utf-8'), None, self.current_room().own_nick.decode('utf-8'))
712
            else:
713
                muc.send_groupchat_message(self.xmpp, self.current_room().name, line)
714
        self.window.input.refresh()
715
        doupdate()
716

717
    def command_help(self, arg):
718 719 720
        """
        /help <command_name>
        """
721
        args = arg.split()
722 723
        room = self.current_room()
        if len(args) == 0:
724
            msg = _('Available commands are: ')
725 726
            for command in self.commands.keys():
                msg += "%s " % command
727
            msg += _("\nType /help <command_name> to know what each command does")
728 729 730 731
        if len(args) == 1:
            if args[0] in self.commands.keys():
                msg = self.commands[args[0]][1]
            else:
732
                msg = _('Unknown command: %s') % args[0]
733
        self.add_message_to_room(room, msg)
734

735 736 737 738
    def command_whois(self, arg):
        """
        /whois <nickname>
        """
739 740
        # TODO
        return
741 742 743 744 745 746 747 748 749 750 751
        args = arg.split()
        room = self.current_room()
        if len(args) != 1:
            self.add_message_to_room(room, _('whois command takes exactly one argument'))
            return
        # check if current room is a MUC
        if room.jid or room.name == 'Info':
            return
        nickname = args[0]
        self.muc.request_vcard(room.name, nickname)

752 753 754 755
    def command_theme(self, arg):
        """
        """
        theme.reload_theme()
756
        self.resize_window()
757

758
    def command_win(self, arg):
759 760 761
        """
        /win <number>
        """
762
        args = arg.split()
763
        if len(args) != 1:
764
            self.command_help('win')
765 766 767 768
            return
        try:
            nb = int(args[0])
        except ValueError:
769
            self.command_help('win')
770 771 772
            return
        if self.current_room().nb == nb:
            return
773
        self.current_room().set_color_state(theme.COLOR_TAB_NORMAL)
774
        self.current_room().remove_line_separator()
775 776 777 778 779
        start = self.current_room()
        self.rooms.append(self.rooms.pop(0))
        while self.current_room().nb != nb:
            self.rooms.append(self.rooms.pop(0))
            if self.current_room() == start:
780
                self.current_room().set_color_state(theme.COLOR_TAB_CURRENT)
781
                self.refresh_window()
782
                return
783
        self.current_room().set_color_state(theme.COLOR_TAB_CURRENT)
784
        self.refresh_window()
785

786
    def command_kick(self, arg):
787 788 789
        """
        /kick <nick> [reason]
        """
790
        args = arg.split()
791
        if len(args) < 1:
792
            self.command_help('kick')
793 794 795 796 797 798 799 800 801
            return
        nick = args[0]
        if len(args) >= 2:
            reason = ' '.join(args[1:])
        else:
            reason = ''
        if self.current_room().name == 'Info' or not self.current_room().joined:
            return
        roomname = self.current_room().name
802 803 804
        res = muc.eject_user(self.xmpp, roomname, nick, reason)
        if res['type'] == 'error':
            self.room_error(res, roomname)
805

806 807 808 809 810 811 812
    def command_say(self, arg):
        """
        /say <message>
        """
        line = arg
        if self.current_room().name != 'Info':
            if self.current_room().jid is not None:
813
                muc.send_private_message(self.xmpp, self.current_room().name, line)
814 815
                self.add_message_to_room(self.current_room(), line.decode('utf-8'), None, self.current_room().own_nick)
            else:
816
                muc.send_groupchat_message(self.xmpp, self.current_room().name, line)
817 818 819
        self.window.input.refresh()
        doupdate()

820
    def command_join(self, arg):
821 822 823
        """
        /join [room][/nick] [password]
        """
824
        args = arg.split()
825
        password = None
826 827 828 829 830 831
        if len(args) == 0:
            r = self.current_room()
            if r.name == 'Info':
                return
            room = r.name
            nick = r.own_nick
832
        else:
833 834
            info = args[0].split('/')
            if len(info) == 1:
835 836 837 838
                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
                nick = config.get('default_nick', '')
                if nick == '':
                    nick = default
839 840
            else:
                nick = info[1]
841
            if info[0] == '':   # happens with /join /nickname, which is OK
842 843 844 845
                r = self.current_room()
                if r.name == 'Info':
                    return
                room = r.name
846 847
                if nick == '':
                    nick = r.own_nick
848 849
            else:
                room = info[0]
850 851 852 853
            if not is_jid(room): # no server is provided, like "/join hello"
                # use the server of the current room if available
                # check if the current room's name has a server
                if is_jid(self.current_room().name):
854
                    room += '@%s' % jid_get_domain(self.current_room().name)
855
                else:           # no server could be found, print a message and return
856
                    self.add_message_to_room(self.current_room(), _("You didn't specify a server for the room you want to join"))
857
                    return
858
            r = self.get_room_by_name(room)
859 860
        if len(args) == 2:       # a password is provided
            password = args[1]
861
        if r and r.joined:       # if we are already in the room
862
            self.command_win('%s' % (r.nb))
863
            return
864 865
        room = room.lower()
        self.xmpp.plugin['xep_0045'].joinMUC(room, nick, password)
866
        if not r:   # if the room window exists, we don't recreate it.
867
            self.open_new_room(room, nick)
868
        else:
869
            r.own_nick = nick
870
            r.users = []
871

872
    def command_bookmark(self, arg):
873 874 875
        """
        /bookmark [room][/nick]
        """
876
        args = arg.split()
877 878 879 880 881 882 883 884 885 886 887 888 889
        nick = None
        if len(args) == 0:
            room = self.current_room()
            if room.name == 'Info':
                return
            roomname = room.name
            if room.joined:
                nick