Make the presence service resolve all shared activity services by default; it's up to the activities to ignore services that aren't theirs. Also add 'joined-activity' and 'left-activity' signals on Buddy objects, mainly for the PresenceWindow's 'Who's here' bits

This commit is contained in:
Dan Williams 2006-06-19 09:49:57 -04:00
parent d08ea50d15
commit c65ef6f9cd
5 changed files with 67 additions and 62 deletions

View File

@ -37,8 +37,8 @@ class BrowserShell(dbus.service.Object):
@dbus.service.method('com.redhat.Sugar.BrowserShell')
def open_browser(self, uri, serialized_service=None):
service = None
# if serialized_service is not None:
# service = Service.deserialize(serialized_service)
if serialized_service is not None:
service = Service.deserialize(serialized_service)
browser = BrowserActivity(uri)
self.__browsers.append(browser)
gobject.idle_add(self._start_browser_cb, browser, service)

View File

@ -19,12 +19,8 @@ class GroupChat(Chat):
self._pservice.connect('service-appeared', self._service_appeared_cb)
self._pservice.track_service_type(GroupChat.SERVICE_TYPE)
# FIXME remove, when we join the activity this will happen automatically
# (Once we have a global presence service)
self._pservice.track_activity(activity.get_id())
def _service_appeared_cb(self, pservice, buddy, service):
if service.get_type() == GroupChat.SERVICE_TYPE:
if service.get_full_type() == GroupChat.SERVICE_TYPE:
logging.debug('Group chat service appeared, setup the stream.')
self._setup_stream(service)

View File

@ -20,7 +20,11 @@ class Buddy(gobject.GObject):
'service-added': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE,
([gobject.TYPE_PYOBJECT])),
'service-removed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE,
([gobject.TYPE_PYOBJECT]))
([gobject.TYPE_PYOBJECT])),
'joined-activity': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE,
([gobject.TYPE_STRING])),
'left-activity': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE,
([gobject.TYPE_STRING]))
}
def __init__(self, service):
@ -69,12 +73,27 @@ class Buddy(gobject.GObject):
if service.get_publisher_address() != self._address:
logging.error('Service publisher and buddy address doesnt match: %s %s' % (service.get_publisher_address(), self._address))
return False
if service.get_type() in self._services.keys():
full_type = service.get_full_type()
if full_type in self._services.keys():
return False
self._services[service.get_full_type()] = service
self._services[full_type] = service
if self._valid:
self.emit("service-added", service)
if service.get_full_type() == PRESENCE_SERVICE_TYPE:
# If this is the first service we've seen that's owned by
# a particular activity, send out the 'joined-activity' signal
(uid, short_stype) = Service._decompose_service_type(full_type)
if uid is not None:
found = False
for serv in self._services.values():
if serv.get_activity_uid() == uid and serv.get_full_type() != full_type:
found = True
break
if not found:
print "Buddy (%s) joined activity %s." % (self._nick_name, service.get_activity_uid())
self.emit("joined-activity", service)
if full_type == PRESENCE_SERVICE_TYPE:
# A buddy isn't valid until its official presence
# service has been found and resolved
self._valid = True
@ -89,11 +108,26 @@ class Buddy(gobject.GObject):
return
if service.get_name() != self._nick_name:
return
if self._services.has_key(service.get_full_type()):
full_type = service.get_full_type()
if self._services.has_key(full_type):
if self._valid:
self.emit("service-removed", service)
del self._services[service.get_full_type()]
if service.get_full_type() == PRESENCE_SERVICE_TYPE:
del self._services[full_type]
# If this is the lase service owned by a particular activity,
# and it's just been removed, send out the 'left-actvity' signal
(uid, short_stype) = Service._decompose_service_type(full_type)
if uid is not None:
found = False
for serv in self._services.values():
if serv.get_activity_uid() == uid:
found = True
break
if not found:
print "Buddy (%s) left activity %s." % (self._nick_name, service.get_activity_uid())
self.emit("left-activity", service)
if full_type == PRESENCE_SERVICE_TYPE:
self._valid = False
def get_service_of_type(self, stype=None, activity=None):

View File

