gui.py 44.1 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 46 47 48 49 50 51 52 53 54 55
# 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',
    }

56 57
def doupdate():
    curses.doupdate()
58

59 60
class Gui(object):
    """
61
    User interface using ncurses
62
    """
63 64 65 66
    def __init__(self, stdscr=None, muc=None):
        self.init_curses(stdscr)
        self.stdscr = stdscr
        self.window = Window(stdscr)
67
        self.rooms = [Room('Info', '', self.window)]
68
        self.ignores = {}
69

70
        self.muc = muc
71

72
        self.commands = {
73
            'help': (self.command_help, u'\_o< KOIN KOIN KOIN'),
74
            '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")),
75 76 77 78 79 80 81 82
            '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.")),
83 84
            '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.")),
85 86 87 88 89 90 91
            '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)")),
92
            'unquery': (self.command_unquery, _("Usage: /unquery\nClose the private conversation window")),
93 94 95
            '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")),
96
            '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.")),
97
            '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')),
98 99
            '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 "/"')),
100
            'whois': (self.command_whois, _('Usage: /whois <nickname>\nWhois: Request many informations about the user.')),
101 102
            }

103 104 105 106 107 108 109
        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,
110 111
            "KEY_PPAGE": self.scroll_page_up,
            "KEY_NPAGE": self.scroll_page_down,
112
            "KEY_DC": self.window.input.key_dc,
113
            "KEY_F(5)": self.rotate_rooms_left,
114
            "^P": self.rotate_rooms_left,
115
            "KEY_F(6)": self.rotate_rooms_right,
116
            "^N": self.rotate_rooms_right,
117 118
            "\t": self.completion,
            "^I": self.completion,
119
            "KEY_BTAB": self.last_words_completion,
120
            "KEY_RESIZE": self.resize_window,
121
            "KEY_BACKSPACE": self.window.input.key_backspace,
122
            '^?': self.window.input.key_backspace,
123 124 125 126 127 128 129 130 131 132 133 134
            '^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
135 136
            }

137 138 139 140 141
        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)
142
        self.handler.connect('private-message', self.private_message)
143
        self.handler.connect('error-message', self.room_error)
144
        self.handler.connect('error', self.information)
145

146 147 148 149 150 151 152
    def resize_window(self):
        """
        Resize the whole screen
        """
        self.window.resize(self.stdscr)
        self.window.refresh(self.rooms)

153
    def main_loop(self, stdscr):
154 155 156
        """
        main loop waiting for the user to press a key
        """
157
        self.refresh_window()
158
        while True:
159
            doupdate()
160
            char=read_char(stdscr)
161
            try: # if this is not a valide utf-8 char, discard it
162
                char.decode('utf-8')
163 164
            except UnicodeDecodeError:
                continue
165 166 167
            # search for keyboard shortcut
            if char in self.key_func.keys():
                self.key_func[char]()
168
            else:
169 170
                if len(char.decode('utf-8')) > 1:
                    continue    # ignore non-handled keyboard shortcuts
171
                self.window.do_command(char)
172

173
    def current_room(self):
174 175 176
        """
        returns the current room, the one we are viewing
        """
177
        return self.rooms[0]
178 179

    def get_room_by_name(self, name):
180 181 182 183
        """
        returns the room that has this name
        """
        for room in self.rooms:
184
            if room.name.decode('utf-8') == name:
185 186
                return room
        return None
187

188
    def init_curses(self, stdscr):
189 190 191
        """
        ncurses initialization
        """
192 193
        curses.start_color()
        curses.noecho()
194
        curses.curs_set(0)
195
        curses.use_default_colors()
196
        stdscr.keypad(True)
197 198
        curses.init_pair(1, curses.COLOR_WHITE,
                         curses.COLOR_BLUE)
199 200 201 202
        curses.init_pair(2, curses.COLOR_WHITE, -1) # Visitor
        curses.init_pair(3, curses.COLOR_CYAN, -1)
        curses.init_pair(4, curses.COLOR_RED, -1) # Admin
        curses.init_pair(5, curses.COLOR_BLUE, -1) # Participant
203 204 205 206
        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)
207 208 209 210 211 212 213 214 215 216
        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)
217
        curses.init_pair(15, curses.COLOR_WHITE, # new message in private room
218
                         curses.COLOR_GREEN)
219 220
        curses.init_pair(16, curses.COLOR_YELLOW,
                         curses.COLOR_BLUE)
221

222
    def reset_curses(self):
