gui.py 40.2 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

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

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

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

58
        self.muc = muc
59

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

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

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

124 125 126 127 128
        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)
129
        self.handler.connect('private-message', self.private_message)
130
        self.handler.connect('error-message', self.room_error)
131
        self.handler.connect('error', self.information)
132

133 134 135 136 137 138 139
    def resize_window(self):
        """
        Resize the whole screen
        """
        self.window.resize(self.stdscr)
        self.window.refresh(self.rooms)

140
    def main_loop(self, stdscr):
141 142 143
        """
        main loop waiting for the user to press a key
        """
144
        while True:
145
            doupdate()
146 147 148 149
            char=read_char(stdscr)
            # search for keyboard shortcut
            if char in self.key_func.keys():
                self.key_func[char]()
150
            else:
151 152
                if len(char.decode('utf-8')) > 1:
                    continue    # ignore non-handled keyboard shortcuts
153
                self.window.do_command(char)
154

155
    def current_room(self):
156 157 158
        """
        returns the current room, the one we are viewing
        """
159
        return self.rooms[0]
160 161

    def get_room_by_name(self, name):
162 163 164 165 166 167 168
        """
        returns the room that has this name
        """
        for room in self.rooms:
            if room.name == name:
                return room
        return None
169

170
    def init_curses(self, stdscr):
171 172 173
        """
        ncurses initialization
        """
174 175
        curses.start_color()
        curses.noecho()
176
        curses.curs_set(0)
177
        curses.use_default_colors()
178
        stdscr.keypad(True)
179 180
        curses.init_pair(1, curses.COLOR_WHITE,
                         curses.COLOR_BLUE)
181 182 183
        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
184 185 186 187 188
        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)
189 190 191 192 193 194 195 196 197 198
        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)
199
        curses.init_pair(15, curses.COLOR_WHITE, # new message in private room
200
                         curses.COLOR_GREEN)
201

202
    def reset_curses(self):
203 204 205 206 207
        """
        Reset terminal capabilities to what they were before ncurses
        init
        """
        curses.echo()
208
        curses.nocbreak()
209
        curses.endwin()
210

211
    def on_connected(self, jid):
212
        """
213
        We are connected when authentification confirmation is received
214
        """
215 216
        self.information(_("Welcome on Poezio \o/!"))
        self.information(_("Your JID is %s") % jid)
217 218

    def join_room(self, room, nick):
219 220 221
        """
        join the specified room (muc), using the specified nick
        """
222
        r = Room(room, nick, self.window)
223 224 225 226 227 228 229 230 231 232 233
        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)
234

235
    def completion(self):
236 237 238
        """
        Called when Tab is pressed, complete the nickname in the input
        """
239 240 241 242 243 244 245 246 247
        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
248 249
        if len(self.window.input.text) == 0:
            self.last_talked_completion()
250
        self.window.input.auto_completion([user.nick for user in sorted(self.current_room().users, compare_users)])
251

252 253 254 255 256 257 258 259 260 261
    def last_talked_completion(self):
        """
        If tab is used while the input is empty, insert the nickname
        of the last person who spoke
        """
        for msg in self.current_room().messages[::-1]:
            if msg.nickname is not None and msg.nickname != self.current_room().own_nick:
                self.window.input.text = msg.nickname+config.get('after_completion', ',')+" "
                self.window.input.key_end()

262 263 264 265 266 267 268
    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()
269
        for msg in self.current_room().messages[:-9:-1]:
270 271 272 273
            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:
274
                    words.append(word.encode('utf-8'))
275 276
        self.window.input.auto_completion(words)

277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295
    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])

296
    def rotate_rooms_right(self, args=None):
297 298 299
        """
        rotate the rooms list to the right
        """
300
        self.current_room().set_color_state(11)
301
        self.rooms.append(self.rooms.pop(0))
302
        self.window.refresh(self.rooms)
303

304
    def rotate_rooms_left(self, args=None):
305 306 307
        """
        rotate the rooms list to the right
        """
308
        self.current_room().set_color_state(11)
309
        self.rooms.insert(0, self.rooms.pop())
310
        self.window.refresh(self.rooms)
311

312 313
    def scroll_page_down(self, args=None):
        self.current_room().scroll_down()
314
        self.window.refresh(self.rooms)
315 316

    def scroll_page_up(self, args=None):
317
        self.current_room().scroll_up(self.window.size)
318
        self.window.refresh(self.rooms)
319

320
    def room_error(self, room, error, msg):
321 322 323
        """
        Display the error on the room window
        """
324 325
        if not error:
            return
326
        room = self.get_room_by_name(room)
