windows.py 50.3 KB
Newer Older
1
# Copyright 2010-2011 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 19 20 21
Define all the windows.
A window is a little part of the screen, for example the input window,
the text window, the roster window, etc.
A Tab (see tab.py) is composed of multiple Windows
22 23
"""

24 25
from gettext import (bindtextdomain, textdomain, bind_textdomain_codeset,
                     gettext as _)
26
from os.path import isfile
27

28 29 30
import logging
log = logging.getLogger(__name__)

31 32 33
import locale
locale.setlocale(locale.LC_ALL, '')

34
import shlex
35
import curses
36
from config import config
37

38 39
from threading import Lock

40
from contact import Contact, Resource
41
from roster import RosterGroup, roster
42

43
from message import Line
44
from tabs import MIN_WIDTH, MIN_HEIGHT
45

46 47
from sleekxmpp.xmlstream.stanzabase import JID

48
import theme
49

50 51
g_lock = Lock()

52
LINES_NB_LIMIT = 4096
53

54
class Win(object):
55 56
    def __init__(self):
        pass
57

58
    def _resize(self, height, width, y, x, parent_win):
59 60
        if height == 0 or width == 0:
            return
61
        self.height, self.width, self.x, self.y = height, width, x, y
62 63 64 65
        self._win = curses.newwin(height, width, y, x)

    def _refresh(self):
        self._win.noutrefresh()
66

67 68
    def addnstr(self, *args):
        """
69
        Safe call to addnstr
70 71
        """
        try:
72
            self._win.addnstr(*args)
73 74 75 76 77
        except:
            pass

    def addstr(self, *args):
        """
78
        Safe call to addstr
