gajim-plural/src/common/connection.py

999 lines
31 KiB
Python
Raw Normal View History

## common/connection.py
##
## Gajim Team:
## - Yann Le Boulanger <asterix@lagaule.org>
## - Vincent Hanquez <tab@snarc.org>
## - Nikos Kouremenos <nkour@jabber.org>
##
## 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 sys
import os
import time
import sre
import traceback
from calendar import timegm
2005-04-26 20:45:54 +02:00
import common.xmpp
2005-05-27 07:56:17 +02:00
from common import helpers
2005-04-14 09:42:26 +02:00
from common import gajim
from common import GnuPG
USE_GPG = GnuPG.USE_GPG
from common import i18n
_ = i18n._
2005-04-30 10:48:50 +02:00
STATUS_LIST = ['offline', 'connecting', 'online', 'chat', 'away', 'xa', 'dnd',
'invisible']
distro_info = {
2005-04-19 15:14:41 +02:00
'Arch Linux': '/etc/arch-release',
'Aurox Linux': '/etc/aurox-release',
'Conectiva Linux': '/etc/conectiva-release',
'Debian GNU/Linux': '/etc/debian_release',
'Debian GNU/Linux': '/etc/debian_version',
'Fedora Linux': '/etc/fedora-release',
'Gentoo Linux': '/etc/gentoo-release',
'Linux from Scratch': '/etc/lfs-release',
'Mandrake Linux': '/etc/mandrake-release',
'Slackware Linux': '/etc/slackware-release',
'Slackware Linux': '/etc/slackware-version',
'Solaris/Sparc': '/etc/release',
'SUSE Linux': '/etc/SuSE-release',
2005-04-19 15:14:41 +02:00
'Sun JDS': '/etc/sun-release',
'PLD Linux': '/etc/pld-release',
'Yellow Dog Linux': '/etc/yellowdog-release',
# many distros use the /etc/redhat-release for compatibility
# so Redhat is the last
2005-04-19 15:14:41 +02:00
'Redhat Linux': '/etc/redhat-release'
}
def get_os_info():
2005-04-17 01:15:03 +02:00
if os.name == 'nt':
2005-04-19 15:14:41 +02:00
win_version = {
(1, 4, 0): '95',
(1, 4, 10): '98',
(1, 4, 90): 'ME',
(2, 4, 0): 'NT',
(2, 5, 0): '2000',
2005-04-17 01:15:03 +02:00
(2, 5, 1): 'XP'
2005-04-19 15:14:41 +02:00
}[ os.sys.getwindowsversion()[3],
os.sys.getwindowsversion()[0],
2005-04-17 01:15:03 +02:00
os.sys.getwindowsversion()[1] ]
return 'Windows' + ' ' + win_version
2005-04-26 20:45:54 +02:00
elif os.name == 'posix':
executable = 'lsb_release'
params = ' --id --codename --release --short'
for path in os.environ['PATH'].split(':'):
full_path_to_executable = os.path.join(path, executable)
if os.path.exists(full_path_to_executable):
command = executable + params
child_stdin, child_stdout = os.popen2(command)
output = child_stdout.readline().strip()
child_stdout.close()
child_stdin.close()
# some distros put n/a in places so remove them
2005-05-20 18:46:07 +02:00
pattern = sre.compile(r' n/a', sre.IGNORECASE)
output = sre.sub(pattern, '', output)
return output
# lsb_release executable not available, so parse files
2005-04-19 15:14:41 +02:00
for distro_name in distro_info:
path_to_file = distro_info[distro_name]
if os.path.exists(path_to_file):
fd = open(path_to_file)
text = fd.read().strip()
fd.close()
if path_to_file.endswith('version'):
2005-04-19 15:14:41 +02:00
text = distro_name + ' ' + text
elif path_to_file.endswith('aurox-release'): # file doesn't have version
text = distro_name
elif path_to_file.endswith('lfs-release'): # file just has version
text = distro_name + ' ' + text
return text
return 'N/A'
2005-04-19 15:14:41 +02:00
class Connection:
"""Connection class"""
def __init__(self, name):
# dict of function to be called for each event
self.handlers = {'ROSTER': [], 'WARNING': [], 'ERROR': [], 'STATUS': [],
'NOTIFY': [], 'MSG': [], 'MSGERROR': [], 'MSGSENT': [] ,
'SUBSCRIBED': [], 'UNSUBSCRIBED': [], 'SUBSCRIBE': [],
'AGENT_INFO': [], 'REGISTER_AGENT_INFO': [], 'AGENT_INFO_ITEMS': [],
'AGENT_INFO_INFO': [], 'QUIT': [], 'ACC_OK': [], 'MYVCARD': [],
'OS_INFO': [], 'VCARD': [], 'GC_MSG': [], 'GC_SUBJECT': [],
2005-05-10 18:53:28 +02:00
'GC_CONFIG': [], 'BAD_PASSPHRASE': [], 'ROSTER_INFO': [],
'ERROR_ANSWER': []}
self.name = name
self.connected = 0 # offline
2005-04-26 20:45:54 +02:00
self.connection = None # xmpppy instance
self.gpg = None
self.status = ''
self.myVCardID = []
self.on_purpose = False
self.password = gajim.config.get_per('accounts', name, 'password')
if USE_GPG:
self.gpg = GnuPG.GnuPG()
gajim.config.set('usegpg', True)
else:
gajim.config.set('usegpg', False)
# END __init__
def dispatch(self, event, data):
if not event in self.handlers:
return
for handler in self.handlers[event]:
handler(self.name, data)
2005-04-26 20:45:54 +02:00
def _discover(self, ns, jid, node = None): #FIXME: this is in features.py but it is blocking
if not self.connection:
return
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'get', to = jid, queryNS = ns)
if node:
iq.setQuerynode(node)
self.connection.send(iq)
def discoverItems(self, jid, node = None):
'''According to JEP-0030: jid is mandatory,
name, node, action is optional.'''
self._discover(common.xmpp.NS_DISCO_ITEMS, jid, node)
def discoverInfo(self, jid, node = None):
'''According to JEP-0030:
For identity: category, name is mandatory, type is optional.
For feature: var is mandatory'''
self._discover(common.xmpp.NS_DISCO_INFO, jid, node)
def _vCardCB(self, con, vc):
"""Called when we recieve a vCard
Parse the vCard and send it to plugins"""
frm = vc.getFrom()
if frm:
frm = frm.getStripped()
else:
name = gajim.config.get_per('accounts', self.name, 'name')
hostname = gajim.config.get_per('accounts', self.name, 'hostname')
frm = name + '@' + hostname
vcard = {'jid': frm}
2005-04-26 20:45:54 +02:00
if vc.getTag('vCard').getNamespace() == common.xmpp.NS_VCARD:
card = vc.getChildren()[0]
for info in card.getChildren():
if info.getChildren() == []:
vcard[info.getName()] = info.getData()
else:
vcard[info.getName()] = {}
for c in info.getChildren():
vcard[info.getName()][c.getName()] = c.getData()
if vc.getID() in self.myVCardID:
self.myVCardID.remove(vc.getID())
self.dispatch('MYVCARD', vcard)
else:
self.dispatch('VCARD', vcard)
2005-04-26 20:45:54 +02:00
def _messageCB(self, con, msg):
"""Called when we recieve a message"""
2005-05-20 13:23:08 +02:00
msgtxt = msg.getBody()
if not msg.getTag('body'): #no <body>
return
2005-04-12 14:23:08 +02:00
mtype = msg.getType()
tim = msg.getTimestamp()
tim = time.strptime(tim, '%Y%m%dT%H:%M:%S')
tim = time.localtime(timegm(tim))
2005-04-26 20:45:54 +02:00
xtags = msg.getTags('x')
encTag = None
decmsg = ''
for xtag in xtags:
2005-04-26 20:45:54 +02:00
if xtag.getNamespace() == common.xmpp.NS_ENCRYPTED:
encTag = xtag
break
if encTag and USE_GPG:
#decrypt
encmsg = encTag.getData()
keyID = gajim.config.get_per('accounts', self.name, 'keyid')
if keyID:
decmsg = self.gpg.decrypt(encmsg, keyID)
if decmsg:
msgtxt = decmsg
2005-04-12 14:23:08 +02:00
if mtype == 'error':
self.dispatch('MSGERROR', (str(msg.getFrom()), \
msg.getErrorCode(), msg.getError(), msgtxt, tim))
2005-04-12 14:23:08 +02:00
elif mtype == 'groupchat':
subject = msg.getSubject()
if subject:
self.dispatch('GC_SUBJECT', (str(msg.getFrom()), subject))
else:
self.dispatch('GC_MSG', (str(msg.getFrom()), msgtxt, tim))
2005-04-16 11:36:18 +02:00
gajim.logger.write('gc', msgtxt, str(msg.getFrom()), tim = tim)
else:
2005-04-16 11:36:18 +02:00
gajim.logger.write('incoming', msgtxt, str(msg.getFrom()), tim = tim)
self.dispatch('MSG', (str(msg.getFrom()), msgtxt, tim))
# END messageCB
def _presenceCB(self, con, prs):
"""Called when we recieve a presence"""
2005-04-26 20:45:54 +02:00
# if prs.getXNode(common.xmpp.NS_DELAY): return
who = str(prs.getFrom())
prio = prs.getPriority()
if not prio:
prio = 0
2005-04-12 14:23:08 +02:00
ptype = prs.getType()
if ptype == None: ptype = 'available'
gajim.log.debug('PresenceCB : %s' % ptype)
2005-04-26 20:45:54 +02:00
xtags = prs.getTags('x')
sigTag = None
keyID = ''
status = prs.getStatus()
for xtag in xtags:
2005-04-26 20:45:54 +02:00
if xtag.getNamespace() == common.xmpp.NS_SIGNED:
sigTag = xtag
break
if sigTag and USE_GPG:
#verify
sigmsg = sigTag.getData()
keyID = self.gpg.verify(status, sigmsg)
2005-04-12 14:23:08 +02:00
if ptype == 'available':
show = prs.getShow()
if not show:
show = 'online'
2005-04-12 14:23:08 +02:00
elif ptype == 'unavailable':
2005-04-16 11:36:18 +02:00
show = 'offline'
2005-04-12 14:23:08 +02:00
elif ptype == 'subscribe':
gajim.log.debug('subscribe request from %s' % who)
if gajim.config.get('alwaysauth') or who.find("@") <= 0:
if self.connection:
2005-04-26 20:45:54 +02:00
self.connection.send(common.xmpp.Presence(who, 'subscribed'))
if who.find("@") <= 0:
self.dispatch('NOTIFY', (prs.getFrom().getStripped(), \
'offline', 'offline', prs.getFrom().getResource(), prio, \
keyID, None, None, None, None, None, None))
else:
if not status:
status = _('I would like to add you to my roster.')
self.dispatch('SUBSCRIBE', (who, status))
2005-04-12 14:23:08 +02:00
elif ptype == 'subscribed':
jid = prs.getFrom()
self.dispatch('SUBSCRIBED', (jid.getStripped(), jid.getResource()))
self.dispatch('UPDUSER', (jid.getStripped(), jid.getNode(), \
2005-04-14 09:42:26 +02:00
['General']))
#BE CAREFUL : no con.updateRosterItem() in a callback
gajim.log.debug('we are now subscribed to %s' % who)
2005-04-12 14:23:08 +02:00
elif ptype == 'unsubscribe':
gajim.log.debug('unsubscribe request from %s' % who)
2005-04-12 14:23:08 +02:00
elif ptype == 'unsubscribed':
gajim.log.debug('we are now unsubscribed to %s' % who)
self.dispatch('UNSUBSCRIBED', prs.getFrom().getStripped())
2005-04-12 14:23:08 +02:00
elif ptype == 'error':
errmsg = prs.getError()
errcode = prs.getErrorCode()
if errcode == '409': #conflict : Nick Conflict
self.dispatch('ERROR', errmsg)
elif errcode == '502': # Internal Timeout:
self.dispatch('NOTIFY', (prs.getFrom().getStripped(),
'error', errmsg, prs.getFrom().getResource(), prio, keyID,
prs.getRole(), prs.getAffiliation(), prs.getJid(),
prs.getReason(), prs.getActor(), prs.getStatusCode()))
else:
self.dispatch('ERROR_ANSWER', (prs.getFrom().getStripped(), errmsg,
errcode))
2005-04-16 11:36:18 +02:00
if ptype == 'available' or ptype == 'unavailable':
gajim.logger.write('status', status, prs.getFrom().getStripped(), show)
self.dispatch('NOTIFY', (prs.getFrom().getStripped(), show, status,
prs.getFrom().getResource(), prio, keyID, prs.getRole(),
prs.getAffiliation(), prs.getJid(), prs.getReason(),
2005-04-16 11:36:18 +02:00
prs.getActor(), prs.getStatusCode()))
# END presenceCB
2005-04-26 20:45:54 +02:00
def _disconnectedCB(self):
"""Called when we are disconnected"""
gajim.log.debug('disconnectedCB')
if not self.connection:
return
self.connected = 0
self.dispatch('STATUS', 'offline')
self.connection = None
if not self.on_purpose:
2005-05-27 14:11:07 +02:00
self.dispatch('ERROR', 'You have been disconected from %s' % self.name)
self.on_purpose = False
# END disconenctedCB
def _rosterSetCB(self, con, iq_obj):
gajim.log.debug('rosterSetCB')
2005-04-26 20:45:54 +02:00
for item in iq_obj.getTag('query').getChildren():
jid = item.getAttr('jid')
name = item.getAttr('name')
sub = item.getAttr('subscription')
ask = item.getAttr('ask')
groups = []
for group in item.getTags('group'):
groups.append(group.getData())
self.dispatch('ROSTER_INFO', (jid, name, sub, ask, groups))
raise common.xmpp.NodeProcessed
def _BrowseResultCB(self, con, iq_obj):
gajim.log.debug('BrowseResultCB')
identities, features, items = [], [], []
2005-05-07 14:24:19 +02:00
for q in iq_obj.getChildren():
if q.getNamespace() != common.xmpp.NS_BROWSE:
continue
attr = {}
for key in q.getAttrs().keys():
attr[key.encode('utf8')] = q.getAttr(key).encode('utf8')
identities = [attr]
for node in q.getChildren():
if node.getName() == 'ns':
features.append(node.getData())
else:
infos = {}
for key in node.getAttrs().keys():
infos[key.encode('utf8')] = node.getAttr(key).encode('utf8')
infos['category'] = node.getName()
items.append(infos)
jid = str(iq_obj.getFrom())
self.dispatch('AGENT_INFO', (jid, identities, features, items))
def _DiscoverItemsCB(self, con, iq_obj):
gajim.log.debug('DiscoverItemsCB')
2005-04-26 20:45:54 +02:00
q = iq_obj.getTag('query')
node = q.getAttr('node')
if not node:
node = ''
qp = iq_obj.getQueryPayload()
items = []
if not qp:
qp = []
for i in qp:
attr = {}
2005-04-26 20:45:54 +02:00
for key in i.getAttrs():
attr[key.encode('utf8')] = i.getAttrs()[key].encode('utf8')
items.append(attr)
jid = str(iq_obj.getFrom())
self.dispatch('AGENT_INFO_ITEMS', (jid, node, items))
def _DiscoverInfoErrorCB(self, con, iq_obj):
gajim.log.debug('DiscoverInfoErrorCB')
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(to = iq_obj.getFrom(), typ = 'get', queryNS =\
common.xmpp.NS_AGENTS)
con.send(iq)
def _DiscoverInfoCB(self, con, iq_obj):
gajim.log.debug('DiscoverInfoCB')
# According to JEP-0030:
# For identity: category, name is mandatory, type is optional.
# For feature: var is mandatory
identities, features = [], []
2005-04-26 20:45:54 +02:00
q = iq_obj.getTag('query')
node = q.getAttr('node')
if not node:
node = ''
2005-04-26 20:45:54 +02:00
qc = iq_obj.getQueryChildren()
if not qc:
qc = []
for i in qc:
if i.getName() == 'identity':
attr = {}
2005-04-26 20:45:54 +02:00
for key in i.getAttrs().keys():
attr[key.encode('utf8')] = i.getAttr(key).encode('utf8')
identities.append(attr)
elif i.getName() == 'feature':
features.append(i.getAttr('var'))
jid = str(iq_obj.getFrom())
if not identities:
self.connection.send(common.xmpp.Iq(typ = 'get', queryNS = \
common.xmpp.NS_AGENTS))
else:
2005-04-24 17:47:08 +02:00
self.dispatch('AGENT_INFO_INFO', (jid, node, identities, features))
2005-04-26 20:45:54 +02:00
self.discoverItems(jid, node)
def _VersionCB(self, con, iq_obj):
gajim.log.debug('VersionCB')
iq_obj = iq_obj.buildReply('result')
qp = iq_obj.getTag('query')
2005-04-26 23:33:01 +02:00
qp.setTagData('name', 'Gajim')
qp.setTagData('version', gajim.version)
send_os = gajim.config.get('send_os_info')
if send_os:
2005-04-26 23:33:01 +02:00
qp.setTagData('os', get_os_info())
con.send(iq_obj)
raise common.xmpp.NodeProcessed
def _VersionResultCB(self, con, iq_obj):
gajim.log.debug('VersionResultCB')
client_info = ''
os_info = ''
qp = iq_obj.getTag('query')
if qp.getTag('name'):
client_info += qp.getTag('name').getData()
if qp.getTag('version'):
client_info += ' ' + qp.getTag('version').getData()
if qp.getTag('os'):
os_info += qp.getTag('os').getData()
jid = iq_obj.getFrom().getStripped()
resource = iq_obj.getFrom().getResource()
self.dispatch('OS_INFO', (jid, resource, client_info, os_info))
def _MucOwnerCB(self, con, iq_obj):
gajim.log.debug('MucOwnerCB')
qp = iq_obj.getQueryPayload()
node = None
for q in qp:
2005-04-26 20:45:54 +02:00
if q.getNamespace() == common.xmpp.NS_DATA:
node = q
if not node:
return
# Parse the form
dic = {}
tag = node.getTag('title')
if tag:
dic['title'] = tag.getData()
tag = node.getTag('instructions')
if tag:
dic['instructions'] = tag.getData()
i = 0
for child in node.getChildren():
if child.getName() != 'field':
continue
var = child.getAttr('var')
ctype = child.getAttr('type')
label = child.getAttr('label')
if not var and ctype != 'fixed': # We must have var if type != fixed
continue
dic[i] = {}
if var:
dic[i]['var'] = var
if ctype:
dic[i]['type'] = ctype
if label:
dic[i]['label'] = label
tags = child.getTags('value')
if len(tags):
dic[i]['values'] = []
for tag in tags:
data = tag.getData()
if ctype == 'boolean':
2005-04-24 18:14:50 +02:00
if data in ['yes', 'true', 'assent', '1']:
data = True
else:
data = False
dic[i]['values'].append(data)
tag = child.getTag('desc')
if tag:
dic[i]['desc'] = tag.getData()
option_tags = child.getTags('option')
if len(option_tags):
dic[i]['options'] = {}
j = 0
for option_tag in option_tags:
dic[i]['options'][j] = {}
label = option_tag.getAttr('label')
if label:
dic[i]['options'][j]['label'] = label
tags = option_tag.getTags('value')
dic[i]['options'][j]['values'] = []
for tag in tags:
dic[i]['options'][j]['values'].append(tag.getData())
j += 1
i += 1
self.dispatch('GC_CONFIG', (str(iq_obj.getFrom()), dic))
def _MucErrorCB(self, con, iq_obj):
gajim.log.debug('MucErrorCB')
jid = str(iq_obj.getFrom())
errmsg = iq_obj.getError()
errcode = iq_obj.getErrorCode()
self.dispatch('MSGERROR', (jid, errcode, errmsg))
2005-05-08 19:00:41 +02:00
def _getRosterCB(self, con, iq_obj):
roster = self.connection.getRoster().getRaw()
if not roster :
roster = {}
name = gajim.config.get_per('accounts', self.name, 'name')
hostname = gajim.config.get_per('accounts', self.name, 'hostname')
if roster.has_key(name + '@' + hostname):
del roster[name + '@' + hostname]
self.dispatch('ROSTER', roster)
2005-05-10 18:53:28 +02:00
def _ErrorCB(self, con, iq_obj):
errmsg = iq_obj.getError()
errcode = iq_obj.getErrorCode()
jid_from = str(iq_obj.getFrom())
self.dispatch('ERROR_ANSWER', (jid_from, errmsg, errcode))
def connect(self):
"""Connect and authenticate to the Jabber server"""
name = gajim.config.get_per('accounts', self.name, 'name')
hostname = gajim.config.get_per('accounts', self.name, 'hostname')
resource = gajim.config.get_per('accounts', self.name, 'resource')
2005-05-09 21:52:43 +02:00
usetls = gajim.config.get_per('accounts', self.name, 'usetls')
#create connection if it doesn't already exist
if self.connection:
return self.connection
self.connected = 1
if gajim.config.get_per('accounts', self.name, 'use_proxy'):
proxy = {'host': gajim.config.get_per('accounts', self.name,
'proxyhost')}
proxy['port'] = gajim.config.get_per('accounts', self.name,
'proxyport')
proxy['user'] = gajim.config.get_per('accounts', self.name,
'proxyuser')
proxy['password'] = gajim.config.get_per('accounts', self.name,
'proxypass')
else:
proxy = None
if gajim.config.get('verbose'):
2005-05-08 19:00:41 +02:00
con = common.xmpp.Client(hostname)
else:
2005-05-03 10:06:59 +02:00
con = common.xmpp.Client(hostname, debug = [])
#debug = [common.jabber.DBG_ALWAYS], log = sys.stderr, \
#connection=common.xmlstream.TCP_SSL, port=5223, proxy = proxy)
common.xmpp.dispatcher.DefaultTimeout = 45
con.UnregisterDisconnectHandler(con.DisconnectHandler)
2005-04-26 20:45:54 +02:00
con.RegisterDisconnectHandler(self._disconnectedCB)
con_type = con.connect(proxy=proxy, tls=usetls) #FIXME: blocking
if not con_type:
gajim.log.debug('Couldn\'t connect to %s' % hostname)
self.connected = 0
self.dispatch('STATUS', 'offline')
self.dispatch('ERROR', _('Couldn\'t connect to %s') % hostname)
return None
2005-04-26 20:45:54 +02:00
con.RegisterHandler('message', self._messageCB)
con.RegisterHandler('presence', self._presenceCB)
con.RegisterHandler('iq', self._vCardCB, 'result',\
common.xmpp.NS_VCARD)
con.RegisterHandler('iq', self._rosterSetCB, 'set',\
common.xmpp.NS_ROSTER)
con.RegisterHandler('iq', self._BrowseResultCB, 'result',\
common.xmpp.NS_BROWSE)
con.RegisterHandler('iq', self._DiscoverItemsCB, 'result',\
common.xmpp.NS_DISCO_ITEMS)
con.RegisterHandler('iq', self._DiscoverInfoCB, 'result',\
common.xmpp.NS_DISCO_INFO)
con.RegisterHandler('iq', self._DiscoverInfoErrorCB, 'error',\
common.xmpp.NS_DISCO_INFO)
con.RegisterHandler('iq', self._VersionCB, 'get',\
common.xmpp.NS_VERSION)
con.RegisterHandler('iq', self._VersionResultCB, 'result',\
common.xmpp.NS_VERSION)
con.RegisterHandler('iq', self._MucOwnerCB, 'result',\
common.xmpp.NS_MUC_OWNER)
2005-05-08 19:00:41 +02:00
con.RegisterHandler('iq', self._getRosterCB, 'result',\
common.xmpp.NS_ROSTER)
2005-05-10 18:53:28 +02:00
con.RegisterHandler('iq', self._ErrorCB, 'error')
2005-04-26 20:45:54 +02:00
gajim.log.debug('Connected to server')
try:
auth = con.auth(name, self.password, resource,
not gajim.config.get_per('accounts', self.name, 'force_nonSASL')) #FIXME: blocking
except IOError: #probably a timeout
self.connected = 0
self.dispatch('STATUS', 'offline')
self.dispatch('ERROR', _('Couldn\'t connect to %s') % hostname)
return None
if auth:
2005-05-08 19:00:41 +02:00
con.initRoster()
self.connected = 2
return con
else:
gajim.log.debug('Couldn\'t authenticate to %s' % hostname)
self.connected = 0
self.dispatch('STATUS', 'offline')
self.dispatch('ERROR', _('Authentication failed with %s, check your login and password') % hostname)
return None
# END connect
def register_handler(self, event, function):
if event in self.handlers:
self.handlers[event].append(function)
def unregister_handler(self, event, function):
if event in self.handlers:
if function in self.handlers[event]:
self.handlers[event].remove(function)
def quit(self, kill_core):
if kill_core:
if self.connected > 1:
self.connected = 0
self.connection.disconnect('Disconnected')
return
def ask_roster(self):
roster = {}
if self.connection:
roster = self.connection.getRoster().getRaw()
return roster
def change_status(self, show, msg):
if not show in STATUS_LIST:
return -1
signed = ''
keyID = gajim.config.get_per('accounts', self.name, 'keyid')
if keyID and USE_GPG:
if not msg:
2005-05-27 00:02:01 +02:00
lowered_uf_status_msg = helpers.get_uf_show(show).lower()
msg = "I'm %s" % lowered_uf_status_msg
signed = self.gpg.sign(msg, keyID)
if signed == 'BAD_PASSPHRASE':
signed = ''
if self.connected < 2:
self.dispatch('BAD_PASSPHRASE', ())
self.status = msg
if show != 'offline' and not self.connected:
self.connection = self.connect()
if self.connected == 2:
self.connected = STATUS_LIST.index(show)
#send our presence
2005-04-12 14:23:08 +02:00
ptype = 'available'
if show == 'invisible':
2005-04-12 14:23:08 +02:00
ptype = 'invisible'
2005-04-17 13:49:39 +02:00
prio = str(gajim.config.get_per('accounts', self.name, 'priority'))
p = common.xmpp.Presence(typ = ptype, priority = prio, show = show)
if msg:
p.setStatus(msg)
2005-04-26 20:45:54 +02:00
if signed:
2005-04-26 23:33:01 +02:00
p.setTag(common.xmpp.NS_SIGNED + ' x').setData(signed)
2005-04-26 20:45:54 +02:00
self.connection.send(p)
self.dispatch('STATUS', show)
#ask our VCard
iq = self.request_vcard(None)
2005-04-26 20:45:54 +02:00
self.myVCardID.append(iq.getID())
elif show == 'offline' and self.connected:
self.connected = 0
2005-05-04 21:20:02 +02:00
if self.connection:
self.on_purpose = True
p = common.xmpp.Presence(typ = 'unavailable')
if msg:
p.setStatus(msg)
self.connection.send(p)
2005-05-04 21:20:02 +02:00
self.connection.disconnect()
self.dispatch('STATUS', 'offline')
self.connection = None
elif show != 'offline' and self.connected:
self.connected = STATUS_LIST.index(show)
2005-04-12 14:23:08 +02:00
ptype = 'available'
if show == 'invisible':
2005-04-12 14:23:08 +02:00
ptype = 'invisible'
2005-04-17 13:49:39 +02:00
prio = str(gajim.config.get_per('accounts', self.name, 'priority'))
p = common.xmpp.Presence(typ = ptype, priority = prio, show = show)
if msg:
p.setStatus(msg)
if signed:
p.setTag(common.xmpp.NS_SIGNED + ' x').setData(signed)
2005-04-26 20:45:54 +02:00
self.connection.send(p)
self.dispatch('STATUS', show)
def send_message(self, jid, msg, keyID):
2005-04-14 09:42:26 +02:00
if not self.connection:
return
msgtxt = msg
msgenc = ''
if keyID and USE_GPG:
#encrypt
2005-05-27 16:16:34 +02:00
msgenc = self.gpg.encrypt(msg, [keyID])
if msgenc: msgtxt = _('[this message is encrypted]')
2005-04-26 20:45:54 +02:00
msg_iq = common.xmpp.Message(to = jid, body = msgtxt, typ = 'chat')
if msgenc:
2005-04-26 20:45:54 +02:00
msg_iq.setTag(common.xmpp.NS_ENCRYPTED + ' x').setData(msgenc)
self.connection.send(msg_iq)
2005-04-16 11:36:18 +02:00
gajim.logger.write('outgoing', msg, jid)
self.dispatch('MSGSENT', (jid, msg, keyID))
2005-04-14 09:42:26 +02:00
def request_subscription(self, jid, msg):
if not self.connection:
return
gajim.log.debug('subscription request for %s' % jid)
2005-04-26 20:45:54 +02:00
pres = common.xmpp.Presence(jid, 'subscribe')
if not msg:
msg = _('I would like to add you to my roster.')
pres.setStatus(msg)
self.connection.send(pres)
2005-04-14 09:42:26 +02:00
def send_authorization(self, jid):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
self.connection.send(common.xmpp.Presence(jid, 'subscribed'))
2005-04-14 09:42:26 +02:00
def refuse_authorization(self, jid):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
self.connection.send(common.xmpp.Presence(jid, 'unsubscribed'))
def unsubscribe(self, jid):
if not self.connection:
return
delauth = gajim.config.get('delauth')
delroster = gajim.config.get('delroster')
if delauth:
2005-04-26 20:45:54 +02:00
self.connection.getRoster().Unsubscribe(jid)
if delroster:
2005-04-26 20:45:54 +02:00
self.connection.getRoster().delItem(jid)
def _continue_unsubscribe(self, con, iq_obj, agent):
self.connection.getRoster().delItem(agent)
self.dispatch('AGENT_REMOVED', agent)
2005-05-08 19:33:08 +02:00
def unsubscribe_agent(self, agent):
if not self.connection:
return
iq = common.xmpp.Iq('set', common.xmpp.NS_REGISTER, to = agent)
iq.getTag('query').setTag('remove')
self.connection.SendAndCallForResponse(iq, self._continue_unsubscribe,
2005-05-08 19:33:08 +02:00
{'agent': agent})
return
def update_user(self, jid, name, groups):
if self.connection:
2005-04-30 10:48:50 +02:00
self.connection.getRoster().setItem(jid = jid, name = name,
2005-04-26 20:45:54 +02:00
groups = groups)
def request_agents(self, jid, node):
if self.connection:
self.connection.send(common.xmpp.Iq(to = jid, typ = 'get',
queryNS = common.xmpp.NS_BROWSE))
2005-04-26 20:45:54 +02:00
self.discoverInfo(jid, node)
def _receive_register_agent_info(self, con, iq_obj, agent):
if not common.xmpp.isResultNode(iq_obj):
return
iq = common.xmpp.Iq('get', common.xmpp.NS_REGISTER, to = agent)
df = iq_obj.getTag('query', namespace = common.xmpp.NS_REGISTER).\
getTag('x', namespace = common.xmpp.NS_DATA)
if df:
df = common.xmpp.DataForm(node = df)
self.dispatch('REGISTER_AGENT_INFO', (agent, df.asDict()))
return
df = common.xmpp.DataForm(typ = 'form')
for i in iq_obj.getQueryPayload():
if type(i) != type(iq):
pass
elif i.getName() == 'instructions':
df.addInstructions(i.getData())
else:
df.setField(i.getName()).setValue(i.getData())
self.dispatch('REGISTER_AGENT_INFO', (agent, df.asDict()))
return
def request_register_agent_info(self, agent):
if not self.connection:
return None
iq = common.xmpp.Iq('get', common.xmpp.NS_REGISTER, to = agent)
self.connection.SendAndCallForResponse(iq,
self._receive_register_agent_info, {'agent': agent})
return
2005-04-26 20:45:54 +02:00
def register_agent(self, agent, info):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
common.xmpp.features.register(self.connection, agent, info) # FIXME: Blocking
def new_account(self, name, config):
# If a connection already exist we cannot create a new account
if self.connection:
return
if config['use_proxy']:
proxy = {'host': config['proxyhost'], 'port': config['proxyport'],
'user': config['proxyuser'], 'password': config['proxypass']}
else:
proxy = None
c = common.xmpp.Client(server = config['hostname'], debug = [])
con_type = c.connect(proxy = proxy)
if not con_type:
gajim.log.debug('Couldn\'t connect to %s' % config['hostname'])
self.dispatch('ERROR', _('Couldn\'t connect to ') + config['hostname'])
return False
else:
2005-05-18 14:02:27 +02:00
gajim.log.debug('Connected to server')
req = common.xmpp.features.getRegInfo(c, config['hostname']).asDict() # FIXME! This blocks!
req['username'] = config['name']
req['password'] = config['password']
if not common.xmpp.features.register(c, config['hostname'], req): #FIXME: error
2005-05-09 12:34:47 +02:00
self.dispatch('ERROR', _('Error: ') + c.lastErr)
else:
2005-05-09 12:34:47 +02:00
self.name = name
self.connected = 0
self.password = config['password']
2005-05-09 12:34:47 +02:00
if USE_GPG:
self.gpg = GnuPG.GnuPG()
gajim.config.set('usegpg', True)
else:
gajim.config.set('usegpg', False)
gajim.connections[name] = self
self.dispatch('ACC_OK', (name, config))
def account_changed(self, new_name):
self.name = new_name
def request_os_info(self, jid, resource):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(to=jid + '/' + resource, typ = 'get', queryNS =\
common.xmpp.NS_VERSION)
self.connection.send(iq)
def request_vcard(self, jid = None):
'''request the VCARD and return the iq'''
if not self.connection:
return
iq = common.xmpp.Iq(typ = 'get')
if jid:
iq.setTo(jid)
2005-04-26 20:45:54 +02:00
iq.setTag(common.xmpp.NS_VCARD + ' vCard')
self.connection.send(iq)
return iq
#('VCARD', {entry1: data, entry2: {entry21: data, ...}, ...})
def send_vcard(self, vcard):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'set')
iq2 = iq.setTag(common.xmpp.NS_VCARD + ' vCard')
for i in vcard.keys():
if i != 'jid':
if type(vcard[i]) == type({}):
2005-04-26 20:45:54 +02:00
iq3 = iq2.addChild(i)
for j in vcard[i].keys():
2005-04-26 20:45:54 +02:00
iq3.addChild(j).setData(vcard[i][j])
else:
2005-04-26 20:45:54 +02:00
iq2.addChild(i).setData(vcard[i])
self.connection.send(iq)
2005-04-12 14:23:08 +02:00
def send_agent_status(self, agent, ptype):
if not self.connection:
return
2005-04-12 14:23:08 +02:00
if not ptype:
ptype = 'available';
2005-04-26 20:45:54 +02:00
p = common.xmpp.Presence(to = agent, typ = ptype)
self.connection.send(p)
2005-04-18 14:17:43 +02:00
def join_gc(self, nick, room, server, password):
if not self.connection:
return
p = common.xmpp.Presence(to = '%s@%s/%s' % (room, server, nick),
show = STATUS_LIST[self.connected], status = self.status)
t = p.setTag(common.xmpp.NS_MUC + ' x')
if password:
t.setTagData('password', password)
self.connection.send(p)
def send_gc_message(self, jid, msg):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
msg_iq = common.xmpp.Message(jid, msg, typ = 'groupchat')
self.connection.send(msg_iq)
self.dispatch('MSGSENT', (jid, msg))
def send_gc_subject(self, jid, subject):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
msg_iq = common.xmpp.Message(jid,typ = 'groupchat', subject = subject)
self.connection.send(msg_iq)
def request_gc_config(self, room_jid):
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'get', queryNS = common.xmpp.NS_MUC_OWNER,\
to = room_jid)
self.connection.send(iq)
def send_gc_status(self, nick, jid, show, status):
if not self.connection:
return
if show == 'offline':
2005-04-12 14:23:08 +02:00
ptype = 'unavailable'
show = None
else:
2005-04-12 14:23:08 +02:00
ptype = 'available'
2005-04-26 20:45:54 +02:00
self.connection.send(common.xmpp.Presence(to = '%s/%s' % (jid, nick), \
typ = ptype, show = show, status = status))
def gc_set_role(self, room_jid, nick, role):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'set', to = room_jid, queryNS =\
common.xmpp.NS_MUC_ADMIN)
item = iq.getTag('query').setTag('item')
item.setAttr('nick', nick)
item.setAttr('role', role)
self.connection.send(iq)
def gc_set_affiliation(self, room_jid, jid, affiliation):
if not self.connection:
return
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'set', to = room_jid, queryNS =\
common.xmpp.NS_MUC_ADMIN)
item = iq.getTag('query').setTag('item')
item.setAttr('jid', jid)
item.setAttr('affiliation', affiliation)
self.connection.send(iq)
def send_gc_config(self, room_jid, config):
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'set', to = room_jid, queryNS =\
common.xmpp.NS_MUC_OWNER)
query = iq.getTag('query')
x = query.setTag(common.xmpp.NS_DATA + ' x', attrs = {'type': 'submit'}) # FIXME: should really use XData class
i = 0
while config.has_key(i):
if not config[i].has_key('type'):
i += 1
continue
if config[i]['type'] == 'fixed':
i += 1
continue
tag = x.addChild('field')
if config[i].has_key('var'):
2005-04-26 20:45:54 +02:00
tag.setAttr('var', config[i]['var'])
if config[i].has_key('values'):
for val in config[i]['values']:
if val == False:
val = '0'
elif val == True:
val = '1'
2005-04-26 20:45:54 +02:00
tag.setTagData('value', val)
i += 1
self.connection.send(iq)
def gpg_passphrase(self, passphrase):
if USE_GPG:
self.gpg.passphrase = passphrase
def ask_gpg_secrete_keys(self):
if USE_GPG:
keys = self.gpg.get_secret_keys()
return keys
return None
def change_password(self, password, username):
if not self.connection:
return
hostname = gajim.config.get_per('accounts', self.name, 'hostname')
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'set', to = hostname)
q = iq.setTag(common.xmpp.NS_REGISTER + ' query')
q.setTagData('username',username)
q.setTagData('password',password)
self.connection.send(iq)
2005-04-20 14:23:41 +02:00
def unregister_account(self):
if self.connected == 0:
self.connection = self.connect()
if self.connected > 1:
hostname = gajim.config.get_per('accounts', self.name, 'hostname')
2005-04-26 20:45:54 +02:00
iq = common.xmpp.Iq(typ = 'set', to = hostname)
q = iq.setTag(common.xmpp.NS_REGISTER + ' query').setTag('remove')
2005-04-20 14:23:41 +02:00
self.connection.send(iq)
def process(self, timeout):
if not self.connection:
return
if self.connected:
try:
self.connection.Process(timeout)
except:
gajim.log.debug('error appeared while processing xmpp:')
traceback.print_exc()
self.connected = 0
self.dispatch('STATUS', 'offline')
if not self.connection:
return
try:
self.connection.disconnect()
except:
gajim.log.debug('error appeared while processing xmpp:')
traceback.print_exc()
self.connection = None
# END GajimCore