core.py 50.5 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
from datetime import datetime
31 32

import common
33
import theme
34 35
import logging

36 37
from sleekxmpp.xmlstream.stanzabase import JID

38
log = logging.getLogger(__name__)
39

40
import multiuserchat as muc
41
import tabs
42
import windows
43

44
from connection import connection
45
from config import config
46
from logger import logger
47 48
from user import User
from room import Room
49
from roster import Roster, RosterGroup, roster
50
from contact import Contact, Resource
51
from message import Message
52
from text_buffer import TextBuffer
53
from keyboard import read_char
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
        self.information_buffer = TextBuffer()
87
        self.information_win_size = config.get('info_win_height', 2, 'var')
88 89
        default_tab = tabs.InfoTab(self, "Info") if self.xmpp.anon\
            else tabs.RosterInfoTab(self)
90
        default_tab.on_gain_focus()
91
        self.tabs = [default_tab]
92 93 94
        # a unique buffer used to store global informations
        # that are displayed in almost all tabs, in an
        # information window.
95

96
        self.resize_timer = None
97
        self.previous_tab_nb = 0
98
        self.own_nick = config.get('own_nick', self.xmpp.boundjid.bare)
99 100 101 102 103 104 105
        # 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
106
        self.commands = {
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),
            '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
            '^L': self.full_screen_redraw,
145 146
            }

147 148 149 150
        # 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)
151
        self.xmpp.add_event_handler("groupchat_subject", self.on_groupchat_subject)
152
        self.xmpp.add_event_handler("message", self.on_message)
153 154
        self.xmpp.add_event_handler("got_online" , self.on_got_online)
        self.xmpp.add_event_handler("got_offline" , self.on_got_offline)
155
        self.xmpp.add_event_handler("roster_update", self.on_roster_update)
156
        self.xmpp.add_event_handler("changed_status", self.on_presence)
157 158 159 160 161 162 163 164

    def grow_information_win(self):
        """
        """
        if self.information_win_size == 14:
            return
        self.information_win_size += 1
        for tab in self.tabs:
165
            tab.on_info_win_size_changed()
166 167 168 169 170 171 172 173 174
        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:
175
            tab.on_info_win_size_changed()
176
        self.refresh_window()
177

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

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

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

215 216 217 218
        if not self.xmpp.anon:
            # request the roster
            self.xmpp.getRoster()
            # send initial presence
219
            self.xmpp.makePresence().send()
220 221 222 223 224
        rooms = config.get('rooms', '')
        if rooms == '' or not isinstance(rooms, str):
            return
        rooms = rooms.split(':')
        for room in rooms:
225 226
            jid = JID(room)
            if jid.bare == '':
227
                return
228 229
            if jid.resource != '':
                nick = jid.resource
230 231 232 233 234
            else:
                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
                nick = config.get('default_nick', '')
                if nick == '':
                    nick = default
235 236
            self.open_new_room(jid.bare, nick, False)
            muc.join_groupchat(self.xmpp, jid.bare, 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
                if isinstance(_tab, tabs.PrivateTab) and JID(_tab.get_name()).bare == room.name:
319
                    _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 = JID(private_room.name).bare+'/'+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.bare and isinstance(tab, tabs.MucTab):
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

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

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

528 529 530 531 532 533 534
    def full_screen_redraw(self):
        """
        Completely erase and redraw the screen
        """
        self.stdscr.clear()
        self.call_for_resize()

535 536 537 538 539 540 541
    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)
        """
542 543 544 545 546 547 548 549 550
        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()
551

552 553 554 555
    def resize_window(self):
        """
        Resize the whole screen
        """
556 557
        with resize_lock:
            for tab in self.tabs:
558
                tab.resize()
559
            self.refresh_window()
560

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

575
    def current_tab(self):
576 577 578
        """
        returns the current room, the one we are viewing
        """
579
        return self.tabs[0]
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:
586
            if isinstance(tab, tabs.ConversationTab):
587 588 589 590
                if tab.get_name() == jid:
                    return tab.get_room()
        return None

591 592 593 594 595 596 597 598 599
    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

600
    def get_room_by_name(self, name):
601 602 603
        """
        returns the room that has this name
        """
604
        for tab in self.tabs:
605
            if (isinstance(tab, tabs.MucTab) or
606
                isinstance(tab, tabs.PrivateTab)) and tab.get_name() == name:
607
                return tab.get_room()
608
        return None
609

610
    def init_curses(self, stdscr):
611 612 613
        """
        ncurses initialization
        """
614
        curses.curs_set(1)
615
        curses.noecho()
616 617
        # curses.raw()
        theme.init_colors()
618
        stdscr.keypad(True)
619

620
    def reset_curses(self):
621 622 623 624 625
        """
        Reset terminal capabilities to what they were before ncurses
        init
        """
        curses.echo()
626
        curses.nocbreak()
627
        curses.endwin()
628

629 630 631 632
    def refresh_window(self):
        """
        Refresh everything
        """
633
        self.current_tab().set_color_state(theme.COLOR_TAB_CURRENT)
634
        self.current_tab().refresh(self.tabs, self.information_buffer, roster)
635
        self.doupdate()
636

637
    def add_tab(self, new_tab, focus=False):
638
        """