79
        """
80
        try:
81
            self._win.addstr(*args)
82 83
        except:
            pass
84

85 86 87 88
    def finish_line(self, color):
        """
        Write colored spaces until the end of line
        """
89
        (y, x) = self._win.getyx()
90 91 92
        size = self.width-x
        self.addnstr(' '*size, size, curses.color_pair(color))

93
class UserList(Win):
94 95
    def __init__(self):
        Win.__init__(self)
96
        self.pos = 0
97 98 99
        self.color_role = {'moderator': theme.COLOR_USER_MODERATOR,
                           'participant':theme.COLOR_USER_PARTICIPANT,
                           'visitor':theme.COLOR_USER_VISITOR,
100 101
                           'none':theme.COLOR_USER_NONE,
                           '':theme.COLOR_USER_NONE
102
                           }
103
        self.color_show = {'xa':theme.COLOR_STATUS_XA,
104 105
                           'none':theme.COLOR_STATUS_NONE,
                           '':theme.COLOR_STATUS_NONE,
106 107 108
                           'dnd':theme.COLOR_STATUS_DND,
                           'away':theme.COLOR_STATUS_AWAY,
                           'chat':theme.COLOR_STATUS_CHAT
109
                           }
110

111 112 113 114 115 116 117 118 119 120 121
    def scroll_up(self):
        self.pos += 4

    def scroll_down(self):
        self.pos -= 4
        if self.pos < 0:
            self.pos = 0

    def draw_plus(self, y):
        self.addstr(y, self.width-2, '++', curses.color_pair(42))

122
    def refresh(self, users):
123
        with g_lock:
124
            self._win.erase()
125
            y = 0
126 127 128 129
            users = sorted(users)
            if self.pos >= len(users) and self.pos != 0:
                self.pos = len(users)-1
            for user in users[self.pos:]:
130 131 132 133 134 135 136 137 138
                if not user.role in self.color_role:
                    role_col = theme.COLOR_USER_NONE
                else:
                    role_col = self.color_role[user.role]
                if not user.show in self.color_show:
                    show_col = theme.COLOR_STATUS_NONE
                else:
                    show_col = self.color_show[user.show]
                self.addstr(y, 0, theme.CHAR_STATUS, curses.color_pair(show_col))
139
                self.addstr(y, 1, user.nick, curses.color_pair(role_col))
140 141 142
                y += 1
                if y == self.height:
                    break
143 144 145 146 147
            # draw indicators of position in the list
            if self.pos > 0:
                self.draw_plus(0)
            if self.pos + self.height < len(users):
                self.draw_plus(self.height-1)
148
            self._refresh()
149

150 151
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
152 153 154
        self._win.attron(curses.color_pair(theme.COLOR_VERTICAL_SEPARATOR))
        self._win.vline(0, 0, curses.ACS_VLINE, self.height)
        self._win.attroff(curses.color_pair(theme.COLOR_VERTICAL_SEPARATOR))
155

156
class Topic(Win):
157 158
    def __init__(self):
        Win.__init__(self)
159
        self._message = ''
160

161 162
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
163

164
    def refresh(self, topic=None):
165
        with g_lock:
166
            self._win.erase()
167 168 169 170 171
            if topic:
                msg = topic[:self.width-1]
            else:
                msg = self._message[:self.width-1]
            self.addstr(0, 0, msg, curses.color_pair(theme.COLOR_TOPIC_BAR))
172
            (y, x) = self._win.getyx()
173 174 175 176
            remaining_size = self.width - x
            if remaining_size:
                self.addnstr(' '*remaining_size, remaining_size,
                             curses.color_pair(theme.COLOR_INFORMATION_BAR))
177
            self._refresh()
178

179 180 181
    def set_message(self, message):
        self._message = message

182
class GlobalInfoBar(Win):
183 184
    def __init__(self):
        Win.__init__(self)
185

186 187
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
188

189
    def refresh(self, tabs, current):
190 191 192 193
        def compare_room(a):
            # return a.nb - b.nb
            return a.nb
        comp = lambda x: x.nb
194
        with g_lock:
195
            self._win.erase()
196
            self.addstr(0, 0, "[", curses.color_pair(theme.COLOR_INFORMATION_BAR))
197 198 199
            sorted_tabs = sorted(tabs, key=comp)
            for tab in sorted_tabs:
                color = tab.get_color_state()
200 201 202
                if config.get('show_inactive_tabs', 'true') == 'false' and\
                        color == theme.COLOR_TAB_NORMAL:
                    continue
203 204 205 206 207
                try:
                    self.addstr("%s" % str(tab.nb), curses.color_pair(color))
                    self.addstr("|", curses.color_pair(theme.COLOR_INFORMATION_BAR))
                except:             # end of line
                    break
208
            (y, x) = self._win.getyx()
209
            self.addstr(y, x-1, '] ', curses.color_pair(theme.COLOR_INFORMATION_BAR))
210
            (y, x) = self._win.getyx()
211 212 213
            remaining_size = self.width - x
            self.addnstr(' '*remaining_size, remaining_size,
                         curses.color_pair(theme.COLOR_INFORMATION_BAR))
214
            self._refresh()
215

216 217 218 219 220
class InfoWin(Win):
    """
    Base class for all the *InfoWin, used in various tabs. For example
    MucInfoWin, etc. Provides some useful methods.
    """
221 222
    def __init__(self):
        Win.__init__(self)
223

224
    def print_scroll_position(self, window):
225 226 227 228 229
        """
        Print, link in Weechat, a -PLUS(n)- where n
        is the number of available lines to scroll
        down
        """
230 231
        if window.pos > 0:
            plus = ' -PLUS(%s)-' % window.pos
232
            self.addstr(plus, curses.color_pair(theme.COLOR_SCROLLABLE_NUMBER) | curses.A_BOLD)
233 234 235 236 237 238

class PrivateInfoWin(InfoWin):
    """
    The live above the information window, displaying informations
    about the MUC user we are talking to
    """
239 240
    def __init__(self):
        InfoWin.__init__(self)
241

242 243
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
244

245
    def refresh(self, room, window):
246

247
        with g_lock:
248
            self._win.erase()
249
            self.write_room_name(room)
250
            self.print_scroll_position(window)
251
            self.finish_line(theme.COLOR_INFORMATION_BAR)
252
            self._refresh()
253 254

    def write_room_name(self, room):
255 256
        jid = JID(room.name)
        room_name, nick = jid.bare, jid.resource
257
        self.addstr(nick, curses.color_pair(13))
258
        txt = ' from room %s' % room_name
259
        self.addstr(txt, curses.color_pair(theme.COLOR_INFORMATION_BAR))
260

261 262 263
class ConversationInfoWin(InfoWin):
    """
    The line above the information window, displaying informations
264
    about the user we are talking to
