X-Git-Url: http://git.maemo.org/git/?a=blobdiff_plain;f=src%2Fcapabilities.py;h=11a7f0cf93f8e73c0fc2cd846b17b02be70b0d44;hb=2b6cb781e1d2b011dab0eecf1008439368e7b045;hp=6c7ba15e5efe9d6aa264d1f2d866ceb90f6153c5;hpb=a840b976430685dbf23bee4e590d1ee23594728c;p=theonering

diff --git a/src/capabilities.py b/src/capabilities.py
index 6c7ba15..11a7f0c 100644
--- a/src/capabilities.py
+++ b/src/capabilities.py
@@ -1,6 +1,9 @@
 import logging
 
+import telepathy
+
 import tp
+import gtk_toolbox
 
 
 _moduleLogger = logging.getLogger('capabilities')
@@ -8,4 +11,84 @@ _moduleLogger = logging.getLogger('capabilities')
 
 class CapabilitiesMixin(tp.ConnectionInterfaceCapabilities):
 
-	pass
+	def __init__(self):
+		tp.ConnectionInterfaceCapabilities.__init__(self)
+		self._implement_property_get(
+			telepathy.interfaces.CONN_INTERFACE_CAPABILITIES,
+			{"caps": self.GetCapabilities},
+		)
+
+	@property
+	def session(self):
+		"""
+		@abstract
+		"""
+		raise NotImplementedError("Abstract property called")
+
+	def get_handle_by_id(self, handleType, handleId):
+		"""
+		@abstract
+		"""
+		raise NotImplementedError("Abstract function called")
+
+	def GetSelfHandle(self):
+		"""
+		@abstract
+		"""
+		raise NotImplementedError("Abstract function called")
+
+	@gtk_toolbox.log_exception(_moduleLogger)
+	def GetCapabilities(self, handleIds):
+		"""
+		@todo HACK Remove this once we are building against a fixed version of python-telepathy
+		"""
+		ret = []
+		for handleId in handleIds:
+			h = self.get_handle_by_id(telepathy.HANDLE_TYPE_CONTACT, handleId)
+			if handleId != 0 and (telepathy.HANDLE_TYPE_CONTACT, handleId) not in self._handles:
+				raise telepathy.errors.InvalidHandle
+			elif h in self._caps:
+				types = self._caps[h]
+				for type in types:
+					for ctype, specs in types.iteritems():
+						ret.append([handleId, type, specs[0], specs[1]])
+			else:
+				# No caps, so just default to the connection's caps
+				types = self._caps[self.GetSelfHandle()]
+				for type in types:
+					for ctype, specs in types.iteritems():
+						ret.append([handleId, type, specs[0], specs[1]])
+		return ret
+
+	@gtk_toolbox.log_exception(_moduleLogger)
+	def AdvertiseCapabilities(self, add, remove):
+		"""
+		@todo HACK Remove this once we are building against a fixed version of python-telepathy
+		"""
+		my_caps = self._caps.setdefault(self.GetSelfHandle(), {})
+
+		changed = {}
+		for ctype, spec_caps in add:
+			changed[ctype] = spec_caps
+		for ctype in remove:
+			changed[ctype] = None
+
+		caps = []
+		for ctype, spec_caps in changed.iteritems():
+			gen_old, spec_old = my_caps.get(ctype, (0, 0))
+			if spec_caps is None:
+				# channel type no longer supported (provider has gone away)
+				gen_new, spec_new = 0, 0
+			else:
+				# channel type supports new capabilities
+				gen_new, spec_new = gen_old, spec_old | spec_caps
+			if spec_old != spec_new or gen_old != gen_new:
+				caps.append((self.GetSelfHandle(), ctype, gen_old, gen_new,
+							spec_old, spec_new))
+
+		self.CapabilitiesChanged(caps)
+		_moduleLogger.info("CapsChanged %r" % self._caps)
+
+		# return all my capabilities
+		ret = [(ctype, caps[1]) for ctype, caps in my_caps.iteritems()]
+		return ret