add property bare_name in zeroconf.contacts[..]
property list. Using it, instead of bad-client
This commit is contained in:
Dimitur Kirov
parent
99265c0430
commit
fd7eead5dd
|
|
@ -164,8 +164,7 @@ class ConnectionZeroconf(ConnectionHandlersZeroconf):
|
|||
diffs = self.roster.getDiffs()
|
||||
for key in diffs:
|
||||
self.roster.setItem(key)
|
||||
display_key = self.zeroconf.check_jid(key)
|
||||
self.dispatch('NOTIFY', (display_key, self.roster.getStatus(key), self.roster.getMessage(key), 'local', 0, None, 0))
|
||||
self.dispatch('NOTIFY', (key, self.roster.getStatus(key), self.roster.getMessage(key), 'local', 0, None, 0))
|
||||
return self.call_resolve_timeout
|
||||
|
||||
# callbacks called from zeroconf
|
||||
|
|
|
|||
|
|
@ -21,8 +21,8 @@ class Roster:
|
|||
self.zeroconf = zeroconf # our zeroconf instance
|
||||
|
||||
def update_roster(self):
|
||||
for (jid, dom, interf, proto, host, addr, port, txt) in self.zeroconf.get_contacts().values():
|
||||
self.setItem(jid)
|
||||
for val in self.zeroconf.contacts.values():
|
||||
self.setItem(val[zeroconf.C_NAME])
|
||||
|
||||
def getRoster(self):
|
||||
#print 'roster_zeroconf.py: getRoster'
|
||||
|
|
@ -47,7 +47,7 @@ class Roster:
|
|||
|
||||
def setItem(self, jid, name = '', groups = ''):
|
||||
#print 'roster_zeroconf.py: setItem %s' % jid
|
||||
(service_jid, domain, interface, protocol, host, address, port, txt) \
|
||||
(service_jid, domain, interface, protocol, host, address, port, bare_jid, txt) \
|
||||
= self.zeroconf.get_contact(jid)
|
||||
|
||||
self._data[jid]={}
|
||||
|
|
@ -63,19 +63,25 @@ class Roster:
|
|||
status = txt_dict['status']
|
||||
else:
|
||||
status = ''
|
||||
if txt_dict.has_key('1st') and txt_dict.has_key('last'):
|
||||
self._data[jid]['name']=txt_dict['1st']+' '+txt_dict['last']
|
||||
nm = ''
|
||||
if txt_dict.has_key('1st'):
|
||||
nm = txt_dict['1st']
|
||||
if txt_dict.has_key('last'):
|
||||
if nm != '':
|
||||
nm = +' '
|
||||
nm += txt_dict['last']
|
||||
if nm:
|
||||
self._data[jid]['name'] = nm
|
||||
else:
|
||||
self._data[jid]['name']=jid
|
||||
if status == 'avail': status = 'online'
|
||||
self._data[jid]['name'] = jid
|
||||
if status == 'avail':
|
||||
status = 'online'
|
||||
self._data[jid]['txt_dict'] = txt_dict
|
||||
if not self._data[jid]['txt_dict'].has_key('msg'):
|
||||
self._data[jid]['txt_dict']['msg'] = ''
|
||||
self._data[jid]['status'] = status
|
||||
self._data[jid]['show'] = status
|
||||
|
||||
# print self._data[jid]
|
||||
|
||||
def delItem(self, jid):
|
||||
#print 'roster_zeroconf.py: delItem %s' % jid
|
||||
if self._data.has_key(jid):
|
||||
|
|
|
|||
|
|
@ -21,13 +21,17 @@ from common import xmpp
|
|||
try:
|
||||
import avahi, gobject, dbus
|
||||
except ImportError:
|
||||
print "Error: python-avahi and python-dbus need to be installed. No zeroconf support."
|
||||
gajim.log.debug("Error: python-avahi and python-dbus need to be installed. No zeroconf support.")
|
||||
|
||||
try:
|
||||
import dbus.glib
|
||||
except ImportError, e:
|
||||
pass
|
||||
|
||||
|
||||
C_NAME, C_DOMAIN, C_INTERFACE, C_PROTOCOL, C_HOST, \
|
||||
C_ADDRESS, C_PORT, C_BARE_NAME, C_TXT = range(9)
|
||||
|
||||
class Zeroconf:
|
||||
def __init__(self, new_serviceCB, remove_serviceCB, name, host, port):
|
||||
self.domain = None # specific domain to browse
|
||||
|
|
@ -66,8 +70,11 @@ class Zeroconf:
|
|||
def remove_service_callback(self, interface, protocol, name, stype, domain, flags):
|
||||
# print "Service '%s' in domain '%s' on %i.%i disappeared." % (name, domain, interface, protocol)
|
||||
if name != self.name:
|
||||
del self.contacts[name]
|
||||
self.remove_serviceCB(name)
|
||||
for key in self.contacts.keys():
|
||||
if self.contacts[key][C_BARE_NAME] == name:
|
||||
del self.contacts[key]
|
||||
self.remove_serviceCB(key)
|
||||
return
|
||||
|
||||
def new_service_type(self, interface, protocol, stype, domain, flags):
|
||||
# Are we already browsing this domain for this type?
|
||||
|
|
@ -88,23 +95,6 @@ class Zeroconf:
|
|||
if domain != "local":
|
||||
self.browse_domain(interface, protocol, domain)
|
||||
|
||||
def check_jid(self, jid):
|
||||
# TODO: at least miranda uses bad service names(only host name), so change them - probabaly not so nice... need to find a better solution
|
||||
# this is necessary so they don't get displayed as a transport
|
||||
# [dkirov] maybe turn it into host+'@'+host, instead ?
|
||||
# [sb] that would mean we can't do recreate below
|
||||
if jid.find('@') == -1:
|
||||
return 'bad-client@' + jid
|
||||
else:
|
||||
return jid
|
||||
|
||||
def recreate_bad_jid(self,jid):
|
||||
at = jid.find('@')
|
||||
if jid[:at] == 'bad-client':
|
||||
return jid[at+1:]
|
||||
else:
|
||||
return jid
|
||||
|
||||
def txt_array_to_dict(self,txt_array):
|
||||
items = {}
|
||||
|
||||
|
|
@ -117,16 +107,23 @@ class Zeroconf:
|
|||
|
||||
def service_resolved_callback(self, interface, protocol, name, stype, domain, host, aprotocol, address, port, txt, flags):
|
||||
print "Service data for service '%s' in domain '%s' on %i.%i:" % (name, domain, interface, protocol)
|
||||
print "\tHost %s (%s), port %i, TXT data: %s" % (host, address, port, str(avahi.txt_array_to_string_array(txt)))
|
||||
self.contacts[name] = (name, domain, interface, protocol, host, address, port, txt)
|
||||
print "\tHost %s (%s), port %i, TXT data: %s" % (host, address, port, avahi.txt_array_to_string_array(txt))
|
||||
bare_name = name
|
||||
if name.find('@') == -1:
|
||||
name = name + '@' + name
|
||||
|
||||
self.contacts[name] = (name, domain, interface, protocol, host, address, port,
|
||||
bare_name, txt)
|
||||
self.new_serviceCB(name)
|
||||
|
||||
# different handler when resolving all contacts
|
||||
def service_resolved_all_callback(self, interface, protocol, name, stype, domain, host, aprotocol, address, port, txt, flags):
|
||||
#print "Service data for service '%s' in domain '%s' on %i.%i:" % (name, domain, interface, protocol)
|
||||
#print "\tHost %s (%s), port %i, TXT data: %s" % (host, address, port, str(avahi.txt_array_to_string_array(txt)))
|
||||
|
||||
self.contacts[name] = (name, domain, interface, protocol, host, address, port, txt)
|
||||
bare_name = name
|
||||
if name.find('@') == -1:
|
||||
name = name + '@' + name
|
||||
self.contacts[name] = (name, domain, interface, protocol, host, address, port, bare_name, txt)
|
||||
|
||||
def service_added_callback(self):
|
||||
# print 'Service successfully added'
|
||||
|
|
@ -255,6 +252,9 @@ class Zeroconf:
|
|||
|
||||
return True
|
||||
|
||||
def check_jid(self, jid):
|
||||
return jid
|
||||
|
||||
def disconnect(self):
|
||||
if self.connected:
|
||||
self.connected = False
|
||||
|
|
@ -263,16 +263,14 @@ class Zeroconf:
|
|||
# refresh txt data of all contacts manually (no callback available)
|
||||
def resolve_all(self):
|
||||
for val in self.contacts.values():
|
||||
#val:(name, domain, interface, protocol, host, address, port, txt)
|
||||
self.server.ResolveService(int(val[2]), int(val[3]), val[0], \
|
||||
self.stype, val[1], avahi.PROTO_UNSPEC, dbus.UInt32(0),\
|
||||
self.server.ResolveService(int(val[C_INTERFACE]), int(val[C_PROTOCOL]), val[C_BARE_NAME], \
|
||||
self.stype, val[C_DOMAIN], avahi.PROTO_UNSPEC, dbus.UInt32(0),\
|
||||
reply_handler=self.service_resolved_all_callback, error_handler=self.print_error_callback)
|
||||
|
||||
def get_contacts(self):
|
||||
return self.contacts
|
||||
|
||||
def get_contact(self, jid):
|
||||
if self.contacts.has_key(jid):
|
||||
return self.contacts[jid]
|
||||
|
||||
def update_txt(self, txt):
|
||||
|
|
@ -301,14 +299,12 @@ class Zeroconf:
|
|||
|
||||
def send_message(self, jid, msg, type = 'chat'):
|
||||
print 'zeroconf.py: send_message:'+ msg
|
||||
jid = self.recreate_bad_jid(jid)
|
||||
|
||||
sock = socket.socket ( socket.AF_INET, socket.SOCK_STREAM )
|
||||
#sock.setblocking(False)
|
||||
sock.connect ( ( self.contacts[jid][C_ADDRESS], self.contacts[jid][C_PORT] ) )
|
||||
|
||||
sock.connect ( ( self.contacts[jid][4], self.contacts[jid][6] ) )
|
||||
|
||||
print (self.txt_array_to_dict(self.contacts[jid][7]))['port.p2pj']
|
||||
#print (self.txt_array_to_dict(self.contacts[jid][C_TXT]))['port.p2pj']
|
||||
|
||||
#was for adium which uses the txt record
|
||||
#sock.connect ( ( self.contacts[jid][5], int((self.txt_array_to_dict(self.contacts[jid][7]))['port.p2pj']) ) )
|
||||
|
|
|
|||
Loading…
Reference in New Issue