265
    """
266 267 268 269 270 271 272 273 274 275
    color_show = {'xa':theme.COLOR_STATUS_XA,
                  'none':theme.COLOR_STATUS_ONLINE,
                  '':theme.COLOR_STATUS_ONLINE,
                  'available':theme.COLOR_STATUS_ONLINE,
                  'dnd':theme.COLOR_STATUS_DND,
                  'away':theme.COLOR_STATUS_AWAY,
                  'chat':theme.COLOR_STATUS_CHAT,
                  'unavailable':theme.COLOR_STATUS_UNAVAILABLE
                  }

276 277
    def __init__(self):
        InfoWin.__init__(self)
278

279 280
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
281

282
    def refresh(self, jid, contact, text_buffer, window):
283 284 285
        # contact can be None, if we receive a message
        # from someone not in our roster. In this case, we display
        # only the maximum information from the message we can get.
286 287 288 289 290 291 292 293 294 295 296 297 298
        jid = JID(jid)
        if contact:
            if jid.resource:
                resource = contact.get_resource_by_fulljid(jid.full)
            else:
                resource = contact.get_highest_priority_resource()
        else:
            resource = None
        # if contact is None, then resource is None too: user is not in the roster
        # so we don't know almost anything about it
        # If contact is a Contact, then
        # resource can now be a Resource: user is in the roster and online
        # or resource is None: user is in the roster but offline
299
        with g_lock:
300
            self._win.erase()
301 302 303
            self.write_contact_jid(jid)
            self.write_contact_informations(contact)
            self.write_resource_information(resource)
304
            self.print_scroll_position(window)
305
            self.finish_line(theme.COLOR_INFORMATION_BAR)
306
            self._refresh()
307

308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324
    def write_resource_information(self, resource):
        """
        Write the informations about the resource
        """
        if not resource:
            presence = "unavailable"
        else:
            presence = resource.get_presence()
        color = RosterWin.color_show[presence]
        self.addstr('[', curses.color_pair(theme.COLOR_INFORMATION_BAR))
        self.addstr(" ", curses.color_pair(color))
        self.addstr(']', curses.color_pair(theme.COLOR_INFORMATION_BAR))

    def write_contact_informations(self, contact):
        """
        Write the informations about the contact
        """
325
        if not contact:
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342
            self.addstr("(contact not in roster)", curses.color_pair(theme.COLOR_INFORMATION_BAR))
            return
        display_name = contact.get_name() or contact.get_bare_jid()
        self.addstr('%s '%(display_name), curses.color_pair(theme.COLOR_INFORMATION_BAR))

    def write_contact_jid(self, jid):
        """
        Just write the jid that we are talking to
        """
        self.addstr('[', curses.color_pair(theme.COLOR_INFORMATION_BAR))
        self.addstr(jid.full, curses.color_pair(10))
        self.addstr('] ', curses.color_pair(theme.COLOR_INFORMATION_BAR))

class ConversationStatusMessageWin(InfoWin):
    """
    The upper bar displaying the status message of the contact
    """
343 344
    def __init__(self):
        InfoWin.__init__(self)
345

346 347
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
348 349 350 351 352 353 354 355

    def refresh(self, jid, contact):
        jid = JID(jid)
        if contact:
            if jid.resource:
                resource = contact.get_resource_by_fulljid(jid.full)
            else:
                resource = contact.get_highest_priority_resource()
356
        else:
357 358 359 360 361 362 363 364 365 366
            resource = None
        with g_lock:
            self._win.erase()
            if resource:
                self.write_status_message(resource)
            self.finish_line(theme.COLOR_INFORMATION_BAR)
            self._refresh()

    def write_status_message(self, resource):
        self.addstr(resource.get_status(), curses.color_pair(theme.COLOR_INFORMATION_BAR))
367

368 369 370 371 372
class MucInfoWin(InfoWin):
    """
    The line just above the information window, displaying informations
    about the MUC we are viewing
    """
373 374
    def __init__(self):
        InfoWin.__init__(self)
375

376 377
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
378

379
    def refresh(self, room, window=None):
380
        with g_lock:
381
            self._win.erase()
382 383 384 385
            self.write_room_name(room)
            self.write_own_nick(room)
            self.write_disconnected(room)
            self.write_role(room)
386 387
            if window:
                self.print_scroll_position(window)
388
            self.finish_line(theme.COLOR_INFORMATION_BAR)
389
            self._refresh()
390 391 392 393

    def write_room_name(self, room):
        """
        """
394
        self.addstr('[', curses.color_pair(theme.COLOR_INFORMATION_BAR))
395
        self.addnstr(room.name, len(room.name), curses.color_pair(13))
396
        self.addstr('] ', curses.color_pair(theme.COLOR_INFORMATION_BAR))
397 398 399 400 401 402

    def write_disconnected(self, room):
        """
        Shows a message if the room is not joined
        """
        if not room.joined:
403 404
            self.addstr(' -!- Not connected ', curses.color_pair(theme.COLOR_INFORMATION_BAR))

405 406 407 408 409 410 411 412 413
    def write_own_nick(self, room):
        """
        Write our own nick in the info bar
        """
        nick = room.own_nick
        if not nick:
            return
        if len(nick) > 13:
            nick = nick[:13]+'…'
414
        self.addstr(nick, curses.color_pair(theme.COLOR_INFORMATION_BAR))
415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430

    def write_role(self, room):
        """
        Write our own role and affiliation
        """
        own_user = None
        for user in room.users:
            if user.nick == room.own_nick:
                own_user = user
                break
        if not own_user:
            return
        txt = ' ('
        if own_user.affiliation != 'none':
            txt += own_user.affiliation+', '
        txt += own_user.role+')'
431
        self.addstr(txt, curses.color_pair(theme.COLOR_INFORMATION_BAR))
432

433
class TextWin(Win):
434 435
    def __init__(self):
        Win.__init__(self)
436 437 438 439 440 441
        self.pos = 0
        self.built_lines = []   # Each new message is built and kept here.
        # on resize, we rebuild all the messages

    def scroll_up(self, dist=14):
        self.pos += dist
442 443 444 445
        if self.pos + self.height > len(self.built_lines):
            self.pos = len(self.built_lines) - self.height
            if self.pos < 0:
                self.pos = 0
446 447 448 449 450 451

    def scroll_down(self, dist=14):
        self.pos -= dist
        if self.pos <= 0:
            self.pos = 0

452 453 454 455 456 457 458 459 460 461 462 463 464 465
    def remove_line_separator(self):
        """
        Remove the line separator
        """
        if None in self.built_lines:
            self.built_lines.remove(None)

    def add_line_separator(self):
        """
        add a line separator at the end of messages list
        """
        if None not in self.built_lines:
            self.built_lines.append(None)

466 467 468 469 470 471 472 473 474 475 476 477
    def build_new_message(self, message):
        """
        Take one message, build it and add it to the list
        Return the number of lines that are built for the given
        message.
        """
        if message == None:  # line separator
            self.built_lines.append(None)
            return 0
        txt = message.txt
        if not txt:
            return 0
478
            # length of the time
479 480 481 482 483 484 485 486 487 488 489 490 491
        offset = 9+len(theme.CHAR_TIME_LEFT[:1])+len(theme.CHAR_TIME_RIGHT[:1])
        if message.nickname and len(message.nickname) >= 30:
            nick = message.nickname[:30]+'…'
        else:
            nick = message.nickname
        if nick:
            offset += len(nick) + 2 # + nick + spaces length
        first = True
        this_line_was_broken_by_space = False
        nb = 0
        while txt != '':
            if txt[:self.width-offset].find('\n') != -1:
                limit = txt[:self.width-offset].find('\n')
492
            else:
493 494 495 496 497 498
                # break between words if possible
                if len(txt) >= self.width-offset:
                    limit = txt[:self.width-offset].rfind(' ')
                    this_line_was_broken_by_space = True
                    if limit <= 0:
                        limit = self.width-offset
499 500
                        this_line_was_broken_by_space = False
                else:
501 502
                    limit = self.width-offset-1
                    this_line_was_broken_by_space = False
503
            color = message.user.color if message.user else message.nick_color
504 505 506
            if not first:
                nick = None
                time = None
507 508 509 510 511 512 513
            else:               # strftime is VERY slow, improve performance
                                # by calling it only one time here, and
                                # not at each refresh
                time = {'hour': '%s'%(message.time.strftime("%H"),),
                        'minute': '%s'%(message.time.strftime("%M"),),
                        'second': '%s'%(message.time.strftime("%S"),),
                        }
514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530
            l = Line(nick, color,
                     time,
                     txt[:limit], message.color,
                     offset,
                     message.colorized)
            self.built_lines.append(l)
            nb += 1
            if this_line_was_broken_by_space:
                txt = txt[limit+1:] # jump the space at the start of the line
            else:
                txt = txt[limit:]
            if txt.startswith('\n'):
                txt = txt[1:]
            first = False
        while len(self.built_lines) > LINES_NB_LIMIT:
            self.built_lines.pop(0)
        return nb
531

532
    def refresh(self, room):
533
        """