@ -93,7 +93,7 @@ class PresenceService(gobject.GObject):
return PresenceService.__instance
get_instance = staticmethod(get_instance)
def __init__(self, debug=False):
def __init__(self, debug=True):
gobject.GObject.__init__(self)
self._debug = debug
@ -113,7 +113,6 @@ class PresenceService(gobject.GObject):
# All the mdns service types we care about
self._allowed_service_types = [] # Short service type
self._allowed_activities = [] # activity UID
# Keep track of stuff we're already browsing with ZC
self._service_type_browsers = {}
@ -163,7 +162,7 @@ class PresenceService(gobject.GObject):
def _resolve_service_error_handler(self, err):
logging.error("error resolving service: %s" % err)
def _find_service_adv(self, interface=None, protocol=None, name=None, stype=None, domain=None):
def _find_service_adv(self, interface=None, protocol=None, name=None, stype=None, domain=None, is_short_stype=False):
"""Search a list of service advertisements for ones matching certain criteria."""
adv_list = []
for adv in self._service_advs:
@ -173,8 +172,13 @@ class PresenceService(gobject.GObject):
continue
if name and adv.name() != name:
continue
if stype and adv.stype() != stype:
continue
if is_short_stype:
(uid, dec_stype) = Service._decompose_service_type(adv.stype())
if uid is None or stype != dec_stype:
continue
else:
if stype and adv.stype() != stype:
continue
if domain and adv.domain() != domain:
continue
adv_list.append(adv)
@ -314,9 +318,8 @@ class PresenceService(gobject.GObject):
# If we care about the service right now, resolve it
resolve = False
if uid in self._allowed_activities:
if short_stype in self._allowed_service_types:
resolve = True
if uid is not None or short_stype in self._allowed_service_types:
resolve = True
if self._is_special_service_type(short_stype):
resolve = True
if resolve and not adv in self._resolve_queue:
@ -416,25 +419,6 @@ class PresenceService(gobject.GObject):
def _new_domain_cb_glue(self, interface, protocol, domain, flags=0):
gobject.idle_add(self._new_domain_cb, interface, protocol, domain, flags)
def track_activity(self, activity_uid):
"""INTERNAL ONLY; register an activity's UID to recognize service
events for that specific activity."""
if not activity_uid or not util.validate_activity_uid(activity_uid):
raise ValueError("activity uid must be a valid activity uid string.")
if activity_uid in self._allowed_activities:
return
self._allowed_activities.append(activity_uid)
self._check_and_resolve_service_advs()
def untrack_activity(self, activity_uid):
"""INTERNAL ONLY; unregister an activity's UID to stop service
events for that specific activity."""
if not activity_uid or not util.validate_activity_uid(activity_uid):
raise ValueError("activity uid must be a valid activity uid string.")
if activity_uid not in self._allowed_activities:
return
self._allowed_activities.remove(activity_uid)
def track_service_type(self, short_stype):
"""Requests that the Presence service look for and recognize
a certain mDNS service types."""
@ -456,22 +440,17 @@ class PresenceService(gobject.GObject):
self._allowed_service_types.append(dec_stype)
self._check_and_resolve_service_advs(dec_stype)
def _check_and_resolve_service_advs(self, specific_stype=None):
def _check_and_resolve_service_advs(self, short_stype):
"""We should only get called with short service types (ie, not
service types that can be decomposed into a UID and a type)."""
# Find unresolved services that match the service type
# we're now interested in, and resolve them
resolv_list = []
# Find all services first by their activity
search_types = self._allowed_service_types
if specific_stype:
search_types = [specific_stype]
for uid in self._allowed_activities:
for short_stype in search_types:
full_stype = Service.compose_service_type(short_stype, uid)
adv_list = self._find_service_adv(stype=full_stype)
resolv_list = resolv_list + adv_list
# Then, find services by just the plain service type
if specific_stype is not None:
resolv_list = resolv_list + self._find_service_adv(stype=specific_stype)
# Find services of this type belonging to specific activities
resolv_list = self._find_service_adv(stype=short_stype, is_short_stype=True)
# And also just plain ones of this type
resolv_list = resolv_list + self._find_service_adv(stype=short_stype)
# Request resolution for them if they aren't in-process already
for adv in resolv_list:
@ -505,6 +484,9 @@ class PresenceService(gobject.GObject):
def share_activity(self, activity, stype, properties={}, address=None, port=None):
"""Convenience function to share an activity with other buddies."""
if not self._started:
raise RuntimeError("presence service must be started first.")
uid = activity.get_id()
logging.debug('Sharing activity uid %s, stype %s' % (uid, stype))
owner_nick = self._owner.get_nick_name()
@ -536,8 +518,6 @@ class PresenceService(gobject.GObject):
rs_name = service.get_name()
rs_stype = service.get_full_type()
rs_port = service.get_port()
if type(rs_port) != type(1) and (rs_port <= 1024 or rs_port > 65536):
raise ValueError("invalid service port.")
rs_props = service.get_properties()
rs_domain = service.get_domain()
rs_address = service.get_address()
@ -560,10 +540,7 @@ class PresenceService(gobject.GObject):
# should un-register it an re-register with the correct info
if str(exc) == "Local name collision":
pass
uid = service.get_activity_uid()
activity_stype = service.get_type()
if uid:
self.track_activity(uid)
self.track_service_type(activity_stype)
return group

View File

@ -236,15 +236,13 @@ class StartPage(gtk.HBox):
self._search(None)
def _on_local_activity_started_cb(self, helper, activity_container, activity_id):
self._pservice.track_activity(activity_id)
print "new local activity %s" % activity_id
def _on_local_activity_ended_cb(self, helper, activity_container, activity_id):
self._pservice.untrack_activity(activity_id)
print "local activity %s disappeared" % activity_id
def _on_new_service_adv_cb(self, pservice, activity_id, short_stype):
if activity_id:
self._pservice.track_activity(activity_id)
self._pservice.track_service_type(short_stype)
def _on_activity_announced_cb(self, pservice, service, buddy):