poezio/src/gui.py
2010-05-11 10:38:57 +00:00

588 lines
25 KiB
Python

#!/usr/bin/python
# -*- coding:utf-8 -*-
#
# Copyright 2010 Le Coz Florent <louizatakk@fedoraproject.org>
#
# This file is part of Poezio.
#
# Poezio is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# Poezio is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Poezio. If not, see <http://www.gnu.org/licenses/>.
from common import debug
from gettext import (bindtextdomain, textdomain, bind_textdomain_codeset,
gettext as _)
bindtextdomain('poezio')
textdomain('poezio')
bind_textdomain_codeset('poezio', 'utf-8')
import locale
locale.setlocale(locale.LC_ALL, '')
import sys
import curses
import xmpp
from datetime import datetime
from time import (altzone, daylight, gmtime, localtime, mktime, strftime,
time as time_time, timezone, tzname)
from calendar import timegm
import common
from handler import Handler
from logging import logger
from random import randrange
from config import config
from window import Window
from user import User
from room import Room
class Gui(object):
"""
Graphical user interface using ncurses
"""
def __init__(self, stdscr=None, muc=None):
self.room_number = 0
self.init_curses(stdscr)
self.stdscr = stdscr
self.rooms = [Room('Info', '', self.next_room_number())] # current_room is self.rooms[0]
self.window = Window(stdscr)
self.window.new_room(self.current_room())
self.window.refresh(self.rooms)
self.muc = muc
self.commands = {
'help': (self.command_help, _('OLOL, this is SOOO recursive')),
'join': (self.command_join, _('Usage: /join [room_name][/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 password to join the room.\nExamples:\n/join room@server.tld\n/join room@server.tld/John\n/join /me_again\n/join\n/join room@server.tld/my_nick password\n/join / pass')),
'quit': (self.command_quit, _('Usage: /quit\nQuit: Just disconnect from the server and exit poezio.')),
'exit': (self.command_quit, _('Usage: /exit\nExit: Just disconnect from the server and exit poezio.')),
'next': (self.rotate_rooms_right, _('Usage: /next\nNext: Go to the next room.')),
'n': (self.rotate_rooms_right, _('Usage: /n\nN: Go to the next room.')),
'prev': (self.rotate_rooms_left, _('Usage: /prev\nPrev: Go to the previous room.')),
'p': (self.rotate_rooms_left, _('Usage: /p\nP: Go to the previous room.')),
'win': (self.command_win, _('Usage: /win <number>\nWin: Go to the specified room.')),
'w': (self.command_win, _('Usage: /w <number>\nW: Go to the specified room.')),
'part': (self.command_part, _('Usage: /part [message]\nPart: disconnect from a room. You can specify an optional message.')),
'show': (self.command_show, _(u'Usage: /show <availability> [status]\nShow: Change your availability and (optionaly) your status. The <availability> argument is one of "avail, available, ok, here, chat, away, afk, dnd, busy, xa" and the optional [message] argument will be your status message')),
'away': (self.command_away, _('Usage: /away [message]\nAway: Sets your availability to away and (optional) sets your status message. This is equivalent to "/show away [message]"')),
'busy': (self.command_busy, _('Usage: /busy [message]\nBusy: Sets your availability to busy and (optional) sets your status message. This is equivalent to "/show busy [message]"')),
'avail': (self.command_avail, _('Usage: /avail [message]\nAvail: Sets your availability to available and (optional) sets your status message. This is equivalent to "/show available [message]"')),
'available': (self.command_avail, _('Usage: /available [message]\nAvailable: Sets your availability to available and (optional) sets your status message. This is equivalent to "/show available [message]"')),
'bookmark': (self.command_bookmark, _('Usage: /bookmark [roomname][/nick]\nBookmark: Bookmark the specified room (you will then auto-join it on each poezio start). This commands uses the same syntaxe as /join. Type /help join for syntaxe examples. Note that when typing "/bookmark" on its own, the room will be bookmarked with the nickname you\'re currently using in this room (instead of default_nick)')),
'set': (self.command_set, _('Usage: /set <option> [value]\nSet: Sets the value to the option in your configuration file. You can, for example, change your default nickname by doing `/set default_nick toto` or your resource with `/set resource blabla`. You can also set an empty value (nothing) by providing no [value] after <option>.')),
'kick': (self.command_kick, _('Usage: /kick <nick> [reason]\nKick: Kick the user with the specified nickname. You also can give an optional reason.')),
'topic': (self.command_topic, _('Usage: /topic <subject>\nTopic: Change the subject of the room')),
'nick': (self.command_nick, _('Usage: /nick <nickname>\nNick: Change your nickname in the current room'))
}
self.key_func = {
"KEY_LEFT": self.window.input.key_left,
"KEY_RIGHT": self.window.input.key_right,
"KEY_UP": self.window.input.key_up,
"KEY_END": self.window.input.key_end,
"KEY_HOME": self.window.input.key_home,
"KEY_DOWN": self.window.input.key_down,
"KEY_DC": self.window.input.key_dc,
"KEY_F(5)": self.rotate_rooms_left,
"KEY_F(6)": self.rotate_rooms_right,
"kLFT5": self.rotate_rooms_left,
"kRIT5": self.rotate_rooms_right,
"\t": self.auto_completion,
"KEY_BACKSPACE": self.window.input.key_backspace
}
self.handler = Handler()
self.handler.connect('on-connected', self.on_connected)
self.handler.connect('join-room', self.join_room)
self.handler.connect('room-presence', self.room_presence)
self.handler.connect('room-message', self.room_message)
self.handler.connect('error-message', self.room_error)
self.handler.connect('error', self.information)
def main_loop(self, stdscr):
while 1:
stdscr.leaveok(1)
curses.doupdate()
try:
key = stdscr.getkey()
except:
debug("main_loop exception")
# self.window.resize(stdscr)
# self.window.refresh(self.rooms)
continue
if str(key) in self.key_func.keys():
self.key_func[key]()
elif str(key) == 'KEY_RESIZE':
self.window.resize(stdscr)
self.window.refresh(self.rooms)
elif len(key) >= 4:
continue
elif ord(key) == 10:
self.execute()
elif ord(key) == 8 or ord(key) == 127:
self.window.input.key_backspace()
elif ord(key) < 32:
continue
else:
if ord(key) == 27 and ord(stdscr.getkey()) == 91:
last = ord(stdscr.getkey()) # FIXME: ugly ugly workaround.
if last == 51:
self.window.input.key_dc()
continue
elif ord(key) > 190 and ord(key) < 225:
key = key+stdscr.getkey()
elif ord(key) == 226:
key = key+stdscr.getkey()
key = key+stdscr.getkey()
self.window.do_command(key)
def next_room_number(self):
nb = self.room_number
self.room_number += 1
return nb
def current_room(self):
return self.rooms[0]
def get_room_by_name(self, name):
for room in self.rooms:
if room.name == name:
return room
return None
def init_curses(self, stdscr):
curses.start_color()
curses.noecho()
# curses.cbreak()
# curses.raw()
curses.use_default_colors()
stdscr.keypad(True)
curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_BLUE)
curses.init_pair(2, curses.COLOR_BLUE, -1)
curses.init_pair(3, curses.COLOR_RED, -1) # Admin
curses.init_pair(4, curses.COLOR_BLUE, -1) # Participant
curses.init_pair(5, curses.COLOR_WHITE, -1) # Visitor
curses.init_pair(6, curses.COLOR_CYAN, -1)
curses.init_pair(7, curses.COLOR_GREEN, -1)
curses.init_pair(8, curses.COLOR_MAGENTA, -1)
curses.init_pair(9, curses.COLOR_YELLOW, -1)
curses.init_pair(10, curses.COLOR_WHITE, curses.COLOR_CYAN) # current room
curses.init_pair(11, curses.COLOR_WHITE, curses.COLOR_BLUE) # normal room
curses.init_pair(12, curses.COLOR_WHITE, curses.COLOR_MAGENTA) # new message room
curses.init_pair(13, curses.COLOR_WHITE, curses.COLOR_RED) # highlight room
curses.init_pair(14, curses.COLOR_WHITE, curses.COLOR_YELLOW)
curses.init_pair(15, curses.COLOR_WHITE, curses.COLOR_GREEN)
def reset_curses(self):
curses.echo()
curses.nocbreak()
curses.endwin()
def on_connected(self, jid):
self.information(_("Welcome on Poezio \o/!"))
self.information(_("Your JID is %s") % jid)
def join_room(self, room, nick):
r = Room(room, nick, self.next_room_number())
self.current_room().set_color_state(11)
if self.current_room().nb == 0:
self.rooms.append(r)
else:
for ro in self.rooms:
if ro.nb == 0:
self.rooms.insert(self.rooms.index(ro), r)
break
while self.current_room().nb != r.nb:
self.rooms.insert(0, self.rooms.pop())
self.window.new_room(r)
self.window.refresh(self.rooms)
def auto_completion(self):
self.window.input.auto_completion(self.current_room().users)
def rotate_rooms_right(self, args=None):
self.current_room().set_color_state(11)
self.rooms.append(self.rooms.pop(0))
self.window.refresh(self.rooms)
def rotate_rooms_left(self, args=None):
self.current_room().set_color_state(11)
self.rooms.insert(0, self.rooms.pop())
self.window.refresh(self.rooms)
def room_error(self, room, error, msg):
r = self.get_room_by_name(room)
code = error.getAttr('code')
typ = error.getAttr('type')
body = error.getTag('text').getData()
self.add_info(r, _('Error: %(code)s-%(msg)s: %(body)s' % {'msg':msg, 'code':code, 'body':body}))
if code == '401':
self.add_info(r, _('To provide a password in order to join the room, type "/join / password" (replace "password" by the real password)'))
def room_message(self, stanza, date=None):
delay_tag = stanza.getTag('delay', namespace='urn:xmpp:delay')
if delay_tag and not date:
delayed = True
date = common.datetime_tuple(delay_tag.getAttr('stamp'))
else:
delayed = False
if stanza.getType() != 'groupchat':
return # ignore all messages not comming from a MUC
nick_from = stanza.getFrom().getResource()
room_from = stanza.getFrom().getStripped()
room = self.get_room_by_name(room_from)
if not room:
self.information(_("message received for a non-existing room: %s") % (room_from))
return
body = stanza.getBody()
subject = stanza.getSubject()
if subject:
if nick_from:
self.add_info(room, _("%(nick)s changed the subject to: %(subject)s") % {'nick':nick_from, 'subject':subject}, date)
else:
self.add_info(room, _("The subject is: %(subject)s") % {'subject':subject}, date)
room.topic = subject.encode('utf-8').replace('\n', '|')
if room == self.current_room():
self.window.topic_win.refresh(room.topic)
else:
if body.startswith('/me '):
self.add_info(room, nick_from + ' ' + body[4:], date)
else:
self.add_message(room, nick_from, body, date, delayed)
self.window.input.refresh()
curses.doupdate()
def room_presence(self, stanza):
if len(sys.argv) > 1:
self.information(str(stanza))
from_nick = stanza.getFrom().getResource()
from_room = stanza.getFrom().getStripped()
room = self.get_room_by_name(from_room)
if not room:
return
else:
msg = None
affiliation = stanza.getAffiliation()
show = stanza.getShow()
status = stanza.getStatus()
role = stanza.getRole()
if not room.joined: # user in the room BEFORE us.
room.users.append(User(from_nick, affiliation, show, status, role))
if from_nick.encode('utf-8') == room.own_nick:
room.joined = True
self.add_info(room, _("Your nickname is %s") % (from_nick))
else:
self.add_info(room, _("%s is in the room") % (from_nick.encode('utf-8')))
else:
change_nick = stanza.getStatusCode() == '303'
kick = stanza.getStatusCode() == '307'
user = room.get_user_by_name(from_nick)
# New user
if not user:
room.users.append(User(from_nick, affiliation, show, status, role))
if not config.get('hide_enter_join', "false") == "true":
self.add_info(room, _('%(nick)s joined the room %(roomname)s') % {'nick':from_nick, 'roomname': room.name})
# nick change
elif change_nick:
if user.nick == room.own_nick:
room.own_nick = stanza.getNick().encode('utf-8')
user.change_nick(stanza.getNick())
self.add_info(room, _('%(old_nick)s is now known as %(new_nick)s') % {'old_nick':from_nick, 'new_nick':stanza.getNick()})
# kick
elif kick:
room.users.remove(user)
try:
reason = stanza.getReason().encode('utf-8')
except:
reason = ''
try:
by = stanza.getActor().encode('utf-8')
except:
by = None
if from_nick == room.own_nick: # we are kicked
room.disconnect()
if by:
self.add_info(room, _('You have been kicked by %(by)s. Reason: %(reason)s') % {'by':by, 'reason':reason})
else:
self.add_info(room, _('You have been kicked. Reason: %s') % (reason))
else:
if by:
self.add_info(room, _('%(nick)s has been kicked by %(by)s. Reason: %(reason)s') % {'nick':from_nick, 'by':by, 'reason':reason})
else:
self.add_info(room, _('%(nick)s has been kicked. Reason: %(reason)s') % {'nick':from_nick, 'reason':reason})
# user quit
elif status == 'offline' or role == 'none':
room.users.remove(user)
if not config.get('hide_enter_join', "false") == "true":
self.add_info(room, _('%s has left the room') % (from_nick))
# status change
else:
user.update(affiliation, show, status, role)
if not config.get('hide_status_change', "false") == "true":
self.add_info(room, _('%(nick)s changed his/her status : %(a)s, %(b)s, %(c)s, %(d)s') % {'nick':from_nick, 'a':affiliation, 'b':role, 'c':show, 'd':status})
if room == self.current_room():
self.window.user_win.refresh(room.users)
self.window.input.refresh()
curses.doupdate()
def add_info(self, room, info, date=None):
"""
add a new information in the specified room
(displays it immediately AND saves it for redisplay
in futur refresh)
"""
if not date:
date = datetime.now()
msg = room.add_info(info, date)
self.window.text_win.add_line(room, (date, msg))
if room.name == self.current_room().name:
self.window.text_win.refresh(room.name)
self.window.input.refresh()
curses.doupdate()
def add_message(self, room, nick_from, body, date=None, delayed=False):
if not date:
date = datetime.now()
color = room.add_message(nick_from, body, date)
self.window.text_win.add_line(room, (date, nick_from.encode('utf-8'), body.encode('utf-8'), color))
if room == self.current_room():
self.window.text_win.refresh(room.name)
elif not delayed:
self.window.info_win.refresh(self.rooms, self.current_room())
def execute(self):
line = self.window.input.get_text()
self.window.input.clear_text()
self.window.input.refresh()
if line == "":
return
if line.startswith('/'):
command = line.strip()[:].split()[0][1:]
args = line.strip()[:].split()[1:]
if command in self.commands.keys():
func = self.commands[command][0]
func(args)
return
else:
self.add_info(self.current_room(), _("Error: unknown command (%s)") % (command))
elif self.current_room().name != 'Info':
self.muc.send_message(self.current_room().name, line)
self.window.input.refresh()
curses.doupdate()
def command_help(self, args):
room = self.current_room()
if len(args) == 0:
msg = _('Available commands are:')
for command in self.commands.keys():
msg += "%s " % command
msg += _("\nType /help <command_name> to know what each command does")
if len(args) == 1:
if args[0] in self.commands.keys():
msg = self.commands[args[0]][1]
else:
msg = _('Unknown command: %s') % args[0]
self.add_info(room, msg)
def command_win(self, args):
if len(args) != 1:
self.command_help(['win'])
return
try:
nb = int(args[0])
except ValueError:
self.command_help(['win'])
return
if self.current_room().nb == nb:
return
self.current_room().set_color_state(11)
start = self.current_room()
self.rooms.append(self.rooms.pop(0))
while self.current_room().nb != nb:
self.rooms.append(self.rooms.pop(0))
if self.current_room() == start:
self.window.refresh(self.rooms)
return
self.window.refresh(self.rooms)
def command_kick(self, args):
if len(args) < 1:
self.command_help(['kick'])
return
nick = args[0]
if len(args) >= 2:
reason = ' '.join(args[1:])
else:
reason = ''
if self.current_room().name == 'Info' or not self.current_room().joined:
return
roomname = self.current_room().name
self.muc.eject_user(roomname, 'kick', nick, reason)
def command_join(self, args):
password = None
if len(args) == 0:
r = self.current_room()
if r.name == 'Info':
return
room = r.name
nick = r.own_nick
else:
info = args[0].split('/')
if len(info) == 1:
nick = config.get('default_nick', 'Poezio')
else:
nick = info[1]
if info[0] == '': # happens with /join /nickname, which is OK
r = self.current_room()
if r.name == 'Info':
return
room = r.name
if nick == '':
nick = r.own_nick
else:
room = info[0]
r = self.get_room_by_name(room)
if len(args) == 2: # a password is provided
password = args[1]
if r and r.joined: # if we are already in the room
self.information(_("already in room [%s]") % room)
return
self.muc.join_room(room, nick, password)
if not r: # if the room window exists, we don't recreate it.
self.join_room(room, nick)
else:
r.users = []
def command_bookmark(self, args):
nick = None
if len(args) == 0:
room = self.current_room()
if room.name == 'Info':
return
roomname = room.name
if room.joined:
nick = room.own_nick
else:
info = args[0].split('/')
if len(info) == 2:
nick = info[1]
roomname = info[0]
if roomname == '':
roomname = self.current_room().name
if nick:
res = roomname+'/'+nick
else:
res = roomname
bookmarked = config.get('rooms', '')
# check if the room is already bookmarked.
# if yes, replace it (i.e., update the associated nick)
bookmarked = bookmarked.split(':')
for room in bookmarked:
if room.split('/')[0] == roomname:
bookmarked.remove(room)
break
bookmarked = ':'.join(bookmarked)
config.set_and_save('rooms', bookmarked+':'+res)
def command_set(self, args):
if len(args) != 2 and len(args) != 1:
self.command_help(['set'])
return
option = args[0]
if len(args) == 2:
value = args[1]
else:
value = ''
config.set_and_save(option, value)
msg = "%s=%s" % (option, value)
room = self.current_room()
self.add_info(room, msg)
def command_show(self, args):
possible_show = {'avail':'None',
'available':'None',
'ok':'None',
'here':'None',
'chat':'chat',
'away':'away',
'afk':'away',
'dnd':'dnd',
'busy':'dnd',
'xa':'xa'
}
if len(args) < 1:
return
if not args[0] in possible_show.keys():
self.command_help(['show'])
return
show = possible_show[args[0]]
if len(args) > 1:
msg = ' '.join(args[1:])
else:
msg = None
for room in self.rooms:
if room.joined:
self.muc.change_show(room.name, room.own_nick, show, msg)
def command_away(self, args):
args.insert(0, 'away')
self.command_show(args)
def command_busy(self, args):
args.insert(0, 'busy')
self.command_show(args)
def command_avail(self, args):
args.insert(0, 'available')
self.command_show(args)
def command_part(self, args):
reason = None
room = self.current_room()
if room.name == 'Info':
return
if len(args):
msg = ' '.join(args)
else:
msg = None
if room.joined:
self.muc.quit_room(room.name, room.own_nick, msg)
self.rooms.remove(self.current_room())
self.window.refresh(self.rooms)
def command_topic(self, args):
subject = ' '.join(args)
room = self.current_room()
if not room.joined or room.name == "Info":
return
self.muc.change_subject(room.name, subject)
def command_nick(self, args):
if len(args) != 1:
return
nick = args[0]
room = self.current_room()
if not room.joined or room.name == "Info":
return
self.muc.change_nick(room.name, nick)
def information(self, msg):
room = self.get_room_by_name("Info")
self.add_info(room, msg)
def command_quit(self, args):
self.reset_curses()
sys.exit()