534 535
        Build the Line objects from the messages, and then write
        them in the text area
536
        """
537
        if self.height <= 0 or not self.built_lines:
538
            return
539 540 541 542 543
        if self.pos != 0:
            lines = self.built_lines[-self.height-self.pos:-self.pos]
        else:
            lines = self.built_lines[-self.height:]
        self._win.move(0, 0)
544
        with g_lock:
545
            self._win.erase()
546 547
            for y, line in enumerate(lines):
                if line is None:
548
                    self.write_line_separator()
549 550 551 552 553 554
                else:
                    if line.time:
                        self.write_time(line.time)
                    if line.nickname:
                        self.write_nickname(line.nickname, line.nickname_color)
                    self.write_text(y, line.text_offset, line.text, line.text_color, line.colorized)
555 556
                if y != self.height - 1:
                    self.addstr('\n')
557
            self._refresh()
558

559 560 561
    def write_line_separator(self):
        """
        """
562
        self.addnstr('- '*(self.width//2-1)+'-', self.width, curses.color_pair(theme.COLOR_NEW_TEXT_SEPARATOR))
563

564
    def write_text(self, y, x, txt, color, colorized):
565
        """
566
        write the text of a line.
567
        """
568
        txt = txt
569 570
        if not colorized:
            if color:
571
                self._win.attron(curses.color_pair(color))
572
            self.addstr(y, x, txt)
573
            if color:
574
                self._win.attroff(curses.color_pair(color))
575 576 577 578 579 580 581

        else:                   # Special messages like join or quit
            special_words = {
                theme.CHAR_JOIN: theme.COLOR_JOIN_CHAR,
                theme.CHAR_QUIT: theme.COLOR_QUIT_CHAR,
                theme.CHAR_KICK: theme.COLOR_KICK_CHAR,
                }
582 583 584
            try:
                splitted = shlex.split(txt)
            except ValueError:
585 586 587 588 589
                # FIXME colors are disabled on too long words
                txt = txt.replace('"[', '').replace(']"', '')\
                    .replace('"{', '').replace('}"', '')\
                    .replace('"(', '').replace(')"', '')
                splitted = txt.split()
590
            for word in splitted:
591
                if word in list(special_words.keys()):
592
                    self.addstr(word, curses.color_pair(special_words[word]))
593
                elif word.startswith('(') and word.endswith(')'):
594 595 596
                    self.addstr('(', curses.color_pair(color))
                    self.addstr(word[1:-1], curses.color_pair(theme.COLOR_CURLYBRACKETED_WORD))
                    self.addstr(')', curses.color_pair(color))
597
                elif word.startswith('{') and word.endswith('}'):
598
                    self.addstr(word[1:-1], curses.color_pair(theme.COLOR_ACCOLADE_WORD))
599
                elif word.startswith('[') and word.endswith(']'):
600
                    self.addstr(word[1:-1], curses.color_pair(theme.COLOR_BRACKETED_WORD))
601
                else:
602
                    self.addstr(word, curses.color_pair(color))
603
                self.addstr(' ')
604

605
    def write_nickname(self, nickname, color):
606 607 608 609
        """
        Write the nickname, using the user's color
        and return the number of written characters
        """
610
        if color:
611
            self._win.attron(curses.color_pair(color))
612
        self.addstr(nickname)
613
        if color:
614
            self._win.attroff(curses.color_pair(color))
615
        self.addstr("> ")
616

617
    def write_time(self, time):
618 619 620
        """
        Write the date on the yth line of the window
        """
621
        self.addstr(theme.CHAR_TIME_LEFT, curses.color_pair(theme.COLOR_TIME_LIMITER))
622
        self.addstr(time['hour'], curses.color_pair(theme.COLOR_TIME_NUMBERS))
623
        self.addstr(':', curses.color_pair(theme.COLOR_TIME_SEPARATOR))
624
        self.addstr(time['minute'], curses.color_pair(theme.COLOR_TIME_NUMBERS))
625
        self.addstr(':', curses.color_pair(theme.COLOR_TIME_SEPARATOR))
626 627
        self.addstr(time['second'], curses.color_pair(theme.COLOR_TIME_NUMBERS))
        self.addstr(theme.CHAR_TIME_RIGHT, curses.color_pair(theme.COLOR_TIME_LIMITER))
628
        self.addstr(' ')
629

630
    def resize(self, height, width, y, x, stdscr, room=None):
631
        self._resize(height, width, y, x, stdscr)
632 633 634 635 636 637 638
        if room:
            self.rebuild_everything(room)

    def rebuild_everything(self, room):
        self.built_lines = []
        for message in room.messages:
            self.build_new_message(message)
639

640 641
class HelpText(Win):
    """