223 224 225 226 227
        """
        Reset terminal capabilities to what they were before ncurses
        init
        """
        curses.echo()
228
        curses.nocbreak()
229
        curses.endwin()
230

231
    def on_connected(self, jid):
232
        """
233
        We are connected when authentification confirmation is received
234
        """
235 236
        self.information(_("Welcome on Poezio \o/!"))
        self.information(_("Your JID is %s") % jid)
237

238 239 240 241 242 243 244
    def refresh_window(self):
        """
        Refresh everything
        """
        self.current_room().set_color_state(common.ROOM_STATE_CURRENT)
        self.window.refresh(self.rooms)

245
    def join_room(self, room, nick):
246 247 248
        """
        join the specified room (muc), using the specified nick
        """
249
        r = Room(room, nick, self.window)
250 251 252 253 254 255 256 257
        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
258 259
        self.command_win("%s" % r.nb)
        self.refresh_window()
260

261
    def completion(self):
262 263 264
        """
        Called when Tab is pressed, complete the nickname in the input
        """
265 266
        def compare_users(a, b):
            """
267
            Used to sort users by their last_talked
268
            """
269
            if not a.last_talked and b.last_talked:
270
                return 0
271 272 273 274 275
            elif not b.last_talked and a.last_talked:
                return 1
            if a.last_talked <  b.last_talked:
                return 1
            else:
276
                return -1
277
        self.window.input.auto_completion([user.nick for user in sorted(self.current_room().users, compare_users)])
278

279 280 281 282 283 284 285
    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()
286
        for msg in self.current_room().messages[:-9:-1]:
287 288 289 290
            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:
291
                    words.append(word.encode('utf-8'))
292 293
        self.window.input.auto_completion(words)

294 295 296 297 298 299 300 301 302
    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:
303
                self.command_win('%s' % room.nb)
304 305 306
                return
        for room in self.rooms:
            if room.color_state == 13:
307
                self.command_win('%s' % room.nb)
308 309 310
                return
        for room in self.rooms:
            if room.color_state == 12:
311 312
                self.command_win('%s' % room.nb)
                return
313

314
    def rotate_rooms_right(self, args=None):
315 316 317
        """
        rotate the rooms list to the right
        """
318
        self.current_room().set_color_state(common.ROOM_STATE_NONE)
319
        self.current_room().remove_line_separator()
320
        self.rooms.append(self.rooms.pop(0))
321 322
        self.current_room().set_color_state(common.ROOM_STATE_CURRENT)
        self.refresh_window()
323

324
    def rotate_rooms_left(self, args=None):
325 326 327
        """
        rotate the rooms list to the right
        """
328
        self.current_room().set_color_state(common.ROOM_STATE_NONE)
329
        self.current_room().remove_line_separator()
330
        self.rooms.insert(0, self.rooms.pop())
331 332
        self.current_room().set_color_state(common.ROOM_STATE_CURRENT)
        self.refresh_window()
333

334
    def scroll_page_down(self, args=None):
335
        self.current_room().scroll_down(self.window.text_win.height-1)
336
        self.refresh_window()
337 338

    def scroll_page_up(self, args=None):
339
        self.current_room().scroll_up(self.window.text_win.height-1)
340
        self.refresh_window()
341

342
    def room_error(self, room, error, msg):
343 344 345
        """
        Display the error on the room window
        """
346 347
        if not error:
            return
348
        room = self.get_room_by_name(room)
349 350
        if not room:
            room = self.get_room_by_name('Info')
351 352
        code = error.getAttr('code')
        typ = error.getAttr('type')
353 354 355 356 357 358 359 360
        if error.getTag('text'):
            body = error.getTag('text').getData()
        else: # No description of the error is provided in the stanza
            # If it's a standard error, use our own messages
            if code in ERROR_AND_STATUS_CODES.keys():
                body = ERROR_AND_STATUS_CODES[code]
            else:
                body = _('Unknown error')
361 362
        self.add_message_to_room(room, _('Error: %(code)s-%(msg)s: %(body)s' %
                                   {'msg':msg, 'code':code, 'body':body}))
363
        if code == '401':
364 365
            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':
366 367 368
            if config.get('alternative_nickname', '') != '':
                self.command_join('%s/%s'% (room.name, room.own_nick+config.get('alternative_nickname', '')))
            else:
369
                self.add_message_to_room(room, _('You can join the room with an other nick, by typing "/join /other_nick"'))
370
        self.refresh_window()
371

