1 files changed, 135 insertions, 0 deletions
diff --git a/resource_manager.py b/resource_manager.py
new file mode 100644
index 0000000..1a6e224
--- /dev/null
+++ b/resource_manager.py
@@ -0,0 +1,135 @@
+import sys
+from BeautifulSoup import BeautifulSoup
+import ParrotProtocol
+
+
+class ResourceManagerBase(object):
+    resources = [
+    ]
+
+    def __init__(self, socket, resource_values=None):
+        self.sock = socket
+        self.resource_values = resource_values or {}
+
+    def get(self, resource):
+        try:
+            return self.resource_values[resource]
+        except KeyError:
+            return self.fetch(resource)
+
+    def fetch(self, resource):
+        assert resource in self.resources, 'Unknown resource {}'.format(resource)
+        message = ParrotProtocol.getRequest(resource + '/get')
+        result = self.send_message(message)
+        self.resource_values[resource] = result
+        return result
+
+    def toggle_on(self, resource):
+        assert resource in self.resources, 'Unknown resource {}'.format(resource)
+        message = ParrotProtocol.getRequest(resource + '/enable')
+        self.send_message(message)
+        self.fetch(resource)
+
+    def toggle_off(self, resource):
+        assert resource in self.resources, 'Unknown resource {}'.format(resource)
+        message = ParrotProtocol.getRequest(resource + '/disable')
+        self.send_message(message)
+        self.fetch(resource)
+
+    def set(self, resource, arg):
+        assert resource in self.resources, 'Unknown resource {}'.format(resource)
+        message = ParrotProtocol.setRequest(resource + '/set', str(arg).lower())
+        self.send_message(message)
+        self.fetch(resource)
+
+    def send_message(self, message):
+        try:
+            self.sock.send(str(message))
+        except Exception:
+            self.sock = ""
+            return
+        else:
+            return self.get_answer()
+
+    def get_answer(self):
+        data = self.receive_message()
+        while not data.answer:
+            if data.notify:
+                self.handle_notification(data.notify)
+            else:
+                raise AssertionError('Unknown response')
+            data = self.receive_message()
+        return data.answer
+
+    def handle_notification(self, notification):
+
+        self.fetch(notification['path'].rsplit('/', 1)[0].encode('utf-8'))
+
+    def receive_message(self):
+        if sys.platform == "darwin":
+            self.sock.recv(30)
+        else:
+            self.sock.recv(7)
+        return BeautifulSoup(self.sock.recv(1024))
+
+    def close(self):
+        self.sock.close()
+
+
+class GenericResourceManager(ResourceManagerBase):
+    resources = {
+        '/api/software/version',
+    }
+
+    def __init__(self, sock):
+        super(GenericResourceManager, self).__init__(sock)
+        self.notifications = []
+
+    def handle_notification(self, notification):
+        self.notifications.append(notification)
+
+    def get_resource_manager(self, resource_manager_class):
+        resource_manager = resource_manager_class(self.sock, self.resource_values)
+        for notitifaction in self.notifications:
+            resource_manager.handle_notification(notitifaction)
+        return resource_manager
+
+    @property
+    def api_version(self):
+        answer = self.get("/api/software/version")
+        try:
+            return answer.software["version"]
+        except KeyError:
+            return answer.software['sip6']
+
+
+class Version1ResourceManager(ResourceManagerBase):
+    resources = {
+        '/api/software/version',
+        '/api/system/battery',
+        '/api/bluetooth/friendlyname',
+        '/api/system/auto_connection/enabled',
+        '/api/system/anc_phone_mode/enabled',
+        '/api/audio/specific_mode/enabled',
+        '/api/audio/sound_effect/enabled',
+        '/api/audio/noise_cancellation/enabled',
+    }
+
+class Version2ResourceManager(ResourceManagerBase):
+    resources = {
+        '/api/software/version',
+        '/api/system/battery',
+        '/api/system/pi',
+        '/api/bluetooth/friendlyname',
+        '/api/system/auto_connection/enabled',
+        '/api/system/anc_phone_mode/enabled',
+        '/api/flight_mode',
+        '/api/audio/sound_effect/enabled',
+        '/api/audio/sound_effect/room_size',
+        '/api/audio/sound_effect/angle',
+        '/api/audio/noise',
+        '/api/audio/noise_control',
+        '/api/audio/noise_control/enabled',
+        '/api/audio/track/metadata',
+    }
+