gui.py 35.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
#!/usr/bin/python
# -*- 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/>.

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

24 25
from time import sleep

26
import sys
27
import os
28 29

import curses
30
from datetime import datetime
31 32

import common
33

34
from handler import Handler
35
from config import config
36
from window import Window
37 38
from user import User
from room import Room
39
from message import Message
40
from keyboard import read_char
41
from common import is_jid_the_same, jid_get_domain, is_jid
42

43 44
def doupdate():
    curses.doupdate()
45

46 47
class Gui(object):
    """
48
    User interface using ncurses
49
    """
50 51 52 53
    def __init__(self, stdscr=None, muc=None):
        self.init_curses(stdscr)
        self.stdscr = stdscr
        self.window = Window(stdscr)
54
        self.rooms = [Room('Info', '', self.window)]
55
        self.ignores = {}
56

57
        self.muc = muc
58

59
        self.commands = {
60
            'help': (self.command_help, u'\_o< KOIN KOIN KOIN'),
61
            '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")),
62 63 64 65 66 67 68 69
            '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.")),
70 71
            '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.")),
72 73 74 75 76 77 78
            '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)")),
79
            'unquery': (self.command_unquery, _("Usage: /unquery\nClose the private conversation window")),
80 81 82
            '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")),
83
            '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')),
84

85
            'nick': (self.command_nick, _("Usage: /nick <nickname> \nNick: Change your nickname in the current room"))
86 87
            }

88 89 90 91 92 93 94
        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,
95 96
            "KEY_PPAGE": self.scroll_page_up,
            "KEY_NPAGE": self.scroll_page_down,
97
            "KEY_DC": self.window.input.key_dc,
98
            "KEY_F(5)": self.rotate_rooms_left,
99
            "^N": self.rotate_rooms_left,
100
            "KEY_F(6)": self.rotate_rooms_right,
101
            "^P": self.rotate_rooms_right,
102
            "\t": self.auto_completion,
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
            "^I": self.auto_completion,
            "KEY_RESIZE": self.window.resize,
            "KEY_BACKSPACE": self.window.input.key_backspace,
            '^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
118 119
            }

120 121 122 123 124
        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)
125
        self.handler.connect('private-message', self.private_message)
126
        self.handler.connect('error-message', self.room_error)
127
        self.handler.connect('error', self.information)
128

129
    def main_loop(self, stdscr):
130 131 132
        """
        main loop waiting for the user to press a key
        """
133
        while True:
134
            doupdate()
135 136 137 138
            char=read_char(stdscr)
            # search for keyboard shortcut
            if char in self.key_func.keys():
                self.key_func[char]()
139
            else:
140 141 142
                # if len(char) > 1:
                #     continue    # ignore non-handled keyboard shortcuts
                self.window.do_command(char)
143

144
    def current_room(self):
145 146 147
        """
        returns the current room, the one we are viewing
        """
148
        return self.rooms[0]
149 150

    def get_room_by_name(self, name):
151 152 153 154 155 156 157
        """
        returns the room that has this name
        """
        for room in self.rooms:
            if room.name == name:
                return room
        return None
158

159
    def init_curses(self, stdscr):
160 161 162
        """
        ncurses initialization
        """
163 164
        curses.start_color()
        curses.noecho()
165
        curses.curs_set(0)
166
        curses.use_default_colors()
167
        stdscr.keypad(True)
168 169
        curses.init_pair(1, curses.COLOR_WHITE,
                         curses.COLOR_BLUE)
170 171 172
        curses.init_pair(4, curses.COLOR_CYAN, -1)
        curses.init_pair(2, curses.COLOR_RED, -1) # Admin
        curses.init_pair(3, curses.COLOR_BLUE, -1) # Participant
173 174 175 176 177
        curses.init_pair(5, curses.COLOR_WHITE, -1) # Visitor
        curses.init_pair(6, curses.COLOR_CYAN, -1)
        curses.init_pair(7, curses.COLOR_GREEN, -1)
        curses.init_pair(8, curses.COLOR_MAGENTA, -1)
        curses.init_pair(9, curses.COLOR_YELLOW, -1)