327 328
        code = error.getAttr('code')
        typ = error.getAttr('type')
329 330 331 332
        if error.getTag('text'):
            body = error.getTag('text').getData()
        else:
            body = _('Unknown error')
333 334
        self.add_message_to_room(room, _('Error: %(code)s-%(msg)s: %(body)s' %
                                   {'msg':msg, 'code':code, 'body':body}))
335
        if code == '401':
336
            room.add(_('To provide a password in order to join the room, type "/join / password" (replace "password" by the real password)'))
337
        self.window.refresh(self.rooms)
338

339 340 341 342 343 344 345 346 347 348
    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)
349 350
            if not room:
                return
351 352 353 354 355 356 357 358 359 360 361 362 363
        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
364 365 366 367
        room = self.get_room_by_name(room_name)
        if not room:
            return None
        own_nick = room.own_nick
368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383
        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

384
    def room_message(self, stanza, date=None):
385 386 387
        """
        Display the message on the room window
        """
388
        delay_tag = stanza.getTag('delay', namespace='urn:xmpp:delay')
389
        if delay_tag:
390 391 392
            delayed = True
            date = common.datetime_tuple(delay_tag.getAttr('stamp'))
        else:
393 394 395 396 397 398 399 400
            # We support the OLD and deprecated XEP: http://xmpp.org/extensions/xep-0091.html
            # But it sucks, please, Jabber servers, don't do this :(
            delay_tag = stanza.getTag('x', namespace='jabber:x:delay')
            if delay_tag:
                delayed = True
                date = common.datetime_tuple(delay_tag.getAttr('stamp'))
            else:
                delayed = False
401 402 403
        if stanza.getType() != 'groupchat':
            return  # ignore all messages not comming from a MUC
        nick_from = stanza.getFrom().getResource()
404
        room_from = stanza.getFrom().getStripped()
405 406
        if (self.ignores.has_key(room_from)) and (nick_from in self.ignores[room_from]):
            return
407
        room = self.get_room_by_name(room_from)
408
	if not room:
409
	    self.information(_("message received for a non-existing room: %s") % (room_from))
410
            return
411
        body = stanza.getBody()
412 413
        subject = stanza.getSubject()
        if subject:
414
            if nick_from:
415
                self.add_message_to_room(room, _("%(nick)s changed the subject to: %(subject)s") % {'nick':nick_from, 'subject':subject}, date)
416
            else:
417
                self.add_message_to_room(room, _("The subject is: %(subject)s") % {'subject':subject}, date)
418
            room.topic = subject.encode('utf-8').replace('\n', '|')
419 420
            if room == self.current_room():
                self.window.topic_win.refresh(room.topic)
421
        elif body:
422
            if body.startswith('/me '):
423
                self.add_message_to_room(room, "* "+nick_from + ' ' + body[4:], date)
424
            else:
425 426
                date = date if delayed == True else None
                self.add_message_to_room(room, body, date, nick_from)
427
        self.window.refresh(self.rooms)
428
        doupdate()
429 430

    def room_presence(self, stanza):
431 432 433 434
        """
        Display the presence on the room window and update the
        presence information of the concerned user
        """
435 436
        from_nick = stanza.getFrom().getResource()
        from_room = stanza.getFrom().getStripped()
437
	room = self.get_room_by_name(from_room)
438
	if not room:
439
            return
440
        else:
441 442 443 444 445 446
            msg = None
            affiliation = stanza.getAffiliation()
            show = stanza.getShow()
            status = stanza.getStatus()
            role = stanza.getRole()
            if not room.joined:     # user in the room BEFORE us.
447 448 449 450 451 452 453
                # ignore redondant presence message, see bug #1509
                if from_nick not in [user.nick for user in room.users]:
                    room.users.append(User(from_nick, affiliation, show, status,
                                           role))
                    if from_nick.encode('utf-8') == room.own_nick:
                        room.joined = True
                        self.add_message_to_room(room, _("Your nickname is %s") % (from_nick))
454 455 456 457 458 459
            else:
                change_nick = stanza.getStatusCode() == '303'
                kick = stanza.getStatusCode() == '307'
                user = room.get_user_by_name(from_nick)
                # New user
                if not user:
460 461
                    room.users.append(User(from_nick, affiliation,
                                           show, status, role))
462 463
                    hide_exit_join = config.get('hide_exit_join', -1)
                    if hide_exit_join != 0:
464
                        self.add_message_to_room(room, _("%(nick)s joined the room %(roomname)s") % {'nick':from_nick, 'roomname': room.name})
465 466 467 468
                # nick change
                elif change_nick:
                    if user.nick == room.own_nick:
                        room.own_nick = stanza.getNick().encode('utf-8')
