core.py 52 KB
Newer Older
1
# Copyright 2010 Le Coz Florent <louiz@louiz.org>
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
#
# 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/>.

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

21 22
from time import sleep

23
import os
24
import re
25 26
import sys
import shlex
27
import curses
28
import threading
29 30
import webbrowser

31
from datetime import datetime
32 33

import common
34
import theme
35 36
import logging

37 38
from sleekxmpp.xmlstream.stanzabase import JID

39
log = logging.getLogger(__name__)
40

41
import multiuserchat as muc
42
from connection import connection
43
from config import config
44
from tab import MucTab, InfoTab, PrivateTab, RosterInfoTab, ConversationTab
45
from logger import logger
46 47
from user import User
from room import Room
48
from roster import Roster, RosterGroup, roster
49
from contact import Contact, Resource
50
from message import Message
51
from text_buffer import TextBuffer
52
from keyboard import read_char
53
from common import jid_get_domain, is_jid
54

55 56
# http://xmpp.org/extensions/xep-0045.html#errorstatus
ERROR_AND_STATUS_CODES = {
57 58 59 60 61 62 63 64
    '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'),
65 66
    }

67 68 69 70 71 72 73
SHOW_NAME = {
    'dnd': _('busy'),
    'away': _('away'),
    'xa': _('not available'),
    'chat': _('chatty'),
    '': _('available')
    }
74 75 76

resize_lock = threading.Lock()

77
class Core(object):
78
    """
79
    User interface using ncurses
80
    """
81
    def __init__(self, xmpp):
82
        self.running = True
83 84 85
        self.stdscr = curses.initscr()
        self.init_curses(self.stdscr)
        self.xmpp = xmpp
86 87
        default_tab = InfoTab(self, "Info") if self.xmpp.anon\
            else RosterInfoTab(self)
88
        default_tab.on_gain_focus()
89
        self.tabs = [default_tab]
90 91 92 93 94
        # a unique buffer used to store global informations
        # that are displayed in almost all tabs, in an
        # information window.
        self.information_buffer = TextBuffer()
        self.information_win_size = 2 # Todo, get this from config
95
        self.resize_timer = None
96
        self.previous_tab_nb = 0
97
        self.own_nick = config.get('own_nick', self.xmpp.boundjid.bare)
98 99 100 101 102 103 104
        # global commands, available from all tabs
        # a command is tuple of the form:
        # (the function executing the command. Takes a string as argument,
        #  a string representing the help message,
        #  a completion function, taking a Input as argument. Can be None)
        #  The completion function should return True if a completion was
        #  made ; False otherwise
105
        self.commands = {
106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
            'help': (self.command_help, '\_o< KOIN KOIN KOIN', None),
            '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"), self.completion_join),
            'exit': (self.command_quit, _("Usage: /exit\nExit: Just disconnect from the server and exit poezio."), None),
            'next': (self.rotate_rooms_right, _("Usage: /next\nNext: Go to the next room."), None),
            'n': (self.rotate_rooms_right, _("Usage: /n\nN: Go to the next room."), None),
            'prev': (self.rotate_rooms_left, _("Usage: /prev\nPrev: Go to the previous room."), None),
            'p': (self.rotate_rooms_left, _("Usage: /p\nP: Go to the previous room."), None),
            'win': (self.command_win, _("Usage: /win <number>\nWin: Go to the specified room."), None),
            'w': (self.command_win, _("Usage: /w <number>\nW: Go to the specified room."), None),
            '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 [status] argument will be your status message"), None),
            '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]'"), None),
            '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]'"), None),
            '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]'"), None),
            '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]'"), None),
           '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)"), None),
            '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>."), None),
            '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."), None),
            'whois': (self.command_whois, _('Usage: /whois <nickname>\nWhois: Request many informations about the user.'), None),
            'theme': (self.command_theme, _('Usage: /theme\nTheme: Reload the theme defined in the config file.'), None),
125
            'list': (self.command_list, _('Usage: /list\n/List: get the list of public chatrooms on the specified server'), self.completion_list),
126 127
            }