178 179 180 181 182 183 184 185 186 187
        curses.init_pair(10, curses.COLOR_WHITE,
                         curses.COLOR_CYAN) # current room
        curses.init_pair(11, curses.COLOR_WHITE,
                         curses.COLOR_BLUE) # normal room
        curses.init_pair(12, curses.COLOR_WHITE,
                         curses.COLOR_MAGENTA) # new message room
        curses.init_pair(13, curses.COLOR_WHITE,
                         curses.COLOR_RED) # highlight room
        curses.init_pair(14, curses.COLOR_WHITE,
                         curses.COLOR_YELLOW)
188
        curses.init_pair(15, curses.COLOR_WHITE, # new message in private room
189
                         curses.COLOR_GREEN)
190

191
    def reset_curses(self):
192 193 194 195 196
        """
        Reset terminal capabilities to what they were before ncurses
        init
        """
        curses.echo()
197
        curses.nocbreak()
198
        curses.endwin()
199

200
    def on_connected(self, jid):
201
        """
202
        We are connected when authentification confirmation is received
203
        """
204 205
        self.information(_("Welcome on Poezio \o/!"))
        self.information(_("Your JID is %s") % jid)
206 207

    def join_room(self, room, nick):
208 209 210
        """
        join the specified room (muc), using the specified nick
        """
211
        r = Room(room, nick, self.window)
212 213 214 215 216 217 218 219 220 221 222
        self.current_room().set_color_state(11)
        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
        while self.current_room().nb != r.nb:
            self.rooms.insert(0, self.rooms.pop())
        self.window.refresh(self.rooms)
223

224
    def auto_completion(self):
225 226 227
        """
        Called when Tab is pressed, complete the nickname in the input
        """
228 229 230 231 232 233 234 235 236 237
        def compare_users(a, b):
            """
            Used to sort users by their availability
            """
            if a.show == b.show:
                return 0
            if a.show is None:
                return -1
            return 1
        self.window.input.auto_completion(sorted(self.current_room().users, compare_users))
238

239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257
    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:
            if room.color_state == 15:
                self.command_win([room.nb])
                return
        for room in self.rooms:
            if room.color_state == 13:
                self.command_win([room.nb])
                return
        for room in self.rooms:
            if room.color_state == 12:
                self.command_win([room.nb])

258
    def rotate_rooms_right(self, args=None):
259 260 261
        """
        rotate the rooms list to the right
        """
262
        self.current_room().set_color_state(11)
263
        self.rooms.append(self.rooms.pop(0))
264
        self.window.refresh(self.rooms)
265

266
    def rotate_rooms_left(self, args=None):
267 268 269
        """
        rotate the rooms list to the right
        """
270
        self.current_room().set_color_state(11)
271
        self.rooms.insert(0, self.rooms.pop())
272
        self.window.refresh(self.rooms)
273

274 275
    def scroll_page_down(self, args=None):
        self.current_room().scroll_down()
276
        self.window.refresh(self.rooms)
277 278

    def scroll_page_up(self, args=None):
279
        self.current_room().scroll_up(self.window.size)
280
        self.window.refresh(self.rooms)
281

282
    def room_error(self, room, error, msg):
283 284 285
        """
        Display the error on the room window
        """
286 287
        if not error:
            return
288
        room = self.get_room_by_name(room)
289 290
        code = error.getAttr('code')
        typ = error.getAttr('type')
291 292 293 294
        if error.getTag('text'):
            body = error.getTag('text').getData()
        else:
            body = _('Unknown error')
295 296
        self.add_message_to_room(room, _('Error: %(code)s-%(msg)s: %(body)s' %
                                   {'msg':msg, 'code':code, 'body':body}))
297
        if code == '401':
298
            room.add(_('To provide a password in order to join the room, type "/join / password" (replace "password" by the real password)'))
299
        self.window.refresh(self.rooms)
300

301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
    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

341
    def room_message(self, stanza, date=None):
342 343 344
        """
        Display the message on the room window
        """