372 373 374 375 376 377 378 379 380 381
    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)
382 383
            if not room:
                return
384 385 386 387 388 389 390 391 392 393
        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:
394
                    self.command_win('%s' % room.nb)
395 396
                    return
        # create the new tab
397 398 399 400
        room = self.get_room_by_name(room_name)
        if not room:
            return None
        own_nick = room.own_nick
401 402 403 404 405 406 407 408 409 410 411 412 413
        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)
414
        self.refresh_window()
415 416
        return r

417
    def room_message(self, stanza, date=None):
418 419 420
        """
        Display the message on the room window
        """
421
        delay_tag = stanza.getTag('delay', namespace='urn:xmpp:delay')
422
        if delay_tag:
423 424 425
            delayed = True
            date = common.datetime_tuple(delay_tag.getAttr('stamp'))
        else:
426 427 428 429 430 431 432 433
            # 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
434 435 436
        if stanza.getType() != 'groupchat':
            return  # ignore all messages not comming from a MUC
        nick_from = stanza.getFrom().getResource()
437
        room_from = stanza.getFrom().getStripped()
438 439
        if (self.ignores.has_key(room_from)) and (nick_from in self.ignores[room_from]):
            return
440
        room = self.get_room_by_name(room_from)
441
	if not room:
442
	    self.information(_("message received for a non-existing room: %s") % (room_from))
443
            return
444
        body = stanza.getBody()
445 446
        subject = stanza.getSubject()
        if subject:
447
            if nick_from:
448
                self.add_message_to_room(room, _("%(nick)s changed the subject to: %(subject)s") % {'nick':nick_from, 'subject':subject}, date)
449
            else:
450
                self.add_message_to_room(room, _("The subject is: %(subject)s") % {'subject':subject}, date)
451
            room.topic = subject.encode('utf-8').replace('\n', '|')
452 453
            if room == self.current_room():
                self.window.topic_win.refresh(room.topic)
454
        elif body:
455
            if body.startswith('/me '):
456
                self.add_message_to_room(room, "* "+nick_from + ' ' + body[4:], date)
457
            else:
458 459
                date = date if delayed == True else None
                self.add_message_to_room(room, body, date, nick_from)
460
        self.refresh_window()
461
        doupdate()
462 463

    def room_presence(self, stanza):
464 465 466 467
        """
        Display the presence on the room window and update the
        presence information of the concerned user
        """
468 469
        from_nick = stanza.getFrom().getResource()
        from_room = stanza.getFrom().getStripped()
470
	room = self.get_room_by_name(from_room)
471
	if not room:
472
            return
473
        else:
474 475 476 477 478
            msg = None
            affiliation = stanza.getAffiliation()
            show = stanza.getShow()
            status = stanza.getStatus()
            role = stanza.getRole()
479
            jid = stanza.getJid()
480
            if not room.joined:     # user in the room BEFORE us.
481 482
                # ignore redondant presence message, see bug #1509
                if from_nick not in [user.nick for user in room.users]:
483 484
                    new_user = User(from_nick, affiliation, show, status, role)
                    room.users.append(new_user)
485 486 487
                    if from_nick.encode('utf-8') == room.own_nick:
                        room.joined = True
                        self.add_message_to_room(room, _("Your nickname is %s") % (from_nick))
488 489 490 491 492
                        # Check for a 170 status code
                        for xtag in stanza.getTags('x'):
                            for child in xtag.getTags('status'):
                                if child.getAttr('code') == '170':
                                    self.add_message_to_room(room, 'Warning: this room is publicly logged')
493
                        new_user.color = 2
494 495 496 497 498 499
            else:
                change_nick = stanza.getStatusCode() == '303'
                kick = stanza.getStatusCode() == '307'
                user = room.get_user_by_name(from_nick)
                # New user
                if not user:
500 501
                    room.users.append(User(from_nick, affiliation,
                                           show, status, role))
502 503
                    hide_exit_join = config.get('hide_exit_join', -1)
                    if hide_exit_join != 0:
504 505 506 507
                        if not jid:
                            self.add_message_to_room(room, _("%(nick)s joined the room") % {'nick':from_nick})
                        else:
                            self.add_message_to_room(room, _("%(nick)s (%(jid)s) joined the room") % {'nick':from_nick, 'jid':jid})
508 509 510 511
                # nick change
                elif change_nick:
                    if user.nick == room.own_nick:
                        room.own_nick = stanza.getNick().encode('utf-8')
512 513 514 515
                        # 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()