469 470 471 472
                        # 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()
473
                    user.change_nick(stanza.getNick())
474
                    self.add_message_to_room(room, _('%(old)s is now known as %(new)s') % {'old':from_nick, 'new':stanza.getNick()})
475 476 477 478 479 480 481 482
                    # 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

483 484 485
                # kick
                elif kick:
                    room.users.remove(user)
486
                    try:
487
                        reason = stanza.getReason()
488 489
                    except:
                        reason = ''
490
                    try:
491
                        by = stanza.getActor()
492 493 494 495 496
                    except:
                        by = None
                    if from_nick == room.own_nick: # we are kicked
                        room.disconnect()
                        if by:
497
                            self.add_message_to_room(room,  _("You have been kicked by %(by)s. Reason: %(reason)s") % {'by':by, 'reason':reason})
498
                        else:
499
                            self.add_message_to_room(room, _("You have been kicked. Reason: %s") % (reason))
500 501
                    else:
                        if by:
502
                            self.add_message_to_room(room, _("%(nick)s has been kicked by %(by)s. Reason: %(reason)s") % {'nick':from_nick, 'by':by, 'reason':reason})
503
                        else:
504
                            self.add_message_to_room(room, _("%(nick)s has been kicked. Reason: %(reason)s") % {'nick':from_nick, 'reason':reason})
505 506 507
                # user quit
                elif status == 'offline' or role == 'none':
                    room.users.remove(user)
508 509 510
                    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))
511 512 513
                    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))
514 515
                # status change
                else:
516 517 518 519 520 521 522 523 524 525 526
                    # 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 ","
527
                    hide_status_change = config.get('hide_status_change', -1) if config.get('hide_status_change', -1) >= -1 else -1
528
                    if (hide_status_change == -1 or \
529
                            user.has_talked_since(hide_status_change) or\
530 531 532 533 534 535
                            user.nick == room.own_nick)\
                            and\
                            (affiliation != user.affiliation or\
                                role != user.role or\
                                show != user.show or\
                                status != user.status):
536
                        # display the message in the room
537
                        self.add_message_to_room(room, msg)
538 539 540 541 542
                    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)
543 544
            if room == self.current_room():
                self.window.user_win.refresh(room.users)
545
        self.window.input.refresh()
546
        doupdate()
547

548
    def add_message_to_room(self, room, txt, time=None, nickname=None):
549
        """
550 551
        Add the message to the room and refresh the associated component
        of the interface
552
        """
553
        room.add_message(txt, time, nickname)
554
        if room == self.current_room():
555 556
            self.window.text_win.refresh(room)
        else:
557
            self.window.info_win.refresh(self.rooms, self.current_room())
558
        self.window.input.refresh()
559 560

    def execute(self):
561 562 563
        """
        Execute the /command or just send the line on the current room
        """
564 565
        line = self.window.input.get_text()
        self.window.input.clear_text()
566
        self.window.input.refresh()
567 568
        if line == "":
            return
569
        if line.startswith('/') and not line.startswith('/me '):
570
            command = line.strip()[:].split()[0][1:]
571 572
            arg = line[1+len(command):]
            # example. on "/link 0 open", command = "link" and arg = "0 open"
573
            if command in self.commands.keys():
574
                func = self.commands[command][0]
575
                func(arg)
576
                return
577
            else:
578
                self.add_message_to_room(self.current_room(), _("Error: unknown command (%s)") % (command))
579
        elif self.current_room().name != 'Info':
580 581 582 583 584
            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)
585
        self.window.input.refresh()
586
        doupdate()
587

588
    def command_help(self, arg):
589 590 591
        """
        /help <command_name>
        """
592
        args = arg.split()
593 594
        room = self.current_room()
        if len(args) == 0:
595
            msg = _('Available commands are: ')
596 597
            for command in self.commands.keys():
                msg += "%s " % command
598
            msg += _("\nType /help <command_name> to know what each command does")
599 600 601 602
        if len(args) == 1:
            if args[0] in self.commands.keys():
                msg = self.commands[args[0]][1]
            else:
603
                msg = _('Unknown command: %s') % args[0]
604
        self.add_message_to_room(room, msg)
605

606
    def command_win(self, arg):
607 608 609
        """
        /win <number>
        """
610
        args = arg.split()
611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629
        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)
630
        self.current_room().set_color_state(11)
631

632
    def command_kick(self, arg):
633 634 635
        """
        /kick <nick> [reason]
        """
636
        args = arg.split()
637 638 639 640 641 642 643 644 645 646 647 648 649
        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)

650
    def command_join(self, arg):
651 652 653
        """
        /join [room][/nick] [password]
        """
