#!/usr/bin/env python ## plugins/gtkgui.py ## ## Gajim Team: ## - Yann Le Boulanger ## - Vincent Hanquez ## - David Ferlier ## ## Copyright (C) 2003 Gajim Team ## ## This program 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 2 only. ## ## This program 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. ## import pygtk pygtk.require('2.0') import gtk from gtk import TRUE, FALSE import gtk.glade,gobject import os,string,time,Queue import common.optparser,common.sleepy CONFPATH = "~/.gajim/config" Wbrowser = 0 Waccounts = 0 class user: """Informations concerning each users""" def __init__(self, *args): if len(args) == 0: self.jid = '' self.name = '' self.groups = [] self.show = '' self.status = '' self.sub == '' self.resource == '' elif len(args) == 7: self.jid = args[0] self.name = args[1] self.groups = args[2] self.show = args[3] self.status = args[4] self.sub = args[5] self.resource = args[6] # elif ((len(args)) and (type (args[0]) == type (self)) and # (self.__class__ == args[0].__class__)): # self.name = args[0].name # self.groups = args[0].groups # self.show = args[0].show # self.status = args[0].status # self.sub = args[0].sub else: raise TypeError, 'bad arguments' class info_user: """Class for user's information window""" def delete_event(self, widget): """close window""" self.window.destroy() def add_grp_to_user(self, model, path, iter): """Insert user to the group in inter""" self.user.groups.append(model.get_value(iter, 0)) def on_close(self, widget): """Save user's informations and update the roster on the Jabber server""" for i in self.r.l_contact[self.user.jid]['iter']: self.r.tree.get_model().remove(i) self.r.l_contact[self.user.jid]['iter'] = [] self.user.groups = [] model = self.list2.get_model() model.foreach(self.add_grp_to_user) self.r.add_user(self.user) self.r.queueOUT.put(('UPDUSER', (self.user.jid, self.user.name, \ self.user.groups))) self.delete_event(self) def add_grp(self, model, path, iter, stors): """Transfert the iter from stors[0] to stors[1]""" i = stors[1].append() stors[1].set(i, 0, stors[0].get_value(iter, 0)) stors[0].remove(iter) def on_add(self, widget): """When Add button is clicked""" model = self.list1.get_model() select = self.list1.get_selection() select.selected_foreach(self.add_grp, (model, self.list2.get_model())) def on_remove(self, widget): """When Remove button is clicked""" model = self.list2.get_model() select = self.list2.get_selection() select.selected_foreach(self.add_grp, (model, self.list1.get_model())) def on_new_key_pressed(self, widget, event): """If enter is pressed in new group entry, add the group""" if event.keyval == gtk.keysyms.Return: model = self.list1.get_model() txt = self.entry_new.get_text() iter = model.append() model.set(iter, 0, txt) self.entry_new.set_text('') return 1 else: return 0 def init_lists(self): """Initialize both available and current listStores""" #list available store = gtk.ListStore(gobject.TYPE_STRING) for i in self.r.l_group.keys(): if i != 'Agents' and i not in self.user.groups: iter = store.append() store.set(iter, 0, i) self.list1.set_model(store) column = gtk.TreeViewColumn('Available', gtk.CellRendererText(), text=0) self.list1.append_column(column) #list_current store = gtk.ListStore(gobject.TYPE_STRING) for i in self.user.groups: iter = store.append() store.set(iter, 0, i) self.list2.set_model(store) column = gtk.TreeViewColumn('Available', gtk.CellRendererText(), text=0) self.list2.append_column(column) def __init__(self, user, roster): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Info_user') self.window = xml.get_widget("Info_user") self.r = roster self.user = user self.list1 = xml.get_widget("treeview_available") self.list2 = xml.get_widget("treeview_current") self.entry_new = xml.get_widget("entry_new") xml.get_widget('label_name').set_text(user.name) xml.get_widget('label_id').set_text(user.jid) xml.get_widget('label_resource').set_text(user.resource) xml.get_widget('entry_name').set_text(user.name) if not user.status: user.status = '' xml.get_widget('label_status').set_text(user.show + ' : ' + \ user.status) self.init_lists() xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_close_clicked', self.on_close) xml.signal_connect('on_add_clicked', self.on_add) xml.signal_connect('on_remove_clicked', self.on_remove) xml.signal_connect('on_entry_new_key_press_event', \ self.on_new_key_pressed) class prefs: """Class for Preferences window""" def delete_event(self, widget): """close window""" self.window.destroy() def on_color_button_clicked(self, widget): """Open a ColorSelectionDialog and change button's color""" if widget.name == 'colorIn': color = self.colorIn da = self.da_in elif widget.name == 'colorOut': color = self.colorOut da = self.da_out elif widget.name == 'colorStatus': color = self.colorStatus da = self.da_status colorseldlg = gtk.ColorSelectionDialog('Select Color') colorsel = colorseldlg.colorsel colorsel.set_previous_color(color) colorsel.set_current_color(color) colorsel.set_has_palette(gtk.TRUE) response = colorseldlg.run() if response == gtk.RESPONSE_OK: color = colorsel.get_current_color() da.modify_bg(gtk.STATE_NORMAL, color) if widget.name == 'colorIn': self.colorIn = color elif widget.name == 'colorOut': self.colorOut = color elif widget.name == 'colorStatus': self.colorStatus = color colorseldlg.destroy() def write_cfg(self): """Save preferences in config File and apply them""" #Color for incomming messages colSt = '#'+(hex(self.colorIn.red)+'0')[2:4]\ +(hex(self.colorIn.green)+'0')[2:4]\ +(hex(self.colorIn.blue)+'0')[2:4] self.r.cfgParser.set('GtkGui', 'inmsgcolor', colSt) for j in self.r.tab_messages.keys(): self.r.tab_messages[j].tagIn.set_property("foreground", colSt) #Color for outgoing messages colSt = '#'+(hex(self.colorOut.red)+'0')[2:4]\ +(hex(self.colorOut.green)+'0')[2:4]\ +(hex(self.colorOut.blue)+'0')[2:4] self.r.cfgParser.set('GtkGui', 'outmsgcolor', colSt) for j in self.r.tab_messages.keys(): self.r.tab_messages[j].tagOut.set_property("foreground", colSt) #Color for status messages colSt = '#'+(hex(self.colorStatus.red)+'0')[2:4]\ +(hex(self.colorStatus.green)+'0')[2:4]\ +(hex(self.colorStatus.blue)+'0')[2:4] self.r.cfgParser.set('GtkGui', 'statusmsgcolor', colSt) for j in self.r.tab_messages.keys(): self.r.tab_messages[j].tagStatus.set_property("foreground", colSt) #IconStyle ist = self.combo_iconstyle.entry.get_text() self.r.cfgParser.set('GtkGui', 'iconstyle', ist) self.r.iconstyle = ist self.r.mkpixbufs() #autopopup pp = self.chk_autopp.get_active() if pp == True: self.r.cfgParser.set('GtkGui', 'autopopup', '1') self.r.autopopup = 1 else: self.r.cfgParser.set('GtkGui', 'autopopup', '0') self.r.autopopup = 0 #autoaway aw = self.chk_autoaway.get_active() if aw == True: self.r.cfgParser.set('GtkGui', 'autoaway', '1') self.r.plugin.autoaway = 1 else: self.r.cfgParser.set('GtkGui', 'autoaway', '0') self.r.plugin.autoaway = 0 aat = self.spin_autoawaytime.get_value_as_int() self.r.plugin.autoawaytime = aat self.r.cfgParser.set('GtkGui', 'autoawaytime', aat) #autoxa xa = self.chk_autoxa.get_active() if xa == True: self.r.cfgParser.set('GtkGui', 'autoxa', '1') self.r.plugin.autoxa = 1 else: self.r.cfgParser.set('GtkGui', 'autoxa', '0') self.r.plugin.autoxa = 0 axt = self.spin_autoxatime.get_value_as_int() self.r.plugin.autoxatime = axt self.r.cfgParser.set('GtkGui', 'autoxatime', axt) if self.r.plugin.sleeper: self.r.plugin.sleeper = common.sleepy.Sleepy(\ self.r.plugin.autoawaytime*60, self.r.plugin.autoxatime*60) self.r.cfgParser.writeCfgFile() self.r.cfgParser.parseCfgFile() self.r.redraw_roster() def on_ok(self, widget): """When Ok button is clicked""" self.write_cfg() self.window.destroy() def __init__(self, roster): """Initialize Preference window""" xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Prefs') self.window = xml.get_widget("Prefs") self.r = roster self.da_in = xml.get_widget("drawing_in") self.da_out = xml.get_widget("drawing_out") self.da_status = xml.get_widget("drawing_status") self.combo_iconstyle = xml.get_widget("combo_iconstyle") self.chk_autopp = xml.get_widget("chk_autopopup") self.chk_autoaway = xml.get_widget("chk_autoaway") self.spin_autoawaytime = xml.get_widget("spin_autoawaytime") self.chk_autoxa = xml.get_widget("chk_autoxa") self.spin_autoxatime = xml.get_widget("spin_autoxatime") #Color for incomming messages colSt = self.r.cfgParser.GtkGui_inmsgcolor if not colSt: colSt = '#ff0000' cmapIn = self.da_in.get_colormap() self.colorIn = cmapIn.alloc_color(colSt) self.da_in.window.set_background(self.colorIn) #Color for outgoing messages colSt = self.r.cfgParser.GtkGui_outmsgcolor if not colSt: colSt = '#0000ff' cmapOut = self.da_out.get_colormap() self.colorOut = cmapOut.alloc_color(colSt) self.da_out.window.set_background(self.colorOut) #Color for status messages colSt = self.r.cfgParser.GtkGui_statusmsgcolor if not colSt: colSt = '#00ff00' cmapStatus = self.da_status.get_colormap() self.colorStatus = cmapStatus.alloc_color(colSt) self.da_status.window.set_background(self.colorStatus) #iconStyle list_style = os.listdir('plugins/gtkgui/icons/') l = [] for i in list_style: if i != 'CVS': l.append(i) if l.count == 0: l.append(" ") self.combo_iconstyle.set_popdown_strings(l) if self.r.iconstyle in l: self.combo_iconstyle.entry.set_text(self.r.iconstyle) #Autopopup st = self.r.cfgParser.GtkGui_autopopup if not st: st = '0' pp = string.atoi(st) self.chk_autopp.set_active(pp) #Autoaway st = self.r.cfgParser.GtkGui_autoaway if not st: st = '1' aw = string.atoi(st) self.chk_autoaway.set_active(aw) st = self.r.cfgParser.GtkGui_autoawaytime if not st: st = '10' ti = string.atoi(st) self.spin_autoawaytime.set_value(ti) #Autoxa st = self.r.cfgParser.GtkGui_autoxa if not st: st = '1' xa = string.atoi(st) self.chk_autoxa.set_active(xa) st = self.r.cfgParser.GtkGui_autoxatime if not st: st = '20' ti = string.atoi(st) self.spin_autoxatime.set_value(ti) xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_but_col_clicked', \ self.on_color_button_clicked) xml.signal_connect('on_ok_clicked', self.on_ok) class away_msg: """Class for Away Message Window""" def delete_event(self, widget): """close window""" self.window.destroy() def on_ok(self): """When Ok button is clicked""" beg, end = self.txtBuffer.get_bounds() self.msg = self.txtBuffer.get_text(beg, end, 0) self.window.destroy() def run(self): """Wait for Ok button to be pressed and return away messsage""" rep = self.window.run() if rep == gtk.RESPONSE_OK: beg, end = self.txtBuffer.get_bounds() msg = self.txtBuffer.get_text(beg, end, 0) self.window.destroy() if not msg: msg = "" return msg def __init__(self): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Away_msg') self.window = xml.get_widget("Away_msg") txt = xml.get_widget("textview") self.txtBuffer = txt.get_buffer() xml.signal_connect('gtk_widget_destroy', self.delete_event) class add: """Class for Add user window""" def delete_event(self, widget): """close window""" self.Wadd.destroy() def on_subscribe(self, widget): """When Subscribe button is clicked""" who = self.xml.get_widget("entry_who").get_text() buf = self.xml.get_widget("textview_sub").get_buffer() start_iter = buf.get_start_iter() end_iter = buf.get_end_iter() txt = buf.get_text(start_iter, end_iter, 0) self.r.req_sub(self, who, txt) self.delete_event(self) def __init__(self, roster, jid=None): self.r = roster xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Add') if jid: xml.get_widget('entry_who').set_text(jid) self.Wadd = xml.get_widget("Add") xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_button_sub_clicked', self.on_subscribe) class warning: """Class for warning window : print a warning message""" def delete_event(self, widget): """close window""" self.window.destroy() def __init__(self, txt): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Warning') self.window = xml.get_widget("Warning") xml.get_widget('label').set_text(txt) xml.signal_connect('gtk_widget_destroy', self.delete_event) class about: """Class for about window""" def delete_event(self, widget): """close window""" self.Wabout.destroy() def __init__(self): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'About') self.Wabout = xml.get_widget("About") xml.signal_connect('gtk_widget_destroy', self.delete_event) class account_pref: """Class for account informations""" def delete_event(self, widget): """close window""" self.window.destroy() def init_account(self, infos): """Initialize window with defaults values""" if infos.has_key('name'): self.entryName.set_text(infos['name']) if infos.has_key('jid'): self.entryJid.set_text(infos['jid']) if infos.has_key('password'): self.entryPass.set_text(infos['password']) if infos.has_key('ressource'): self.entryRessource.set_text(infos['ressource']) def on_save_clicked(self, widget): """When save button is clicked : Save informations in config file""" name = self.entryName.get_text() jid = self.entryJid.get_text() if (name == ''): warning('You must enter a name for this account') return 0 if (jid == '') or (string.count(jid, '@') != 1): warning('You must enter a Jabber ID for this account\nFor example : login@hostname') return 0 else: (login, hostname) = string.split(jid, '@') #if we are modifying an account if self.mod: #if we modify the name of the account if name != self.acc: self.accs.r.cfgParser.remove_section(self.acc) self.accs.r.accounts.remove(self.acc) self.accs.r.cfgParser.add_section(name) self.accs.r.accounts.append(name) accountsStr = string.join(self.accs.accounts) self.accs.r.cfgParser.set('Profile', 'accounts', accountsStr) #if it's a new account else: if name in self.accs.r.accounts: warning('An account already has this name') return 0 #if we neeed to register a new account if self.check.get_active(): self.accs.r.queueOUT.put(('NEW_ACC', (hostname, login, \ self.entryPass.get_text(), name, \ self.entryRessource.get_text()))) self.check.set_active(FALSE) return 1 self.accs.r.cfgParser.add_section(name) self.accs.r.accounts.append(name) accountsStr = string.join(self.accs.accounts) self.accs.r.cfgParser.set('Profile', 'accounts', accountsStr) self.accs.r.cfgParser.set(name, 'name', login) self.accs.r.cfgParser.set(name, 'hostname', hostname) self.accs.r.cfgParser.set(name, 'password', self.entryPass.get_text()) self.accs.r.cfgParser.set(name, 'ressource', self.entryRessource.get_text()) self.accs.r.cfgParser.writeCfgFile() self.accs.r.cfgParser.parseCfgFile() self.accs.init_accounts() self.delete_event(self) #info must be a dictionnary def __init__(self, accs, infos = {}): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Account') self.window = xml.get_widget("Account") self.entryPass = xml.get_widget("entry_password") self.entryRessource = xml.get_widget("entry_ressource") self.check = xml.get_widget("checkbutton") self.entryName = xml.get_widget("entry_name") self.entryJid = xml.get_widget("entry_jid") self.accs = accs if infos: self.mod = TRUE self.acc = infos['name'] self.init_account(infos) else: self.mod = FALSE if self.mod: self.check.set_sensitive(FALSE) xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_save_clicked', self.on_save_clicked) class accounts: """Class for accounts window : lists of accounts""" def delete_event(self, widget): """close window""" global Waccounts Waccounts = 0 self.window.destroy() def init_accounts(self): """initialize listStore with existing accounts""" model = self.treeview.get_model() model.clear() for account in self.r.accounts: iter = model.append() model.set(iter, 0, account, 1, self.r.cfgParser.__getattr__("%s" % account+"_hostname")) def on_row_activated(self, widget): """Activate delete and modify buttons when a row is selected""" self.modButt.set_sensitive(TRUE) self.delButt.set_sensitive(TRUE) def on_new_clicked(self, widget): """When new button is clicked : open an account information window""" account_pref(self) def on_delete_clicked(self, widget): """When delete button is clicked : Remove an account from the listStore and from the config file""" sel = self.treeview.get_selection() (mod, iter) = sel.get_selected() model = self.treeview.get_model() account = model.get_value(iter, 0) self.r.cfgParser.remove_section(account) self.r.accounts.remove(account) accountsStr = string.join(self.r.accounts) self.r.cfgParser.set('Profile', 'accounts', accountsStr) self.r.cfgParser.writeCfgFile() self.r.cfgParser.parseCfgFile() self.init_accounts() def on_modify_clicked(self, widget): """When modify button is clicked : open the account information window for this account""" infos = {} sel = self.treeview.get_selection() model = self.treeview.get_model() (mod, iter) = sel.get_selected() account = model.get_value(iter, 0) infos['name'] = account infos['jid'] = self.r.cfgParser.__getattr__("%s" % account+"_name") + \ '@' + self.r.cfgParser.__getattr__("%s" % account+"_hostname") infos['password'] = self.r.cfgParser.__getattr__("%s" % account+"_password") infos['ressource'] = self.r.cfgParser.__getattr__("%s" % account+"_ressource") account_pref(self, infos) def __init__(self, roster): self.r = roster xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Accounts') self.window = xml.get_widget("Accounts") self.treeview = xml.get_widget("treeview") self.modButt = xml.get_widget("modify_button") self.delButt = xml.get_widget("delete_button") model = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING) self.treeview.set_model(model) #columns renderer = gtk.CellRendererText() renderer.set_data('column', 0) self.treeview.insert_column_with_attributes(-1, 'Name', renderer, text=0) renderer = gtk.CellRendererText() renderer.set_data('column', 1) self.treeview.insert_column_with_attributes(-1, 'Server', renderer, text=1) xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_row_activated', self.on_row_activated) xml.signal_connect('on_new_clicked', self.on_new_clicked) xml.signal_connect('on_delete_clicked', self.on_delete_clicked) xml.signal_connect('on_modify_clicked', self.on_modify_clicked) self.init_accounts() class confirm: def delete_event(self, widget): """close window""" self.window.destroy() def req_usub(self, widget): model = self.r.tree.get_model() jid = model.get_value(self.iter, 2) self.r.queueOUT.put(('UNSUB', jid)) del self.r.l_contact[jid] model.remove(self.iter) self.delete_event(self) def __init__(self, roster, iter): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Confirm') self.window = xml.get_widget('Confirm') self.r = roster self.iter = iter jid = self.r.tree.get_model().get_value(iter, 2) xml.get_widget('label_confirm').set_text('Are you sure you want to remove ' + jid + ' from your roster ?') xml.signal_connect('on_okbutton_clicked', self.req_usub) xml.signal_connect('gtk_widget_destroy', self.delete_event) class authorize: def delete_event(self, widget): """close window""" self.window.destroy() def auth(self, widget): self.r.queueOUT.put(('AUTH', self.jid)) self.delete_event(self) if not self.r.l_contact.has_key(self.jid): add(self.r, self.jid) def deny(self, widget): self.r.queueOUT.put(('DENY', self.jid)) self.delete_event(self) def __init__(self, roster, jid): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Sub_req') self.window = xml.get_widget('Sub_req') self.r = roster self.jid = jid xml.get_widget('label').set_text('Subscription request from ' + self.jid) xml.signal_connect('on_button_auth_clicked', self.auth) xml.signal_connect('on_button_deny_clicked', self.deny) xml.signal_connect('on_button_close_clicked', self.delete_event) class agent_reg: def delete_event(self, widget): """close window""" self.window.destroy() def draw_table(self): nbrow = 0 for name in self.infos.keys(): if name != 'key' and name != 'instructions' and name != 'x': nbrow = nbrow + 1 self.table.resize(rows=nbrow, columns=2) label = gtk.Label(name) self.table.attach(label, 0, 1, nbrow-1, nbrow, 0, 0, 0, 0) entry = gtk.Entry() entry.set_text(self.infos[name]) self.table.attach(entry, 1, 2, nbrow-1, nbrow, 0, 0, 0, 0) self.entries[name] = entry if nbrow == 1: entry.grab_focus() self.table.show_all() def on_ok(self, widget): for name in self.entries.keys(): self.infos[name] = self.entries[name].get_text() self.r.queueOUT.put(('REG_AGENT', self.agent)) self.delete_event(self) def __init__(self, agent, infos, roster): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'agent_reg') self.agent = agent self.infos = infos self.r = roster self.window = xml.get_widget('agent_reg') self.table = xml.get_widget('table') self.window.set_title('Register to ' + agent) xml.get_widget('label').set_text(infos['instructions']) self.entries = {} self.draw_table() xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_button_cancel_clicked', self.delete_event) xml.signal_connect('on_button_ok_clicked', self.on_ok) class browser: def delete_event(self, widget): """close window""" global Wbrowser Wbrowser = 0 self.window.destroy() def browse(self): self.r.queueOUT.put(('REQ_AGENTS', None)) def agents(self, agents): model = self.treeview.get_model() for jid in agents.keys(): iter = model.append() model.set(iter, 0, agents[jid]['name'], 1, agents[jid]['service']) def on_refresh(self, widget): self.treeview.get_model().clear() self.browse() def on_row_activated(self, widget, path, col=0): model = self.treeview.get_model() iter = model.get_iter(path) service = model.get_value(iter, 1) self.r.queueOUT.put(('REQ_AGENT_INFO', service)) self.delete_event(self) def __init__(self, roster): xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'browser') self.window = xml.get_widget('browser') self.treeview = xml.get_widget('treeview') self.r = roster model = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING) self.treeview.set_model(model) #columns renderer = gtk.CellRendererText() renderer.set_data('column', 0) self.treeview.insert_column_with_attributes(-1, 'Name', renderer, text=0) renderer = gtk.CellRendererText() renderer.set_data('column', 1) self.treeview.insert_column_with_attributes(-1, 'Service', renderer, text=1) xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_refresh_clicked', self.on_refresh) xml.signal_connect('on_row_activated', self.on_row_activated) #TODO: Si connecte self.browse() class message: def delete_event(self, widget): """close window""" del self.r.tab_messages[self.user.jid] self.window.destroy() def print_conversation(self, txt, contact = None, tim = None): buffer = self.conversation.get_buffer() if not txt: txt = "" end_iter = buffer.get_end_iter() if not tim: tim = time.strftime("[%H:%M:%S]") buffer.insert(end_iter, tim) if contact: if contact == 'status': buffer.insert_with_tags_by_name(end_iter, txt+'\n', \ 'status') else: buffer.insert_with_tags_by_name(end_iter, ' ', 'outgoing') buffer.insert(end_iter, txt+'\n') else: buffer.insert_with_tags_by_name(end_iter, '<' + self.user.name + '> ', 'incoming') buffer.insert(end_iter, txt+'\n') self.conversation.scroll_to_mark(\ buffer.get_mark('end'), 0.1, 0, 0, 0) def read_queue(self, q): while not q.empty(): evt = q.get() self.print_conversation(evt[0], tim = evt[1]) del self.r.tab_queues[self.user.jid] for i in self.r.l_contact[self.user.jid]['iter']: if self.r.pixbufs.has_key(self.user.show): self.r.tree.get_model().set_value(i, 0, self.r.pixbufs[self.user.show]) def on_msg_key_press_event(self, widget, event): if event.keyval == gtk.keysyms.Return: if (event.state & gtk.gdk.SHIFT_MASK): return 0 txt_buffer = widget.get_buffer() start_iter = txt_buffer.get_start_iter() end_iter = txt_buffer.get_end_iter() txt = txt_buffer.get_text(start_iter, end_iter, 0) if txt != '': self.r.queueOUT.put(('MSG',(self.user.jid, txt))) txt_buffer.set_text('', -1) self.print_conversation(txt, self.user.jid) widget.grab_focus() return 1 return 0 def on_clear(self, widget): buffer = self.conversation.get_buffer() deb, end = buffer.get_bounds() buffer.delete(deb, end) def __init__(self, user, roster): self.user = user self.r = roster xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Chat') self.window = xml.get_widget('Chat') self.window.set_title('Chat with ' + user.name) self.img = xml.get_widget('image') self.img.set_from_pixbuf(self.r.pixbufs[user.show]) xml.get_widget('label_contact').set_text(user.name + ' <'\ + user.jid + '>') self.message = xml.get_widget('message') self.message.grab_focus() self.conversation = xml.get_widget('conversation') buffer = self.conversation.get_buffer() end_iter = buffer.get_end_iter() buffer.create_mark('end', end_iter, 0) # self.window.show() xml.signal_connect('gtk_widget_destroy', self.delete_event) xml.signal_connect('on_clear_button_clicked', self.on_clear) xml.signal_connect('on_msg_key_press_event', self.on_msg_key_press_event) self.tagIn = buffer.create_tag("incoming") color = self.r.cfgParser.GtkGui_inmsgcolor if not color: color = '#ff0000' #red self.tagIn.set_property("foreground", color) self.tagOut = buffer.create_tag("outgoing") color = self.r.cfgParser.GtkGui_outmsgcolor if not color: color = '#0000ff' #blue self.tagOut.set_property("foreground", color) self.tagStatus = buffer.create_tag("status") color = self.r.cfgParser.GtkGui_statusmsgcolor if not color: color = 'green' self.tagStatus.set_property("foreground", color) class roster: def get_icon_pixbuf(self, stock): return self.tree.render_icon(stock, size = gtk.ICON_SIZE_MENU, detail = None) def add_user(self, u): """ add a ligne to the roster """ newgrp = 0 self.l_contact[u.jid] = {'user': u, 'iter': []} if u.groups == []: if string.find(u.jid, "@") <= 0: u.groups.append('Agents') else: u.groups.append('general') if u.show != 'offline' or self.showOffline or 'Agents' in u.groups: model = self.tree.get_model() for g in u.groups: if not self.l_group.has_key(g): self.l_group[g] = {'iter':None, 'hide':False} if not self.l_group[g]['iter']: iterG = model.append(None, (self.pixbufs['closed'], g, 'group', FALSE, self.grpbgcolor, TRUE)) self.l_group[g] = {'iter':iterG, 'hide':False} newgrp = 1 if g == 'Agents': iterU = model.append(self.l_group[g]['iter'], \ (self.pixbufs[u.show], u.name, 'agent', FALSE, \ self.userbgcolor, TRUE)) else: iterU = model.append(self.l_group[g]['iter'], \ (self.pixbufs[u.show], u.name, u.jid, TRUE, \ self.userbgcolor, TRUE)) self.l_contact[u.jid]['iter'].append(iterU) if newgrp == 1: #expand new groups self.tree.expand_row(model.get_path(iterG), FALSE) def redraw_roster(self): for j in self.l_contact.keys(): self.l_contact[j]['iter'] = [] for g in self.l_group.keys(): self.l_group[g]['iter'] = None self.draw_roster() def draw_roster(self): self.tree.get_model().clear() for j in self.l_contact.keys(): self.add_user(self.l_contact[j]['user']) def mklists(self, tab): for jid in tab.keys(): #remove ressource from jid string ji = string.split(jid, '/')[0] name = tab[jid]['name'] if not name: if string.find(ji, "@") <= 0: name = ji else: name = string.split(jid, '@')[0] show = tab[jid]['show'] if not show: show = 'offline' user1 = user(ji, name, tab[jid]['groups'], show, tab[jid]['status'], tab[jid]['sub'], '') self.l_contact[ji] = {'user':user1, 'iter':[]} for i in tab[jid]['groups'] : if not i in self.l_group.keys(): self.l_group[i] = {'iter':None, 'hide':False} def update_iter(self, widget, path, iter, data): model = self.tree.get_model() jid = model.get_value(iter, 2) if jid == data[0]: if data[1] == 'offline': model.remove(iter) if not self.showOffline: self.found = 1 else: model.set_value(iter, 0, self.pixbufs[data[1]]) self.found = 1 return 1 return 0 def chg_status(self, jid, show, status): u = self.l_contact[jid]['user'] if self.l_contact[jid]['iter'] == []: self.add_user(u) else: model = self.tree.get_model() if show == 'offline' and not self.showOffline: for i in self.l_contact[jid]['iter']: parent_i = model.iter_parent(i) if model.iter_n_children(parent_i) == 1: model.remove(i) grp = model.get_value(parent_i, 1) model.remove(parent_i) self.l_group[grp]['iter'] = None else: model.remove(i) self.l_contact[jid]['iter'] = [] else: for i in self.l_contact[jid]['iter']: if self.pixbufs.has_key(show): model.set_value(i, 0, self.pixbufs[show]) #update icon in chat window if self.tab_messages.has_key(jid): self.tab_messages[jid].img.set_from_pixbuf(self.pixbufs[show]) u.show = show u.status = status def on_info(self, widget, jid): info_user(self.l_contact[jid]['user'], self) def mk_menu_c(self, event, iter): model = self.tree.get_model() jid = model.get_value(iter, 2) path = model.get_path(iter) self.menu_c = gtk.Menu() item = gtk.MenuItem("Start chat") self.menu_c.append(item) item.connect("activate", self.on_row_activated, path) item = gtk.MenuItem("Rename") self.menu_c.append(item) # item.connect("activate", self.on_rename, iter) item = gtk.MenuItem() self.menu_c.append(item) item = gtk.MenuItem("Subscription") self.menu_c.append(item) menu_sub = gtk.Menu() item.set_submenu(menu_sub) item = gtk.MenuItem("Resend authorization to") menu_sub.append(item) item.connect("activate", self.authorize, jid) item = gtk.MenuItem("Rerequest authorization from") menu_sub.append(item) item.connect("activate", self.req_sub, jid, 'I would like to add you to my contact list, please.') item = gtk.MenuItem() self.menu_c.append(item) item = gtk.MenuItem("Remove") self.menu_c.append(item) item.connect("activate", self.on_req_usub, iter) item = gtk.MenuItem() self.menu_c.append(item) item = gtk.MenuItem("Informations") self.menu_c.append(item) item.connect("activate", self.on_info, jid) self.menu_c.popup(None, None, None, event.button, event.time) self.menu_c.show_all() def mk_menu_g(self, event): self.menu_c = gtk.Menu() item = gtk.MenuItem("grp1") self.menu_c.append(item) item = gtk.MenuItem("grp2") self.menu_c.append(item) item = gtk.MenuItem("grp3") self.menu_c.append(item) self.menu_c.popup(None, None, None, event.button, event.time) self.menu_c.show_all() def authorize(self, widget, jid): self.queueOUT.put(('AUTH', jid)) # def rename(self, widget, jid, name): # u = self.l_contact[jid]['user'] # u.name = name # for i in self.l_contact[jid]['iter']: # self.tree.get_model().set_value(i, 1, name) def req_sub(self, widget, jid, txt): self.queueOUT.put(('SUB', (jid, txt))) if not self.l_contact.has_key(jid): user1 = user(jid, jid, ['general'], 'requested', 'requested', 'sub', '') self.add_user(user1) def init_tree(self): self.tree.get_model().clear() #l_contact = {jid:{'user':_, 'iter':[iter1, ...]] self.l_contact = {} #l_group = {name:{'iter':_, 'hide':Bool} self.l_group = {} def on_treeview_event(self, widget, event): if (event.button == 3) & (event.type == gtk.gdk.BUTTON_PRESS): try: path, column, x, y = self.tree.get_path_at_pos(int(event.x), int(event.y)) except TypeError: return model = self.tree.get_model() iter = model.get_iter(path) data = model.get_value(iter, 2) if data == 'group': self.mk_menu_g(event) elif data == 'agent': #TODO pass else: self.mk_menu_c(event, iter) return gtk.TRUE return gtk.FALSE def on_req_usub(self, widget, iter): window_confirm = confirm(self, iter) def on_status_changed(self, widget): accountsStr = self.cfgParser.Profile_accounts accounts = string.split(accountsStr, ' ') if widget.name != 'online' and widget.name != 'offline': w = away_msg() txt = w.run() else: txt = widget.name self.queueOUT.put(('STATUS',(widget.name, txt, accounts[0]))) def on_prefs(self, widget): window = prefs(self) def on_add(self, widget): window_add = add(self) def on_about(self, widget): window_about = about() def on_accounts(self, widget): global Waccounts if not Waccounts: Waccounts = accounts(self) def on_quit(self, widget): self.queueOUT.put(('QUIT','')) gtk.mainquit() # sys.exit(0) def on_row_activated(self, widget, path, col=0): model = self.tree.get_model() iter = model.get_iter(path) jid = model.get_value(iter, 2) if (jid == 'group'): if (self.tree.row_expanded(path)): self.tree.collapse_row(path) else: self.tree.expand_row(path, FALSE) else: if self.tab_messages.has_key(jid): self.tab_messages[jid].window.present() elif self.l_contact.has_key(jid): self.tab_messages[jid] = message(self.l_contact[jid]['user'], self) if self.tab_queues.has_key(jid): self.tab_messages[jid].read_queue(self.tab_queues[jid]) def on_row_expanded(self, widget, iter, path): self.tree.get_model().set_value(iter, 0, self.pixbufs['opened']) def on_row_collapsed(self, widget, iter, path): self.tree.get_model().set_value(iter, 0, self.pixbufs['closed']) def on_cell_edited (self, cell, row, new_text): model = self.tree.get_model() iter = model.get_iter_from_string(row) jid = model.get_value(iter, 2) old_text = self.l_contact[jid]['user'].name #If it is a double click, old_text == new_text if old_text == new_text: if self.tab_messages.has_key(jid): self.tab_messages[jid].window.present() elif self.l_contact.has_key(jid): self.tab_messages[jid] = message(self.l_contact[jid]['user'], self) if self.tab_queues.has_key(jid): self.tab_messages[jid].read_queue(self.tab_queues[jid]) else: model.set_value(iter, 1, new_text) self.l_contact[jid]['user'].name = new_text self.queueOUT.put(('UPDUSER', (jid, new_text, self.l_contact[jid]['user'].groups))) def on_browse(self, widget): global Wbrowser if not Wbrowser: Wbrowser = browser(self) def mkpixbufs(self): self.path = 'plugins/gtkgui/icons/' + self.iconstyle + '/' self.pixbufs = {} for state in ('online', 'away', 'xa', 'dnd', 'offline', 'requested', 'message', 'opened', 'closed'): if not os.path.exists(self.path + state + '.xpm'): print 'No such file : ' + self.path + state + '.xpm' self.pixbufs[state] = None else: pix = gtk.gdk.pixbuf_new_from_file (self.path + state + '.xpm') self.pixbufs[state] = pix def on_show_off(self, widget): self.showOffline = 1 - self.showOffline self.redraw_roster() def __init__(self, queueOUT, plug): # FIXME : handle no file ... self.cfgParser = common.optparser.OptionsParser(CONFPATH) self.cfgParser.parseCfgFile() xml = gtk.glade.XML('plugins/gtkgui/gtkgui.glade', 'Gajim') self.window = xml.get_widget('Gajim') self.tree = xml.get_widget('treeview') self.plugin = plug self.connected = 0 #(icon, name, jid, editable, background color, show_icon) model = gtk.TreeStore(gtk.gdk.Pixbuf, str, str, gobject.TYPE_BOOLEAN, str, gobject.TYPE_BOOLEAN) self.tree.set_model(model) self.init_tree() self.iconstyle = self.cfgParser.GtkGui_iconstyle if not self.iconstyle: self.iconstyle = 'sun' self.mkpixbufs() # map = self.tree.get_colormap() # colour = map.alloc_color("red") # light red # colour2 = map.alloc_color("blue") # light red # colour = map.alloc_color("#FF9999") # light red # st = self.tree.get_style().copy() # st.bg[gtk.STATE_NORMAL] = colour # st.fg[gtk.STATE_NORMAL] = colour # st.bg[gtk.STATE_ACTIVE] = colour2 # st.fg[gtk.STATE_ACTIVE] = colour2 # st.bg[gtk.STATE_INSENSITIVE] = colour # st.bg[gtk.STATE_PRELIGHT] = colour # st.bg[gtk.STATE_SELECTED] = colour # st.fg[gtk.STATE_SELECTED] = colour2 # st.white = colour # print st.bg # print self.tree.get_property('expander-column') # self.tree.set_style(st) self.queueOUT = queueOUT self.optionmenu = xml.get_widget('optionmenu') self.optionmenu.set_history(6) self.tab_messages = {} self.tab_queues = {} accountsStr = self.cfgParser.Profile_accounts self.accounts = string.split(accountsStr, ' ') showOffline = self.cfgParser.GtkGui_showoffline if showOffline: self.showOffline = string.atoi(showOffline) else: self.showOffline = 0 xml.get_widget('show_offline').set_active(self.showOffline) self.grpbgcolor = 'gray50' self.userbgcolor = 'white' #columns col = gtk.TreeViewColumn() render_pixbuf = gtk.CellRendererPixbuf() col.pack_start(render_pixbuf, expand = False) col.add_attribute(render_pixbuf, 'pixbuf', 0) col.add_attribute(render_pixbuf, 'cell-background', 4) col.add_attribute(render_pixbuf, 'visible', 5) render_text = gtk.CellRendererText() render_text.connect('edited', self.on_cell_edited) col.pack_start(render_text, expand = True) col.add_attribute(render_text, 'text', 1) col.add_attribute(render_text, 'cell-background', 4) col.add_attribute(render_text, 'editable', 3) self.tree.append_column(col) col = gtk.TreeViewColumn() render_pixbuf = gtk.CellRendererPixbuf() col.pack_start(render_pixbuf, expand = False) self.tree.append_column(col) col.set_visible(FALSE) self.tree.set_expander_column(col) #signals xml.signal_connect('gtk_main_quit', self.on_quit) xml.signal_connect('on_preferences_activate', self.on_prefs) xml.signal_connect('on_accounts_activate', self.on_accounts) xml.signal_connect('on_browse_agents_activate', self.on_browse) xml.signal_connect('on_add_activate', self.on_add) xml.signal_connect('on_show_offline_activate', self.on_show_off) xml.signal_connect('on_about_activate', self.on_about) xml.signal_connect('on_quit_activate', self.on_quit) xml.signal_connect('on_treeview_event', self.on_treeview_event) xml.signal_connect('on_status_changed', self.on_status_changed) xml.signal_connect('on_row_activated', self.on_row_activated) xml.signal_connect('on_row_expanded', self.on_row_expanded) xml.signal_connect('on_row_collapsed', self.on_row_collapsed) class plugin: def read_queue(self): global Wbrowser model = self.r.tree.get_model() while self.queueIN.empty() == 0: ev = self.queueIN.get() if ev[0] == 'ROSTER': self.r.init_tree() self.r.mklists(ev[1]) self.r.draw_roster() elif ev[0] == 'WARNING': warning(ev[1]) elif ev[0] == 'STATUS': st = "" for i in range(7): if self.r.optionmenu.get_menu().get_children()[i].name == ev[1]: st = self.r.optionmenu.get_menu().get_children()[i].name self.r.optionmenu.set_history(i) break if st == 'offline': self.r.connected = 0 self.sleeper = None for j in self.r.l_contact.keys(): self.r.chg_status(j, 'offline', 'Disconnected') elif self.r.connected == 0: # self.tree self.r.connected = 1 self.r.plugin.sleeper = common.sleepy.Sleepy(\ self.autoawaytime*60, self.autoxatime*60) elif ev[0] == 'NOTIFY': jid = string.split(ev[1][0], '/')[0] res = ev[1][3] if not res: res = '' if string.find(jid, "@") <= 0: #It must be an agent ji = string.replace(jid, '@', '') else: ji = jid #Update user if self.r.l_contact.has_key(ji): u = self.r.l_contact[ji]['user'] u.show = ev[1][1] u.status = ev[1][2] u.resource = res #Print status in chat window if self.r.tab_messages.has_key(ji): self.r.tab_messages[ji].print_conversation(\ "%s is now %s (%s)" % (u.name, ev[1][1], ev[1][2]), 'status') if string.find(jid, "@") <= 0: #It must be an agent if not self.r.l_contact.has_key(ji): user1 = user(ji, ji, ['Agents'], ev[1][1], ev[1][2], 'from', res) self.r.add_user(user1) else: #Update existing line # self.l_contact[jid]['user'].show = show # self.l_contact[jid]['user'].status = status for i in self.r.l_contact[ji]['iter']: if self.r.pixbufs.has_key(ev[1][1]): model.set_value(i, 0, self.r.pixbufs[ev[1][1]]) elif self.r.l_contact.has_key(ji): #It isn't an agent self.r.chg_status(jid, ev[1][1], ev[1][2]) elif ev[0] == 'MSG': if string.find(ev[1][0], "@") <= 0: jid = string.replace(ev[1][0], '@', '') else: jid = ev[1][0] autopopup = self.r.cfgParser.GtkGui_autopopup if autopopup: self.autopopup = string.atoi(autopopup) else: self.autopopup = 0 if self.autopopup == 0 and not self.r.tab_messages.has_key(jid): #We save it in a queue if not self.r.tab_queues.has_key(jid): self.r.tab_queues[jid] = Queue.Queue(50) for i in self.r.l_contact[jid]['iter']: model.set_value(i, 0, self.r.pixbufs['message']) tim = time.strftime("[%H:%M:%S]") self.r.tab_queues[jid].put((ev[1][1], tim)) else: if not self.r.tab_messages.has_key(jid): #FIXME:message from unknown if self.r.l_contact.has_key(jid): self.r.tab_messages[jid] = message(self.r.l_contact[jid]['user'], self.r) if self.r.tab_messages.has_key(jid): self.r.tab_messages[jid].print_conversation(ev[1][1]) elif ev[0] == 'SUBSCRIBE': authorize(self.r, ev[1]) elif ev[0] == 'SUBSCRIBED': jid = ev[1]['jid'] if self.r.l_contact.has_key(jid): u = self.r.l_contact[jid]['user'] u.name = ev[1]['nom'] for i in self.r.l_contact[u.jid]['iter']: model.set_value(i, 1, u.name) else: user1 = user(jid, jid, ['general'], 'online', 'online', 'to', ev[1]['ressource']) self.r.add_user(user1) #TODO: print 'you are now authorized' elif ev[0] == 'AGENTS': if Wbrowser: Wbrowser.agents(ev[1]) elif ev[0] == 'AGENT_INFO': if not ev[1][1].has_key('instructions'): warning('error contacting %s' % ev[1][0]) else: Wreg = agent_reg(ev[1][0], ev[1][1], self.r) #('ACC_OK', (hostname, login, pasword, name, ressource)) elif ev[0] == 'ACC_OK': self.r.cfgParser.add_section(ev[1][3]) self.r.accounts.append(ev[1][3]) accountsStr = string.join(self.r.accounts) self.r.cfgParser.set('Profile', 'accounts', accountsStr) self.r.cfgParser.set(ev[1][3], 'name', ev[1][1]) self.r.cfgParser.set(ev[1][3], 'hostname', ev[1][0]) self.r.cfgParser.set(ev[1][3], 'password', ev[1][2]) self.r.cfgParser.set(ev[1][3], 'ressource', ev[1][4]) self.r.cfgParser.writeCfgFile() self.r.cfgParser.parseCfgFile() if (Waccounts != 0): Waccounts.init_accounts() elif ev[0] == 'QUIT': self.r.on_quit(self) return 1 def read_sleepy(self): if self.sleeper and (self.autoaway or self.autoxa) and \ (self.r.optionmenu.get_history()==0 or \ self.sleeper_state!=common.sleepy.STATE_AWAKE): self.sleeper.poll() state = self.sleeper.getState() if state != self.sleeper_state: accountsStr = self.r.cfgParser.Profile_accounts accounts = string.split(accountsStr, ' ') if state == common.sleepy.STATE_AWAKE: #on repasse online self.r.optionmenu.set_history(0) self.r.queueOUT.put(('STATUS',('online', '', accounts[0]))) if state == common.sleepy.STATE_AWAY and self.autoaway: #on passe away self.r.optionmenu.set_history(1) self.r.queueOUT.put(('STATUS',('away', 'auto away (idle)', accounts[0]))) if state == common.sleepy.STATE_XAWAY and self.autoxa: #on passe away self.r.optionmenu.set_history(2) self.r.queueOUT.put(('STATUS',('xa', 'auto away (idel)', accounts[0]))) self.sleeper_state = state return 1 def __init__(self, quIN, quOUT): gtk.threads_init() gtk.threads_enter() self.queueIN = quIN self.r = roster(quOUT, self) st = self.r.cfgParser.GtkGui_autoaway if not st: st = '1' self.autoaway = string.atoi(st) st = self.r.cfgParser.GtkGui_autoawaytime if not st: st = '10' self.autoawaytime = string.atoi(st) st = self.r.cfgParser.GtkGui_autoxa if not st: st = '1' self.autoxa = string.atoi(st) st = self.r.cfgParser.GtkGui_autoxatime if not st: st = '20' self.autoxatime = string.atoi(st) self.time = gtk.timeout_add(200, self.read_queue) gtk.timeout_add(1000, self.read_sleepy) self.sleeper = None self.sleeper_state = None gtk.main() # while 1: # while gtk.events_pending(): gtk.mainiteration() gtk.threads_leave() if __name__ == "__main__": plugin(None, None) print "plugin gtkgui loaded"