516
                    user.change_nick(stanza.getNick())
517
                    self.add_message_to_room(room, _('%(old)s is now known as %(new)s') % {'old':from_nick, 'new':stanza.getNick()})
518 519 520 521 522 523 524 525
                    # 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

526 527 528
                # kick
                elif kick:
                    room.users.remove(user)
529
                    try:
530
                        reason = stanza.getReason()
531 532
                    except:
                        reason = ''
533
                    try:
534
                        by = stanza.getActor()
535 536 537 538 539
                    except:
                        by = None
                    if from_nick == room.own_nick: # we are kicked
                        room.disconnect()
                        if by:
540
                            self.add_message_to_room(room,  _("You have been kicked by %(by)s. Reason: %(reason)s") % {'by':by, 'reason':reason})
541
                        else:
542
                            self.add_message_to_room(room, _("You have been kicked. Reason: %s") % (reason))
543 544 545
                        # try to auto-rejoin
                        if config.get('autorejoin', 'false') == 'true':
                            self.muc.join_room(room.name, room.own_nick)
546 547
                    else:
                        if by:
548
                            self.add_message_to_room(room, _("%(nick)s has been kicked by %(by)s. Reason: %(reason)s") % {'nick':from_nick, 'by':by, 'reason':reason})
549
                        else:
550
                            self.add_message_to_room(room, _("%(nick)s has been kicked. Reason: %(reason)s") % {'nick':from_nick, 'reason':reason})
551 552 553
                # user quit
                elif status == 'offline' or role == 'none':
                    room.users.remove(user)
554 555
                    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):
556 557 558 559
                        if not jid:
                            self.add_message_to_room(room, _('%s has left the room') % (from_nick))
                        else:
                            self.add_message_to_room(room, _('%(nick)s (%(jid)s) has left the room') % {'nick':from_nick, 'jid':jid})
560 561 562
                    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))
563 564
                # status change
                else:
565 566 567 568 569 570 571 572 573 574 575
                    # 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 ","
576
                    hide_status_change = config.get('hide_status_change', -1) if config.get('hide_status_change', -1) >= -1 else -1
577
                    if (hide_status_change == -1 or \
578
                            user.has_talked_since(hide_status_change) or\
579 580 581 582 583 584
                            user.nick == room.own_nick)\
                            and\
                            (affiliation != user.affiliation or\
                                role != user.role or\
                                show != user.show or\
                                status != user.status):
585
                        # display the message in the room
586
                        self.add_message_to_room(room, msg)
587 588 589 590 591
                    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)
592 593
            if room == self.current_room():
                self.window.user_win.refresh(room.users)
594
        self.window.input.refresh()
595
        doupdate()
596

597
    def add_message_to_room(self, room, txt, time=None, nickname=None):
598
        """
599 600
        Add the message to the room and refresh the associated component
        of the interface
601
        """
602 603
        if room != self.current_room():
            room.add_line_separator()
604
        room.add_message(txt, time, nickname)
605
        if room == self.current_room():
606 607
            self.window.text_win.refresh(room)
        else:
608
            self.window.info_win.refresh(self.rooms, self.current_room())
609
        self.window.input.refresh()
610 611

    def execute(self):
612 613 614
        """
        Execute the /command or just send the line on the current room
        """
615 616
        line = self.window.input.get_text()
        self.window.input.clear_text()
617
        self.window.input.refresh()
618 619
        if line == "":
            return
620
        if line.startswith('/') and not line.startswith('/me '):
621
            command = line.strip()[:].split()[0][1:]
622
            arg = line[2+len(command):] # jump the '/' and the ' '
623
            # example. on "/link 0 open", command = "link" and arg = "0 open"
624
            if command in self.commands.keys():
625
                func = self.commands[command][0]
626
                func(arg)
627
                return
628
            else:
629
                self.add_message_to_room(self.current_room(), _("Error: unknown command (%s)") % (command))
630
        elif self.current_room().name != 'Info':
631 632 633 634 635
            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)
636
        self.window.input.refresh()
637
        doupdate()
638

639
    def command_help(self, arg):
640 641 642
        """
        /help <command_name>
        """
643
        args = arg.split()
644 645
        room = self.current_room()
        if len(args) == 0:
646
            msg = _('Available commands are: ')
647 648
            for command in self.commands.keys():
                msg += "%s " % command
649
            msg += _("\nType /help <command_name> to know what each command does")