654
        args = arg.split()
655
        password = None
656 657 658 659 660 661
        if len(args) == 0:
            r = self.current_room()
            if r.name == 'Info':
                return
            room = r.name
            nick = r.own_nick
662
        else:
663 664
            info = args[0].split('/')
            if len(info) == 1:
665 666 667 668
                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
                nick = config.get('default_nick', '')
                if nick == '':
                    nick = default
669 670
            else:
                nick = info[1]
671
            if info[0] == '':   # happens with /join /nickname, which is OK
672 673 674 675
                r = self.current_room()
                if r.name == 'Info':
                    return
                room = r.name
676 677
                if nick == '':
                    nick = r.own_nick
678 679
            else:
                room = info[0]
680 681 682 683 684 685
            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
686
                    self.add_message_to_room(self.current_room(), _("You didn't specify a server for the room you want to join"))
687
                    return
688
            r = self.get_room_by_name(room)
689 690
        if len(args) == 2:       # a password is provided
            password = args[1]
691
        if r and r.joined:       # if we are already in the room
692
            self.add_message_to_room(self.current_room(), _("already in room [%s]") % room)
693
            return
694
        self.muc.join_room(room, nick, password)
695
        if not r:   # if the room window exists, we don't recreate it.
696
            self.join_room(room, nick)
697
        else:
698
            r.own_nick = nick
699
            # r.own_nick = nick
700
            r.users = []
701

702
    def command_bookmark(self, arg):
703 704 705
        """
        /bookmark [room][/nick]
        """
706
        args = arg.split()
707 708 709 710 711 712 713 714 715 716 717 718 719
        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]
720 721
            if roomname == '':
                roomname = self.current_room().name
722 723 724 725
        if nick:
            res = roomname+'/'+nick
        else:
            res = roomname
726 727 728 729 730 731 732 733 734
        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)
735 736 737
        bookmarks = bookmarked+':'+res
        config.set_and_save('rooms', bookmarks)
        self.add_message_to_room(self.current_room(), _('Your bookmarks are now: %s') % bookmarks)
738

739
    def command_set(self, arg):
740 741 742
        """
        /set <option> [value]
        """
743
        args = arg.split()
744
        if len(args) != 2 and len(args) != 1:
745 746 747
            self.command_help(['set'])
            return
        option = args[0]
748 749 750 751
        if len(args) == 2:
            value = args[1]
        else:
            value = ''
752
        config.set_and_save(option, value)
753 754
        msg = "%s=%s" % (option, value)
        room = self.current_room()
755
        self.add_message_to_room(room, msg)
756

757
    def command_show(self, arg):
758 759 760
        """
        /show <status> [msg]
        """
761
        args = arg.split()
762 763 764 765
        possible_show = {'avail':None,
                         'available':None,
                         'ok':None,
                         'here':None,
766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786
                         '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)

787
    def command_ignore(self, arg):
788 789 790
        """
        /ignore <nick>
        """
791
        args = arg.split()
792
        if len(args) != 1:
793 794 795
            self.command_help(['ignore'])
            return
        if self.current_room().name == 'Info' or not self.current_room().joined:
796
            return
797 798 799 800 801 802 803 804 805
        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)
806

807
    def command_unignore(self, arg):
808 809 810
        """
        /unignore <nick>
        """
811
        args = arg.split()
812
        if len(args) != 1:
813 814 815 816 817 818 819 820
            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)
821
            return
822 823 824 825
        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)
826

827
    def command_away(self, arg):
828 829 830
        """
        /away [msg]
        """
831
        self.command_show("away "+arg)
832

833
    def command_busy(self, arg):
834 835 836
        """
        /busy [msg]
        """
837
        self.command_show("busy "+args)
838

839
    def command_avail(self, arg):
840 841 842
        """
        /avail [msg]
        """
843
        self.command_show("available "+args)
844

845
    def command_part(self, arg):
846 847 848
        """
        /part [msg]
        """
849
        args = arg.split()
850 851 852 853 854 855 856 857
        reason = None
        room = self.current_room()
        if room.name == 'Info':
            return
        if len(args):
            msg = ' '.join(args)
        else:
            msg = None
858 859
        if room.joined:
            self.muc.quit_room(room.name, room.own_nick, msg)
860
        self.rooms.remove(self.current_room())
861
        self.window.refresh(self.rooms)
862

863
    def command_unquery(self, arg):
864 865 866 867 868 869 870 871
        """
        /unquery
        """
        room = self.current_room()
        if room.jid is not None:
            self.rooms.remove(room)
            self.window.refresh(self.rooms)

872
    def command_query(self, arg):