642
    A Window just displaying a read-only message.
643 644 645
    Usually used to replace an Input when the tab is in
    command mode.
    """
646 647
    def __init__(self, text=''):
        Win.__init__(self)
648 649
        self.txt = text

650 651
    def resize(self, height, width, y, x, stdscr):
        self._resize(height, width, y, x, stdscr)
652 653 654 655 656 657 658 659 660 661 662

    def refresh(self):
        with g_lock:
            self._win.erase()
            self.addstr(0, 0, self.txt[:self.width-1], curses.color_pair(theme.COLOR_INFORMATION_BAR))
            self.finish_line(theme.COLOR_INFORMATION_BAR)
            self._refresh()

    def do_command(self, key):
        return False

663 664
class Input(Win):
    """
665 666 667 668 669 670 671 672 673
    The simplest Input possible, provides just a way to edit a single line
    of text. It also has a clipboard, common to all Inputs.
    Doesn't have any history.
    It doesn't do anything when enter is pressed either.
    This should be herited for all kinds of Inputs, for example MessageInput
    or the little inputs in dataforms, etc, adding specific features (completion etc)
    It features two kinds of completion, but they have to be called from outside (the Tab),
    passing the list of items that can be used to complete. The completion can be used
    in a very flexible way.
674
    """
675 676
    clipboard = '' # A common clipboard for all the inputs, this makes
    # it easy cut and paste text between various input
677
    def __init__(self):
678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698
        self.key_func = {
            "KEY_LEFT": self.key_left,
            "M-D": self.key_left,
            "KEY_RIGHT": self.key_right,
            "M-C": self.key_right,
            "KEY_END": self.key_end,
            "KEY_HOME": self.key_home,
            "KEY_DC": self.key_dc,
            '^D': self.key_dc,
            'M-b': self.jump_word_left,
            '^W': self.delete_word,
            '^K': self.delete_end_of_line,
            '^U': self.delete_begining_of_line,
            '^Y': self.paste_clipboard,
            '^A': self.key_home,
            '^E': self.key_end,
            'M-f': self.jump_word_right,
            "KEY_BACKSPACE": self.key_backspace,
            '^?': self.key_backspace,
            }

699
        Win.__init__(self)
700
        self.text = ''
701 702
        self.pos = 0            # cursor position
        self.line_pos = 0 # position (in self.text) of
703

704 705 706
    def is_empty(self):
        return len(self.text) == 0

707 708 709
    def resize(self, height, width, y, x, stdscr):

        self._resize(height, width, y, x, stdscr)
710
        self._win.erase()
711
        self.addnstr(0, 0, self.text, self.width-1)
712

713 714 715 716 717 718 719 720 721 722
    def jump_word_left(self):
        """
        Move the cursor one word to the left
        """
        if not len(self.text) or self.pos == 0:
            return
        previous_space = self.text[:self.pos+self.line_pos].rfind(' ')
        if previous_space == -1:
            previous_space = 0
        diff = self.pos+self.line_pos-previous_space
723
        for i in range(diff):
724
            self.key_left()
725
        return True
726 727 728 729 730 731 732 733 734 735 736

    def jump_word_right(self):
        """
        Move the cursor one word to the right
        """
        if len(self.text) == self.pos+self.line_pos or not len(self.text):
            return
        next_space = self.text.find(' ', self.pos+self.line_pos+1)
        if next_space == -1:
            next_space = len(self.text)
        diff = next_space - (self.pos+self.line_pos)
737
        for i in range(diff):
738
            self.key_right()
739
        return True
740 741 742 743 744 745 746 747 748 749 750

    def delete_word(self):
        """
        Delete the word just before the cursor
        """
        if not len(self.text) or self.pos == 0:
            return
        previous_space = self.text[:self.pos+self.line_pos].rfind(' ')
        if previous_space == -1:
            previous_space = 0
        diff = self.pos+self.line_pos-previous_space
751
        for i in range(diff):
752 753
            self.key_backspace(False)
        self.rewrite_text()
754
        return True
755 756 757 758 759 760 761

    def delete_end_of_line(self):
        """
        Cut the text from cursor to the end of line
        """
        if len(self.text) == self.pos+self.line_pos:
            return              # nothing to cut
762
        Input.clipboard = self.text[self.pos+self.line_pos:]
763 764
        self.text = self.text[:self.pos+self.line_pos]
        self.key_end()
765
        return True
766 767 768 769 770 771 772

    def delete_begining_of_line(self):
        """
        Cut the text from cursor to the begining of line
        """
        if self.pos+self.line_pos == 0:
            return
773
        Input.clipboard = self.text[:self.pos+self.line_pos]
774 775
        self.text = self.text[self.pos+self.line_pos:]
        self.key_home()
776
        return True
777 778 779 780 781

    def paste_clipboard(self):
        """
        Insert what is in the clipboard at the cursor position
        """
782
        if not Input.clipboard or len(Input.clipboard) == 0:
783
            return
784
        for letter in Input.clipboard:
785
            self.do_command(letter)
786
        return True
787

788
    def key_dc(self):
789 790 791
        """
        delete char just after the cursor
        """
792
        self.reset_completion()
793 794 795 796
        if self.pos + self.line_pos == len(self.text):
            return              # end of line, nothing to delete
        self.text = self.text[:self.pos+self.line_pos]+self.text[self.pos+self.line_pos+1:]
        self.rewrite_text()
797
        return True
798 799

    def key_home(self):
800 801 802
        """
        Go to the begining of line
        """
803
        self.reset_completion()
804
        self.pos = 0
805 806
        self.line_pos = 0
        self.rewrite_text()
807
        return True
808

809 810 811 812 813 814
    def key_end(self, reset=False):
        """
        Go to the end of line
        """
        if reset:
            self.reset_completion()
815
        if len(self.text) >= self.width-1:
816 817
            self.pos = self.width-1
            self.line_pos = len(self.text)-self.pos
818
        else:
819 820 821
            self.pos = len(self.text)
            self.line_pos = 0
        self.rewrite_text()
822
        return True
823 824

    def key_left(self):
825 826 827
        """
        Move the cursor one char to the left
        """
828
        self.reset_completion()
829
        (y, x) = self._win.getyx()
830 831 832
        if self.pos == self.width-1 and self.line_pos > 0:
            self.line_pos -= 1
        elif self.pos >= 1:
833
            self.pos -= 1
834
        self.rewrite_text()
835
        return True
836 837

    def key_right(self):
838 839 840
        """
        Move the cursor one char to the right
        """
841
        self.reset_completion()
842
        (y, x) = self._win.getyx()
843 844 845 846
        if self.pos == self.width-1:
            if self.line_pos + self.width-1 < len(self.text):
                self.line_pos += 1
        elif self.pos < len(self.text):
847
            self.pos += 1
848
        self.rewrite_text()
849
        return True
850

851
    def key_backspace(self, reset=True):
852 853 854
        """
        Delete the char just before the cursor
        """
855
        self.reset_completion()
856
        (y, x) = self._win.getyx()
857 858 859 860
        if self.pos == 0:
            return
        self.text = self.text[:self.pos+self.line_pos-1]+self.text[self.pos+self.line_pos:]
        self.key_left()
861 862
        if reset:
            self.rewrite_text()
863
        return True
864

865
    def auto_completion(self, word_list, add_after):
866
        """
867 868 869 870
        Complete the input, from a list of words
        if add_after is None, we use the value defined in completion
        plus a space, after the completion. If it's a string, we use it after the
        completion (with no additional space)
871
        """
872
        if self.pos+self.line_pos != len(self.text): # or len(self.text) == 0
873
            return # we don't complete if cursor is not at the end of line
874
        completion_type = config.get('completion', 'normal')
875
        if completion_type == 'shell' and self.text != '':
876
            self.shell_completion(word_list, add_after)
877
        else