345 346 347 348 349 350
        delay_tag = stanza.getTag('delay', namespace='urn:xmpp:delay')
        if delay_tag and not date:
            delayed = True
            date = common.datetime_tuple(delay_tag.getAttr('stamp'))
        else:
            delayed = False
351 352 353
        if stanza.getType() != 'groupchat':
            return  # ignore all messages not comming from a MUC
        nick_from = stanza.getFrom().getResource()
354
        room_from = stanza.getFrom().getStripped()
355 356
        if (self.ignores.has_key(room_from)) and (nick_from in self.ignores[room_from]):
            return
357
        room = self.get_room_by_name(room_from)
358
	if not room:
359
	    self.information(_("message received for a non-existing room: %s") % (room_from))
360
            return
361
        body = stanza.getBody()
362 363
        subject = stanza.getSubject()
        if subject:
364
            if nick_from:
365
                self.add_message_to_room(room, _("%(nick)s changed the subject to: %(subject)s") % {'nick':nick_from, 'subject':subject}, date)
366
            else:
367
                self.add_message_to_room(room, _("The subject is: %(subject)s") % {'subject':subject}, date)
368
            room.topic = subject.encode('utf-8').replace('\n', '|')
369 370
            if room == self.current_room():
                self.window.topic_win.refresh(room.topic)
371
        elif body:
372
            if body.startswith('/me '):
373
                self.add_message_to_room(room, "* "+nick_from + ' ' + body[4:], date)
374
            else:
375 376
                date = date if delayed == True else None
                self.add_message_to_room(room, body, date, nick_from)
377
        self.window.refresh(self.rooms)
378
        doupdate()
379 380

    def room_presence(self, stanza):
381 382 383 384
        """
        Display the presence on the room window and update the
        presence information of the concerned user
        """
385 386
        from_nick = stanza.getFrom().getResource()
        from_room = stanza.getFrom().getStripped()
387
	room = self.get_room_by_name(from_room)
388
	if not room:
389
            return
390
        else:
391 392 393 394 395 396
            msg = None
            affiliation = stanza.getAffiliation()
            show = stanza.getShow()
            status = stanza.getStatus()
            role = stanza.getRole()
            if not room.joined:     # user in the room BEFORE us.
397 398
                room.users.append(User(from_nick, affiliation, show, status,
                                       role))
399 400
                if from_nick.encode('utf-8') == room.own_nick:
                    room.joined = True
401
                    self.add_message_to_room(room, _("Your nickname is %s") % (from_nick))
402
                else:
403
                    self.add_message_to_room(room, _("%s is in the room") % (from_nick))
404 405 406 407 408 409
            else:
                change_nick = stanza.getStatusCode() == '303'
                kick = stanza.getStatusCode() == '307'
                user = room.get_user_by_name(from_nick)
                # New user
                if not user:
410 411
                    room.users.append(User(from_nick, affiliation,
                                           show, status, role))
412 413
                    hide_exit_join = config.get('hide_exit_join', -1)
                    if hide_exit_join != 0:
414
                        self.add_message_to_room(room, _("%(nick)s joined the room %(roomname)s") % {'nick':from_nick, 'roomname': room.name})
415 416 417 418
                # nick change
                elif change_nick:
                    if user.nick == room.own_nick:
                        room.own_nick = stanza.getNick().encode('utf-8')
419 420 421 422
                        # 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 = stanza.getNick()
423
                    user.change_nick(stanza.getNick())
424
                    self.add_message_to_room(room, _('%(old)s is now known as %(new)s') % {'old':from_nick, 'new':stanza.getNick()})
425 426 427 428 429 430 431 432
                    # 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

433 434 435
                # kick
                elif kick:
                    room.users.remove(user)
436
                    try:
437
                        reason = stanza.getReason()
438 439
                    except:
                        reason = ''
440
                    try:
441
                        by = stanza.getActor()
442 443 444 445 446
                    except:
                        by = None
                    if from_nick == room.own_nick: # we are kicked
                        room.disconnect()
                        if by:
447
                            self.add_message_to_room(room,  _("You have been kicked by %(by)s. Reason: %(reason)s") % {'by':by, 'reason':reason})
448
                        else:
