Initial openvim v0.4.6 upload

[osm/openvim.git] / charm / openvim / interface-openvim / requires.py

diff --git a/charm/openvim/interface-openvim/requires.py b/charm/openvim/interface-openvim/requires.py
new file mode 100644 (file)
index 0000000..77da9ca
--- /dev/null
+++ b/charm/openvim/interface-openvim/requires.py
@@ -0,0 +1,61 @@
+from charms.reactive import hook
+from charms.reactive import RelationBase
+from charms.reactive import scopes
+
+
+class OpenVimRequires(RelationBase):
+    scope = scopes.UNIT
+
+    @hook('{requires:openvim}-relation-{joined,changed}')
+    def changed(self):
+        conv = self.conversation()
+        if conv.get_remote('port'):
+            # this unit's conversation has a port, so
+            # it is part of the set of available units
+            conv.set_state('{relation_name}.available')
+
+    @hook('{requires:openvim}-relation-{departed,broken}')
+    def broken(self):
+        conv = self.conversation()
+        conv.remove_state('{relation_name}.available')
+
+    def services(self):
+        """
+        Returns a list of available openvim services and their associated hosts
+        and ports.
+
+        The return value is a list of dicts of the following form::
+
+            [
+                {
+                    'service_name': name_of_service,
+                    'hosts': [
+                        {
+                            'hostname': address_of_host,
+                            'port': port_for_host,
+                            'user': user_for_host,
+                        },
+                        # ...
+                    ],
+                },
+                # ...
+            ]
+        """
+        services = {}
+        for conv in self.conversations():
+            service_name = conv.scope.split('/')[0]
+            service = services.setdefault(service_name, {
+                'service_name': service_name,
+                'hosts': [],
+            })
+            host = conv.get_remote('hostname') or \
+                conv.get_remote('private-address')
+            port = conv.get_remote('port')
+            user = conv.get_remote('user')
+            if host and port:
+                service['hosts'].append({
+                    'hostname': host,
+                    'port': port,
+                    'user': user,
+                })
+        return [s for s in services.values() if s['hosts']]