diff options
| author | Jonas Dellinger <jonas.dellinger@2trde.com> | 2022-06-16 18:33:43 +0200 |
|---|---|---|
| committer | Jonas Dellinger <jonas.dellinger@2trde.com> | 2022-06-16 18:33:43 +0200 |
| commit | 0a12fe6102da33977548ba0c277bd4fe34e262ab (patch) | |
| tree | 7ff803d0d106db43ce206a6cdfc74c187f0d901a /backend/plugin.py | |
| parent | a95bf94d878f61869895bb22cbff1b4f524c5dca (diff) | |
| download | decky-loader-0a12fe6102da33977548ba0c277bd4fe34e262ab.tar.gz decky-loader-0a12fe6102da33977548ba0c277bd4fe34e262ab.zip | |
First draft of backend independent plugins
Diffstat (limited to 'backend/plugin.py')
| -rw-r--r-- | backend/plugin.py | 115 |
1 files changed, 0 insertions, 115 deletions
diff --git a/backend/plugin.py b/backend/plugin.py deleted file mode 100644 index 11c94a05..00000000 --- a/backend/plugin.py +++ /dev/null @@ -1,115 +0,0 @@ -import multiprocessing -from asyncio import (Lock, get_event_loop, new_event_loop, - open_unix_connection, set_event_loop, sleep, - start_unix_server) -from concurrent.futures import ProcessPoolExecutor -from importlib.util import module_from_spec, spec_from_file_location -from json import dumps, load, loads -from os import path, setuid -from signal import SIGINT, signal -from sys import exit -from time import time - -multiprocessing.set_start_method("fork") - -class PluginWrapper: - def __init__(self, file, plugin_directory, plugin_path) -> None: - self.file = file - self.plugin_directory = plugin_directory - self.reader = None - self.writer = None - self.socket_addr = f"/tmp/plugin_socket_{time()}" - self.method_call_lock = Lock() - - json = load(open(path.join(plugin_path, plugin_directory, "plugin.json"), "r")) - - self.legacy = False - self.main_view_html = json["main_view_html"] if "main_view_html" in json else "" - self.tile_view_html = json["tile_view_html"] if "tile_view_html" in json else "" - self.legacy = self.main_view_html or self.tile_view_html - - self.name = json["name"] - self.author = json["author"] - self.flags = json["flags"] - - self.passive = not path.isfile(self.file) - - def __str__(self) -> str: - return self.name - - def _init(self): - signal(SIGINT, lambda s, f: exit(0)) - - set_event_loop(new_event_loop()) - if self.passive: - return - setuid(0 if "root" in self.flags else 1000) - spec = spec_from_file_location("_", self.file) - module = module_from_spec(spec) - spec.loader.exec_module(module) - self.Plugin = module.Plugin - - if hasattr(self.Plugin, "_main"): - get_event_loop().create_task(self.Plugin._main(self.Plugin)) - get_event_loop().create_task(self._setup_socket()) - get_event_loop().run_forever() - - async def _setup_socket(self): - self.socket = await start_unix_server(self._listen_for_method_call, path=self.socket_addr) - - async def _listen_for_method_call(self, reader, writer): - while True: - data = loads((await reader.readline()).decode("utf-8")) - if "stop" in data: - get_event_loop().stop() - while get_event_loop().is_running(): - await sleep(0) - get_event_loop().close() - return - d = {"res": None, "success": True} - try: - d["res"] = await getattr(self.Plugin, data["method"])(self.Plugin, **data["args"]) - except Exception as e: - d["res"] = str(e) - d["success"] = False - finally: - writer.write((dumps(d)+"\n").encode("utf-8")) - await writer.drain() - - async def _open_socket_if_not_exists(self): - if not self.reader: - while True: - try: - self.reader, self.writer = await open_unix_connection(self.socket_addr) - break - except: - await sleep(0) - - def start(self): - if self.passive: - return self - multiprocessing.Process(target=self._init).start() - return self - - def stop(self): - if self.passive: - return - async def _(self): - await self._open_socket_if_not_exists() - self.writer.write((dumps({"stop": True})+"\n").encode("utf-8")) - await self.writer.drain() - self.writer.close() - get_event_loop().create_task(_(self)) - - async def execute_method(self, method_name, kwargs): - if self.passive: - raise RuntimeError("This plugin is passive (aka does not implement main.py)") - async with self.method_call_lock: - await self._open_socket_if_not_exists() - self.writer.write( - (dumps({"method": method_name, "args": kwargs})+"\n").encode("utf-8")) - await self.writer.drain() - res = loads((await self.reader.readline()).decode("utf-8")) - if not res["success"]: - raise Exception(res["res"]) - return res["res"] |
