gajim-plural/plugins/gtkgui/roster_window.py

1413 lines
49 KiB
Python

## plugins/gtkgui/roster_window.py
##
## Gajim Team:
## - Yann Le Boulanger <asterix@lagaule.org>
## - Vincent Hanquez <tab@snarc.org>
## - Nikos Kouremenos <kourem@gmail.com>
##
## Copyright (C) 2003-2005 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 gtk
import gtk.glade
import pango
import gobject
import os
import Queue
import common.sleepy
from tabbed_chat_window import *
from groupchat_window import *
from history_window import *
from gtkgui import CellRendererImage, User
from dialogs import *
from config import *
from common import i18n
_ = i18n._
APP = i18n.APP
gtk.glade.bindtextdomain(APP, i18n.DIR)
gtk.glade.textdomain(APP)
GTKGUI_GLADE='plugins/gtkgui/gtkgui.glade'
class Roster_window:
"""Class for main window of gtkgui plugin"""
def get_account_iter(self, name):
if self.regroup:
return None
model = self.tree.get_model()
fin = False
account = model.get_iter_root()
if not account:
return None
while not fin:
account_name = model.get_value(account, 3)
if name == account_name:
return account
account = model.iter_next(account)
if not account:
fin = True
return None
def get_group_iter(self, name, account):
model = self.tree.get_model()
root = self.get_account_iter(account)
fin = False
group = model.iter_children(root)
if not group:
fin = True
while not fin:
group_name = model.get_value(group, 3)
if name == group_name:
return group
group = model.iter_next(group)
if not group:
fin = True
return None
def get_user_iter(self, jid, account):
model = self.tree.get_model()
acct = self.get_account_iter(account)
found = []
fin = False
group = model.iter_children(acct)
if not group:
return found
while not fin:
fin2 = False
user = model.iter_children(group)
if not user:
fin2=True
while not fin2:
if jid == model.get_value(user, 3):
found.append(user)
user = model.iter_next(user)
if not user:
fin2 = True
group = model.iter_next(group)
if not group:
fin = True
return found
def add_account_to_roster(self, account):
if self.regroup:
return
model = self.tree.get_model()
if self.get_account_iter(account):
return
statuss = ['offline', 'connecting', 'online', 'away', 'xa', 'dnd',\
'invisible']
status = statuss[self.plugin.connected[account]]
model.append(None, (self.pixbufs[status], account, 'account', account,\
account, False))
def remove_newly_added(self, jid, account):
if jid in self.newly_added[account]:
self.newly_added[account].remove(jid)
self.redraw_jid(jid, account)
def add_user_to_roster(self, jid, account):
"""Add a user to the roster and add groups if they aren't in roster"""
showOffline = self.plugin.config['showoffline']
if not self.contacts[account].has_key(jid):
return
users = self.contacts[account][jid]
user = users[0]
if user.jid.find("@") <= 0:
user.groups = ['Agents']
elif user.groups == []:
user.groups.append('General')
if (user.show == 'offline' or user.show == 'error') and not showOffline\
and not 'Agents' in user.groups and \
not self.plugin.queues[account].has_key(user.jid):
return
model = self.tree.get_model()
for g in user.groups:
iterG = self.get_group_iter(g, account)
if not iterG:
IterAcct = self.get_account_iter(account)
iterG = model.append(IterAcct, \
(self.pixbufs['closed'], g, 'group', \
g, account, False))
if not self.groups[account].has_key(g): #It can probably never append
if account+g in self.hidden_lines:
self.groups[account][g] = {'expand': False}
else:
self.groups[account][g] = {'expand': True}
if not account in self.hidden_lines and not self.plugin.config['mergeaccounts']:
self.tree.expand_row((model.get_path(iterG)[0]), False)
typestr = 'user'
if g == 'Agents':
typestr = 'agent'
model.append(iterG, (self.pixbufs[user.show], \
user.name, typestr, user.jid, account, False))
if self.groups[account][g]['expand']:
self.tree.expand_row(model.get_path(iterG), False)
self.redraw_jid(jid, account)
def really_remove_user(self, user, account):
if user.jid in self.to_be_removed[account]:
self.to_be_removed[account].remove(user.jid)
self.remove_user(user, account)
def remove_user(self, user, account):
"""Remove a user from the roster"""
if user.jid in self.to_be_removed[account]:
return
model = self.tree.get_model()
for i in self.get_user_iter(user.jid, account):
parent_i = model.iter_parent(i)
group = model.get_value(parent_i, 3)
model.remove(i)
if model.iter_n_children(parent_i) == 0:
model.remove(parent_i)
# We need to check all contacts, even offline contacts
group_empty = True
for jid in self.contacts[account]:
if group in self.contacts[account][jid][0].groups:
group_empty = False
break
if group_empty:
del self.groups[account][group]
def redraw_jid(self, jid, account):
"""draw the correct pixbuf and name"""
model = self.tree.get_model()
iters = self.get_user_iter(jid, account)
if len(iters) == 0:
return
users = self.contacts[account][jid]
name = users[0].name
if len(users) > 1:
name += " (" + str(len(users)) + ")"
prio = 0
user = users[0]
for u in users:
if u.priority > prio:
prio = u.priority
user = u
for iter in iters:
if self.plugin.queues[account].has_key(jid):
img = self.pixbufs['message']
else:
if user.sub != 'both':
if user.ask == 'subscribe':
img = self.pixbufs['requested']
else:
img = self.pixbufs['not in the roster']
else:
img = self.pixbufs[user.show]
model.set_value(iter, 0, img)
model.set_value(iter, 1, name)
def make_menu(self):
"""create the main_window's menus"""
# try to avoid WIDGET_REALIZED_FOR_EVENT failed which freezes gajim
new_message_menuitem = self.xml.get_widget('new_message_menuitem')
join_gc_menuitem = self.xml.get_widget('join_gc_menuitem')
add_new_contact_menuitem = self.xml.get_widget('add_new_contact_menuitem')
service_disco_menuitem = self.xml.get_widget('service_disco_menuitem')
if self.add_new_contact_handler_id:
add_new_contact_menuitem.handler_disconnect(self.add_new_contact_handler_id)
self.add_new_contact_handler_id = None
if self.service_disco_handler_id:
service_disco_menuitem.handler_disconnect(\
self.service_disco_handler_id)
self.service_disco_handler_id = None
if self.join_gc_handler_id:
join_gc_menuitem.handler_disconnect(self.join_gc_handler_id)
self.join_gc_handler_id = None
if self.new_message_menuitem_handler_id:
new_message_menuitem.handler_disconnect(\
self.new_message_menuitem_handler_id)
self.new_message_menuitem_handler_id = None
#remove the existing submenus
if add_new_contact_menuitem.get_submenu():
add_new_contact_menuitem.remove_submenu()
if service_disco_menuitem.get_submenu():
service_disco_menuitem.remove_submenu()
if join_gc_menuitem.get_submenu():
join_gc_menuitem.remove_submenu()
if new_message_menuitem.get_submenu():
new_message_menuitem.remove_submenu()
if len(self.plugin.accounts.keys()) > 0:
new_message_menuitem.set_sensitive(True)
join_gc_menuitem.set_sensitive(True)
add_new_contact_menuitem.set_sensitive(True)
service_disco_menuitem.set_sensitive(True)
else:
new_message_menuitem.set_sensitive(False)
join_gc_menuitem.set_sensitive(False)
add_new_contact_menuitem.set_sensitive(False)
service_disco_menuitem.set_sensitive(False)
if len(self.plugin.accounts.keys()) >= 2: # 2 or more accounts? make submenus
#add
sub_menu = gtk.Menu()
add_new_contact_menuitem.set_submenu(sub_menu)
for account in self.plugin.accounts.keys():
item = gtk.MenuItem(_('to ') + account + _(' account'))
sub_menu.append(item)
item.connect("activate", self.on_add_new_contact, account)
sub_menu.show_all()
#disco
sub_menu = gtk.Menu()
service_disco_menuitem.set_submenu(sub_menu)
for account in self.plugin.accounts.keys():
our_jid = self.plugin.accounts[account]['name'] + '@' +\
self.plugin.accounts[account]['hostname']
item = gtk.MenuItem(_('using ') + account + _(' account'))
sub_menu.append(item)
item.connect('activate', self.on_service_disco_menuitem_activate, account)
sub_menu.show_all()
#join gc
sub_menu = gtk.Menu()
join_gc_menuitem.set_submenu(sub_menu)
for account in self.plugin.accounts.keys():
our_jid = self.plugin.accounts[account]['name'] + '@' +\
self.plugin.accounts[account]['hostname']
item = gtk.MenuItem(_('as ') + our_jid)
sub_menu.append(item)
item.connect("activate", self.on_join_gc_activate, account)
sub_menu.show_all()
#new message
sub_menu = gtk.Menu()
new_message_menuitem.set_submenu(sub_menu)
for account in self.plugin.accounts.keys():
our_jid = self.plugin.accounts[account]['name'] + '@' +\
self.plugin.accounts[account]['hostname']
item = gtk.MenuItem(_('as ') + our_jid)
sub_menu.append(item)
item.connect('activate', self.on_new_message_menuitem_activate, account)
sub_menu.show_all()
elif len(self.plugin.accounts.keys()) == 1: # one account
#add
if not self.add_new_contact_handler_id:
self.add_new_contact_handler_id = add_new_contact_menuitem.connect(\
'activate', self.on_add_new_contact, self.plugin.accounts.keys()[0])
#disco
if not self.service_disco_handler_id:
self.service_disco_handler_id = service_disco_menuitem.connect(\
'activate', self.on_service_disco_menuitem_activate, self.plugin.accounts.keys()[0])
#join_gc
if not self.join_gc_handler_id:
self.join_gc_handler_id = join_gc_menuitem.connect(\
'activate', self.on_join_gc_activate, self.plugin.accounts.keys()[0])
if not self.new_message_menuitem_handler_id:
self.new_message_menuitem_handler_id = new_message_menuitem.connect(\
'activate', self.on_new_message_menuitem_activate, self.plugin.accounts.keys()[0])
def draw_roster(self):
"""Clear and draw roster"""
self.make_menu()
self.tree.get_model().clear()
for acct in self.contacts.keys():
self.add_account_to_roster(acct)
for jid in self.contacts[acct].keys():
self.add_user_to_roster(jid, acct)
def mklists(self, array, account):
"""fill self.contacts and self.groups"""
if not self.contacts.has_key(account):
self.contacts[account] = {}
if not self.groups.has_key(account):
self.groups[account] = {}
for jid in array.keys():
jids = jid.split('/')
#get jid
ji = jids[0]
#get resource
resource = ''
if len(jids) > 1:
resource = jids[1:]
#get name
name = array[jid]['name']
if not name:
if ji.find("@") <= 0:
name = ji
else:
name = jid.split('@')[0]
#get show
show = array[jid]['show']
if not show:
show = 'offline'
user1 = User(ji, name, array[jid]['groups'], show, \
array[jid]['status'], array[jid]['sub'], array[jid]['ask'], \
resource, 0, '')
#when we draw the roster, we can't have twice the same user with
# 2 resources
self.contacts[account][ji] = [user1]
for g in array[jid]['groups'] :
if not g in self.groups[account].keys():
if account+g in self.hidden_lines:
self.groups[account][g] = {'expand': False}
else:
self.groups[account][g] = {'expand': True}
def chg_user_status(self, user, show, status, account):
"""When a user change his status"""
showOffline = self.plugin.config['showoffline']
model = self.tree.get_model()
luser = self.contacts[account][user.jid]
user.show = show
user.status = status
if (show == 'offline' or show == 'error') and \
not self.plugin.queues[account].has_key(user.jid):
if len(luser) > 1:
luser.remove(user)
self.redraw_jid(user.jid, account)
elif not showOffline:
self.remove_user(user, account)
else:
self.redraw_jid(user.jid, account)
else:
if not self.get_user_iter(user.jid, account):
self.add_user_to_roster(user.jid, account)
self.redraw_jid(user.jid, account)
#Print status in chat window
if self.plugin.windows[account]['chats'].has_key(user.jid):
self.plugin.windows[account]['chats'][user.jid].set_image(user.jid)
name = user.name
if user.resource != '':
name += '/'+user.resource
self.plugin.windows[account]['chats'][user.jid].print_conversation(\
_("%s is now %s (%s)") % (name, show, status), user.jid, 'status')
def on_info(self, widget, user, account):
"""Call vcard_information_window class to display user's information"""
if not self.plugin.windows[account]['infos'].has_key(user.jid):
self.plugin.windows[account]['infos'][user.jid] = \
vcard_information_window(user, self.plugin, account)
def on_agent_logging(self, widget, jid, state, account):
"""When an agent is requested to log in or off"""
self.plugin.send('AGENT_LOGGING', account, (jid, state))
def on_remove_agent(self, widget, jid, account):
"""When an agent is requested to log in or off"""
window = Confirmation_dialog(_('Are you sure you want to remove the agent %s from your roster?') % jid)
if window.get_response() == gtk.RESPONSE_YES:
self.plugin.send('UNSUB_AGENT', account, jid)
for u in self.contacts[account][jid]:
self.remove_user(u, account)
del self.contacts[account][u.jid]
def on_rename(self, widget, iter, path):
model = self.tree.get_model()
model.set_value(iter, 5, True)
self.tree.set_cursor(path, self.tree.get_column(0), True)
def on_edit_groups(self, widget, user, account):
dlg = Edit_groups_dialog(user, account, self.plugin)
dlg.run()
def on_history(self, widget, user):
"""When history button is pressed : call log window"""
if not self.plugin.windows['logs'].has_key(user.jid):
self.plugin.windows['logs'][user.jid] = history_window(self.plugin, \
user.jid)
def mk_menu_user(self, event, iter):
"""Make user's popup menu"""
model = self.tree.get_model()
jid = model.get_value(iter, 3)
path = model.get_path(iter)
account = model.get_value(iter, 4)
user = self.contacts[account][jid][0]
menu = gtk.Menu()
item = gtk.MenuItem(_('Start chat'))
menu.append(item)
item.connect('activate', self.on_roster_treeview_row_activated, path)
item = gtk.MenuItem(_('Rename'))
menu.append(item)
item.connect('activate', self.on_rename, iter, path)
if not 'not in the roster' in user.groups:
item = gtk.MenuItem(_('Edit groups'))
menu.append(item)
item.connect('activate', self.on_edit_groups, user, account)
item = gtk.MenuItem()
menu.append(item)
item = gtk.MenuItem(_('Subscription'))
menu.append(item)
sub_menu = gtk.Menu()
item.set_submenu(sub_menu)
item = gtk.MenuItem(_('Resend authorization to'))
sub_menu.append(item)
item.connect('activate', self.authorize, jid, account)
item = gtk.MenuItem(_('Rerequest authorization from'))
sub_menu.append(item)
item.connect('activate', self.req_sub, jid, \
_('I would like to add you to my contact list.'), account)
item = gtk.MenuItem(_('Remove'))
menu.append(item)
item.connect('activate', self.on_req_usub, user, account)
item = gtk.MenuItem()
menu.append(item)
item = gtk.MenuItem(_('Information'))
menu.append(item)
item.connect('activate', self.on_info, user, account)
item = gtk.MenuItem(_('History'))
menu.append(item)
item.connect('activate', self.on_history, user)
menu.popup(None, None, None, event.button, event.time)
menu.show_all()
menu.reposition()
def mk_menu_g(self, event, iter):
"""Make group's popup menu"""
model = self.tree.get_model()
path = model.get_path(iter)
menu = gtk.Menu()
item = gtk.MenuItem(_('Rename'))
menu.append(item)
item.connect('activate', self.on_rename, iter, path)
menu.popup(None, None, None, event.button, event.time)
menu.show_all()
menu.reposition()
def mk_menu_agent(self, event, iter):
"""Make agent's popup menu"""
model = self.tree.get_model()
jid = model.get_value(iter, 3)
path = model.get_path(iter)
account = model.get_value(iter, 4)
menu = gtk.Menu()
item = gtk.MenuItem(_('Log on'))
if self.contacts[account][jid][0].show != 'offline':
item.set_sensitive(False)
menu.append(item)
item.connect('activate', self.on_agent_logging, jid, 'available', account)
item = gtk.MenuItem(_('Log off'))
if self.contacts[account][jid][0].show == 'offline':
item.set_sensitive(False)
menu.append(item)
item.connect('activate', self.on_agent_logging, jid, 'unavailable', \
account)
item = gtk.MenuItem()
menu.append(item)
item = gtk.MenuItem(_('Remove'))
menu.append(item)
item.connect('activate', self.on_remove_agent, jid, account)
menu.popup(None, None, None, event.button, event.time)
menu.show_all()
menu.reposition()
def on_edit_account(self, widget, account):
if not self.plugin.windows.has_key('account_modification_window'):
infos = self.plugin.accounts[account]
infos['accname'] = account
infos['jid'] = self.plugin.accounts[account]["name"] + \
'@' + self.plugin.accounts[account]["hostname"]
self.plugin.windows['account_modification'] = \
Account_modification_window(self.plugin, infos)
def mk_menu_account(self, event, iter):
"""Make account's popup menu"""
model = self.tree.get_model()
account = model.get_value(iter, 3)
menu = gtk.Menu()
item = gtk.MenuItem(_('Status'))
menu.append(item)
sub_menu = gtk.Menu()
item.set_submenu(sub_menu)
item = gtk.MenuItem(_('Online'))
sub_menu.append(item)
item.connect('activate', self.change_status, account, 'online')
item = gtk.MenuItem(_('Away'))
sub_menu.append(item)
item.connect('activate', self.change_status, account, 'away')
item = gtk.MenuItem(_('NA'))
sub_menu.append(item)
item.connect('activate', self.change_status, account, 'xa')
item = gtk.MenuItem(_('DND'))
sub_menu.append(item)
item.connect('activate', self.change_status, account, 'dnd')
item = gtk.MenuItem(_('Invisible'))
sub_menu.append(item)
item.connect('activate', self.change_status, account, 'invisible')
item = gtk.MenuItem()
sub_menu.append(item)
item = gtk.MenuItem(_('Offline'))
sub_menu.append(item)
item.connect('activate', self.change_status, account, 'offline')
item = gtk.MenuItem()
menu.append(item)
item = gtk.MenuItem(_('_Edit account'))
menu.append(item)
item.connect('activate', self.on_edit_account, account)
item = gtk.MenuItem(_('_Service discovery'))
menu.append(item)
item.connect('activate', self.on_service_disco_menuitem_activate, account)
item = gtk.MenuItem(_('_Add contact'))
menu.append(item)
item.connect('activate', self.on_add_new_contact, account)
item = gtk.MenuItem(_('Join _groupchat'))
menu.append(item)
item.connect('activate', self.on_join_gc_activate, account)
item = gtk.MenuItem(_('_New message'))
menu.append(item)
item.connect('activate', self.on_new_message_menuitem_activate, account)
menu.popup(None, None, None, event.button, event.time)
menu.show_all()
menu.reposition()
def authorize(self, widget, jid, account):
"""Authorize a user"""
self.plugin.send('AUTH', account, jid)
def req_sub(self, widget, jid, txt, account, group=None, pseudo=None):
"""Request subscription to a user"""
if not pseudo:
pseudo = jid
self.plugin.send('SUB', account, (jid, txt))
if not self.contacts[account].has_key(jid):
if not group:
group = 'General'
user1 = User(jid, pseudo, [group], 'requested', \
'requested', 'none', 'subscribe', '', 0, '')
self.contacts[account][jid] = [user1]
self.add_user_to_roster(jid, account)
def on_roster_treeview_key_press_event(self, widget, event):
"""when a key is pressed in the treeviews"""
if event.keyval == gtk.keysyms.Escape:
self.tree.get_selection().unselect_all()
if event.keyval == gtk.keysyms.F2:
treeselection = self.tree.get_selection()
model, iter = treeselection.get_selected()
if not iter:
return
type = model.get_value(iter, 2)
if type == 'user' or type == 'group':
path = model.get_path(iter)
model.set_value(iter, 5, True)
self.tree.set_cursor(path, self.tree.get_column(0), True)
if event.keyval == gtk.keysyms.Delete:
treeselection = self.tree.get_selection()
model, iter = treeselection.get_selected()
if not iter:
return
jid = model.get_value(iter, 3)
account = model.get_value(iter, 4)
type = model.get_value(iter, 2)
if type == 'user':
user = self.contacts[account][jid][0]
self.on_req_usub(widget, user, account)
elif type == 'agent':
self.on_remove_agent(widget, jid, account)
return False
def on_roster_treeview_button_press_event(self, widget, event):
"""popup contact's , group's or agent's menu"""
if event.type == gtk.gdk.BUTTON_PRESS:
if event.button == 3: # Right click
try:
path, column, x, y = self.tree.get_path_at_pos(int(event.x), \
int(event.y))
except TypeError:
self.tree.get_selection().unselect_all()
return
model = self.tree.get_model()
iter = model.get_iter(path)
type = model.get_value(iter, 2)
if type == 'group':
self.mk_menu_g(event, iter)
elif type == 'agent':
self.mk_menu_agent(event, iter)
elif type == 'user':
self.mk_menu_user(event, iter)
elif type == 'account':
self.mk_menu_account(event, iter)
#return True
return False
if event.button == 1: # Left click
try:
path, column, x, y = self.tree.get_path_at_pos(int(event.x), \
int(event.y))
except TypeError:
self.tree.get_selection().unselect_all()
return False
model = self.tree.get_model()
iter = model.get_iter(path)
type = model.get_value(iter, 2)
if (type == 'group' or type == 'account'):
if x < 20: # first cell in 1st column (the arrow SINGLE clicked)
if (self.tree.row_expanded(path)):
self.tree.collapse_row(path)
else:
self.tree.expand_row(path, False)
return False
def on_req_usub(self, widget, user, account):
"""Remove a user"""
window = Confirmation_dialog(_("Are you sure you want to remove %s (%s) from your roster?") % (user.name, user.jid))
if window.get_response() == gtk.RESPONSE_YES:
self.plugin.send('UNSUB', account, user.jid)
for u in self.contacts[account][user.jid]:
self.remove_user(u, account)
del self.contacts[account][u.jid]
def send_status(self, account, status, txt, autoconnect=0):
if status != 'offline':
if self.plugin.connected[account] < 2:
model = self.tree.get_model()
accountIter = self.get_account_iter(account)
if accountIter:
model.set_value(accountIter, 0, self.pixbufs['connecting'])
self.plugin.connected[account] = 1
if self.plugin.systray_enabled:
self.plugin.systray.set_status('connecting')
save_pass = 0
if self.plugin.accounts[account].has_key('savepass'):
save_pass = self.plugin.accounts[account]['savepass']
if not save_pass and self.plugin.connected[account] < 2:
passphrase = ''
w = Passphrase_dialog(_('Enter your password for account %s') \
% account, 'Save password', autoconnect)
passphrase, save = w.run()
if passphrase == -1:
if accountIter:
model.set_value(accountIter, 0, self.pixbufs['offline'])
self.plugin.connected[account] = 0
self.plugin.systray.set_status('offline')
self.update_status_comboxbox()
return
self.plugin.send('PASSPHRASE', account, passphrase)
if save:
self.plugin.accounts[account]['savepass'] = 1
self.plugin.accounts[account]['password'] = passphrase
keyid = None
save_gpg_pass = 0
if self.plugin.accounts[account].has_key('savegpgpass'):
save_gpg_pass = self.plugin.accounts[account]['savegpgpass']
if self.plugin.accounts[account].has_key('keyid'):
keyid = self.plugin.accounts[account]['keyid']
if keyid and self.plugin.connected[account] < 2 and \
self.plugin.config['usegpg']:
if save_gpg_pass:
passphrase = self.plugin.accounts[account]['gpgpassword']
else:
passphrase = ''
w = Passphrase_dialog(\
_('Enter GPG key passphrase for account %s') % account, \
'Save passphrase', autoconnect)
passphrase, save = w.run()
if passphrase == -1:
passphrase = ''
if save:
self.plugin.accounts[account]['savegpgpass'] = 1
self.plugin.accounts[account]['gpgpassword'] = passphrase
self.plugin.send('GPGPASSPHRASE', account, passphrase)
self.plugin.send('STATUS', account, (status, txt))
for room_jid in self.plugin.windows[account]['gc']:
if room_jid != 'tabbed':
nick = self.plugin.windows[account]['gc'][room_jid].nicks[room_jid]
self.plugin.send('GC_STATUS', account, (nick, room_jid, status, \
txt))
if status == 'online' and self.plugin.sleeper.getState() != \
common.sleepy.STATE_UNKNOWN:
self.plugin.sleeper_state[account] = 1
else:
self.plugin.sleeper_state[account] = 0
def get_status_message(self, status, autoconnect = 0):
if (status == 'online' and not self.plugin.config['ask_online_status']) \
or (status == 'offline' and not \
self.plugin.config['ask_offline_status']):
return status
dlg = Change_status_message_dialog(self.plugin, status, autoconnect)
message = dlg.run()
return message
def change_status(self, widget, account, status):
message = self.get_status_message(status)
if message == -1:
return
self.send_status(account, status, message)
def on_status_combobox_changed(self, widget):
"""When we change our status"""
model = self.status_combobox.get_model()
active = self.status_combobox.get_active()
if active < 0:
return
accounts = self.plugin.accounts.keys()
if len(accounts) == 0:
Error_dialog(_('You must create an account before connecting to jabber network.'))
self.update_status_comboxbox()
return
status = model[active][2]
message = self.get_status_message(status)
if message == -1:
self.update_status_comboxbox()
return
for acct in accounts:
if self.plugin.accounts[acct].has_key('sync_with_global_status'):
if not self.plugin.accounts[acct]['sync_with_global_status']:
continue
self.send_status(acct, status, message)
def update_status_comboxbox(self):
#table to change index in plugin.connected to index in combobox
table = {0:5, 1:5, 2:0, 3:1, 4:2, 5:3, 6:4}
maxi = 0
if len(self.plugin.connected.values()):
maxi = max(self.plugin.connected.values())
#temporarily block signal in order not to send status that we show
#in the combobox
self.status_combobox.handler_block(self.id_signal_cb)
self.status_combobox.set_active(table[maxi])
self.status_combobox.handler_unblock(self.id_signal_cb)
statuss = ['offline', 'connecting', 'online', 'away', 'xa', 'dnd',\
'invisible']
if self.plugin.systray_enabled:
self.plugin.systray.set_status(statuss[maxi])
image = self.pixbufs[statuss[maxi]]
if image.get_storage_type() == gtk.IMAGE_ANIMATION:
pixbuf = image.get_animation().get_static_image()
self.window.set_icon(pixbuf)
elif image.get_storage_type() == gtk.IMAGE_PIXBUF:
self.window.set_icon(image.get_pixbuf())
def on_status_changed(self, account, status):
"""the core tells us that our status has changed"""
if not self.contacts.has_key(account):
return
model = self.tree.get_model()
accountIter = self.get_account_iter(account)
if accountIter:
model.set_value(accountIter, 0, self.pixbufs[status])
statuss = ['offline', 'connecting', 'online', 'away', 'xa', 'dnd',\
'invisible']
if status == 'offline':
for jid in self.contacts[account]:
luser = self.contacts[account][jid]
luser_copy = []
for user in luser:
luser_copy.append(user)
for user in luser_copy:
self.chg_user_status(user, 'offline', 'Disconnected', account)
self.plugin.connected[account] = statuss.index(status)
self.update_status_comboxbox()
def new_chat(self, user, account):
if self.plugin.config['usetabbedchat']:
if not self.plugin.windows[account]['chats'].has_key('tabbed'):
self.plugin.windows[account]['chats']['tabbed'] = \
Tabbed_chat_window(user, self.plugin, account)
else:
self.plugin.windows[account]['chats']['tabbed'].new_user(user)
self.plugin.windows[account]['chats'][user.jid] = \
self.plugin.windows[account]['chats']['tabbed']
self.plugin.windows[account]['chats']['tabbed'].window.present()
else:
self.plugin.windows[account]['chats'][user.jid] = \
Tabbed_chat_window(user, self.plugin, account)
def new_group(self, jid, nick, account):
if self.plugin.config['usetabbedchat']:
if not self.plugin.windows[account]['gc'].has_key('tabbed'):
self.plugin.windows[account]['gc']['tabbed'] = \
Groupchat_window(jid, nick, self.plugin, account)
else:
self.plugin.windows[account]['gc']['tabbed'].new_group(jid, nick)
self.plugin.windows[account]['gc'][jid] = \
self.plugin.windows[account]['gc']['tabbed']
self.plugin.windows[account]['gc']['tabbed'].window.present()
self.plugin.windows[account]['gc']['tabbed'].active_tab(jid)
else:
self.plugin.windows[account]['gc'][jid] = \
Groupchat_window(jid, nick, self.plugin, account)
def on_message(self, jid, msg, tim, account):
"""when we receive a message"""
if not self.contacts[account].has_key(jid):
user1 = User(jid, jid, ['not in the roster'], \
'not in the roster', 'not in the roster', 'none', None, '', 0, '')
self.contacts[account][jid] = [user1]
self.add_user_to_roster(jid, account)
iters = self.get_user_iter(jid, account)
if iters:
path = self.tree.get_model().get_path(iters[0])
else:
path = None
autopopup = self.plugin.config['autopopup']
autopopupaway = self.plugin.config['autopopupaway']
if (autopopup == 0 or ( not autopopupaway and \
self.plugin.connected[account] > 2)) and not \
self.plugin.windows[account]['chats'].has_key(jid):
#We save it in a queue
if not self.plugin.queues[account].has_key(jid):
model = self.tree.get_model()
self.plugin.queues[account][jid] = Queue.Queue(50)
self.redraw_jid(jid, account)
if self.plugin.systray_enabled:
self.plugin.systray.add_jid(jid, account)
self.plugin.queues[account][jid].put((msg, tim))
self.nb_unread += 1
self.show_title()
if not path:
self.add_user_to_roster(jid, account)
iters = self.get_user_iter(jid, account)
path = self.tree.get_model().get_path(iters[0])
self.tree.expand_row(path[0:1], False)
self.tree.expand_row(path[0:2], False)
self.tree.scroll_to_cell(path)
self.tree.set_cursor(path)
else:
if not self.plugin.windows[account]['chats'].has_key(jid):
self.new_chat(self.contacts[account][jid][0], account)
if path:
self.tree.expand_row(path[0:1], False)
self.tree.expand_row(path[0:2], False)
self.tree.scroll_to_cell(path)
self.tree.set_cursor(path)
self.plugin.windows[account]['chats'][jid].print_conversation(msg, \
jid, tim = tim)
def on_preferences_menuitem_activate(self, widget):
if self.plugin.windows['preferences'].window.get_property('visible'):
self.plugin.windows['preferences'].window.present() # give focus
else:
self.plugin.windows['preferences'].window.show_all()
def on_add_new_contact(self, widget, account):
Add_new_contact_window(self.plugin, account)
def on_join_gc_activate(self, widget, account):
Join_groupchat_window(self.plugin, account)
def on_new_message_menuitem_activate(self, widget, account):
New_message_dialog(self.plugin, account)
def on_about_menuitem_activate(self, widget):
#About_dialog()
Popup_window(self.plugin)
def on_accounts_menuitem_activate(self, widget):
if self.plugin.windows.has_key('accounts'):
self.plugin.windows['accounts'].present()
else:
self.plugin.windows['accounts'] = Accounts_window(self.plugin)
def close_all(self, dic):
"""close all the windows in the given dictionary"""
for w in dic.values():
if type(w) == type({}):
self.close_all(w)
else:
w.window.destroy()
def on_roster_window_delete_event(self, widget, event):
"""When we want to close the window"""
if self.plugin.systray_enabled:
self.window.hide()
else:
accounts = self.plugin.accounts.keys()
get_msg = False
for acct in accounts:
if self.plugin.connected[acct]:
get_msg = True
break
if get_msg:
message = self.get_status_message('offline')
if message == -1:
message = ''
for acct in accounts:
if self.plugin.connected[acct]:
self.send_status(acct, 'offline', message)
self.quit_gtkgui_plugin()
return True # do NOT destory the window
def quit_gtkgui_plugin(self):
"""When we quit the gtk plugin :
tell that to the core and exit gtk"""
if self.plugin.config.has_key('saveposition'):
if self.plugin.config['saveposition']:
self.plugin.config['x-position'], self.plugin.config['y-position']=\
self.window.get_position()
self.plugin.config['width'], self.plugin.config['height'] = \
self.window.get_size()
self.plugin.config['hiddenlines'] = '\t'.join(self.hidden_lines)
self.plugin.send('CONFIG', None, ('GtkGui', self.plugin.config, 'GtkGui'))
self.plugin.send('QUIT', None, ('gtkgui', 1))
print _("plugin gtkgui stopped")
self.close_all(self.plugin.windows)
if self.plugin.systray_enabled:
self.plugin.hide_systray()
gtk.main_quit()
def on_quit_menuitem_activate(self, widget):
accounts = self.plugin.accounts.keys()
get_msg = False
for acct in accounts:
if self.plugin.connected[acct]:
get_msg = True
break
if get_msg:
message = self.get_status_message('offline')
if message == -1:
return
for acct in accounts:
if self.plugin.connected[acct]:
self.send_status(acct, 'offline', message)
self.quit_gtkgui_plugin()
def on_roster_treeview_row_activated(self, widget, path, col=0):
"""When an iter is dubble clicked :
open the chat window"""
model = self.tree.get_model()
iter = model.get_iter(path)
account = model.get_value(iter, 4)
type = model.get_value(iter, 2)
jid = model.get_value(iter, 3)
if (type == 'group') or (type == 'account'):
if (self.tree.row_expanded(path)):
self.tree.collapse_row(path)
else:
self.tree.expand_row(path, False)
else:
if self.plugin.windows[account]['chats'].has_key(jid):
if self.plugin.config['usetabbedchat']:
self.plugin.windows[account]['chats'][jid].active_tab(jid)
self.plugin.windows[account]['chats'][jid].window.present()
elif self.contacts[account].has_key(jid):
self.new_chat(self.contacts[account][jid][0], account)
self.plugin.windows[account]['chats'][jid].active_tab(jid)
def on_roster_treeview_row_expanded(self, widget, iter, path):
"""When a row is expanded change the icon of the arrow"""
model = self.tree.get_model()
account = model.get_value(iter, 4)
type = model.get_value(iter, 2)
if type == 'group':
model.set_value(iter, 0, self.pixbufs['opened'])
jid = model.get_value(iter, 3)
self.groups[account][jid]['expand'] = True
if account+jid in self.hidden_lines:
self.hidden_lines.remove(account+jid)
elif type == 'account':
if account in self.hidden_lines:
self.hidden_lines.remove(account)
for g in self.groups[account]:
groupIter = self.get_group_iter(g, account)
if groupIter and self.groups[account][g]['expand']:
pathG = model.get_path(groupIter)
self.tree.expand_row(pathG, False)
def on_roster_treeview_row_collapsed(self, widget, iter, path):
"""When a row is collapsed :
change the icon of the arrow"""
model = self.tree.get_model()
account = model.get_value(iter, 4)
type = model.get_value(iter, 2)
if type == 'group':
model.set_value(iter, 0, self.pixbufs['closed'])
jid = model.get_value(iter, 3)
self.groups[account][jid]['expand'] = False
if not account+jid in self.hidden_lines:
self.hidden_lines.append(account+jid)
elif type == 'account':
if not account in self.hidden_lines:
self.hidden_lines.append(account)
def on_editing_canceled (self, cell):
"""editing has been canceled"""
#TODO: get iter
#model.set_value(iter, 5, False)
pass
def on_cell_edited (self, cell, row, new_text):
"""When an iter is editer :
if text has changed, rename the user"""
model = self.tree.get_model()
iter = model.get_iter_from_string(row)
path = model.get_path(iter)
account = model.get_value(iter, 4)
jid = model.get_value(iter, 3)
type = model.get_value(iter, 2)
if type == 'user':
old_text = self.contacts[account][jid][0].name
if old_text != new_text:
for u in self.contacts[account][jid]:
u.name = new_text
self.plugin.send('UPDUSER', account, (jid, new_text, \
self.contacts[account][jid][0].groups))
self.redraw_jid(jid, account)
elif type == 'group':
old_name = model.get_value(iter, 1)
#get all users in that group
for jid in self.contacts[account]:
user = self.contacts[account][jid][0]
if old_name in user.groups:
#set them in the new one and remove it from the old
self.remove_user(user, account)
user.groups.remove(old_name)
user.groups.append(new_text)
self.add_user_to_roster(user.jid, account)
self.plugin.send('UPDUSER', account, (user.jid, user.name, \
user.groups))
model.set_value(iter, 5, False)
def on_service_disco_menuitem_activate(self, widget, account):
"""When Service Discovery is selected:
Call browse class"""
if self.plugin.windows[account].has_key('disco'):
self.plugin.windows[account]['disco'].present()
else:
self.plugin.windows[account]['disco'] = \
Service_discovery_window(self.plugin, account)
def mkpixbufs(self):
"""initialise pixbufs array"""
iconset = self.plugin.config['iconset']
if not iconset:
iconset = 'sun'
self.path = 'plugins/gtkgui/iconsets/' + iconset + '/'
self.pixbufs = {}
for state in ('connecting', 'online', 'chat', 'away', 'xa', 'dnd', \
'invisible', 'offline', 'error', 'requested', 'message', \
'opened', 'closed', 'not in the roster'):
# try to open a pixfile with the correct method
state_file = state.replace(' ', '_')
files = []
files.append(self.path + state_file + '.gif')
files.append(self.path + state_file + '.png')
files.append(self.path + state_file + '.xpm')
image = gtk.Image()
image.show()
self.pixbufs[state] = image
for file in files:
if os.path.exists(file):
image.set_from_file(file)
break
def reload_pixbufs(self):
self.mkpixbufs()
# Update the roster
self.draw_roster()
# Update the status combobox
model = self.status_combobox.get_model()
iter = model.get_iter_root()
while iter:
model.set_value(iter, 1, self.pixbufs[model.get_value(iter, 2)])
iter = model.iter_next(iter)
# Update the systray
if self.plugin.systray_enabled:
self.plugin.systray.set_img()
for account in self.plugin.accounts.keys():
# Update opened chat windows
for jid in self.plugin.windows[account]['chats']:
if jid != 'tabbed':
self.plugin.windows[account]['chats'][jid].set_image(jid)
# Update opened groupchat windows
for jid in self.plugin.windows[account]['gc']:
if jid != 'tabbed':
self.plugin.windows[account]['gc'][jid].udpate_pixbufs()
# Update windows icons
image = self.pixbufs['online']
if image.get_storage_type() == gtk.IMAGE_ANIMATION:
pixbuf = image.get_animation().get_static_image()
elif image.get_storage_type() == gtk.IMAGE_PIXBUF:
pixbuf = image.get_pixbuf()
for win in gtk.window_list_toplevels():
win.set_icon(pixbuf)
# Update roster_window icon with the status image
self.update_status_comboxbox()
def on_show_offline_contacts_menuitem_activate(self, widget):
"""when show offline option is changed:
redraw the treeview"""
self.plugin.config['showoffline'] = 1 - self.plugin.config['showoffline']
self.plugin.send('CONFIG', None, ('GtkGui', self.plugin.config, 'GtkGui'))
self.draw_roster()
def iconCellDataFunc(self, column, renderer, model, iter, data=None):
"""When a row is added, set properties for icon renderer"""
if model.get_value(iter, 2) == 'account':
renderer.set_property('cell-background', \
self.plugin.config['accountbgcolor'])
renderer.set_property('xalign', 0)
elif model.get_value(iter, 2) == 'group':
renderer.set_property('cell-background', \
self.plugin.config['groupbgcolor'])
renderer.set_property('xalign', 0.5)
else:
renderer.set_property('cell-background', \
self.plugin.config['userbgcolor'])
renderer.set_property('xalign', 1)
renderer.set_property('width', 20)
def nameCellDataFunc(self, column, renderer, model, iter, data=None):
"""When a row is added, set properties for name renderer"""
if model.get_value(iter, 2) == 'account':
renderer.set_property('foreground', \
self.plugin.config['accounttextcolor'])
renderer.set_property('cell-background', \
self.plugin.config['accountbgcolor'])
renderer.set_property('font', self.plugin.config['accountfont'])
renderer.set_property('xpad', 0)
elif model.get_value(iter, 2) == 'group':
renderer.set_property('foreground', \
self.plugin.config['grouptextcolor'])
renderer.set_property('cell-background', \
self.plugin.config['groupbgcolor'])
renderer.set_property('font', self.plugin.config['groupfont'])
renderer.set_property('xpad', 4)
else:
jid = model.get_value(iter, 3)
account = model.get_value(iter, 4)
if jid in self.newly_added[account]:
renderer.set_property('foreground', 'green')
elif jid in self.to_be_removed[account]:
renderer.set_property('foreground', 'red')
else:
renderer.set_property('foreground', \
self.plugin.config['usertextcolor'])
renderer.set_property('cell-background', \
self.plugin.config['userbgcolor'])
renderer.set_property('font', self.plugin.config['userfont'])
renderer.set_property('xpad', 8)
def compareIters(self, model, iter1, iter2, data = None):
"""Compare two iters to sort them"""
name1 = model.get_value(iter1, 1)
name2 = model.get_value(iter2, 1)
if not name1 or not name2:
return 0
type = model.get_value(iter1, 2)
if type == 'group':
if name1 == 'Agents':
return 1
if name2 == 'Agents':
return -1
if name1.lower() < name2.lower():
return -1
if name2.lower < name1.lower():
return 1
return 0
def drag_data_get_data(self, treeview, context, selection, target_id, etime):
treeselection = treeview.get_selection()
model, iter = treeselection.get_selected()
path = model.get_path(iter)
data = ""
if len(path) == 3:
data = model.get_value(iter, 3)
selection.set(selection.target, 8, data)
def drag_data_received_data(self, treeview, context, x, y, selection, info,
etime):
merge = 0
if self.plugin.config['mergeaccounts']:
merge = 1
model = treeview.get_model()
data = selection.data
if not data:
return
drop_info = treeview.get_dest_row_at_pos(x, y)
if not drop_info:
return
path_dest, position = drop_info
if position == gtk.TREE_VIEW_DROP_BEFORE and len(path_dest) == 2 - merge\
and path_dest[1 - merge] == 0: #droped before the first group
return
if position == gtk.TREE_VIEW_DROP_BEFORE and len(path_dest) == 2 - merge:
#droped before a group : we drop it in the previous group
path_dest = (path_dest[1 - merge], path_dest[1 - merge]-1)
iter_dest = model.get_iter(path_dest)
iter_source = treeview.get_selection().get_selected()[1]
path_source = model.get_path(iter_source)
if len(path_dest) == 1 and not merge: #droped on an account
return
if path_dest[0] != path_source[0] and not merge:
#droped in another account
return
iter_group_source = model.iter_parent(iter_source)
grp_source = model.get_value(iter_group_source, 3)
if grp_source == 'Agents' or grp_source == 'not in the roster':
return
account = model.get_value(iter_dest, 4)
type_dest = model.get_value(iter_dest, 2)
if type_dest == 'group':
grp_dest = model.get_value(iter_dest, 3)
else:
grp_dest = model.get_value(model.iter_parent(iter_dest), 3)
if grp_source == grp_dest:
return
for u in self.contacts[account][data]:
u.groups.remove(grp_source)
u.groups.append(grp_dest)
self.plugin.send('UPDUSER', account, (u.jid, u.name, u.groups))
if model.iter_n_children(iter_group_source) == 1: #this was the only child
model.remove(iter_group_source)
#delete the group if it is empty (need to look for offline users too)
group_empty = True
for jid in self.contacts[account]:
if grp_source in self.contacts[account][jid][0].groups:
group_empty = False
break
if group_empty:
del self.groups[account][grp_source]
self.add_user_to_roster(data, account)
if context.action == gtk.gdk.ACTION_MOVE:
context.finish(True, True, etime)
return
def show_title(self):
start = ""
if self.nb_unread > 1:
start = "[" + str(self.nb_unread) + "] "
elif self.nb_unread == 1:
start = "* "
self.window.set_title(start + " Gajim")
def __init__(self, plugin):
self.xml = gtk.glade.XML(GTKGUI_GLADE, 'roster_window', APP)
self.window = self.xml.get_widget('roster_window')
self.tree = self.xml.get_widget('roster_treeview')
self.plugin = plugin
self.nb_unread = 0
self.add_new_contact_handler_id = False
self.service_disco_handler_id = False
self.join_gc_handler_id = False
self.new_message_menuitem_handler_id = False
self.regroup = 0
if self.plugin.config.has_key('mergeaccounts'):
self.regroup = self.plugin.config['mergeaccounts']
if self.plugin.config.has_key('saveposition'):
if self.plugin.config['saveposition']:
if self.plugin.config.has_key('x-position') and \
self.plugin.config.has_key('y-position'):
self.window.move(self.plugin.config['x-position'], \
self.plugin.config['y-position'])
if self.plugin.config.has_key('width') and \
self.plugin.config.has_key('height'):
self.window.resize(self.plugin.config['width'], \
self.plugin.config['height'])
self.window.show_all()
self.groups = {}
self.contacts = {}
self.newly_added = {}
self.to_be_removed = {}
self.popups_height = 0
for a in self.plugin.accounts.keys():
self.contacts[a] = {}
self.groups[a] = {}
self.newly_added[a] = []
self.to_be_removed[a] = []
#(icon, name, type, jid, account, editable)
model = gtk.TreeStore(gtk.Image, str, str, str, str, gobject.TYPE_BOOLEAN)
model.set_sort_func(1, self.compareIters)
model.set_sort_column_id(1, gtk.SORT_ASCENDING)
self.tree.set_model(model)
self.mkpixbufs()
liststore = gtk.ListStore(gobject.TYPE_STRING, gtk.Image, \
gobject.TYPE_STRING)
self.status_combobox = gtk.ComboBox()
self.xml.get_widget('vbox1').pack_end(self.status_combobox, False)
cell = CellRendererImage()
self.status_combobox.pack_start(cell, False)
self.status_combobox.add_attribute(cell, 'image', 1)
cell = gtk.CellRendererText()
cell.set_property('xpad', 5) # padding for status text
self.status_combobox.pack_start(cell, True)
self.status_combobox.add_attribute(cell, 'text', 0)
for status in ['online', 'away', 'xa', 'dnd', 'invisible', 'offline']:
if status == 'dnd':
status_better = 'Busy'
elif status == 'xa':
status_better = 'Extended Away'
else:
status_better = status.capitalize()
iter = liststore.append([status_better, self.pixbufs[status], status])
self.status_combobox.show_all()
self.status_combobox.set_model(liststore)
self.status_combobox.set_active(5)
showOffline = self.plugin.config['showoffline']
self.xml.get_widget('show_offline_contacts_menuitem').set_active(showOffline)
#columns
#this col has two cells: first one img, second one text
col = gtk.TreeViewColumn()
render_pixbuf = CellRendererImage()
col.pack_start(render_pixbuf, expand = False)
col.add_attribute(render_pixbuf, 'image', 0)
col.set_cell_data_func(render_pixbuf, self.iconCellDataFunc, None)
render_text = gtk.CellRendererText()
render_text.connect('edited', self.on_cell_edited)
render_text.connect('editing-canceled', self.on_editing_canceled)
col.pack_start(render_text, expand = True)
col.add_attribute(render_text, 'text', 1)
col.add_attribute(render_text, 'editable', 5) #5th column?? I don't understand :(
col.set_cell_data_func(render_text, self.nameCellDataFunc, None)
self.tree.append_column(col)
#do not show gtk arrows workaround
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
TARGETS = [('MY_TREE_MODEL_ROW', gtk.TARGET_SAME_WIDGET, 0)]
self.tree.enable_model_drag_source( gtk.gdk.BUTTON1_MASK, TARGETS,
gtk.gdk.ACTION_DEFAULT| gtk.gdk.ACTION_MOVE)
self.tree.enable_model_drag_dest(TARGETS, gtk.gdk.ACTION_DEFAULT)
self.tree.connect("drag_data_get", self.drag_data_get_data)
self.tree.connect("drag_data_received", self.drag_data_received_data)
self.xml.signal_autoconnect(self)
self.id_signal_cb = self.status_combobox.connect('changed',\
self.on_status_combobox_changed)
self.hidden_lines = self.plugin.config['hiddenlines'].split('\t')
self.draw_roster()
if len(self.plugin.accounts) == 0: # if no account
self.plugin.windows['account_modification'] = \
Account_modification_window(self.plugin, {})