650 651 652 653
        if len(args) == 1:
            if args[0] in self.commands.keys():
                msg = self.commands[args[0]][1]
            else:
654
                msg = _('Unknown command: %s') % args[0]
655
        self.add_message_to_room(room, msg)
656

657 658 659 660 661 662 663 664 665 666 667 668 669 670 671
    def command_whois(self, arg):
        """
        /whois <nickname>
        """
        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)

672
    def command_win(self, arg):
673 674 675
        """
        /win <number>
        """
676
        args = arg.split()
677 678 679 680 681 682 683 684 685 686
        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
687
        self.current_room().set_color_state(common.ROOM_STATE_NONE)
688
        self.current_room().remove_line_separator()
689 690 691 692 693
        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:
694 695
                self.current_room().set_color_state(common.ROOM_STATE_CURRENT)
                self.refresh_window()
696
                return
697 698
        self.current_room().set_color_state(common.ROOM_STATE_CURRENT)
        self.refresh_window()
699

700
    def command_kick(self, arg):
701 702 703
        """
        /kick <nick> [reason]
        """
704
        args = arg.split()
705 706 707 708 709 710 711 712 713 714 715 716 717
        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)

718 719 720 721 722 723 724 725 726 727 728 729 730 731
    def command_say(self, arg):
        """
        /say <message>
        """
        line = arg
        if self.current_room().name != 'Info':
            if self.current_room().jid is not None:
                self.muc.send_private_message(self.current_room().name, line)
                self.add_message_to_room(self.current_room(), line.decode('utf-8'), None, self.current_room().own_nick)
            else:
                self.muc.send_message(self.current_room().name, line)
        self.window.input.refresh()
        doupdate()

732
    def command_join(self, arg):
733 734 735
        """
        /join [room][/nick] [password]
        """
736
        args = arg.split()
737
        password = None
738 739 740 741 742 743
        if len(args) == 0:
            r = self.current_room()
            if r.name == 'Info':
                return
            room = r.name
            nick = r.own_nick
744
        else:
745 746
            info = args[0].split('/')
            if len(info) == 1:
747 748 749 750
                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
                nick = config.get('default_nick', '')
                if nick == '':
                    nick = default
751 752
            else:
                nick = info[1]
753
            if info[0] == '':   # happens with /join /nickname, which is OK
754 755 756 757
                r = self.current_room()
                if r.name == 'Info':
                    return
                room = r.name
758 759
                if nick == '':
                    nick = r.own_nick
760 761
            else:
                room = info[0]
762 763 764 765 766 767
            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
768
                    self.add_message_to_room(self.current_room(), _("You didn't specify a server for the room you want to join"))
769
                    return
770
            r = self.get_room_by_name(room)
771 772
        if len(args) == 2:       # a password is provided
            password = args[1]
773
        if r and r.joined:       # if we are already in the room
774
            self.add_message_to_room(self.current_room(), _("already in room [%s]") % room)
775
            return
776
        self.muc.join_room(room, nick, password)
777
        if not r:   # if the room window exists, we don't recreate it.
778
            self.join_room(room, nick)
779
        else:
780
            r.own_nick = nick
781
            # r.own_nick = nick
782
            r.users = []
783

784
    def command_bookmark(self, arg):
785 786 787
        """
        /bookmark [room][/nick]
        """
788
        args = arg.split()
789 790 791 792 793 794 795 796 797 798 799 800 801
        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]
802 803
            if roomname == '':
                roomname = self.current_room().name
804 805 806 807
        if nick:
            res = roomname+'/'+nick
        else:
            res = roomname
808 809 810 811 812 813 814 815 816
        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)
817 818 819
        bookmarks = bookmarked+':'+res
        config.set_and_save('rooms', bookmarks)
        self.add_message_to_room(self.current_room(), _('Your bookmarks are now: %s') % bookmarks)
820

821
    def command_set(self, arg):
822 823 824
        """
        /set <option> [value]
        """
825
        args = arg.split()
826
        if len(args) != 2 and len(args) != 1:
827 828 829
            self.command_help(['set'])
            return
        option = args[0]
830 831 832 833
        if len(args) == 2:
            value = args[1]
        else:
            value = ''
834
        config.set_and_save(option, value)
835 836
        msg = "%s=%s" % (option, value)
        room = self.current_room()
837
        self.add_message_to_room(room, msg)
838

839
    def command_show(self, arg):
840 841 842
        """
        /show <status> [msg]
        """
louiz@4325f9fc-e183-4c21-96ce-0ab188b42d13's avatar