#!/usr/bin/python # -*- coding:utf-8 -*- # # Copyright 2010 Le Coz Florent # # 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 . import curses from logging import logger def get_next_line(str, length): pos = str.rfind(' ', 0, 30) if pos == -1: return str[:length], str[length:] else: return str[:pos], str[pos+1:] def cut_line(str, length): tab = [] while len(str) > length: cut, str = get_next_line(str, length) tab.append(cut) tab.append(str) return tab class Win(object): def __init__(self, height, width, y, x, parent_win): self._resize(height, width, y, x, parent_win) def _resize(self, height, width, y, x, parent_win): self.height, self.width, self.x, self.y = height, width, x, y try: self.win = parent_win.subwin(height, width, y, x) except: pass class UserList(Win): def __init__(self, height, width, y, x, parent_win): Win.__init__(self, height, width, y, x, parent_win) self.win.attron(curses.color_pair(2)) self.win.vline(0, 0, curses.ACS_VLINE, self.height) self.win.attroff(curses.color_pair(2)) self.color_dict = {'moderator': 3, 'participant':2, 'visitor':4} def refresh(self, users): self.win.clear() self.win.attron(curses.color_pair(2)) self.win.vline(0, 0, curses.ACS_VLINE, self.height) self.win.attroff(curses.color_pair(2)) y = 0 for user in users: color = self.color_dict[user.role] self.win.attron(curses.color_pair(color)) self.win.addstr(y, 1, user.nick) self.win.attroff(curses.color_pair(color)) y += 1 self.win.noutrefresh() def resize(self, height, width, y, x, stdscr): self._resize(height, width, y, x, stdscr) class Info(Win): def __init__(self, height, width, y, x, parent_win): Win.__init__(self, height, width, y, x, parent_win) def resize(self, height, width, y, x, stdscr): self._resize(height, width, y, x, stdscr) def refresh(self, room_name): self.win.clear() self.win.addstr(0, 0, room_name + " "*(self.width-len(room_name)-1) , curses.color_pair(1)) self.win.noutrefresh() class TextWin(Win): def __init__(self, height, width, y, x, parent_win): Win.__init__(self, height, width, y, x, parent_win) self.win.idlok(True) self.pos = 0 def refresh(self, lines, users): self.win.clear() y = 0 # logger.info(str(self.height)) # logger.info(str(lines[-self.height/2:])) for line in lines[-self.height:]: # FIXME if len(line) == 2: try: self.win.addstr(y, 0, '['+line[0].strftime("%H:%M:%S") + "] *" + line[1]+"*") except: logger.error(str(line)) raise y += 1 elif len(line) == 3: for user in users: if user.nick == line[1]: break self.win.addstr(y, 0, '['+line[0].strftime("%H:%M:%S") + "] <") length = len('['+line[0].strftime("%H:%M:%S") + "] <") self.win.attron(curses.color_pair(user.color)) self.win.addstr(y, length,line[1]) self.win.attroff(curses.color_pair(user.color)) self.win.addstr(y, length+len(line[1]), ">: ") remaining_width = self.width-(length+len(line[1])+3) tab = cut_line(line[2], remaining_width-1) for l in tab: try: self.win.addstr(y, length+len(line[1])+3, l) y += 1 except:pass elif len(line) == 1: self.win.addstr(y, 0, line[0]) y += 1 self.win.noutrefresh() def resize(self, height, width, y, x, stdscr): self._resize(height, width, y, x, stdscr) class Input(Win): """ """ def __init__(self, height, width, y, x, stdscr): Win.__init__(self, height, width, y, x, stdscr) self.input = curses.textpad.Textbox(self.win) self.input.insert_mode = True def resize(self, height, width, y, x, stdscr): self._resize(height, width, y, x, stdscr) self.input = curses.textpad.Textbox(self.win) self.input.insert_mode = True self.win.clear() def do_command(self, key): self.input.do_command(key) def get_text(self): return self.input.gather() def save_text(self): self.txt = self.input.gather() def refresh(self): self.win.noutrefresh() def clear_text(self): self.win.clear() class Window(object): """ The whole "screen" that can be seen at once in the terminal. It contains an userlist, an input zone and a chat zone """ def __init__(self, stdscr): """ name is the name of the Tab, and it's also the JID of the chatroom. A particular tab is the "Info" tab which has no name (None). This info tab should be unique. The stdscr should be passed to know the size of the terminal """ self.size = (self.height, self.width) = stdscr.getmaxyx() self.user_win = UserList(self.height-3, self.width/7, 1, 6*(self.width/7), stdscr) self.topic_win = Info(1, self.width, 0, 0, stdscr) self.info_win = Info(1, self.width, self.height-2, 0, stdscr) self.text_win = TextWin(self.height-3, (self.width/7)*6, 1, 0, stdscr) self.input = Input(1, self.width, self.height-1, 0, stdscr) # debug def resize(self, stdscr): """ Resize the whole tabe. i.e. all its sub-windows """ self.size = (self.height, self.width) = stdscr.getmaxyx() self.user_win.resize(self.height-3, self.width/7, 1, 6*(self.width/7), stdscr) self.topic_win.resize(1, self.width, 0, 0, stdscr) self.info_win.resize(1, self.width, self.height-2, 0, stdscr) self.text_win.resize(self.height-3, (self.width/7)*6, 1, 0, stdscr) self.input.resize(1, self.width, self.height-1, 0, stdscr) def refresh(self, room): self.text_win.refresh(room.lines, room.users) self.user_win.refresh(room.users) self.topic_win.refresh(room.topic) self.info_win.refresh(room.name) self.input.refresh() def do_command(self, key): self.input.do_command(key) self.input.refresh()