449
                            self.add_message_to_room(room, _("You have been kicked. Reason: %s") % (reason))
450 451
                    else:
                        if by:
452
                            self.add_message_to_room(room, _("%(nick)s has been kicked by %(by)s. Reason: %(reason)s") % {'nick':from_nick, 'by':by, 'reason':reason})
453
                        else:
454
                            self.add_message_to_room(room, _("%(nick)s has been kicked. Reason: %(reason)s") % {'nick':from_nick, 'reason':reason})
455 456 457
                # user quit
                elif status == 'offline' or role == 'none':
                    room.users.remove(user)
458 459 460
                    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))
461 462 463
                    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))
464 465
                # status change
                else:
466 467 468 469 470 471 472 473 474 475 476
                    # 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 ","
477
                    hide_status_change = config.get('hide_status_change', -1) if config.get('hide_status_change', -1) >= -1 else -1
478
                    if (hide_status_change == -1 or \
479
                            user.has_talked_since(hide_status_change) or\
480 481 482 483 484 485
                            user.nick == room.own_nick)\
                            and\
                            (affiliation != user.affiliation or\
                                role != user.role or\
                                show != user.show or\
                                status != user.status):
486
                        # display the message in the room
487
                        self.add_message_to_room(room, msg)
488 489 490 491 492
                    private_room = self.get_room_by_name(stanza.getFrom())
                    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)
493 494
            if room == self.current_room():
                self.window.user_win.refresh(room.users)
495
        self.window.input.refresh()
496
        doupdate()
497

498
    def add_message_to_room(self, room, txt, time=None, nickname=None):
499
        """
500 501
        Add the message to the room and refresh the associated component
        of the interface
502
        """
503
        room.add_message(txt, time, nickname)
504
        if room == self.current_room():
505 506
            self.window.text_win.refresh(room)
        else:
507
            self.window.info_win.refresh(self.rooms, self.current_room())
508
        self.window.input.refresh()
509 510

    def execute(self):
511 512 513
        """
        Execute the /command or just send the line on the current room
        """
514 515
        line = self.window.input.get_text()
        self.window.input.clear_text()
516
        self.window.input.refresh()
517 518
        if line == "":
            return
519
        if line.startswith('/') and not line.startswith('/me '):
520 521 522
            command = line.strip()[:].split()[0][1:]
            args = line.strip()[:].split()[1:]
            if command in self.commands.keys():
523
                func = self.commands[command][0]
524
                func(args)
525
                return
526
            else:
527
                self.add_message_to_room(self.current_room(), _("Error: unknown command (%s)") % (command))
528
        elif self.current_room().name != 'Info':
529 530 531 532 533
            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)
534
        self.window.input.refresh()
535
        doupdate()
536

537
    def command_help(self, args):
538 539 540
        """
        /help <command_name>
        """
541 542
        room = self.current_room()
        if len(args) == 0:
543
            msg = _('Available commands are: ')
544 545
            for command in self.commands.keys():
                msg += "%s " % command
546
            msg += _("\nType /help <command_name> to know what each command does")
547 548 549 550
        if len(args) == 1:
            if args[0] in self.commands.keys():
                msg = self.commands[args[0]][1]
            else:
551
                msg = _('Unknown command: %s') % args[0]
552
        self.add_message_to_room(room, msg)
553

554
    def command_win(self, args):
555 556 557
        """
        /win <number>
        """
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576
        if len(args) != 1:
            self.command_help(['win'])
            return
        try:
            nb = int(args[0])
        except ValueError:
            self.command_help(['win'])
            return
        if self.current_room().nb == nb:
            return
        self.current_room().set_color_state(11)
        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:
                self.window.refresh(self.rooms)
                return
        self.window.refresh(self.rooms)
577
        self.current_room().set_color_state(11)
578

579
    def command_kick(self, args):
580 581 582
        """
        /kick <nick> [reason]
        """
583 584 585 586 587 588 589 590 591 592 593 594 595
        if len(args) < 1:
            self.command_help(['kick'])
            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
        self.muc.eject_user(roomname, 'kick', nick, reason)

596
    def command_join(self, args):
597 598 599
        """
        /join [room][/nick] [password]
        """
