Allow plugin bundle uploads via websocket

Signed-off-by: falkTX <[email protected]>

Author: falkTX

html/js/host.js

@@ -42,7 +42,7 @@ $('document').ready(function() {
         }
     }
 
-    ws.onclose = function (evt) {
+    ws.onclose = function () {
         desktop && desktop.blockUI()
     }
 

mod/webserver.py

@@ -791,6 +791,18 @@ def get(self):
         self.write(True)
 
 class SDKEffectInstaller(EffectInstaller):
+    def set_default_headers(self):
+        if 'Origin' not in self.request.headers.keys():
+            return
+        origin = self.request.headers['Origin']
+        match  = re.match(r'^(\w+)://([^/]*)/?', origin)
+        if match is None:
+            return
+        protocol, domain = match.groups()
+        if protocol != "http" and not domain.endswith(":9000") and not domain.endswith(".mod.audio"):
+            return
+        self.set_header("Access-Control-Allow-Origin", origin)
+
     @web.asynchronous
     @gen.engine
     def post(self):
@@ -816,7 +828,7 @@ def set_default_headers(self):
         if match is None:
             return
         protocol, domain = match.groups()
-        if protocol != "http" or not domain.endswith(":9000"):
+        if protocol != "http" and not domain.endswith(":9000") and not domain.endswith(".mod.audio"):
             return
         self.set_header("Access-Control-Allow-Origin", origin)
 
@@ -1135,7 +1147,7 @@ def get(self, instance):
             logging.exception(e)
         self.write(ok)
 
-class RemoteWebSocket(websocket.WebSocketHandler):
+class RemotePedalboardWebSocket(websocket.WebSocketHandler):
     def check_origin(self, origin):
         match = re.match(r'^(\w+)://([^/]*)/?', origin)
         if match is None:
@@ -1156,6 +1168,41 @@ def on_message(self, pedalboard_id):
         self.write_message("true")
         self.close()
 
+class RemotePluginWebSocket(websocket.WebSocketHandler):
+    def check_origin(self, origin):
+        match = re.match(r'^(\w+)://([^/]*)/?', origin)
+        if match is None:
+            return False
+        protocol, domain = match.groups()
+        if protocol not in ("http", "https"):
+            return False
+        if domain != "mod.audio" and not domain.endswith(".mod.audio"):
+            return False
+        return True
+
+    @gen.coroutine
+    def on_message(self, package):
+        if not package:
+            hwdesc = get_hardware_descriptor()
+            self.write_message(json.dumps({
+                'bin-compat': hwdesc.get('bin-compat', "Unknown"),
+                'platform': hwdesc.get('platform', "Unknown"),
+                'version': IMAGE_VERSION,
+            }))
+            return
+
+        filename = os.path.join(DOWNLOAD_TMP_DIR, "remote.tar.gz")
+
+        with open(filename, 'wb') as fh:
+            fh.write(b64decode(package))
+
+        resp = yield gen.Task(install_package, filename)
+
+        if resp['ok']:
+            SESSION.msg_callback("rescan " + b64encode(json.dumps(resp).encode("utf-8")).decode("utf-8"))
+
+        self.write_message(json.dumps(resp))
+
 class ServerWebSocket(websocket.WebSocketHandler):
     @gen.coroutine
     def open(self):
@@ -2368,7 +2415,8 @@ def get(self):
             (r"/js/templates.js$", BulkTemplateLoader),
 
             (r"/websocket/?$", ServerWebSocket),
-            (r"/rpbsocket/?$", RemoteWebSocket),
+            (r"/rpbsocket/?$", RemotePedalboardWebSocket),
+            (r"/rplsocket/?$", RemotePluginWebSocket),
 
             (r"/(.*)", TimelessStaticFileHandler, {"path": HTML_DIR}),
         ],