128
        self.key_func = {
129 130
            "KEY_PPAGE": self.scroll_page_up,
            "KEY_NPAGE": self.scroll_page_down,
131
            "KEY_F(5)": self.rotate_rooms_left,
132
            "^P": self.rotate_rooms_left,
133
            'kLFT3': self.rotate_rooms_left,
134
            "KEY_F(6)": self.rotate_rooms_right,
135 136
            "^N": self.rotate_rooms_right,
            'kRIT3': self.rotate_rooms_right,
137 138
            "KEY_F(7)": self.shrink_information_win,
            "KEY_F(8)": self.grow_information_win,
139
            "KEY_RESIZE": self.call_for_resize,
140
            'M-e': self.go_to_important_room,
141
            'M-r': self.go_to_roster,
142 143
            'M-z': self.go_to_previous_tab,
            'M-v': self.move_separator,
144 145
            }

146 147 148 149
        # Add handlers
        self.xmpp.add_event_handler("session_start", self.on_connected)
        self.xmpp.add_event_handler("groupchat_presence", self.on_groupchat_presence)
        self.xmpp.add_event_handler("groupchat_message", self.on_groupchat_message)
150
        self.xmpp.add_event_handler("groupchat_subject", self.on_groupchat_subject)
151
        self.xmpp.add_event_handler("message", self.on_message)
152 153
        self.xmpp.add_event_handler("got_online" , self.on_got_online)
        self.xmpp.add_event_handler("got_offline" , self.on_got_offline)
154
        self.xmpp.add_event_handler("roster_update", self.on_roster_update)
155
        self.xmpp.add_event_handler("changed_status", self.on_presence)
156 157 158 159 160 161 162 163

    def grow_information_win(self):
        """
        """
        if self.information_win_size == 14:
            return
        self.information_win_size += 1
        for tab in self.tabs:
164
            tab.on_info_win_size_changed()
165 166 167 168 169 170 171 172 173
        self.refresh_window()

    def shrink_information_win(self):
        """
        """
        if self.information_win_size == 0:
            return
        self.information_win_size -= 1
        for tab in self.tabs:
174
            tab.on_info_win_size_changed()
175
        self.refresh_window()
176

177 178
    def on_got_offline(self, presence):
        jid = presence['from']
179
        contact = roster.get_contact_by_jid(jid.bare)
180 181
        if not contact:
            return
182 183 184 185 186 187
        resource = contact.get_resource_by_fulljid(jid.full)
        assert resource
        self.information('%s is offline' % (resource.get_jid()), "Roster")
        contact.remove_resource(resource)
        if isinstance(self.current_tab(), RosterInfoTab):
            self.refresh_window()
188 189 190

    def on_got_online(self, presence):
        jid = presence['from']
191
        contact = roster.get_contact_by_jid(jid.bare)
192
        if not contact:
193
            # Todo, handle presence comming from contacts not in roster
194
            return
195 196 197
        resource = contact.get_resource_by_fulljid(jid.full)
        assert not resource
        resource = Resource(jid.full)
198
        status = presence['type']
199
        status_message = presence['status']
200
        priority = presence.getPriority() or 0
201
        resource.set_status(status_message)
202 203 204 205
        resource.set_presence(status)
        resource.set_priority(priority)
        contact.add_resource(resource)
        self.information("%s is online (%s)" % (resource.get_jid().full, status), "Roster")
206

207 208 209 210 211
    def on_connected(self, event):
        """
        Called when we are connected and authenticated
        """
        self.information(_("Welcome on Poezio \o/!"))
212
        self.information(_("Your JID is %s") % self.xmpp.boundjid.full)
213

214 215 216 217
        if not self.xmpp.anon:
            # request the roster
            self.xmpp.getRoster()
            # send initial presence
218
            self.xmpp.makePresence().send()
219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234
        rooms = config.get('rooms', '')
        if rooms == '' or not isinstance(rooms, str):
            return
        rooms = rooms.split(':')
        for room in rooms:
            args = room.split('/')
            if args[0] == '':
                return
            roomname = args[0]
            if len(args) == 2:
                nick = args[1]
            else:
                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
                nick = config.get('default_nick', '')
                if nick == '':
                    nick = default