600
        password = None
601 602 603 604 605 606
        if len(args) == 0:
            r = self.current_room()
            if r.name == 'Info':
                return
            room = r.name
            nick = r.own_nick
607
        else:
608 609
            info = args[0].split('/')
            if len(info) == 1:
610 611 612 613
                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
                nick = config.get('default_nick', '')
                if nick == '':
                    nick = default
614 615
            else:
                nick = info[1]
616
            if info[0] == '':   # happens with /join /nickname, which is OK
617 618 619 620
                r = self.current_room()
                if r.name == 'Info':
                    return
                room = r.name
621 622
                if nick == '':
                    nick = r.own_nick
623 624
            else:
                room = info[0]
625 626 627 628 629 630 631 632
            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):
                    room += '@%s' % jid_get_domain(self.current_room().name)
                else:           # no server could be found, print a message and return
                    self.information(_("You didn't specify a server for the room you want to join"))
                    return
633
            r = self.get_room_by_name(room)
634 635
        if len(args) == 2:       # a password is provided
            password = args[1]
636
        if r and r.joined:       # if we are already in the room
637
            self.information(_("already in room [%s]") % room)
638
            return
639
        self.muc.join_room(room, nick, password)
640
        if not r:   # if the room window exists, we don't recreate it.
641
            self.join_room(room, nick)
642
        else:
643
            r.own_nick = nick
644
            # r.own_nick = nick
645
            r.users = []
646

647
    def command_bookmark(self, args):
648 649 650
        """
        /bookmark [room][/nick]
        """
651 652 653 654 655 656 657 658 659 660 661 662 663
        nick = None
        if len(args) == 0:
            room = self.current_room()
            if room.name == 'Info':
                return
            roomname = room.name
            if room.joined:
                nick = room.own_nick
        else:
            info = args[0].split('/')
            if len(info) == 2:
                nick = info[1]
            roomname = info[0]
664 665
            if roomname == '':
                roomname = self.current_room().name
666 667 668 669
        if nick:
            res = roomname+'/'+nick
        else:
            res = roomname
670 671 672 673 674 675 676 677 678
        bookmarked = config.get('rooms', '')
        # check if the room is already bookmarked.
        # if yes, replace it (i.e., update the associated nick)
        bookmarked = bookmarked.split(':')
        for room in bookmarked:
            if room.split('/')[0] == roomname:
                bookmarked.remove(room)
                break
        bookmarked = ':'.join(bookmarked)
679
        config.set_and_save('rooms', bookmarked+':'+res)
680

681
    def command_set(self, args):
682 683 684
        """
        /set <option> [value]
        """
685
        if len(args) != 2 and len(args) != 1:
686 687 688
            self.command_help(['set'])
            return
        option = args[0]
689 690 691 692
        if len(args) == 2:
            value = args[1]
        else:
            value = ''
693
        config.set_and_save(option, value)
694 695
        msg = "%s=%s" % (option, value)
        room = self.current_room()
696
        self.add_message_to_room(room, msg)
697

698
    def command_show(self, args):
699 700 701
        """
        /show <status> [msg]
        """
702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726
        possible_show = {'avail':'None',
                         'available':'None',
                         'ok':'None',
                         'here':'None',
                         'chat':'chat',
                         'away':'away',
                         'afk':'away',
                         'dnd':'dnd',
                         'busy':'dnd',
                         'xa':'xa'
                         }
        if len(args) < 1:
            return
        if not args[0] in possible_show.keys():
            self.command_help(['show'])
            return
        show = possible_show[args[0]]
        if len(args) > 1:
            msg = ' '.join(args[1:])
        else:
            msg = None
        for room in self.rooms:
            if room.joined:
                self.muc.change_show(room.name, room.own_nick, show, msg)

727 728 729 730 731
    def command_ignore(self, args):
        """
        /ignore <nick>
        """
        if len(args) != 1:
732 733 734
            self.command_help(['ignore'])
            return
        if self.current_room().name == 'Info' or not self.current_room().joined:
735
            return