639 640
        Appends the new_tab in the tab list and
        focus it if focus==True
641
        """
642 643
        if self.current_tab().nb == 0:
            self.tabs.append(new_tab)
644
        else:
645 646 647
            for ta in self.tabs:
                if ta.nb == 0:
                    self.tabs.insert(self.tabs.index(ta), new_tab)
648
                    break
649
        if focus:
650
            self.command_win("%s" % new_tab.nb)
651 652 653 654 655 656 657 658

    def open_new_room(self, room, nick, focus=True):
        """
        Open a new tab.MucTab containing a muc Room, using the specified nick
        """
        r = Room(room, nick)
        new_tab = tabs.MucTab(self, r)
        self.add_tab(new_tab, focus)
659
        self.refresh_window()
660

661 662 663 664 665 666
    def go_to_roster(self):
        self.command_win('0')

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

667 668 669 670 671 672 673
    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
        """
674 675 676
        for tab in self.tabs:
            if tab.get_color_state() == theme.COLOR_TAB_PRIVATE:
                self.command_win('%s' % tab.nb)
677
                return
678 679 680
        for tab in self.tabs:
            if tab.get_color_state() == theme.COLOR_TAB_HIGHLIGHT:
                self.command_win('%s' % tab.nb)
681
                return
682 683 684
        for tab in self.tabs:
            if tab.get_color_state() == theme.COLOR_TAB_NEW_MESSAGE:
                self.command_win('%s' % tab.nb)
685
                return
686 687 688 689
        for tab in self.tabs:
            if isinstance(tab, tabs.ChatTab) and not tab.input.is_empty():
                self.command_win('%s' % tab.nb)
                return
690

691
    def rotate_rooms_right(self, args=None):
692 693 694
        """
        rotate the rooms list to the right
        """
695 696 697
        self.current_tab().on_lose_focus()
        self.tabs.append(self.tabs.pop(0))
        self.current_tab().on_gain_focus()
698
        self.refresh_window()
699

700
    def rotate_rooms_left(self, args=None):
701 702 703
        """
        rotate the rooms list to the right
        """
704 705 706
        self.current_tab().on_lose_focus()
        self.tabs.insert(0, self.tabs.pop())
        self.current_tab().on_gain_focus()
707
        self.refresh_window()
708

709
    def scroll_page_down(self, args=None):
710
        self.current_tab().on_scroll_down()
711
        self.refresh_window()
712 713

    def scroll_page_up(self, args=None):
714
        self.current_tab().on_scroll_up()
715
        self.refresh_window()
716

717
    def room_error(self, error, room_name):
718 719 720
        """
        Display the error on the room window
        """
721 722 723 724 725 726
        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:
727
            if code in list(ERROR_AND_STATUS_CODES.keys()):
728 729
                body = ERROR_AND_STATUS_CODES[code]
            else:
730 731
                body = condition or _('Unknown error')
        if code:
732 733
            msg = _('Error: %(code)s - %(msg)s: %(body)s') % {'msg':msg, 'body':body, 'code':code}
            self.add_message_to_text_buffer(room, msg)
734
        else:
735 736
            msg = _('Error: %(msg)s: %(body)s') % {'msg':msg, 'body':body}
            self.add_message_to_text_buffer(room, msg)
737
        if code == '401':
738 739
            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)
740
        if code == '409':
741 742 743
            if config.get('alternative_nickname', '') != '':
                self.command_join('%s/%s'% (room.name, room.own_nick+config.get('alternative_nickname', '')))
            else:
744
                self.add_message_to_text_buffer(room, _('You can join the room with an other nick, by typing "/join /other_nick"'))
745
        self.refresh_window()
746

747
    def open_conversation_window(self, jid, focus=True):
748 749 750
        """
        open a new conversation tab and focus it if needed
        """
751
        text_buffer = TextBuffer()
752
        new_tab = tabs.ConversationTab(self, text_buffer, jid)
753
        # insert it in the rooms
754
        self.add_tab(new_tab, focus)
755
        self.refresh_window()
756
        return new_tab
757

758
    def open_private_window(self, room_name, user_nick, focus=True):
759
        complete_jid = room_name+'/'+user_nick
760
        for tab in self.tabs: # if the room exists, focus it and return
761
            if isinstance(tab, tabs.PrivateTab):
762 763
                if tab.get_name() == complete_jid:
                    self.command_win('%s' % tab.nb)
764
                    return tab.get_room()
765
        # create the new tab
766
        room = self.get_room_by_name(room_name)
767 768 769
        if not room:
            return None
        own_nick = room.own_nick
770
        r = Room(complete_jid, own_nick) # PrivateRoom here
771
        new_tab = tabs.PrivateTab(self, r)
772
        # insert it in the tabs
773
        self.add_tab(new_tab, focus)
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
        """