1 files changed, 77 insertions, 0 deletions
diff --git a/plugin_loader/main.py b/plugin_loader/main.py
new file mode 100644
index 00000000..4ecb5158
--- /dev/null
+++ b/plugin_loader/main.py
@@ -0,0 +1,77 @@
+from aiohttp.web import Application, run_app, static
+from aiohttp_jinja2 import setup as jinja_setup
+from jinja2 import FileSystemLoader
+from os import getenv, path
+from asyncio import get_event_loop
+from json import loads, dumps
+
+from loader import Loader
+from injector import inject_to_tab, get_tabs
+from utilities import util_methods
+
+CONFIG = {
+    "plugin_path": getenv("PLUGIN_PATH", "/home/deck/homebrew/plugins"),
+    "server_host": getenv("SERVER_HOST", "127.0.0.1"),
+    "server_port": int(getenv("SERVER_PORT", "1337"))
+}
+
+class PluginManager:
+    def __init__(self) -> None:
+        self.loop = get_event_loop()
+        self.web_app = Application()
+        self.plugin_loader = Loader(self.web_app, CONFIG["plugin_path"])
+
+        jinja_setup(self.web_app, loader=FileSystemLoader(path.join(path.dirname(__file__), 'templates')))
+        self.web_app.on_startup.append(self.inject_javascript)
+        self.web_app.add_routes([static("/static", path.join(path.dirname(__file__), 'static'))])
+        self.loop.create_task(self.method_call_listener())
+
+    async def resolve_method_call(self, tab, call_id, response):
+        await tab._send_devtools_cmd({
+            "id": 1,
+            "method": "Runtime.evaluate",
+            "params": {
+                "expression": "resolveMethodCall({}, {})".format(call_id, dumps(response)),
+                "userGesture": True
+            }
+        }, receive=False)
+
+    async def handle_method_call(self, method, tab):
+        res = {}
+        try:
+            if method["method"] == "plugin_method":
+                res["result"] = await self.plugin_loader.handle_plugin_method_call(
+                    method["args"]["plugin_name"],
+                    method["args"]["method_name"],
+                    **method["args"]["args"]
+                )
+                res["success"] = True
+            else:
+                r = await util_methods[method["method"]](**method["args"])
+                res["result"] = r
+                res["success"] = True
+        except Exception as e:
+            res["result"] = str(e)
+            res["success"] = False
+        finally:
+            await self.resolve_method_call(tab, method["id"], res)
+
+    async def method_call_listener(self):
+        tab = next((i for i in await get_tabs() if i.title == "QuickAccess"), None)
+        await tab.open_websocket()
+        await tab._send_devtools_cmd({"id": 1, "method": "Runtime.discardConsoleEntries"})
+        await tab._send_devtools_cmd({"id": 1, "method": "Runtime.enable"})
+        async for message in tab.listen_for_message():
+            data = message.json()
+            if not "id" in data and data["method"] == "Runtime.consoleAPICalled" and data["params"]["type"] == "debug":
+                method = loads(data["params"]["args"][0]["value"])
+                self.loop.create_task(self.handle_method_call(method, tab))
+                
+    async def inject_javascript(self, request=None):
+        await inject_to_tab("QuickAccess", open(path.join(path.dirname(__file__), "static/plugin_page.js"), "r").read())
+
+    def run(self):
+        return run_app(self.web_app, host=CONFIG["server_host"], port=CONFIG["server_port"], loop=self.loop)
+
+if __name__ == "__main__":
+    PluginManager().run()
+\ No newline at end of file