736 737 738 739 740 741 742 743 744
        roomname = self.current_room().name
        nick = args[0]
        if not self.ignores.has_key(roomname):
            self.ignores[roomname] = set() # no need for any order
        if nick not in self.ignores[roomname]:
            self.ignores[roomname].add(nick)
            self.add_message_to_room(self.current_room(), _("%s is now ignored") % nick)
        else:
            self.add_message_to_room(self.current_room(), _("%s is already ignored") % nick)
745 746 747 748 749 750

    def command_unignore(self, args):
        """
        /unignore <nick>
        """
        if len(args) != 1:
751 752 753 754 755 756 757 758
            self.command_help(['unignore'])
            return
        if self.current_room().name == 'Info' or not self.current_room().joined:
            return
        roomname = self.current_room().name
        nick = args[0]
        if not self.ignores.has_key(roomname) or (nick not in self.ignores[roomname]):
            self.add_message_to_room(self.current_room(), _("%s was not ignored") % nick)
759
            return
760 761 762 763
        self.ignores[roomname].remove(nick)
        if self.ignores[roomname] == set():
            del self.ignores[roomname]
        self.add_message_to_room(self.current_room(), _("%s is now unignored") % nick)
764

765
    def command_away(self, args):
766 767 768
        """
        /away [msg]
        """
769 770 771 772
        args.insert(0, 'away')
        self.command_show(args)

    def command_busy(self, args):
773 774 775
        """
        /busy [msg]
        """
776 777 778 779
        args.insert(0, 'busy')
        self.command_show(args)

    def command_avail(self, args):
780 781 782
        """
        /avail [msg]
        """
783 784
        args.insert(0, 'available')
        self.command_show(args)
785

786
    def command_part(self, args):
787 788 789
        """
        /part [msg]
        """
790 791 792 793 794 795 796 797
        reason = None
        room = self.current_room()
        if room.name == 'Info':
            return
        if len(args):
            msg = ' '.join(args)
        else:
            msg = None
798 799
        if room.joined:
            self.muc.quit_room(room.name, room.own_nick, msg)
800
        self.rooms.remove(self.current_room())
801
        self.window.refresh(self.rooms)
802

803 804 805 806 807 808 809 810 811
    def command_unquery(self, args):
        """
        /unquery
        """
        room = self.current_room()
        if room.jid is not None:
            self.rooms.remove(room)
            self.window.refresh(self.rooms)

812
    def command_query(self, args):
813
        """
814
        /query <nick> [message]
815
        """
816
        if len(args) < 1:
817 818 819 820 821 822 823
            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:
824 825 826 827 828
                r = self.open_private_window(room.name, user.nick)
        if room and len(args) > 1:
            msg = ' '.join(args[1:])
            self.muc.send_private_message(r.name, msg)
            self.add_message_to_room(r, msg.decode('utf-8'), None, r.own_nick)
829

830
    def command_topic(self, args):
831 832 833
        """
        /topic [new topic]
        """
834
        room = self.current_room()
835
        if len(args) == 0:
836
            self.add_message_to_room(room, _("The subject of the room is: %s") % room.topic.decode('utf-8'))
837
        subject = ' '.join(args)
838 839 840 841
        if not room.joined or room.name == "Info":
            return
        self.muc.change_subject(room.name, subject)

842
    def command_nick(self, args):
843 844 845
        """
        /nick <nickname>
        """
846 847 848 849 850 851 852 853
        if len(args) != 1:
            return
        nick = args[0]
        room = self.current_room()
        if not room.joined or room.name == "Info":
            return
        self.muc.change_nick(room.name, nick)

854
    def information(self, msg):
855 856 857
        """
        Displays an informational message in the "Info" room window
        """
858
        room = self.get_room_by_name("Info")
859
        self.add_message_to_room(room, msg)
860
        self.window.input.refresh()
861

862
    def command_quit(self, args):
863 864 865
        """
        /quit
        """
866 867 868 869 870 871 872
        if len(args):
            msg = ' '.join(args)
        else:
            msg = None
        if msg:
            self.muc.disconnect(self.rooms, msg)
            sleep(0.2)          # :(
873
	self.reset_curses()
874
        sys.exit()