235
            self.open_new_room(roomname, nick, False)
236
            muc.join_groupchat(self.xmpp, roomname, nick)
237
        # if not self.xmpp.anon:
238 239 240 241 242 243 244 245 246 247 248 249 250
        # Todo: SEND VCARD
        return
        if config.get('jid', '') == '': # Don't send the vcard if we're not anonymous
            self.vcard_sender.start()   # because the user ALREADY has one on the server

    def on_groupchat_presence(self, presence):
        """
        Triggered whenever a presence stanza is received from a user in a multi-user chat room.
        Display the presence on the room window and update the
        presence information of the concerned user
        """
        from_nick = presence['from'].resource
        from_room = presence['from'].bare
251
        room = self.get_room_by_name(from_room)
252 253 254 255 256
        code = presence.find('{jabber:client}status')
        status_codes = set([s.attrib['code'] for s in presence.findall('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}status')])
        # Check if it's not an error presence.
        if presence['type'] == 'error':
            return self.room_error(presence, from_room)
257
        if not room:
258
            return
259 260 261 262 263 264 265 266 267 268 269 270
        msg = None
        affiliation = presence['muc']['affiliation']
        show = presence['show']
        status = presence['status']
        role = presence['muc']['role']
        jid = presence['muc']['jid']
        typ = presence['type']
        if not room.joined:     # user in the room BEFORE us.
            # ignore redondant presence message, see bug #1509
            if from_nick not in [user.nick for user in room.users]:
                new_user = User(from_nick, affiliation, show, status, role)
                room.users.append(new_user)
271
                if from_nick == room.own_nick:
272 273
                    room.joined = True
                    new_user.color = theme.COLOR_OWN_NICK
274
                    self.add_message_to_text_buffer(room, _("Your nickname is %s") % (from_nick))
275
                    if '170' in status_codes:
276
                        self.add_message_to_text_buffer(room, 'Warning: this room is publicly logged')
277
        else:
278 279 280 281 282 283 284 285
            change_nick = '303' in status_codes
            kick = '307' in status_codes and typ == 'unavailable'
            user = room.get_user_by_name(from_nick)
            # New user
            if not user:
                self.on_user_join(room, from_nick, affiliation, show, status, role, jid)
            # nick change
            elif change_nick:
286
                self.on_user_nick_change(room, presence, user, from_nick, from_room)
287 288
            # kick
            elif kick:
289
                self.on_user_kicked(room, presence, user, from_nick)
290 291
            # user quit
            elif typ == 'unavailable':
292
                self.on_user_leave_groupchat(room, user, jid, status, from_nick, from_room)
293 294
            # status change
            else:
295
                self.on_user_change_status(room, user, from_nick, from_room, affiliation, role, show, status)
296
        self.refresh_window()
297
        self.doupdate()
298

299 300 301 302 303 304 305 306 307
    def on_user_join(self, room, from_nick, affiliation, show, status, role, jid):
        """
        When a new user joins a groupchat
        """
        room.users.append(User(from_nick, affiliation,
                               show, status, role))
        hide_exit_join = config.get('hide_exit_join', -1)
        if hide_exit_join != 0:
            if not jid.full:
308
                self.add_message_to_text_buffer(room, _('%(spec)s "[%(nick)s]" joined the room') % {'nick':from_nick.replace('"', '\\"'), 'spec':theme.CHAR_JOIN.replace('"', '\\"')}, colorized=True)
309
            else:
310
                self.add_message_to_text_buffer(room, _('%(spec)s "[%(nick)s]" "(%(jid)s)" joined the room') % {'spec':theme.CHAR_JOIN.replace('"', '\\"'), 'nick':from_nick.replace('"', '\\"'), 'jid':jid.full}, colorized=True)
311

312 313 314 315 316
    def on_user_nick_change(self, room, presence, user, from_nick, from_room):
        new_nick = presence.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item').attrib['nick']
        if user.nick == room.own_nick:
            room.own_nick = new_nick
            # also change our nick in all private discussion of this room
317
            for _tab in self.tabs:
318 319
                if isinstance(_tab, PrivateTab) and _tab.get_name().split('/', 1)[0] == room.name:
                    _tab.get_room().own_nick = new_nick
320
        user.change_nick(new_nick)
321
        self.add_message_to_text_buffer(room, _('"[%(old)s]" is now known as "[%(new)s]"') % {'old':from_nick.replace('"', '\\"'), 'new':new_nick.replace('"', '\\"')}, colorized=True)
322 323 324
        # rename the private tabs if needed
        private_room = self.get_room_by_name('%s/%s' % (from_room, from_nick))
        if private_room:
325
            self.add_message_to_text_buffer(private_room, _('"[%(old_nick)s]" is now known as "[%(new_nick)s]"') % {'old_nick':from_nick.replace('"', '\\"'), 'new_nick':new_nick.replace('"', '\\"')}, colorized=True)
326
            new_jid = private_room.name.split('/', 1)[0]+'/'+new_nick
327
            private_room.name = new_jid
328

329 330 331 332 333 334 335
    def on_user_kicked(self, room, presence, user, from_nick):
        """
        When someone is kicked
        """
        room.users.remove(user)
        by = presence.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item/{http://jabber.org/protocol/muc#user}actor')
        reason = presence.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item/{http://jabber.org/protocol/muc#user}reason')
336
        by = by.attrib['jid'] if by is not None else None
337
        reason = reason.text if reason else ''
338 339 340
        if from_nick == room.own_nick: # we are kicked
            room.disconnect()
            if by:
341
                kick_msg = _('%(spec)s [You] have been kicked by "[%(by)s]"') % {'spec': theme.CHAR_KICK.replace('"', '\\"'), 'by':by}
342
            else:
343
                kick_msg = _('%(spec)s [You] have been kicked.') % {'spec':theme.CHAR_KICK.replace('"', '\\"')}
344 345 346 347 348
            # try to auto-rejoin
            if config.get('autorejoin', 'false') == 'true':
                muc.join_groupchat(self.xmpp, room.name, room.own_nick)
        else:
            if by:
349
                kick_msg = _('%(spec)s "[%(nick)s]" has been kicked by "[%(by)s]"') % {'spec':theme.CHAR_KICK.replace('"', '\\"'), 'nick':from_nick.replace('"', '\\"'), 'by':by.replace('"', '\\"')}
350
            else:
351
                kick_msg = _('%(spec)s "[%(nick)s]" has been kicked') % {'spec':theme.CHAR_KICK, 'nick':from_nick.replace('"', '\\"')}
352 353
        if reason:
            kick_msg += _(' Reason: %(reason)s') % {'reason': reason}
354
        self.add_message_to_text_buffer(room, kick_msg, colorized=True)
355

356 357 358 359 360
    def on_user_leave_groupchat(self, room, user, jid, status, from_nick, from_room):
        """
        When an user leaves a groupchat
        """
        room.users.remove(user)
361 362 363
        if room.own_nick == user.nick:
            # We are now out of the room. Happens with some buggy (? not sure) servers
            room.disconnect()
364 365 366
        hide_exit_join = config.get('hide_exit_join', -1) if config.get('hide_exit_join', -1) >= -1 else -1
        if hide_exit_join == -1 or user.has_talked_since(hide_exit_join):
            if not jid.full:
367
                leave_msg = _('%(spec)s "[%(nick)s]" has left the room') % {'nick':from_nick.replace('"', '\\"'), 'spec':theme.CHAR_QUIT.replace('"', '\\"')}
368
            else:
369
                leave_msg = _('%(spec)s "[%(nick)s]" "(%(jid)s)" has left the room') % {'spec':theme.CHAR_QUIT.replace('"', '\\"'), 'nick':from_nick.replace('"', '\\"'), 'jid':jid.full.replace('"', '\\"')}
370 371
            if status:
                leave_msg += ' (%s)' % status
372
            self.add_message_to_text_buffer(room, leave_msg, colorized=True)
373 374 375
        private_room = self.get_room_by_name('%s/%s' % (from_room, from_nick))
        if private_room:
            if not status:
376
                self.add_message_to_text_buffer(private_room, _('%(spec)s "[%(nick)s]" has left the room') % {'nick':from_nick.replace('"', '\\"'), 'spec':theme.CHAR_QUIT.replace('"', '\\"')}, colorized=True)
377
            else:
378
                self.add_message_to_text_buffer(private_room, _('%(spec)s "[%(nick)s]" has left the room "(%(status)s)"') % {'nick':from_nick.replace('"', '\\"'), 'spec':theme.CHAR_QUIT, 'status': status.replace('"', '\\"')}, colorized=True)
379

380 381 382 383 384
    def on_user_change_status(self, room, user, from_nick, from_room, affiliation, role, show, status):
        """
        When an user changes her status
        """
        # build the message
385 386
        display_message = False # flag to know if something significant enough
        # to be displayed has changed
387
        msg = _('"%s" changed: ')% from_nick.replace('"', '\\"')
388
        if affiliation != user.affiliation:
389
            msg += _('affiliation: %s, ') % affiliation
390
            display_message = True
391
        if role != user.role:
392
            msg += _('role: %s, ') % role
393
            display_message = True
394
        if show != user.show and show in list(SHOW_NAME.keys()):
395
            msg += _('show: %s, ') % SHOW_NAME[show]
396 397
            display_message = True
        if status and status != user.status:
398
            msg += _('status: %s, ') % status
399 400 401
            display_message = True
        if not display_message:
            return
402
        msg = msg[:-2] # remove the last ", "
403 404 405 406 407 408 409 410 411 412
        hide_status_change = config.get('hide_status_change', -1) if config.get('hide_status_change', -1) >= -1 else -1
        if (hide_status_change == -1 or \
                user.has_talked_since(hide_status_change) or\
                user.nick == room.own_nick)\
                and\
                (affiliation != user.affiliation or\
                    role != user.role or\
                    show != user.show or\
                    status != user.status):
            # display the message in the room
413
            self.add_message_to_text_buffer(room, msg, colorized=True)
414 415
        private_room = self.get_room_by_name('%s/%s' % (from_room, from_nick))
        if private_room: # display the message in private
416
            self.add_message_to_text_buffer(private_room, msg, colorized=True)
417 418 419
        # finally, effectively change the user status
        user.update(affiliation, show, status, role)

420 421 422 423 424 425
    def on_message(self, message):
        """
        When receiving private message from a muc OR a normal message
        (from one of our contacts)
        """
        if message['type'] == 'groupchat':
426
            return
427 428
        # Differentiate both type of messages, and call the appropriate handler.
        jid_from = message['from']
429
        for tab in self.tabs:
430
            if tab.get_name() == jid_from.full:
431
                if message['type'] == 'error':
432
                    return self.room_error(message, tab.get_room().name)
433 434
                else:
                    return self.on_groupchat_private_message(message)
435 436 437 438 439 440 441
        return self.on_normal_message(message)

    def on_groupchat_private_message(self, message):
        """
        We received a Private Message (from someone in a Muc)
        """
        jid = message['from']
442
        nick_from = jid.resource
443
        room_from = jid.bare
444 445
        room = self.get_room_by_name(jid.full) # get the tab with the private conversation
        if not room: # It's the first message we receive: create the tab
446
            room = self.open_private_window(room_from, nick_from, False)
447 448 449
            if not room:
                return
        body = message['body']
450 451 452
        room.add_message(body, time=None, nickname=nick_from,
                         colorized=False,
                         forced_user=self.get_room_by_name(room_from).get_user_by_name(nick_from))
453
        self.refresh_window()
454
        self.doupdate()
455

456 457 458 459 460
    def focus_tab_named(self, tab_name):
        for tab in self.tabs:
            if tab.get_name() == tab_name:
                self.command_win('%s' % (tab.nb,))

461 462 463 464
    def on_normal_message(self, message):
        """
        When receiving "normal" messages (from someone in our roster)
        """
465
        jid = message['from']
466 467 468
        body = message['body']
        if not body:
            return
469 470 471 472 473 474 475 476 477 478 479 480
        # We first check if we have a conversation opened with this precise resource
        conversation = self.get_tab_by_name(jid.full)
        if not conversation:
            # If not, we search for a conversation with the bare jid
            conversation = self.get_tab_by_name(jid.bare)
            if not conversation:
                # We create the conversation with the bare Jid if nothing was found
                conversation = self.open_conversation_window(jid.bare, False)
            # room = self.open_conversation_window(jid, False)
        self.add_message_to_text_buffer(conversation.get_room(), body, None, jid.full)
        if self.current_tab() is not conversation:
            conversation.set_color_state(theme.COLOR_TAB_PRIVATE)
481
        self.refresh_window()
482 483 484 485 486
        return

    def on_presence(self, presence):
        """
        """
487
        jid = presence['from']
488
        contact = roster.get_contact_by_jid(jid.bare)
489 490 491 492 493 494
        if not contact:
            return
        resource = contact.get_resource_by_fulljid(jid.full)
        if not resource:
            return
        status = presence['type']
495
        status_message = presence['status']
496 497 498
        priority = presence.getPriority() or 0
        resource.set_presence(status)
        resource.set_priority(priority)
499
        resource.set_status(status_message)
500 501
        if isinstance(self.current_tab(), RosterInfoTab):
            self.refresh_window()
502

503 504 505 506 507
    def on_roster_update(self, iq):
        """
        A subscription changed, or we received a roster item
        after a roster request, etc
        """
508 509
        for item in iq.findall('{jabber:iq:roster}query/{jabber:iq:roster}item'):
            jid = item.attrib['jid']
510
            contact = roster.get_contact_by_jid(jid)
511 512
            if not contact:
                contact = Contact(jid)
513
                roster.add_contact(contact, jid)
514 515 516 517
            if 'ask' in item.attrib:
                contact.set_ask(item.attrib['ask'])
            else:
                contact.set_ask(None)
518 519 520 521
            if 'name' in item.attrib:
                contact.set_name(item.attrib['name'])
            else:
                contact.set_name(None)
522 523 524
            if item.attrib['subscription']:
                contact.set_subscription(item.attrib['subscription'])
            groups = item.findall('{jabber:iq:roster}group')
525
            roster.edit_groups_of_contact(contact, [group.text for group in groups])
526 527
        if isinstance(self.current_tab(), RosterInfoTab):
            self.refresh_window()
528

529 530 531 532 533 534 535
    def call_for_resize(self):
        """
        Starts a very short timer. If no other terminal resize
        occured in this delay then poezio is REALLY resize.
        This is to avoid multiple unnecessary software resizes (this
        can be heavy on resource on slow computers or networks)
        """
536 537 538 539 540 541 542 543 544
        with resize_lock:
            if self.resize_timer:
                # a recent terminal resize occured.
                # Cancel the programmed software resize
                self.resize_timer.cancel()
            # add the new timer
            self.resize_timer = threading.Timer(0.1, self.resize_window)
            self.resize_timer.start()
        # self.resize_window()
545

546 547 548 549
    def resize_window(self):
        """
        Resize the whole screen
        """
550
        with resize_lock:
551
           # self.resize_timer = None
552
            for tab in self.tabs:
553
                tab.resize()
554
            self.refresh_window()
555

556
    def main_loop(self):
557 558 559
        """
        main loop waiting for the user to press a key
        """
560
        self.refresh_window()
561
        while self.running:
562
            self.doupdate()
563
            char=read_char(self.stdscr)
564
            # search for keyboard shortcut
565
            if char in list(self.key_func.keys()):
566
                self.key_func[char]()
567
            else:
568
                self.do_command(char)
569

570
    def current_tab(self):
571 572 573
        """
        returns the current room, the one we are viewing
        """
574
        return self.tabs[0]
575

576 577 578 579 580 581 582 583 584 585
    def get_conversation_by_jid(self, jid):
        """
        Return the room of the ConversationTab with the given jid
        """
        for tab in self.tabs:
            if isinstance(tab, ConversationTab):
                if tab.get_name() == jid:
                    return tab.get_room()
        return None

586 587 588 589 590 591 592 593 594
    def get_tab_by_name(self, name):
        """
        Get the tab with the given name.
        """
        for tab in self.tabs:
            if tab.get_name() == name:
                return tab
        return None

595
    def get_room_by_name(self, name):
596 597 598
        """
        returns the room that has this name
        """
599 600 601 602
        for tab in self.tabs:
            if (isinstance(tab, MucTab) or
                isinstance(tab, PrivateTab)) and tab.get_name() == name:
                return tab.get_room()
603
        return None
604

605
    def init_curses(self, stdscr):
606 607 608
        """
        ncurses initialization
        """
609
        curses.curs_set(1)
610
        curses.noecho()
611 612
        # curses.raw()
        theme.init_colors()
613
        stdscr.keypad(True)
614

615
    def reset_curses(self):
616 617 618 619 620
        """
        Reset terminal capabilities to what they were before ncurses
        init
        """
        curses.echo()
621
        curses.nocbreak()
622
        curses.endwin()
623

624 625 626 627
    def refresh_window(self):
        """
        Refresh everything
        """
628
        self.current_tab().set_color_state(theme.COLOR_TAB_CURRENT)
629
        self.current_tab().refresh(self.tabs, self.information_buffer, roster)
630
        self.doupdate()
631

632
    def open_new_room(self, room, nick, focus=True):
633
        """
634
        Open a new MucTab containing a muc Room, using the specified nick
635
        """
636
        r = Room(room, nick)
637
        new_tab = MucTab(self, r)
638 639
        if self.current_tab().nb == 0:
            self.tabs.append(new_tab)
640
        else:
641 642 643
            for ta in self.tabs:
                if ta.nb == 0:
                    self.tabs.insert(self.tabs.index(ta), new_tab)
644
                    break
645
        if focus:
646
            self.command_win("%s" % new_tab.nb)
647
        self.refresh_window()
648

649 650 651 652 653 654
    def go_to_roster(self):
        self.command_win('0')

    def go_to_previous_tab(self):
        self.command_win('%s' % (self.previous_tab_nb,))

655 656 657 658 659 660 661
    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
        """
662 663 664
        for tab in self.tabs:
            if tab.get_color_state() == theme.COLOR_TAB_PRIVATE:
                self.command_win('%s' % tab.nb)
665
                return
666 667 668
        for tab in self.tabs:
            if tab.get_color_state() == theme.COLOR_TAB_HIGHLIGHT:
                self.command_win('%s' % tab.nb)
669
                return
670 671 672
        for tab in self.tabs:
            if tab.get_color_state() == theme.COLOR_TAB_NEW_MESSAGE:
                self.command_win('%s' % tab.nb)
673
                return
674

675
    def rotate_rooms_right(self, args=None):
676 677 678
        """
        rotate the rooms list to the right
        """
679 680 681
        self.current_tab().on_lose_focus()
        self.tabs.append(self.tabs.pop(0))
        self.current_tab().on_gain_focus()
682
        self.refresh_window()
683

684
    def rotate_rooms_left(self, args=None):
685 686 687
        """
        rotate the rooms list to the right
        """
688 689 690
        self.current_tab().on_lose_focus()
        self.tabs.insert(0, self.tabs.pop())
        self.current_tab().on_gain_focus()
691
        self.refresh_window()
692

693
    def scroll_page_down(self, args=None):
694
        self.current_tab().on_scroll_down()
695
        self.refresh_window()
696 697

    def scroll_page_up(self, args=None):
698
        self.current_tab().on_scroll_up()
699
        self.refresh_window()
700

701
    def room_error(self, error, room_name):
702 703 704
        """
        Display the error on the room window
        """
705 706 707 708 709 710
        room = self.get_room_by_name(room_name)
        msg = error['error']['type']
        condition = error['error']['condition']
        code = error['error']['code']
        body = error['error']['text']
        if not body:
711
            if code in list(ERROR_AND_STATUS_CODES.keys()):
712 713
                body = ERROR_AND_STATUS_CODES[code]
            else:
714 715
                body = condition or _('Unknown error')
        if code:
716 717
            msg = _('Error: %(code)s - %(msg)s: %(body)s') % {'msg':msg, 'body':body, 'code':code}
            self.add_message_to_text_buffer(room, msg)
718
        else:
719 720
            msg = _('Error: %(msg)s: %(body)s') % {'msg':msg, 'body':body}
            self.add_message_to_text_buffer(room, msg)
721
        if code == '401':
722 723
            msg = _('To provide a password in order to join the room, type "/join / password" (replace "password" by the real password)')
            self.add_message_to_text_buffer(room, msg)
724
        if code == '409':
725 726 727
            if config.get('alternative_nickname', '') != '':
                self.command_join('%s/%s'% (room.name, room.own_nick+config.get('alternative_nickname', '')))
            else:
728
                self.add_message_to_text_buffer(room, _('You can join the room with an other nick, by typing "/join /other_nick"'))
729
        self.refresh_window()
730

731
    def open_conversation_window(self, jid, focus=True):
732 733 734
        """
        open a new conversation tab and focus it if needed
        """
735
        text_buffer = TextBuffer()
736
        new_tab = ConversationTab(self, text_buffer, jid)
737 738 739 740 741 742 743 744 745 746 747
        # insert it in the rooms
        if self.current_tab().nb == 0:
            self.tabs.append(new_tab)
        else:
            for ta in self.tabs:
                if ta.nb == 0:
                    self.tabs.insert(self.tabs.index(ta), new_tab)
                    break
        if focus:               # focus the room if needed
            self.command_win('%s' % (new_tab.nb))
        self.refresh_window()
748
        return new_tab
749

750
    def open_private_window(self, room_name, user_nick, focus=True):
751
        complete_jid = room_name+'/'+user_nick
752 753 754 755
        for tab in self.tabs: # if the room exists, focus it and return
            if isinstance(tab, PrivateTab):
                if tab.get_name() == complete_jid:
                    self.command_win('%s' % tab.nb)
756 757
                    return
        # create the new tab
758
        room = self.get_room_by_name(room_name)
759 760 761
        if not room:
            return None
        own_nick = room.own_nick
762
        r = Room(complete_jid, own_nick) # PrivateRoom here
763
        new_tab = PrivateTab(self, r)
764
        # insert it in the tabs
765 766
        if self.current_tab().nb == 0:
            self.tabs.append(new_tab)
767
        else:
768 769 770
            for ta in self.tabs:
                if ta.nb == 0:
                    self.tabs.insert(self.tabs.index(ta), new_tab)
771 772
                    break
        if focus:               # focus the room if needed
773
            self.command_win('%s' % (new_tab.nb))
774
        # self.window.new_room(r)
775
        self.refresh_window()
776 777
        return r

778 779 780 781 782 783 784 785 786 787
    def on_groupchat_subject(self, message):
        """
        triggered when the topic is changed
        """
        nick_from = message['mucnick']
        room_from = message.getMucroom()
        room = self.get_room_by_name(room_from)
        subject = message['subject']
        if not subject:
            return
788
        self.add_message_to_text_buffer(room, _("%(nick)s set the subject to: %(subject)s") % {'nick':nick_from, 'subject':subject}, time=None)
789 790 791
        room.topic = subject.replace('\n', '|')
        self.refresh_window()

792
    def on_groupchat_message(self, message):
793
        """
794
        Triggered whenever a message is received from a multi-user chat room.
795
        """
796 797
        delay_tag = message.find('{urn:xmpp:delay}delay')
        if delay_tag is not None:
798
            delayed = True
799
            date = common.datetime_tuple(delay_tag.attrib['stamp'])
800
        else:
801 802
            # We support the OLD and deprecated XEP: http://xmpp.org/extensions/xep-0091.html
            # But it sucks, please, Jabber servers, don't do this :(
803 804
            delay_tag = message.find('{jabber:x:delay}x')
            if delay_tag is not None:
805
                delayed = True
806
                date = common.datetime_tuple(delay_tag.attrib['stamp'])
807 808
            else:
                delayed = False
809
                date = None
810
        nick_from = message['mucnick']
811
        room_from = message.getMucroom()
812 813
        if message['type'] == 'error': # Check if it's an error