Compare commits

..

17 Commits

Author SHA1 Message Date
Sky Leite f1db083749 Merge branch 'main' into python-lint-format-test 2023-02-04 00:42:38 -03:00
Sky Leite 897e1773a5 Disable SyntaxError lint rule due to ruff not supporting pattern matching 2023-02-04 00:31:42 -03:00
Sky Leite 4e92d4bfc5 Fix unescaped curly braces in javascript formatting
This would previously throw an error due to `format` interpreting the
curly braces as placeholders
2023-02-04 00:29:34 -03:00
Sky Leite 1edcc09020 Fix F821 lint rule 2023-02-04 00:22:17 -03:00
Sky Leite e3e1cf2df7 Fix F811 linter rule
get_user_group was defined twice, so I merged both definitions
2023-02-04 00:21:31 -03:00
Sky Leite 93587fe33b Fix E722 linter rule 2023-02-04 00:12:00 -03:00
Sky Leite 8fab487153 Add python linting actions 2023-02-04 00:06:53 -03:00
Sky Leite b6fce46081 Fix most formatting and linting errors 2023-02-03 23:59:43 -03:00
Sky Leite e20fd5042c Revert "Test that prettier properly fails the lint job"
This reverts commit 1f596e5a10.
2023-02-03 10:36:05 -03:00
Sky Leite 1f596e5a10 Test that prettier properly fails the lint job 2023-02-03 10:34:50 -03:00
Sky Leite 6edc3bb658 Remove popd 2023-02-03 10:31:07 -03:00
Sky Leite b33d44c53d Remove Python and Shell linters 2023-02-03 10:29:40 -03:00
Sky Leite 1379a40a89 Move linters to separate workflow 2023-02-03 10:07:16 -03:00
Sky Leite d48fc885a3 Use lint script from package.json 2023-02-03 09:51:28 -03:00
Sky Leite 7675775527 Install prettier plugins before linting 2023-02-02 18:59:23 -03:00
Sky Leite 1320b13507 Add prettier-plugin-import-sort 2023-02-02 18:50:33 -03:00
Sky Leite 52777bc2a4 Add lint job to build workflow 2023-02-02 18:46:57 -03:00
32 changed files with 827 additions and 833 deletions
+1 -2
View File
@@ -12,7 +12,6 @@ body:
- label: I have searched existing issues - label: I have searched existing issues
- label: This issue is not a duplicate of an existing one - label: This issue is not a duplicate of an existing one
- label: I have checked the [common issues section in the readme file](https://github.com/SteamDeckHomebrew/decky-loader#-common-issues) - label: I have checked the [common issues section in the readme file](https://github.com/SteamDeckHomebrew/decky-loader#-common-issues)
- label: I have attached logs to this bug report (failure to include logs will mean your issue will not be responded too).
- type: textarea - type: textarea
attributes: attributes:
@@ -71,4 +70,4 @@ body:
description: Please reboot your deck (if possible) when attempting to recreate the issue, then run ``cd ~ && journalctl -b0 -u plugin_loader.service > deckylog.txt``. This will save the log file to ``~`` aka ``/home/deck``. Please upload the file here description: Please reboot your deck (if possible) when attempting to recreate the issue, then run ``cd ~ && journalctl -b0 -u plugin_loader.service > deckylog.txt``. This will save the log file to ``~`` aka ``/home/deck``. Please upload the file here
placeholder: deckylog.txt placeholder: deckylog.txt
validations: validations:
required: true required: false
+1 -1
View File
@@ -69,7 +69,7 @@ jobs:
run: pnpm run build run: pnpm run build
- name: Build Python Backend 🛠️ - name: Build Python Backend 🛠️
run: pyinstaller --noconfirm --onefile --name "PluginLoader" --add-data ./backend/static:/static --add-data ./backend/legacy:/legacy --add-data ./plugin:/plugin ./backend/*.py run: pyinstaller --noconfirm --onefile --name "PluginLoader" --add-data ./backend/static:/static --add-data ./backend/legacy:/legacy ./backend/*.py
- name: Upload package artifact ⬆️ - name: Upload package artifact ⬆️
if: ${{ !env.ACT }} if: ${{ !env.ACT }}
+10
View File
@@ -15,3 +15,13 @@ jobs:
pushd frontend pushd frontend
npm install npm install
npm run lint npm run lint
- name: Run black (Python formatting)
uses: lgeiger/black-action@v1.0.1
with:
args: "./backend --experimental-string-processing --config ./backend/pyproject.toml"
- name: Run ruff (Python linting)
uses: jpetrucciani/ruff-check@main
with:
path: "./backend"
+4 -5
View File
@@ -11,7 +11,7 @@
<a href="https://github.com/SteamDeckHomebrew/decky-loader/stargazers"><img src="https://img.shields.io/github/stars/SteamDeckHomebrew/decky-loader" /></a> <a href="https://github.com/SteamDeckHomebrew/decky-loader/stargazers"><img src="https://img.shields.io/github/stars/SteamDeckHomebrew/decky-loader" /></a>
<a href="https://github.com/SteamDeckHomebrew/decky-loader/commits/main"><img src="https://img.shields.io/github/last-commit/SteamDeckHomebrew/decky-loader.svg" /></a> <a href="https://github.com/SteamDeckHomebrew/decky-loader/commits/main"><img src="https://img.shields.io/github/last-commit/SteamDeckHomebrew/decky-loader.svg" /></a>
<a href="https://github.com/SteamDeckHomebrew/decky-loader/blob/main/LICENSE"><img src="https://img.shields.io/github/license/SteamDeckHomebrew/decky-loader" /></a> <a href="https://github.com/SteamDeckHomebrew/decky-loader/blob/main/LICENSE"><img src="https://img.shields.io/github/license/SteamDeckHomebrew/decky-loader" /></a>
<a href="https://deckbrew.xyz/discord"><img src="https://img.shields.io/discord/960281551428522045?color=%235865F2&label=discord" /></a> <a href="https://discord.gg/ZU74G2NJzk"><img src="https://img.shields.io/discord/960281551428522045?color=%235865F2&label=discord" /></a>
<br> <br>
<br> <br>
<img src="https://media.discordapp.net/attachments/966017112244125756/1012466063893610506/main.jpg" alt="Decky screenshot" width="80%"> <img src="https://media.discordapp.net/attachments/966017112244125756/1012466063893610506/main.jpg" alt="Decky screenshot" width="80%">
@@ -62,7 +62,7 @@ For more information about Decky Loader as well as documentation and development
### 👋 Uninstallation ### 👋 Uninstallation
We are sorry to see you go! If you are considering uninstalling because you are having issues, please consider [opening an issue](https://github.com/SteamDeckHomebrew/decky-loader/issues) or [joining our Discord](https://deckbrew.xyz/discord) so we can help you and other users. We are sorry to see you go! If you are considering uninstalling because you are having issues, please consider [opening an issue](https://github.com/SteamDeckHomebrew/decky-loader/issues) or [joining our Discord](https://discord.gg/ZU74G2NJzk) so we can help you and other users.
1. Press the <img src="./docs/images/light/steam.svg#gh-dark-mode-only" height=16><img src="./docs/images/dark/steam.svg#gh-light-mode-only" height=16> button and open the Power menu. 1. Press the <img src="./docs/images/light/steam.svg#gh-dark-mode-only" height=16><img src="./docs/images/dark/steam.svg#gh-light-mode-only" height=16> button and open the Power menu.
1. Select "Switch to Desktop". 1. Select "Switch to Desktop".
@@ -84,16 +84,15 @@ Now that you have Decky Loader installed, you can start using plugins. Each plug
### 🛠️ Plugin Development ### 🛠️ Plugin Development
There is no complete plugin development documentation yet. However a good starting point is the [plugin template repository](https://github.com/SteamDeckHomebrew/decky-plugin-template). Consider [joining our Discord](https://deckbrew.xyz/discord) if you have any questions. There is no complete plugin development documentation yet. However a good starting point is the [plugin template repository](https://github.com/SteamDeckHomebrew/decky-plugin-template). Consider [joining our Discord](https://discord.gg/ZU74G2NJzk) if you have any questions.
### 🤝 Contributing ### 🤝 Contributing
Please consult [the wiki page regarding development](https://wiki.deckbrew.xyz/en/loader-dev/development) for more information on installing development versions of Decky Loader. You can also install the Steam Deck UI on a Windows or Linux computer for testing by following [this YouTube guide](https://youtu.be/1IAbZte8e7E?t=112). Please consult [the wiki page regarding development](https://deckbrew.xyz/en/loader-dev/development) for more information on installing development versions of Decky Loader. You can also install the Steam Deck UI on a Windows or Linux computer for testing by following [this YouTube guide](https://youtu.be/1IAbZte8e7E?t=112).
1. Clone the repository using the latest commit to main before starting your PR. 1. Clone the repository using the latest commit to main before starting your PR.
1. In your clone of the repository, run these commands. 1. In your clone of the repository, run these commands.
```bash ```bash
cd frontend
pnpm i pnpm i
pnpm run build pnpm run build
``` ```
+83 -35
View File
@@ -1,27 +1,33 @@
# Full imports # Full imports
import json import json
# import pprint # import pprint
# from pprint import pformat # from pprint import pformat
# Partial imports # Partial imports
from aiohttp import ClientSession, web from aiohttp import ClientSession
from asyncio import get_event_loop, sleep from asyncio import sleep
from concurrent.futures import ProcessPoolExecutor
from hashlib import sha256 from hashlib import sha256
from io import BytesIO from io import BytesIO
from logging import getLogger from logging import getLogger
from os import R_OK, W_OK, path, rename, listdir, access, mkdir from os import R_OK, W_OK, path, listdir, access, mkdir
from shutil import rmtree from shutil import rmtree
from subprocess import call from subprocess import call
from time import time from time import time
from zipfile import ZipFile from zipfile import ZipFile
# Local modules # Local modules
from helpers import get_ssl_context, get_user, get_user_group, download_remote_binary_to_path from helpers import (
get_ssl_context,
get_user,
get_user_group,
download_remote_binary_to_path,
)
from injector import get_gamepadui_tab from injector import get_gamepadui_tab
logger = getLogger("Browser") logger = getLogger("Browser")
class PluginInstallContext: class PluginInstallContext:
def __init__(self, artifact, name, version, hash) -> None: def __init__(self, artifact, name, version, hash) -> None:
self.artifact = artifact self.artifact = artifact
@@ -29,6 +35,7 @@ class PluginInstallContext:
self.version = version self.version = version
self.hash = hash self.hash = hash
class PluginBrowser: class PluginBrowser:
def __init__(self, plugin_path, plugins, loader) -> None: def __init__(self, plugin_path, plugins, loader) -> None:
self.plugin_path = plugin_path self.plugin_path = plugin_path
@@ -43,32 +50,40 @@ class PluginBrowser:
zip_file = ZipFile(zip) zip_file = ZipFile(zip)
zip_file.extractall(self.plugin_path) zip_file.extractall(self.plugin_path)
plugin_dir = self.find_plugin_folder(name) plugin_dir = self.find_plugin_folder(name)
code_chown = call(["chown", "-R", get_user()+":"+get_user_group(), plugin_dir]) code_chown = call(
["chown", "-R", get_user() + ":" + get_user_group(), plugin_dir]
)
code_chmod = call(["chmod", "-R", "555", plugin_dir]) code_chmod = call(["chmod", "-R", "555", plugin_dir])
if code_chown != 0 or code_chmod != 0: if code_chown != 0 or code_chmod != 0:
logger.error(f"chown/chmod exited with a non-zero exit code (chown: {code_chown}, chmod: {code_chmod})") logger.error(
f"chown/chmod exited with a non-zero exit code (chown: {code_chown},"
f" chmod: {code_chmod})"
)
return False return False
return True return True
async def _download_remote_binaries_for_plugin_with_name(self, pluginBasePath): async def _download_remote_binaries_for_plugin_with_name(self, pluginBasePath):
rv = False rv = False
try: try:
packageJsonPath = path.join(pluginBasePath, 'package.json') packageJsonPath = path.join(pluginBasePath, "package.json")
pluginBinPath = path.join(pluginBasePath, 'bin') pluginBinPath = path.join(pluginBasePath, "bin")
if access(packageJsonPath, R_OK): if access(packageJsonPath, R_OK):
with open(packageJsonPath, "r", encoding="utf-8") as f: with open(packageJsonPath, "r", encoding="utf-8") as f:
packageJson = json.load(f) packageJson = json.load(f)
if "remote_binary" in packageJson and len(packageJson["remote_binary"]) > 0: if (
"remote_binary" in packageJson
and len(packageJson["remote_binary"]) > 0
):
# create bin directory if needed. # create bin directory if needed.
rc=call(["chmod", "-R", "777", pluginBasePath]) call(["chmod", "-R", "777", pluginBasePath])
if access(pluginBasePath, W_OK): if access(pluginBasePath, W_OK):
if not path.exists(pluginBinPath): if not path.exists(pluginBinPath):
mkdir(pluginBinPath) mkdir(pluginBinPath)
if not access(pluginBinPath, W_OK): if not access(pluginBinPath, W_OK):
rc=call(["chmod", "-R", "777", pluginBinPath]) call(["chmod", "-R", "777", pluginBinPath])
rv = True rv = True
for remoteBinary in packageJson["remote_binary"]: for remoteBinary in packageJson["remote_binary"]:
@@ -76,16 +91,29 @@ class PluginBrowser:
binName = remoteBinary["name"] binName = remoteBinary["name"]
binURL = remoteBinary["url"] binURL = remoteBinary["url"]
binHash = remoteBinary["sha256hash"] binHash = remoteBinary["sha256hash"]
if not await download_remote_binary_to_path(binURL, binHash, path.join(pluginBinPath, binName)): if not await download_remote_binary_to_path(
binURL, binHash, path.join(pluginBinPath, binName)
):
rv = False rv = False
raise Exception(f"Error Downloading Remote Binary {binName}@{binURL} with hash {binHash} to {path.join(pluginBinPath, binName)}") raise Exception(
"Error Downloading Remote Binary"
f" {binName}@{binURL} with hash {binHash} to"
f" {path.join(pluginBinPath, binName)}"
)
code_chown = call(["chown", "-R", get_user()+":"+get_user_group(), self.plugin_path]) call(
rc=call(["chmod", "-R", "555", pluginBasePath]) [
"chown",
"-R",
get_user() + ":" + get_user_group(),
self.plugin_path,
]
)
call(["chmod", "-R", "555", pluginBasePath])
else: else:
rv = True rv = True
logger.debug(f"No Remote Binaries to Download") logger.debug("No Remote Binaries to Download")
except Exception as e: except Exception as e:
rv = False rv = False
logger.debug(str(e)) logger.debug(str(e))
@@ -95,12 +123,16 @@ class PluginBrowser:
def find_plugin_folder(self, name): def find_plugin_folder(self, name):
for folder in listdir(self.plugin_path): for folder in listdir(self.plugin_path):
try: try:
with open(path.join(self.plugin_path, folder, 'plugin.json'), "r", encoding="utf-8") as f: with open(
path.join(self.plugin_path, folder, "plugin.json"),
"r",
encoding="utf-8",
) as f:
plugin = json.load(f) plugin = json.load(f)
if plugin['name'] == name: if plugin["name"] == name:
return str(path.join(self.plugin_path, folder)) return str(path.join(self.plugin_path, folder))
except: except Exception:
logger.debug(f"skipping {folder}") logger.debug(f"skipping {folder}")
async def uninstall_plugin(self, name): async def uninstall_plugin(self, name):
@@ -127,8 +159,10 @@ class PluginBrowser:
except FileNotFoundError: except FileNotFoundError:
logger.warning(f"Plugin {name} not installed, skipping uninstallation") logger.warning(f"Plugin {name} not installed, skipping uninstallation")
except Exception as e: except Exception as e:
logger.error(f"Plugin {name} in {self.find_plugin_folder(name)} was not uninstalled") logger.error(
logger.error(f"Error at %s", exc_info=e) f"Plugin {name} in {self.find_plugin_folder(name)} was not uninstalled"
)
logger.error("Error at %s", exc_info=e)
if self.loader.watcher: if self.loader.watcher:
self.loader.watcher.disabled = False self.loader.watcher.disabled = False
@@ -140,8 +174,11 @@ class PluginBrowser:
pluginFolderPath = self.find_plugin_folder(name) pluginFolderPath = self.find_plugin_folder(name)
if pluginFolderPath: if pluginFolderPath:
isInstalled = True isInstalled = True
except: except Exception:
logger.error(f"Failed to determine if {name} is already installed, continuing anyway.") logger.error(
f"Failed to determine if {name} is already installed, continuing"
" anyway."
)
logger.info(f"Installing {name} (Version: {version})") logger.info(f"Installing {name} (Version: {version})")
async with ClientSession() as client: async with ClientSession() as client:
logger.debug(f"Fetching {artifact}") logger.debug(f"Fetching {artifact}")
@@ -155,22 +192,26 @@ class PluginBrowser:
try: try:
logger.debug("Uninstalling existing plugin...") logger.debug("Uninstalling existing plugin...")
await self.uninstall_plugin(name) await self.uninstall_plugin(name)
except: except Exception:
logger.error(f"Plugin {name} could not be uninstalled.") logger.error(f"Plugin {name} could not be uninstalled.")
logger.debug("Unzipping...") logger.debug("Unzipping...")
ret = self._unzip_to_plugin_dir(res_zip, name, hash) ret = self._unzip_to_plugin_dir(res_zip, name, hash)
if ret: if ret:
plugin_dir = self.find_plugin_folder(name) plugin_dir = self.find_plugin_folder(name)
ret = await self._download_remote_binaries_for_plugin_with_name(plugin_dir) ret = await self._download_remote_binaries_for_plugin_with_name(
plugin_dir
)
if ret: if ret:
logger.info(f"Installed {name} (Version: {version})") logger.info(f"Installed {name} (Version: {version})")
if name in self.loader.plugins: if name in self.loader.plugins:
self.loader.plugins[name].stop() self.loader.plugins[name].stop()
self.loader.plugins.pop(name, None) self.loader.plugins.pop(name, None)
await sleep(1) await sleep(1)
self.loader.import_plugin(path.join(plugin_dir, "main.py"), plugin_dir) self.loader.import_plugin(
path.join(plugin_dir, "main.py"), plugin_dir
)
else: else:
logger.fatal(f"Failed Downloading Remote Binaries") logger.fatal("Failed Downloading Remote Binaries")
else: else:
self.log.fatal(f"SHA-256 Mismatch!!!! {name} (Version: {version})") self.log.fatal(f"SHA-256 Mismatch!!!! {name} (Version: {version})")
if self.loader.watcher: if self.loader.watcher:
@@ -180,14 +221,21 @@ class PluginBrowser:
async def request_plugin_install(self, artifact, name, version, hash): async def request_plugin_install(self, artifact, name, version, hash):
request_id = str(time()) request_id = str(time())
self.install_requests[request_id] = PluginInstallContext(artifact, name, version, hash) self.install_requests[request_id] = PluginInstallContext(
artifact, name, version, hash
)
tab = await get_gamepadui_tab() tab = await get_gamepadui_tab()
await tab.open_websocket() await tab.open_websocket()
await tab.evaluate_js(f"DeckyPluginLoader.addPluginInstallPrompt('{name}', '{version}', '{request_id}', '{hash}')") await tab.evaluate_js(
f"DeckyPluginLoader.addPluginInstallPrompt('{name}', '{version}',"
f" '{request_id}', '{hash}')"
)
async def confirm_plugin_install(self, request_id): async def confirm_plugin_install(self, request_id):
request = self.install_requests.pop(request_id) request = self.install_requests.pop(request_id)
await self._install(request.artifact, request.name, request.version, request.hash) await self._install(
request.artifact, request.name, request.version, request.hash
)
def cancel_plugin_install(self, request_id): def cancel_plugin_install(self, request_id):
self.install_requests.pop(request_id) self.install_requests.pop(request_id)
+61 -29
View File
@@ -6,8 +6,6 @@ import subprocess
import uuid import uuid
import os import os
import sys import sys
from subprocess import check_output
from time import sleep
from hashlib import sha256 from hashlib import sha256
from io import BytesIO from io import BytesIO
@@ -24,22 +22,38 @@ ssl_ctx = ssl.create_default_context(cafile=certifi.where())
assets_regex = re.compile("^/plugins/.*/assets/.*") assets_regex = re.compile("^/plugins/.*/assets/.*")
frontend_regex = re.compile("^/frontend/.*") frontend_regex = re.compile("^/frontend/.*")
def get_ssl_context(): def get_ssl_context():
return ssl_ctx return ssl_ctx
def get_csrf_token(): def get_csrf_token():
return csrf_token return csrf_token
@middleware @middleware
async def csrf_middleware(request, handler): async def csrf_middleware(request, handler):
if str(request.method) == "OPTIONS" or request.headers.get('Authentication') == csrf_token or str(request.rel_url) == "/auth/token" or str(request.rel_url).startswith("/plugins/load_main/") or str(request.rel_url).startswith("/static/") or str(request.rel_url).startswith("/legacy/") or str(request.rel_url).startswith("/steam_resource/") or str(request.rel_url).startswith("/frontend/") or assets_regex.match(str(request.rel_url)) or frontend_regex.match(str(request.rel_url)): if (
str(request.method) == "OPTIONS"
or request.headers.get("Authentication") == csrf_token
or str(request.rel_url) == "/auth/token"
or str(request.rel_url).startswith("/plugins/load_main/")
or str(request.rel_url).startswith("/static/")
or str(request.rel_url).startswith("/legacy/")
or str(request.rel_url).startswith("/steam_resource/")
or str(request.rel_url).startswith("/frontend/")
or assets_regex.match(str(request.rel_url))
or frontend_regex.match(str(request.rel_url))
):
return await handler(request) return await handler(request)
return Response(text='Forbidden', status='403') return Response(text="Forbidden", status="403")
# Deprecated # Deprecated
def set_user(): def set_user():
pass pass
# Get the user id hosting the plugin loader # Get the user id hosting the plugin loader
def get_user_id() -> int: def get_user_id() -> int:
proc_path = os.path.realpath(sys.argv[0]) proc_path = os.path.realpath(sys.argv[0])
@@ -47,60 +61,73 @@ def get_user_id() -> int:
for pw in pws: for pw in pws:
if proc_path.startswith(os.path.realpath(pw.pw_dir)): if proc_path.startswith(os.path.realpath(pw.pw_dir)):
return pw.pw_uid return pw.pw_uid
raise PermissionError("The plugin loader does not seem to be hosted by any known user.") raise PermissionError(
"The plugin loader does not seem to be hosted by any known user."
)
# Get the user hosting the plugin loader # Get the user hosting the plugin loader
def get_user() -> str: def get_user() -> str:
return pwd.getpwuid(get_user_id()).pw_name return pwd.getpwuid(get_user_id()).pw_name
# Get the effective user id of the running process # Get the effective user id of the running process
def get_effective_user_id() -> int: def get_effective_user_id() -> int:
return os.geteuid() return os.geteuid()
# Get the effective user of the running process # Get the effective user of the running process
def get_effective_user() -> str: def get_effective_user() -> str:
return pwd.getpwuid(get_effective_user_id()).pw_name return pwd.getpwuid(get_effective_user_id()).pw_name
# Get the effective user group id of the running process # Get the effective user group id of the running process
def get_effective_user_group_id() -> int: def get_effective_user_group_id() -> int:
return os.getegid() return os.getegid()
# Get the effective user group of the running process # Get the effective user group of the running process
def get_effective_user_group() -> str: def get_effective_user_group() -> str:
return grp.getgrgid(get_effective_user_group_id()).gr_name return grp.getgrgid(get_effective_user_group_id()).gr_name
# Get the user owner of the given file path. # Get the user owner of the given file path.
def get_user_owner(file_path) -> str: def get_user_owner(file_path) -> str:
return pwd.getpwuid(os.stat(file_path).st_uid).pw_name return pwd.getpwuid(os.stat(file_path).st_uid).pw_name
# Get the user group of the given file path.
def get_user_group(file_path) -> str:
return grp.getgrgid(os.stat(file_path).st_gid).gr_name
# Deprecated # Deprecated
def set_user_group() -> str: def set_user_group() -> str:
return get_user_group() return get_user_group()
# Get the group id of the user hosting the plugin loader # Get the group id of the user hosting the plugin loader
def get_user_group_id() -> int: def get_user_group_id() -> int:
return pwd.getpwuid(get_user_id()).pw_gid return pwd.getpwuid(get_user_id()).pw_gid
# Get the group of the user hosting the plugin loader # Get the group of the user hosting the plugin loader
def get_user_group() -> str: def get_user_group(file_path) -> str:
return grp.getgrgid(get_user_group_id()).gr_name if file_path:
return grp.getgrgid(os.stat(file_path).st_gid).gr_name
else:
return grp.getgrgid(get_user_group_id()).gr_name
# Get the default home path unless a user is specified # Get the default home path unless a user is specified
def get_home_path(username = None) -> str: def get_home_path(username=None) -> str:
if username == None: if username is None:
username = get_user() username = get_user()
return pwd.getpwnam(username).pw_dir return pwd.getpwnam(username).pw_dir
# Get the default homebrew path unless a home_path is specified # Get the default homebrew path unless a home_path is specified
def get_homebrew_path(home_path = None) -> str: def get_homebrew_path(home_path=None) -> str:
if home_path == None: if home_path is None:
home_path = get_home_path() home_path = get_home_path()
return os.path.join(home_path, "homebrew") return os.path.join(home_path, "homebrew")
# Recursively create path and chown as user # Recursively create path and chown as user
def mkdir_as_user(path): def mkdir_as_user(path):
path = os.path.realpath(path) path = os.path.realpath(path)
@@ -113,20 +140,16 @@ def mkdir_as_user(path):
chown_path = os.path.join(chown_path, p) chown_path = os.path.join(chown_path, p)
os.chown(chown_path, uid, gid) os.chown(chown_path, uid, gid)
# Fetches the version of loader # Fetches the version of loader
def get_loader_version() -> str: def get_loader_version() -> str:
try: with open(
with open(os.path.join(os.getcwd(), ".loader.version"), "r", encoding="utf-8") as version_file: os.path.join(os.path.dirname(sys.argv[0]), ".loader.version"),
return version_file.readline().strip() "r",
except: encoding="utf-8",
return "unknown" ) as version_file:
return version_file.readline().replace("\n", "")
# returns the appropriate system python paths
def get_system_pythonpaths() -> list[str]:
# run as normal normal user to also include user python paths
proc = subprocess.run(["python3", "-c", "import sys; print(':'.join(x for x in sys.path if x))"],
user=get_user_id(), env={}, capture_output=True)
return proc.stdout.decode().strip().split(":")
# Download Remote Binaries to local Plugin # Download Remote Binaries to local Plugin
async def download_remote_binary_to_path(url, binHash, path) -> bool: async def download_remote_binary_to_path(url, binHash, path) -> bool:
@@ -140,27 +163,36 @@ async def download_remote_binary_to_path(url, binHash, path) -> bool:
remoteHash = sha256(data.getbuffer()).hexdigest() remoteHash = sha256(data.getbuffer()).hexdigest()
if binHash == remoteHash: if binHash == remoteHash:
data.seek(0) data.seek(0)
with open(path, 'wb') as f: with open(path, "wb") as f:
f.write(data.getbuffer()) f.write(data.getbuffer())
rv = True rv = True
else: else:
raise Exception(f"Fatal Error: Hash Mismatch for remote binary {path}@{url}") raise Exception(
f"Fatal Error: Hash Mismatch for remote binary {path}@{url}"
)
else: else:
rv = False rv = False
except: except Exception:
rv = False rv = False
return rv return rv
async def is_systemd_unit_active(unit_name: str) -> bool: async def is_systemd_unit_active(unit_name: str) -> bool:
res = subprocess.run(["systemctl", "is-active", unit_name], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) res = subprocess.run(
["systemctl", "is-active", unit_name],
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL,
)
return res.returncode == 0 return res.returncode == 0
async def stop_systemd_unit(unit_name: str) -> subprocess.CompletedProcess: async def stop_systemd_unit(unit_name: str) -> subprocess.CompletedProcess:
cmd = ["systemctl", "stop", unit_name] cmd = ["systemctl", "stop", unit_name]
return subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) return subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
async def start_systemd_unit(unit_name: str) -> subprocess.CompletedProcess: async def start_systemd_unit(unit_name: str) -> subprocess.CompletedProcess:
cmd = ["systemctl", "start", unit_name] cmd = ["systemctl", "start", unit_name]
+178 -122
View File
@@ -2,10 +2,9 @@
from asyncio import sleep from asyncio import sleep
from logging import getLogger from logging import getLogger
from traceback import format_exc
from typing import List from typing import List
from aiohttp import ClientSession, WSMsgType from aiohttp import ClientSession
from aiohttp.client_exceptions import ClientConnectorError, ClientOSError from aiohttp.client_exceptions import ClientConnectorError, ClientOSError
from asyncio.exceptions import TimeoutError from asyncio.exceptions import TimeoutError
import uuid import uuid
@@ -39,9 +38,12 @@ class Tab:
async for message in self.websocket: async for message in self.websocket:
data = message.json() data = message.json()
yield data yield data
logger.warn(f"The Tab {self.title} socket has been disconnected while listening for messages.") logger.warn(
f"The Tab {self.title} socket has been disconnected while listening for"
" messages."
)
await self.close_websocket() await self.close_websocket()
async def _send_devtools_cmd(self, dc, receive=True): async def _send_devtools_cmd(self, dc, receive=True):
if self.websocket: if self.websocket:
self.cmd_id += 1 self.cmd_id += 1
@@ -54,19 +56,24 @@ class Tab:
return None return None
raise RuntimeError("Websocket not opened") raise RuntimeError("Websocket not opened")
async def evaluate_js(self, js, run_async=False, manage_socket=True, get_result=True): async def evaluate_js(
self, js, run_async=False, manage_socket=True, get_result=True
):
try: try:
if manage_socket: if manage_socket:
await self.open_websocket() await self.open_websocket()
res = await self._send_devtools_cmd({ res = await self._send_devtools_cmd(
"method": "Runtime.evaluate", {
"params": { "method": "Runtime.evaluate",
"expression": js, "params": {
"userGesture": True, "expression": js,
"awaitPromise": run_async "userGesture": True,
} "awaitPromise": run_async,
}, get_result) },
},
get_result,
)
finally: finally:
if manage_socket: if manage_socket:
@@ -74,9 +81,17 @@ class Tab:
return res return res
async def has_global_var(self, var_name, manage_socket=True): async def has_global_var(self, var_name, manage_socket=True):
res = await self.evaluate_js(f"window['{var_name}'] !== null && window['{var_name}'] !== undefined", False, manage_socket) res = await self.evaluate_js(
f"window['{var_name}'] !== null && window['{var_name}'] !== undefined",
False,
manage_socket,
)
if not "result" in res or not "result" in res["result"] or not "value" in res["result"]["result"]: if (
"result" not in res
or "result" not in res["result"]
or "value" not in res["result"]["result"]
):
return False return False
return res["result"]["result"]["value"] return res["result"]["result"]["value"]
@@ -86,9 +101,12 @@ class Tab:
if manage_socket: if manage_socket:
await self.open_websocket() await self.open_websocket()
res = await self._send_devtools_cmd({ res = await self._send_devtools_cmd(
"method": "Page.close", {
}, False) "method": "Page.close",
},
False,
)
finally: finally:
if manage_socket: if manage_socket:
@@ -99,32 +117,42 @@ class Tab:
""" """
Enables page domain notifications. Enables page domain notifications.
""" """
await self._send_devtools_cmd({ await self._send_devtools_cmd(
"method": "Page.enable", {
}, False) "method": "Page.enable",
},
False,
)
async def disable(self): async def disable(self):
""" """
Disables page domain notifications. Disables page domain notifications.
""" """
await self._send_devtools_cmd({ await self._send_devtools_cmd(
"method": "Page.disable", {
}, False) "method": "Page.disable",
},
False,
)
async def refresh(self): async def refresh(self, manage_socket=False):
try: try:
if manage_socket: if manage_socket:
await self.open_websocket() await self.open_websocket()
await self._send_devtools_cmd({ await self._send_devtools_cmd(
"method": "Page.reload", {
}, False) "method": "Page.reload",
},
False,
)
finally: finally:
if manage_socket: if manage_socket:
await self.close_websocket() await self.close_websocket()
return return
async def reload_and_evaluate(self, js, manage_socket=True): async def reload_and_evaluate(self, js, manage_socket=True):
""" """
Reloads the current tab, with JS to run on load via debugger Reloads the current tab, with JS to run on load via debugger
@@ -133,64 +161,70 @@ class Tab:
if manage_socket: if manage_socket:
await self.open_websocket() await self.open_websocket()
await self._send_devtools_cmd({ await self._send_devtools_cmd({"method": "Debugger.enable"}, True)
"method": "Debugger.enable"
}, True)
await self._send_devtools_cmd({ await self._send_devtools_cmd(
"method": "Runtime.evaluate", {
"params": { "method": "Runtime.evaluate",
"expression": "location.reload();", "params": {
"userGesture": True, "expression": "location.reload();",
"awaitPromise": False "userGesture": True,
} "awaitPromise": False,
}, False) },
},
False,
)
breakpoint_res = await self._send_devtools_cmd({ breakpoint_res = await self._send_devtools_cmd(
"method": "Debugger.setInstrumentationBreakpoint", {
"params": { "method": "Debugger.setInstrumentationBreakpoint",
"instrumentation": "beforeScriptExecution" "params": {"instrumentation": "beforeScriptExecution"},
} },
}, True) True,
)
logger.info(breakpoint_res) logger.info(breakpoint_res)
# Page finishes loading when breakpoint hits # Page finishes loading when breakpoint hits
for x in range(20): for x in range(20):
# this works around 1/5 of the time, so just send it 8 times. # this works around 1/5 of the time, so just send it 8 times.
# the js accounts for being injected multiple times allowing only one instance to run at a time anyway # the js accounts for being injected multiple times allowing only one instance to run at a time anyway
await self._send_devtools_cmd({ await self._send_devtools_cmd(
"method": "Runtime.evaluate", {
"params": { "method": "Runtime.evaluate",
"expression": js, "params": {
"userGesture": True, "expression": js,
"awaitPromise": False "userGesture": True,
} "awaitPromise": False,
}, False) },
},
False,
)
await self._send_devtools_cmd({ await self._send_devtools_cmd(
"method": "Debugger.removeBreakpoint", {
"params": { "method": "Debugger.removeBreakpoint",
"breakpointId": breakpoint_res["result"]["breakpointId"] "params": {
} "breakpointId": breakpoint_res["result"]["breakpointId"]
}, False) },
},
False,
)
for x in range(4): for x in range(4):
await self._send_devtools_cmd({ await self._send_devtools_cmd({"method": "Debugger.resume"}, False)
"method": "Debugger.resume"
}, False)
await self._send_devtools_cmd({ await self._send_devtools_cmd({"method": "Debugger.disable"}, True)
"method": "Debugger.disable"
}, True)
finally: finally:
if manage_socket: if manage_socket:
await self.close_websocket() await self.close_websocket()
return return
async def add_script_to_evaluate_on_new_document(self, js, add_dom_wrapper=True, manage_socket=True, get_result=True): async def add_script_to_evaluate_on_new_document(
self, js, add_dom_wrapper=True, manage_socket=True, get_result=True
):
""" """
How the underlying call functions is not particularly clear from the devtools docs, so stealing puppeteer's description: How the underlying call functions is not particularly clear from the devtools docs, so stealing puppeteer's description:
@@ -225,35 +259,44 @@ class Tab:
""" """
try: try:
wrappedjs = """ wrappedjs = (
function scriptFunc() { """
function scriptFunc() {{
{js} {js}
} }}
if (document.readyState === 'loading') { if (document.readyState === 'loading') {{
addEventListener('DOMContentLoaded', () => { addEventListener('DOMContentLoaded', () => {{
scriptFunc(); scriptFunc();
}); }});
} else { }} else {{
scriptFunc(); scriptFunc();
} }}
""".format(js=js) if add_dom_wrapper else js """.format(
js=js
)
if add_dom_wrapper
else js
)
if manage_socket: if manage_socket:
await self.open_websocket() await self.open_websocket()
res = await self._send_devtools_cmd({ res = await self._send_devtools_cmd(
"method": "Page.addScriptToEvaluateOnNewDocument", {
"params": { "method": "Page.addScriptToEvaluateOnNewDocument",
"source": wrappedjs "params": {"source": wrappedjs},
} },
}, get_result) get_result,
)
finally: finally:
if manage_socket: if manage_socket:
await self.close_websocket() await self.close_websocket()
return res return res
async def remove_script_to_evaluate_on_new_document(self, script_id, manage_socket=True): async def remove_script_to_evaluate_on_new_document(
self, script_id, manage_socket=True
):
""" """
Removes a script from a page that was added with `add_script_to_evaluate_on_new_document` Removes a script from a page that was added with `add_script_to_evaluate_on_new_document`
@@ -267,21 +310,28 @@ class Tab:
if manage_socket: if manage_socket:
await self.open_websocket() await self.open_websocket()
res = await self._send_devtools_cmd({ await self._send_devtools_cmd(
"method": "Page.removeScriptToEvaluateOnNewDocument", {
"params": { "method": "Page.removeScriptToEvaluateOnNewDocument",
"identifier": script_id "params": {"identifier": script_id},
} },
}, False) False,
)
finally: finally:
if manage_socket: if manage_socket:
await self.close_websocket() await self.close_websocket()
async def has_element(self, element_name, manage_socket=True): async def has_element(self, element_name, manage_socket=True):
res = await self.evaluate_js(f"document.getElementById('{element_name}') != null", False, manage_socket) res = await self.evaluate_js(
f"document.getElementById('{element_name}') != null", False, manage_socket
)
if not "result" in res or not "result" in res["result"] or not "value" in res["result"]["result"]: if (
"result" not in res
or "result" not in res["result"]
or "value" not in res["result"]["result"]
):
return False return False
return res["result"]["result"]["value"] return res["result"]["result"]["value"]
@@ -298,23 +348,17 @@ class Tab:
document.head.append(style); document.head.append(style);
style.textContent = `{style}`; style.textContent = `{style}`;
}})() }})()
""", False, manage_socket) """,
False,
manage_socket,
)
if "exceptionDetails" in result["result"]: if "exceptionDetails" in result["result"]:
return { return {"success": False, "result": result["result"]}
"success": False,
"result": result["result"]
}
return { return {"success": True, "result": css_id}
"success": True,
"result": css_id
}
except Exception as e: except Exception as e:
return { return {"success": False, "result": e}
"success": False,
"result": e
}
async def remove_css(self, css_id, manage_socket=True): async def remove_css(self, css_id, manage_socket=True):
try: try:
@@ -326,25 +370,24 @@ class Tab:
if (style.nodeName.toLowerCase() == 'style') if (style.nodeName.toLowerCase() == 'style')
style.parentNode.removeChild(style); style.parentNode.removeChild(style);
}})() }})()
""", False, manage_socket) """,
False,
manage_socket,
)
if "exceptionDetails" in result["result"]: if "exceptionDetails" in result["result"]:
return { return {"success": False, "result": result}
"success": False,
"result": result
}
return { return {"success": True}
"success": True
}
except Exception as e: except Exception as e:
return { return {"success": False, "result": e}
"success": False,
"result": e
}
async def get_steam_resource(self, url): async def get_steam_resource(self, url):
res = await self.evaluate_js(f'(async function test() {{ return await (await fetch("{url}")).text() }})()', True) res = await self.evaluate_js(
f'(async function test() {{ return await (await fetch("{url}")).text()'
" })()",
True,
)
return res["result"]["result"]["value"] return res["result"]["result"]["value"]
def __repr__(self): def __repr__(self):
@@ -387,32 +430,45 @@ async def get_tab(tab_name) -> Tab:
raise ValueError(f"Tab {tab_name} not found") raise ValueError(f"Tab {tab_name} not found")
return tab return tab
async def get_tab_lambda(test) -> Tab: async def get_tab_lambda(test) -> Tab:
tabs = await get_tabs() tabs = await get_tabs()
tab = next((i for i in tabs if test(i)), None) tab = next((i for i in tabs if test(i)), None)
if not tab: if not tab:
raise ValueError(f"Tab not found by lambda") raise ValueError("Tab not found by lambda")
return tab return tab
def tab_is_gamepadui(t: Tab) -> bool: def tab_is_gamepadui(t: Tab) -> bool:
return "https://steamloopback.host/routes/" in t.url and (t.title == "Steam Shared Context presented by Valve™" or t.title == "Steam" or t.title == "SP") return "https://steamloopback.host/routes/" in t.url and (
t.title == "Steam Shared Context presented by Valve™"
or t.title == "Steam"
or t.title == "SP"
)
async def get_gamepadui_tab() -> Tab: async def get_gamepadui_tab() -> Tab:
tabs = await get_tabs() tabs = await get_tabs()
tab = next((i for i in tabs if tab_is_gamepadui(i)), None) tab = next((i for i in tabs if tab_is_gamepadui(i)), None)
if not tab: if not tab:
raise ValueError(f"GamepadUI Tab not found") raise ValueError("GamepadUI Tab not found")
return tab return tab
async def inject_to_tab(tab_name, js, run_async=False): async def inject_to_tab(tab_name, js, run_async=False):
tab = await get_tab(tab_name) tab = await get_tab(tab_name)
return await tab.evaluate_js(js, run_async) return await tab.evaluate_js(js, run_async)
async def close_old_tabs(): async def close_old_tabs():
tabs = await get_tabs() tabs = await get_tabs()
for t in tabs: for t in tabs:
if not t.title or (t.title != "Steam Shared Context presented by Valve™" and t.title != "Steam" and t.title != "SP"): if not t.title or (
t.title != "Steam Shared Context presented by Valve™"
and t.title != "Steam"
and t.title != "SP"
):
logger.debug("Closing tab: " + getattr(t, "title", "Untitled")) logger.debug("Closing tab: " + getattr(t, "title", "Untitled"))
await t.close() await t.close()
await sleep(0.5) await sleep(0.5)
+102 -35
View File
@@ -6,9 +6,14 @@ from pathlib import Path
from traceback import print_exc from traceback import print_exc
from aiohttp import web from aiohttp import web
from os.path import exists from genericpath import exists
from watchdog.events import RegexMatchingEventHandler from watchdog.events import RegexMatchingEventHandler
from watchdog.observers import Observer from watchdog.utils import UnsupportedLibc
try:
from watchdog.observers.inotify import InotifyObserver as Observer
except UnsupportedLibc:
from watchdog.observers.fsevents import FSEventsObserver as Observer
from injector import get_tab, get_gamepadui_tab from injector import get_tab, get_gamepadui_tab
from plugin import PluginWrapper from plugin import PluginWrapper
@@ -16,7 +21,7 @@ from plugin import PluginWrapper
class FileChangeHandler(RegexMatchingEventHandler): class FileChangeHandler(RegexMatchingEventHandler):
def __init__(self, queue, plugin_path) -> None: def __init__(self, queue, plugin_path) -> None:
super().__init__(regexes=[r'^.*?dist\/index\.js$', r'^.*?main\.py$']) super().__init__(regexes=[r"^.*?dist\/index\.js$", r"^.*?main\.py$"])
self.logger = getLogger("file-watcher") self.logger = getLogger("file-watcher")
self.plugin_path = plugin_path self.plugin_path = plugin_path
self.queue = queue self.queue = queue
@@ -27,7 +32,9 @@ class FileChangeHandler(RegexMatchingEventHandler):
return return
plugin_dir = Path(path.relpath(src_path, self.plugin_path)).parts[0] plugin_dir = Path(path.relpath(src_path, self.plugin_path)).parts[0]
if exists(path.join(self.plugin_path, plugin_dir, "plugin.json")): if exists(path.join(self.plugin_path, plugin_dir, "plugin.json")):
self.queue.put_nowait((path.join(self.plugin_path, plugin_dir, "main.py"), plugin_dir, True)) self.queue.put_nowait(
(path.join(self.plugin_path, plugin_dir, "main.py"), plugin_dir, True)
)
def on_created(self, event): def on_created(self, event):
src_path = event.src_path src_path = event.src_path
@@ -57,6 +64,7 @@ class FileChangeHandler(RegexMatchingEventHandler):
self.logger.debug(f"file modified: {src_path}") self.logger.debug(f"file modified: {src_path}")
self.maybe_reload(src_path) self.maybe_reload(src_path)
class Loader: class Loader:
def __init__(self, server_instance, plugin_path, loop, live_reload=False) -> None: def __init__(self, server_instance, plugin_path, loop, live_reload=False) -> None:
self.loop = loop self.loop = loop
@@ -76,18 +84,30 @@ class Loader:
self.loop.create_task(self.handle_reloads()) self.loop.create_task(self.handle_reloads())
self.loop.create_task(self.enable_reload_wait()) self.loop.create_task(self.enable_reload_wait())
server_instance.add_routes([ server_instance.add_routes(
web.get("/frontend/{path:.*}", self.handle_frontend_assets), [
web.get("/plugins", self.get_plugins), web.get("/frontend/{path:.*}", self.handle_frontend_assets),
web.get("/plugins/{plugin_name}/frontend_bundle", self.handle_frontend_bundle), web.get("/plugins", self.get_plugins),
web.post("/plugins/{plugin_name}/methods/{method_name}", self.handle_plugin_method_call), web.get(
web.get("/plugins/{plugin_name}/assets/{path:.*}", self.handle_plugin_frontend_assets), "/plugins/{plugin_name}/frontend_bundle",
self.handle_frontend_bundle,
# The following is legacy plugin code. ),
web.get("/plugins/load_main/{name}", self.load_plugin_main_view), web.post(
web.get("/plugins/plugin_resource/{name}/{path:.+}", self.handle_sub_route), "/plugins/{plugin_name}/methods/{method_name}",
web.get("/steam_resource/{path:.+}", self.get_steam_resource) self.handle_plugin_method_call,
]) ),
web.get(
"/plugins/{plugin_name}/assets/{path:.*}",
self.handle_plugin_frontend_assets,
),
# The following is legacy plugin code.
web.get("/plugins/load_main/{name}", self.load_plugin_main_view),
web.get(
"/plugins/plugin_resource/{name}/{path:.+}", self.handle_sub_route
),
web.get("/steam_resource/{path:.+}", self.get_steam_resource),
]
)
async def enable_reload_wait(self): async def enable_reload_wait(self):
if self.live_reload: if self.live_reload:
@@ -102,36 +122,63 @@ class Loader:
async def get_plugins(self, request): async def get_plugins(self, request):
plugins = list(self.plugins.values()) plugins = list(self.plugins.values())
return web.json_response([{"name": str(i) if not i.legacy else "$LEGACY_"+str(i), "version": i.version} for i in plugins]) return web.json_response(
[
{
"name": str(i) if not i.legacy else "$LEGACY_" + str(i),
"version": i.version,
}
for i in plugins
]
)
def handle_plugin_frontend_assets(self, request): def handle_plugin_frontend_assets(self, request):
plugin = self.plugins[request.match_info["plugin_name"]] plugin = self.plugins[request.match_info["plugin_name"]]
file = path.join(self.plugin_path, plugin.plugin_directory, "dist/assets", request.match_info["path"]) file = path.join(
self.plugin_path,
plugin.plugin_directory,
"dist/assets",
request.match_info["path"],
)
return web.FileResponse(file, headers={"Cache-Control": "no-cache"}) return web.FileResponse(file, headers={"Cache-Control": "no-cache"})
def handle_frontend_bundle(self, request): def handle_frontend_bundle(self, request):
plugin = self.plugins[request.match_info["plugin_name"]] plugin = self.plugins[request.match_info["plugin_name"]]
with open(path.join(self.plugin_path, plugin.plugin_directory, "dist/index.js"), "r", encoding="utf-8") as bundle: with open(
return web.Response(text=bundle.read(), content_type="application/javascript") path.join(self.plugin_path, plugin.plugin_directory, "dist/index.js"),
"r",
encoding="utf-8",
) as bundle:
return web.Response(
text=bundle.read(), content_type="application/javascript"
)
def import_plugin(self, file, plugin_directory, refresh=False, batch=False): def import_plugin(self, file, plugin_directory, refresh=False, batch=False):
try: try:
plugin = PluginWrapper(file, plugin_directory, self.plugin_path) plugin = PluginWrapper(file, plugin_directory, self.plugin_path)
if plugin.name in self.plugins: if plugin.name in self.plugins:
if not "debug" in plugin.flags and refresh: if "debug" not in plugin.flags and refresh:
self.logger.info(f"Plugin {plugin.name} is already loaded and has requested to not be re-loaded") self.logger.info(
return f"Plugin {plugin.name} is already loaded and has requested to"
else: " not be re-loaded"
self.plugins[plugin.name].stop() )
self.plugins.pop(plugin.name, None) return
else:
self.plugins[plugin.name].stop()
self.plugins.pop(plugin.name, None)
if plugin.passive: if plugin.passive:
self.logger.info(f"Plugin {plugin.name} is passive") self.logger.info(f"Plugin {plugin.name} is passive")
self.plugins[plugin.name] = plugin.start() self.plugins[plugin.name] = plugin.start()
self.logger.info(f"Loaded {plugin.name}") self.logger.info(f"Loaded {plugin.name}")
if not batch: if not batch:
self.loop.create_task(self.dispatch_plugin(plugin.name if not plugin.legacy else "$LEGACY_" + plugin.name, plugin.version)) self.loop.create_task(
self.dispatch_plugin(
plugin.name if not plugin.legacy else "$LEGACY_" + plugin.name,
plugin.version,
)
)
except Exception as e: except Exception as e:
self.logger.error(f"Could not load {file}. {e}") self.logger.error(f"Could not load {file}. {e}")
print_exc() print_exc()
@@ -143,10 +190,20 @@ class Loader:
def import_plugins(self): def import_plugins(self):
self.logger.info(f"import plugins from {self.plugin_path}") self.logger.info(f"import plugins from {self.plugin_path}")
directories = [i for i in listdir(self.plugin_path) if path.isdir(path.join(self.plugin_path, i)) and path.isfile(path.join(self.plugin_path, i, "plugin.json"))] directories = [
i
for i in listdir(self.plugin_path)
if path.isdir(path.join(self.plugin_path, i))
and path.isfile(path.join(self.plugin_path, i, "plugin.json"))
]
for directory in directories: for directory in directories:
self.logger.info(f"found plugin: {directory}") self.logger.info(f"found plugin: {directory}")
self.import_plugin(path.join(self.plugin_path, directory, "main.py"), directory, False, True) self.import_plugin(
path.join(self.plugin_path, directory, "main.py"),
directory,
False,
True,
)
async def handle_reloads(self): async def handle_reloads(self):
while True: while True:
@@ -163,10 +220,10 @@ class Loader:
except JSONDecodeError: except JSONDecodeError:
args = {} args = {}
try: try:
if method_name.startswith("_"): if method_name.startswith("_"):
raise RuntimeError("Tried to call private method") raise RuntimeError("Tried to call private method")
res["result"] = await plugin.execute_method(method_name, args) res["result"] = await plugin.execute_method(method_name, args)
res["success"] = True res["success"] = True
except Exception as e: except Exception as e:
res["result"] = str(e) res["result"] = str(e)
res["success"] = False res["success"] = False
@@ -179,9 +236,14 @@ class Loader:
can introduce it more smoothly and give people the chance to sample the new features even can introduce it more smoothly and give people the chance to sample the new features even
without plugin support. They will be removed once legacy plugins are no longer relevant. without plugin support. They will be removed once legacy plugins are no longer relevant.
""" """
async def load_plugin_main_view(self, request): async def load_plugin_main_view(self, request):
plugin = self.plugins[request.match_info["name"]] plugin = self.plugins[request.match_info["name"]]
with open(path.join(self.plugin_path, plugin.plugin_directory, plugin.main_view_html), "r", encoding="utf-8") as template: with open(
path.join(self.plugin_path, plugin.plugin_directory, plugin.main_view_html),
"r",
encoding="utf-8",
) as template:
template_data = template.read() template_data = template.read()
ret = f""" ret = f"""
<script src="/legacy/library.js"></script> <script src="/legacy/library.js"></script>
@@ -205,6 +267,11 @@ class Loader:
async def get_steam_resource(self, request): async def get_steam_resource(self, request):
tab = await get_tab("SP") tab = await get_tab("SP")
try: try:
return web.Response(text=await tab.get_steam_resource(f"https://steamloopback.host/{request.match_info['path']}"), content_type="text/html") return web.Response(
text=await tab.get_steam_resource(
f"https://steamloopback.host/{request.match_info['path']}"
),
content_type="text/html",
)
except Exception as e: except Exception as e:
return web.Response(text=str(e), status=400) return web.Response(text=str(e), status=400)
+72 -31
View File
@@ -1,27 +1,35 @@
# Change PyInstaller files permissions # Change PyInstaller files permissions
import sys import sys
from subprocess import call from subprocess import call
if hasattr(sys, '_MEIPASS'):
call(['chmod', '-R', '755', sys._MEIPASS]) if hasattr(sys, "_MEIPASS"):
call(["chmod", "-R", "755", sys._MEIPASS])
# Full imports # Full imports
from asyncio import new_event_loop, set_event_loop, sleep from asyncio import new_event_loop, set_event_loop, sleep
from json import dumps, loads from logging import basicConfig, getLogger
from logging import DEBUG, INFO, basicConfig, getLogger from os import getenv, path
from os import getenv, chmod, path
from traceback import format_exc from traceback import format_exc
import aiohttp_cors import aiohttp_cors
# Partial imports # Partial imports
from aiohttp import client_exceptions, WSMsgType from aiohttp import client_exceptions
from aiohttp.web import Application, Response, get, run_app, static from aiohttp.web import Application, Response, get, run_app, static
from aiohttp_jinja2 import setup as jinja_setup from aiohttp_jinja2 import setup as jinja_setup
# local modules # local modules
from browser import PluginBrowser from browser import PluginBrowser
from helpers import (REMOTE_DEBUGGER_UNIT, csrf_middleware, get_csrf_token, from helpers import (
get_home_path, get_homebrew_path, get_user, get_user_group, REMOTE_DEBUGGER_UNIT,
stop_systemd_unit, start_systemd_unit) csrf_middleware,
from injector import get_gamepadui_tab, Tab, get_tabs, close_old_tabs get_csrf_token,
get_homebrew_path,
get_user,
get_user_group,
stop_systemd_unit,
start_systemd_unit,
)
from injector import get_gamepadui_tab, Tab, close_old_tabs
from loader import Loader from loader import Loader
from settings import SettingsManager from settings import SettingsManager
from updater import Updater from updater import Updater
@@ -42,35 +50,45 @@ CONFIG = {
} }
basicConfig( basicConfig(
level=CONFIG["log_level"], level=CONFIG["log_level"], format="[%(module)s][%(levelname)s]: %(message)s"
format="[%(module)s][%(levelname)s]: %(message)s"
) )
logger = getLogger("Main") logger = getLogger("Main")
def chown_plugin_dir(): def chown_plugin_dir():
code_chown = call(["chown", "-R", USER+":"+GROUP, CONFIG["plugin_path"]]) code_chown = call(["chown", "-R", USER + ":" + GROUP, CONFIG["plugin_path"]])
code_chmod = call(["chmod", "-R", "555", CONFIG["plugin_path"]]) code_chmod = call(["chmod", "-R", "555", CONFIG["plugin_path"]])
if code_chown != 0 or code_chmod != 0: if code_chown != 0 or code_chmod != 0:
logger.error(f"chown/chmod exited with a non-zero exit code (chown: {code_chown}, chmod: {code_chmod})") logger.error(
f"chown/chmod exited with a non-zero exit code (chown: {code_chown}, chmod:"
f" {code_chmod})"
)
if CONFIG["chown_plugin_path"] == True:
if CONFIG["chown_plugin_path"] is True:
chown_plugin_dir() chown_plugin_dir()
class PluginManager: class PluginManager:
def __init__(self, loop) -> None: def __init__(self, loop) -> None:
self.loop = loop self.loop = loop
self.web_app = Application() self.web_app = Application()
self.web_app.middlewares.append(csrf_middleware) self.web_app.middlewares.append(csrf_middleware)
self.cors = aiohttp_cors.setup(self.web_app, defaults={ self.cors = aiohttp_cors.setup(
"https://steamloopback.host": aiohttp_cors.ResourceOptions( self.web_app,
expose_headers="*", defaults={
allow_headers="*", "https://steamloopback.host": aiohttp_cors.ResourceOptions(
allow_credentials=True expose_headers="*", allow_headers="*", allow_credentials=True
) )
}) },
self.plugin_loader = Loader(self.web_app, CONFIG["plugin_path"], self.loop, CONFIG["live_reload"]) )
self.plugin_browser = PluginBrowser(CONFIG["plugin_path"], self.plugin_loader.plugins, self.plugin_loader) self.plugin_loader = Loader(
self.web_app, CONFIG["plugin_path"], self.loop, CONFIG["live_reload"]
)
self.plugin_browser = PluginBrowser(
CONFIG["plugin_path"], self.plugin_loader.plugins, self.plugin_loader
)
self.settings = SettingsManager("loader", path.join(HOMEBREW_PATH, "settings")) self.settings = SettingsManager("loader", path.join(HOMEBREW_PATH, "settings"))
self.utilities = Utilities(self) self.utilities = Utilities(self)
self.updater = Updater(self) self.updater = Updater(self)
@@ -92,8 +110,12 @@ class PluginManager:
for route in list(self.web_app.router.routes()): for route in list(self.web_app.router.routes()):
self.cors.add(route) self.cors.add(route)
self.web_app.add_routes([static("/static", path.join(path.dirname(__file__), 'static'))]) self.web_app.add_routes(
self.web_app.add_routes([static("/legacy", path.join(path.dirname(__file__), 'legacy'))]) [static("/static", path.join(path.dirname(__file__), "static"))]
)
self.web_app.add_routes(
[static("/legacy", path.join(path.dirname(__file__), "legacy"))]
)
def exception_handler(self, loop, context): def exception_handler(self, loop, context):
if context["message"] == "Unclosed connection": if context["message"] == "Unclosed connection":
@@ -117,7 +139,10 @@ class PluginManager:
while not tab: while not tab:
try: try:
tab = await get_gamepadui_tab() tab = await get_gamepadui_tab()
except (client_exceptions.ClientConnectorError, client_exceptions.ServerDisconnectedError): except (
client_exceptions.ClientConnectorError,
client_exceptions.ServerDisconnectedError,
):
if not dc: if not dc:
logger.debug("Couldn't connect to debugger, waiting...") logger.debug("Couldn't connect to debugger, waiting...")
dc = True dc = True
@@ -148,7 +173,7 @@ class PluginManager:
# This is because of https://github.com/aio-libs/aiohttp/blob/3ee7091b40a1bc58a8d7846e7878a77640e96996/aiohttp/client_ws.py#L321 # This is because of https://github.com/aio-libs/aiohttp/blob/3ee7091b40a1bc58a8d7846e7878a77640e96996/aiohttp/client_ws.py#L321
logger.info("CEF has disconnected...") logger.info("CEF has disconnected...")
# At this point the loop starts again and we connect to the freshly started Steam client once it is ready. # At this point the loop starts again and we connect to the freshly started Steam client once it is ready.
except Exception as e: except Exception:
logger.error("Exception while reading page events " + format_exc()) logger.error("Exception while reading page events " + format_exc())
await tab.close_websocket() await tab.close_websocket()
pass pass
@@ -164,13 +189,29 @@ class PluginManager:
if first: if first:
if await tab.has_global_var("deckyHasLoaded", False): if await tab.has_global_var("deckyHasLoaded", False):
await close_old_tabs() await close_old_tabs()
await tab.evaluate_js("try{if (window.deckyHasLoaded){setTimeout(() => location.reload(), 100)}else{window.deckyHasLoaded = true;(async()=>{try{while(!window.SP_REACT){await new Promise(r => setTimeout(r, 10))};await import('http://localhost:1337/frontend/index.js')}catch(e){console.error(e)};})();}}catch(e){console.error(e)}", False, False, False) await tab.evaluate_js(
except: "try{if (window.deckyHasLoaded){setTimeout(() => location.reload(),"
" 100)}else{window.deckyHasLoaded ="
" true;(async()=>{try{while(!window.SP_REACT){await new Promise(r =>"
" setTimeout(r, 10))};await"
" import('http://localhost:1337/frontend/index.js')}catch(e){console.error(e)};})();}}catch(e){console.error(e)}",
False,
False,
False,
)
except Exception:
logger.info("Failed to inject JavaScript into tab\n" + format_exc()) logger.info("Failed to inject JavaScript into tab\n" + format_exc())
pass pass
def run(self): def run(self):
return run_app(self.web_app, host=CONFIG["server_host"], port=CONFIG["server_port"], loop=self.loop, access_log=None) return run_app(
self.web_app,
host=CONFIG["server_host"],
port=CONFIG["server_port"],
loop=self.loop,
access_log=None,
)
if __name__ == "__main__": if __name__ == "__main__":
loop = new_event_loop() loop = new_event_loop()
+79 -36
View File
@@ -1,22 +1,29 @@
import multiprocessing import multiprocessing
from asyncio import (Lock, get_event_loop, new_event_loop, from asyncio import (
open_unix_connection, set_event_loop, sleep, Lock,
start_unix_server, IncompleteReadError, LimitOverrunError) get_event_loop,
from concurrent.futures import ProcessPoolExecutor new_event_loop,
open_unix_connection,
set_event_loop,
sleep,
start_unix_server,
IncompleteReadError,
LimitOverrunError,
)
from importlib.util import module_from_spec, spec_from_file_location from importlib.util import module_from_spec, spec_from_file_location
from json import dumps, load, loads from json import dumps, load, loads
from logging import getLogger from logging import getLogger
from traceback import format_exc from traceback import format_exc
from os import path, setgid, setuid, environ from os import path, setgid, setuid, environ
from signal import SIGINT, signal from signal import SIGINT, signal
from sys import exit, path as syspath from sys import exit
from time import time from time import time
import helpers import helpers
from updater import Updater
multiprocessing.set_start_method("fork") multiprocessing.set_start_method("fork")
BUFFER_LIMIT = 2 ** 20 # 1 MiB BUFFER_LIMIT = 2**20 # 1 MiB
class PluginWrapper: class PluginWrapper:
def __init__(self, file, plugin_directory, plugin_path) -> None: def __init__(self, file, plugin_directory, plugin_path) -> None:
@@ -30,12 +37,23 @@ class PluginWrapper:
self.version = None self.version = None
json = load(open(path.join(plugin_path, plugin_directory, "plugin.json"), "r", encoding="utf-8")) json = load(
open(
path.join(plugin_path, plugin_directory, "plugin.json"),
"r",
encoding="utf-8",
)
)
if path.isfile(path.join(plugin_path, plugin_directory, "package.json")): if path.isfile(path.join(plugin_path, plugin_directory, "package.json")):
package_json = load(open(path.join(plugin_path, plugin_directory, "package.json"), "r", encoding="utf-8")) package_json = load(
open(
path.join(plugin_path, plugin_directory, "package.json"),
"r",
encoding="utf-8",
)
)
self.version = package_json["version"] self.version = package_json["version"]
self.legacy = False self.legacy = False
self.main_view_html = json["main_view_html"] if "main_view_html" in json else "" 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.tile_view_html = json["tile_view_html"] if "tile_view_html" in json else ""
@@ -62,57 +80,66 @@ class PluginWrapper:
setgid(0 if "root" in self.flags else helpers.get_user_group_id()) setgid(0 if "root" in self.flags else helpers.get_user_group_id())
setuid(0 if "root" in self.flags else helpers.get_user_id()) setuid(0 if "root" in self.flags else helpers.get_user_id())
# export a bunch of environment variables to help plugin developers # export a bunch of environment variables to help plugin developers
environ["HOME"] = helpers.get_home_path("root" if "root" in self.flags else helpers.get_user()) environ["HOME"] = helpers.get_home_path(
"root" if "root" in self.flags else helpers.get_user()
)
environ["USER"] = "root" if "root" in self.flags else helpers.get_user() environ["USER"] = "root" if "root" in self.flags else helpers.get_user()
environ["DECKY_VERSION"] = helpers.get_loader_version() environ["DECKY_VERSION"] = helpers.get_loader_version()
environ["DECKY_USER"] = helpers.get_user() environ["DECKY_USER"] = helpers.get_user()
environ["DECKY_USER_HOME"] = helpers.get_home_path()
environ["DECKY_HOME"] = helpers.get_homebrew_path() environ["DECKY_HOME"] = helpers.get_homebrew_path()
environ["DECKY_PLUGIN_SETTINGS_DIR"] = path.join(environ["DECKY_HOME"], "settings", self.plugin_directory) environ["DECKY_PLUGIN_SETTINGS_DIR"] = path.join(
environ["DECKY_HOME"], "settings", self.plugin_directory
)
helpers.mkdir_as_user(environ["DECKY_PLUGIN_SETTINGS_DIR"]) helpers.mkdir_as_user(environ["DECKY_PLUGIN_SETTINGS_DIR"])
environ["DECKY_PLUGIN_RUNTIME_DIR"] = path.join(environ["DECKY_HOME"], "data", self.plugin_directory) environ["DECKY_PLUGIN_RUNTIME_DIR"] = path.join(
environ["DECKY_HOME"], "data", self.plugin_directory
)
helpers.mkdir_as_user(environ["DECKY_PLUGIN_RUNTIME_DIR"]) helpers.mkdir_as_user(environ["DECKY_PLUGIN_RUNTIME_DIR"])
environ["DECKY_PLUGIN_LOG_DIR"] = path.join(environ["DECKY_HOME"], "logs", self.plugin_directory) environ["DECKY_PLUGIN_LOG_DIR"] = path.join(
environ["DECKY_HOME"], "logs", self.plugin_directory
)
helpers.mkdir_as_user(environ["DECKY_PLUGIN_LOG_DIR"]) helpers.mkdir_as_user(environ["DECKY_PLUGIN_LOG_DIR"])
environ["DECKY_PLUGIN_DIR"] = path.join(self.plugin_path, self.plugin_directory) environ["DECKY_PLUGIN_DIR"] = path.join(
self.plugin_path, self.plugin_directory
)
environ["DECKY_PLUGIN_NAME"] = self.name environ["DECKY_PLUGIN_NAME"] = self.name
environ["DECKY_PLUGIN_VERSION"] = self.version environ["DECKY_PLUGIN_VERSION"] = self.version
environ["DECKY_PLUGIN_AUTHOR"] = self.author environ["DECKY_PLUGIN_AUTHOR"] = self.author
# append the loader's plugin path to the recognized python paths
syspath.append(path.realpath(path.join(path.dirname(__file__), "plugin")))
# append the plugin's `py_modules` to the recognized python paths
syspath.append(path.join(environ["DECKY_PLUGIN_DIR"], "py_modules"))
# append the system and user python paths
syspath.extend(helpers.get_system_pythonpaths())
spec = spec_from_file_location("_", self.file) spec = spec_from_file_location("_", self.file)
module = module_from_spec(spec) module = module_from_spec(spec)
spec.loader.exec_module(module) spec.loader.exec_module(module)
self.Plugin = module.Plugin self.Plugin = module.Plugin
if hasattr(self.Plugin, "_migration"):
get_event_loop().run_until_complete(self.Plugin._migration(self.Plugin))
if hasattr(self.Plugin, "_main"): if hasattr(self.Plugin, "_main"):
get_event_loop().create_task(self.Plugin._main(self.Plugin)) get_event_loop().create_task(self.Plugin._main(self.Plugin))
get_event_loop().create_task(self._setup_socket()) get_event_loop().create_task(self._setup_socket())
get_event_loop().run_forever() get_event_loop().run_forever()
except: except Exception:
self.log.error("Failed to start " + self.name + "!\n" + format_exc()) self.log.error("Failed to start " + self.name + "!\n" + format_exc())
exit(0) exit(0)
async def _unload(self): async def _unload(self):
try: try:
self.log.info("Attempting to unload with plugin " + self.name + "'s \"_unload\" function.\n") self.log.info(
"Attempting to unload with plugin "
+ self.name
+ '\'s "_unload" function.\n'
)
if hasattr(self.Plugin, "_unload"): if hasattr(self.Plugin, "_unload"):
await self.Plugin._unload(self.Plugin) await self.Plugin._unload(self.Plugin)
self.log.info("Unloaded " + self.name + "\n") self.log.info("Unloaded " + self.name + "\n")
else: else:
self.log.info("Could not find \"_unload\" in " + self.name + "'s main.py" + "\n") self.log.info(
except: 'Could not find "_unload" in ' + self.name + "'s main.py" + "\n"
)
except Exception:
self.log.error("Failed to unload " + self.name + "!\n" + format_exc()) self.log.error("Failed to unload " + self.name + "!\n" + format_exc())
exit(0) exit(0)
async def _setup_socket(self): async def _setup_socket(self):
self.socket = await start_unix_server(self._listen_for_method_call, path=self.socket_addr, limit=BUFFER_LIMIT) self.socket = await start_unix_server(
self._listen_for_method_call, path=self.socket_addr, limit=BUFFER_LIMIT
)
async def _listen_for_method_call(self, reader, writer): async def _listen_for_method_call(self, reader, writer):
while True: while True:
@@ -139,12 +166,14 @@ class PluginWrapper:
return return
d = {"res": None, "success": True} d = {"res": None, "success": True}
try: try:
d["res"] = await getattr(self.Plugin, data["method"])(self.Plugin, **data["args"]) d["res"] = await getattr(self.Plugin, data["method"])(
self.Plugin, **data["args"]
)
except Exception as e: except Exception as e:
d["res"] = str(e) d["res"] = str(e)
d["success"] = False d["success"] = False
finally: finally:
writer.write((dumps(d, ensure_ascii=False)+"\n").encode("utf-8")) writer.write((dumps(d, ensure_ascii=False) + "\n").encode("utf-8"))
await writer.drain() await writer.drain()
async def _open_socket_if_not_exists(self): async def _open_socket_if_not_exists(self):
@@ -152,9 +181,11 @@ class PluginWrapper:
retries = 0 retries = 0
while retries < 10: while retries < 10:
try: try:
self.reader, self.writer = await open_unix_connection(self.socket_addr, limit=BUFFER_LIMIT) self.reader, self.writer = await open_unix_connection(
self.socket_addr, limit=BUFFER_LIMIT
)
return True return True
except: except Exception:
await sleep(2) await sleep(2)
retries += 1 retries += 1
return False return False
@@ -170,20 +201,32 @@ class PluginWrapper:
def stop(self): def stop(self):
if self.passive: if self.passive:
return return
async def _(self): async def _(self):
if await self._open_socket_if_not_exists(): if await self._open_socket_if_not_exists():
self.writer.write((dumps({ "stop": True }, ensure_ascii=False)+"\n").encode("utf-8")) self.writer.write(
(dumps({"stop": True}, ensure_ascii=False) + "\n").encode("utf-8")
)
await self.writer.drain() await self.writer.drain()
self.writer.close() self.writer.close()
get_event_loop().create_task(_(self)) get_event_loop().create_task(_(self))
async def execute_method(self, method_name, kwargs): async def execute_method(self, method_name, kwargs):
if self.passive: if self.passive:
raise RuntimeError("This plugin is passive (aka does not implement main.py)") raise RuntimeError(
"This plugin is passive (aka does not implement main.py)"
)
async with self.method_call_lock: async with self.method_call_lock:
if await self._open_socket_if_not_exists(): if await self._open_socket_if_not_exists():
self.writer.write( self.writer.write(
(dumps({ "method": method_name, "args": kwargs }, ensure_ascii=False) + "\n").encode("utf-8")) (
dumps(
{"method": method_name, "args": kwargs}, ensure_ascii=False
)
+ "\n"
).encode("utf-8")
)
await self.writer.drain() await self.writer.drain()
line = bytearray() line = bytearray()
while True: while True:
+15
View File
@@ -0,0 +1,15 @@
[flake8]
max-line-length = 88
[tool.ruff]
ignore = [
# Ignore line length check and let Black handle it
"E501",
# Ignore SyntaxError due to ruff not supporting pattern matching
# https://github.com/charliermarsh/ruff/issues/282
"E999",
]
# Assume Python 3.10.
target-version = "py310"
+13 -10
View File
@@ -2,33 +2,36 @@ from json import dump, load
from os import mkdir, path, listdir, rename from os import mkdir, path, listdir, rename
from shutil import chown from shutil import chown
from helpers import get_home_path, get_homebrew_path, get_user, get_user_group, get_user_owner from helpers import (
get_homebrew_path,
get_user,
get_user_group,
get_user_owner,
)
class SettingsManager: class SettingsManager:
def __init__(self, name, settings_directory = None) -> None: def __init__(self, name, settings_directory=None) -> None:
USER = get_user() USER = get_user()
GROUP = get_user_group() GROUP = get_user_group()
wrong_dir = get_homebrew_path() wrong_dir = get_homebrew_path()
if settings_directory == None: if settings_directory is None:
settings_directory = path.join(wrong_dir, "settings") settings_directory = path.join(wrong_dir, "settings")
self.path = path.join(settings_directory, name + ".json") self.path = path.join(settings_directory, name + ".json")
#Create the folder with the correct permission # Create the folder with the correct permission
if not path.exists(settings_directory): if not path.exists(settings_directory):
mkdir(settings_directory) mkdir(settings_directory)
chown(settings_directory, USER, GROUP) chown(settings_directory, USER, GROUP)
#Copy all old settings file in the root directory to the correct folder # Copy all old settings file in the root directory to the correct folder
for file in listdir(wrong_dir): for file in listdir(wrong_dir):
if file.endswith(".json"): if file.endswith(".json"):
rename(path.join(wrong_dir,file), rename(path.join(wrong_dir, file), path.join(settings_directory, file))
path.join(settings_directory, file))
self.path = path.join(settings_directory, name + ".json") self.path = path.join(settings_directory, name + ".json")
# If the owner of the settings directory is not the user, then set it as the user:
#If the owner of the settings directory is not the user, then set it as the user:
if get_user_owner(settings_directory) != USER: if get_user_owner(settings_directory) != USER:
chown(settings_directory, USER, GROUP) chown(settings_directory, USER, GROUP)
@@ -36,7 +39,7 @@ class SettingsManager:
try: try:
open(self.path, "x", encoding="utf-8") open(self.path, "x", encoding="utf-8")
except FileExistsError as e: except FileExistsError:
self.read() self.read()
pass pass
+93 -32
View File
@@ -16,6 +16,7 @@ from settings import SettingsManager
logger = getLogger("Updater") logger = getLogger("Updater")
class Updater: class Updater:
def __init__(self, context) -> None: def __init__(self, context) -> None:
self.context = context self.context = context
@@ -26,22 +27,27 @@ class Updater:
"get_version": self.get_version, "get_version": self.get_version,
"do_update": self.do_update, "do_update": self.do_update,
"do_restart": self.do_restart, "do_restart": self.do_restart,
"check_for_updates": self.check_for_updates "check_for_updates": self.check_for_updates,
} }
self.remoteVer = None self.remoteVer = None
self.allRemoteVers = None self.allRemoteVers = None
self.localVer = helpers.get_loader_version() try:
self.localVer = helpers.get_loader_version()
except:
self.localVer = False
try: try:
self.currentBranch = self.get_branch(self.context.settings) self.currentBranch = self.get_branch(self.context.settings)
except: except:
self.currentBranch = 0 self.currentBranch = 0
logger.error("Current branch could not be determined, defaulting to \"Stable\"") logger.error(
'Current branch could not be determined, defaulting to "Stable"'
)
if context: if context:
context.web_app.add_routes([ context.web_app.add_routes(
web.post("/updater/{method_name}", self._handle_server_method_call) [web.post("/updater/{method_name}", self._handle_server_method_call)]
]) )
context.loop.create_task(self.version_reloader()) context.loop.create_task(self.version_reloader())
async def _handle_server_method_call(self, request): async def _handle_server_method_call(self, request):
@@ -65,7 +71,7 @@ class Updater:
logger.debug("current branch: %i" % ver) logger.debug("current branch: %i" % ver)
if ver == -1: if ver == -1:
logger.info("Current branch is not set, determining branch from version...") logger.info("Current branch is not set, determining branch from version...")
if self.localVer.startswith("v") and "-pre" in self.localVer: if self.localVer.startswith("v") and self.localVer.find("-pre"):
logger.info("Current version determined to be pre-release") logger.info("Current version determined to be pre-release")
return 1 return 1
else: else:
@@ -86,38 +92,71 @@ class Updater:
case 1 | 2: case 1 | 2:
url = "https://raw.githubusercontent.com/SteamDeckHomebrew/decky-loader/main/dist/plugin_loader-prerelease.service" url = "https://raw.githubusercontent.com/SteamDeckHomebrew/decky-loader/main/dist/plugin_loader-prerelease.service"
case _: case _:
logger.error("You have an invalid branch set... Defaulting to prerelease service, please send the logs to the devs!") logger.error(
"You have an invalid branch set... Defaulting to prerelease"
" service, please send the logs to the devs!"
)
url = "https://raw.githubusercontent.com/SteamDeckHomebrew/decky-loader/main/dist/plugin_loader-prerelease.service" url = "https://raw.githubusercontent.com/SteamDeckHomebrew/decky-loader/main/dist/plugin_loader-prerelease.service"
return str(url) return str(url)
async def get_version(self): async def get_version(self):
return { if self.localVer:
"current": self.localVer, return {
"remote": self.remoteVer, "current": self.localVer,
"all": self.allRemoteVers, "remote": self.remoteVer,
"updatable": self.localVer != "unknown" "all": self.allRemoteVers,
} "updatable": self.localVer != None,
}
else:
return {
"current": "unknown",
"remote": self.remoteVer,
"all": self.allRemoteVers,
"updatable": False,
}
async def check_for_updates(self): async def check_for_updates(self):
logger.debug("checking for updates") logger.debug("checking for updates")
selectedBranch = self.get_branch(self.context.settings) selectedBranch = self.get_branch(self.context.settings)
async with ClientSession() as web: async with ClientSession() as web:
async with web.request("GET", "https://api.github.com/repos/SteamDeckHomebrew/decky-loader/releases", ssl=helpers.get_ssl_context()) as res: async with web.request(
"GET",
"https://api.github.com/repos/SteamDeckHomebrew/decky-loader/releases",
ssl=helpers.get_ssl_context(),
) as res:
remoteVersions = await res.json() remoteVersions = await res.json()
self.allRemoteVers = remoteVersions self.allRemoteVers = remoteVersions
logger.debug("determining release type to find, branch is %i" % selectedBranch) logger.debug("determining release type to find, branch is %i" % selectedBranch)
if selectedBranch == 0: if selectedBranch == 0:
logger.debug("release type: release") logger.debug("release type: release")
self.remoteVer = next(filter(lambda ver: ver["tag_name"].startswith("v") and not ver["prerelease"] and not ver["tag_name"].find("-pre") > 0 and ver["tag_name"], remoteVersions), None) self.remoteVer = next(
filter(
lambda ver: ver["tag_name"].startswith("v")
and not ver["prerelease"]
and ver["tag_name"],
remoteVersions,
),
None,
)
elif selectedBranch == 1: elif selectedBranch == 1:
logger.debug("release type: pre-release") logger.debug("release type: pre-release")
self.remoteVer = next(filter(lambda ver:ver["tag_name"].startswith("v"), remoteVersions), None) self.remoteVer = next(
filter(
lambda ver: ver["prerelease"]
and ver["tag_name"].startswith("v")
and ver["tag_name"].find("-pre"),
remoteVersions,
),
None,
)
else: else:
logger.error("release type: NOT FOUND") logger.error("release type: NOT FOUND")
raise ValueError("no valid branch found") raise ValueError("no valid branch found")
logger.info("Updated remote version information") logger.info("Updated remote version information")
tab = await get_gamepadui_tab() tab = await get_gamepadui_tab()
await tab.evaluate_js(f"window.DeckyPluginLoader.notifyUpdates()", False, True, False) await tab.evaluate_js(
f"window.DeckyPluginLoader.notifyUpdates()", False, True, False
)
return await self.get_version() return await self.get_version()
async def version_reloader(self): async def version_reloader(self):
@@ -127,7 +166,7 @@ class Updater:
await self.check_for_updates() await self.check_for_updates()
except: except:
pass pass
await sleep(60 * 60 * 6) # 6 hours await sleep(60 * 60 * 6) # 6 hours
async def do_update(self): async def do_update(self):
logger.debug("Starting update.") logger.debug("Starting update.")
@@ -141,7 +180,9 @@ class Updater:
async with ClientSession() as web: async with ClientSession() as web:
logger.debug("Downloading systemd service") logger.debug("Downloading systemd service")
# download the relevant systemd service depending upon branch # download the relevant systemd service depending upon branch
async with web.request("GET", service_url, ssl=helpers.get_ssl_context(), allow_redirects=True) as res: async with web.request(
"GET", service_url, ssl=helpers.get_ssl_context(), allow_redirects=True
) as res:
logger.debug("Downloading service file") logger.debug("Downloading service file")
data = await res.content.read() data = await res.content.read()
logger.debug(str(data)) logger.debug(str(data))
@@ -151,22 +192,33 @@ class Updater:
out.write(data) out.write(data)
except Exception as e: except Exception as e:
logger.error(f"Error at %s", exc_info=e) logger.error(f"Error at %s", exc_info=e)
with open(path.join(getcwd(), "plugin_loader.service"), "r", encoding="utf-8") as service_file: with open(
path.join(getcwd(), "plugin_loader.service"), "r", encoding="utf-8"
) as service_file:
service_data = service_file.read() service_data = service_file.read()
service_data = service_data.replace("${HOMEBREW_FOLDER}", helpers.get_homebrew_path()) service_data = service_data.replace(
with open(path.join(getcwd(), "plugin_loader.service"), "w", encoding="utf-8") as service_file: "${HOMEBREW_FOLDER}", helpers.get_homebrew_path()
service_file.write(service_data) )
with open(
path.join(getcwd(), "plugin_loader.service"), "w", encoding="utf-8"
) as service_file:
service_file.write(service_data)
logger.debug("Saved service file") logger.debug("Saved service file")
logger.debug("Copying service file over current file.") logger.debug("Copying service file over current file.")
shutil.copy(service_file_path, "/etc/systemd/system/plugin_loader.service") shutil.copy(service_file_path, "/etc/systemd/system/plugin_loader.service")
if not os.path.exists(path.join(getcwd(), ".systemd")): if not os.path.exists(path.join(getcwd(), ".systemd")):
os.mkdir(path.join(getcwd(), ".systemd")) os.mkdir(path.join(getcwd(), ".systemd"))
shutil.move(service_file_path, path.join(getcwd(), ".systemd")+"/plugin_loader.service") shutil.move(
service_file_path,
path.join(getcwd(), ".systemd") + "/plugin_loader.service",
)
logger.debug("Downloading binary") logger.debug("Downloading binary")
async with web.request("GET", download_url, ssl=helpers.get_ssl_context(), allow_redirects=True) as res: async with web.request(
total = int(res.headers.get('content-length', 0)) "GET", download_url, ssl=helpers.get_ssl_context(), allow_redirects=True
) as res:
total = int(res.headers.get("content-length", 0))
# we need to not delete the binary until we have downloaded the new binary! # we need to not delete the binary until we have downloaded the new binary!
try: try:
remove(path.join(getcwd(), "PluginLoader")) remove(path.join(getcwd(), "PluginLoader"))
@@ -180,13 +232,22 @@ class Updater:
raw += len(c) raw += len(c)
new_progress = round((raw / total) * 100) new_progress = round((raw / total) * 100)
if progress != new_progress: if progress != new_progress:
self.context.loop.create_task(tab.evaluate_js(f"window.DeckyUpdater.updateProgress({new_progress})", False, False, False)) self.context.loop.create_task(
tab.evaluate_js(
f"window.DeckyUpdater.updateProgress({new_progress})",
False,
False,
False,
)
)
progress = new_progress progress = new_progress
with open(path.join(getcwd(), ".loader.version"), "w", encoding="utf-8") as out: with open(
path.join(getcwd(), ".loader.version"), "w", encoding="utf-8"
) as out:
out.write(version) out.write(version)
call(['chmod', '+x', path.join(getcwd(), "PluginLoader")]) call(["chmod", "+x", path.join(getcwd(), "PluginLoader")])
logger.info("Updated loader installation.") logger.info("Updated loader installation.")
await tab.evaluate_js("window.DeckyUpdater.finish()", False, False) await tab.evaluate_js("window.DeckyUpdater.finish()", False, False)
await self.do_restart() await self.do_restart()
+55 -74
View File
@@ -3,13 +3,12 @@ import os
from json.decoder import JSONDecodeError from json.decoder import JSONDecodeError
from traceback import format_exc from traceback import format_exc
from asyncio import sleep, start_server, gather, open_connection from asyncio import start_server, gather, open_connection
from aiohttp import ClientSession, web from aiohttp import ClientSession, web
from logging import getLogger from logging import getLogger
from injector import inject_to_tab, get_gamepadui_tab, close_old_tabs from injector import inject_to_tab, get_gamepadui_tab, close_old_tabs
import helpers import helpers
import subprocess
class Utilities: class Utilities:
@@ -31,7 +30,7 @@ class Utilities:
"get_setting": self.get_setting, "get_setting": self.get_setting,
"filepicker_ls": self.filepicker_ls, "filepicker_ls": self.filepicker_ls,
"disable_rdt": self.disable_rdt, "disable_rdt": self.disable_rdt,
"enable_rdt": self.enable_rdt "enable_rdt": self.enable_rdt,
} }
self.logger = getLogger("Utilities") self.logger = getLogger("Utilities")
@@ -41,9 +40,9 @@ class Utilities:
self.rdt_proxy_task = None self.rdt_proxy_task = None
if context: if context:
context.web_app.add_routes([ context.web_app.add_routes(
web.post("/methods/{method_name}", self._handle_server_method_call) [web.post("/methods/{method_name}", self._handle_server_method_call)]
]) )
async def _handle_server_method_call(self, request): async def _handle_server_method_call(self, request):
method_name = request.match_info["method_name"] method_name = request.match_info["method_name"]
@@ -61,12 +60,11 @@ class Utilities:
res["success"] = False res["success"] = False
return web.json_response(res) return web.json_response(res)
async def install_plugin(self, artifact="", name="No name", version="dev", hash=False): async def install_plugin(
self, artifact="", name="No name", version="dev", hash=False
):
return await self.context.plugin_browser.request_plugin_install( return await self.context.plugin_browser.request_plugin_install(
artifact=artifact, artifact=artifact, name=name, version=version, hash=hash
name=name,
version=version,
hash=hash
) )
async def confirm_plugin_install(self, request_id): async def confirm_plugin_install(self, request_id):
@@ -80,13 +78,11 @@ class Utilities:
async def http_request(self, method="", url="", **kwargs): async def http_request(self, method="", url="", **kwargs):
async with ClientSession() as web: async with ClientSession() as web:
res = await web.request(method, url, ssl=helpers.get_ssl_context(), **kwargs) res = await web.request(
method, url, ssl=helpers.get_ssl_context(), **kwargs
)
text = await res.text() text = await res.text()
return { return {"status": res.status, "headers": dict(res.headers), "body": text}
"status": res.status,
"headers": dict(res.headers),
"body": text
}
async def ping(self, **kwargs): async def ping(self, **kwargs):
return "pong" return "pong"
@@ -95,26 +91,18 @@ class Utilities:
try: try:
result = await inject_to_tab(tab, code, run_async) result = await inject_to_tab(tab, code, run_async)
if "exceptionDetails" in result["result"]: if "exceptionDetails" in result["result"]:
return { return {"success": False, "result": result["result"]}
"success": False,
"result": result["result"]
}
return { return {"success": True, "result": result["result"]["result"].get("value")}
"success": True,
"result": result["result"]["result"].get("value")
}
except Exception as e: except Exception as e:
return { return {"success": False, "result": e}
"success": False,
"result": e
}
async def inject_css_into_tab(self, tab, style): async def inject_css_into_tab(self, tab, style):
try: try:
css_id = str(uuid.uuid4()) css_id = str(uuid.uuid4())
result = await inject_to_tab(tab, result = await inject_to_tab(
tab,
f""" f"""
(function() {{ (function() {{
const style = document.createElement('style'); const style = document.createElement('style');
@@ -122,27 +110,21 @@ class Utilities:
document.head.append(style); document.head.append(style);
style.textContent = `{style}`; style.textContent = `{style}`;
}})() }})()
""", False) """,
False,
)
if "exceptionDetails" in result["result"]: if "exceptionDetails" in result["result"]:
return { return {"success": False, "result": result["result"]}
"success": False,
"result": result["result"]
}
return { return {"success": True, "result": css_id}
"success": True,
"result": css_id
}
except Exception as e: except Exception as e:
return { return {"success": False, "result": e}
"success": False,
"result": e
}
async def remove_css_from_tab(self, tab, css_id): async def remove_css_from_tab(self, tab, css_id):
try: try:
result = await inject_to_tab(tab, result = await inject_to_tab(
tab,
f""" f"""
(function() {{ (function() {{
let style = document.getElementById("{css_id}"); let style = document.getElementById("{css_id}");
@@ -150,22 +132,16 @@ class Utilities:
if (style.nodeName.toLowerCase() == 'style') if (style.nodeName.toLowerCase() == 'style')
style.parentNode.removeChild(style); style.parentNode.removeChild(style);
}})() }})()
""", False) """,
False,
)
if "exceptionDetails" in result["result"]: if "exceptionDetails" in result["result"]:
return { return {"success": False, "result": result}
"success": False,
"result": result
}
return { return {"success": True}
"success": True
}
except Exception as e: except Exception as e:
return { return {"success": False, "result": e}
"success": False,
"result": e
}
async def get_setting(self, key, default): async def get_setting(self, key, default):
return self.context.settings.getSetting(key, default) return self.context.settings.getSetting(key, default)
@@ -187,7 +163,7 @@ class Utilities:
# return os.path.getmtime(os.path.join(path, file)) # return os.path.getmtime(os.path.join(path, file))
# return 0 # return 0
# file_names = sorted(os.listdir(path), key=sorter, reverse=True) # TODO provide more sort options # file_names = sorted(os.listdir(path), key=sorter, reverse=True) # TODO provide more sort options
file_names = sorted(os.listdir(path)) # Alphabetical file_names = sorted(os.listdir(path)) # Alphabetical
files = [] files = []
@@ -196,16 +172,15 @@ class Utilities:
is_dir = os.path.isdir(full_path) is_dir = os.path.isdir(full_path)
if is_dir or include_files: if is_dir or include_files:
files.append({ files.append(
"isdir": is_dir, {
"name": file, "isdir": is_dir,
"realpath": os.path.realpath(full_path) "name": file,
}) "realpath": os.path.realpath(full_path),
}
)
return { return {"realpath": os.path.realpath(path), "files": files}
"realpath": os.path.realpath(path),
"files": files
}
# Based on https://stackoverflow.com/a/46422554/13174603 # Based on https://stackoverflow.com/a/46422554/13174603
def start_rdt_proxy(self, ip, port): def start_rdt_proxy(self, ip, port):
@@ -215,10 +190,10 @@ class Utilities:
writer.write(await reader.read(2048)) writer.write(await reader.read(2048))
finally: finally:
writer.close() writer.close()
async def handle_client(local_reader, local_writer): async def handle_client(local_reader, local_writer):
try: try:
remote_reader, remote_writer = await open_connection( remote_reader, remote_writer = await open_connection(ip, port)
ip, port)
pipe1 = pipe(local_reader, remote_writer) pipe1 = pipe(local_reader, remote_writer)
pipe2 = pipe(remote_reader, local_writer) pipe2 = pipe(remote_reader, local_writer)
await gather(pipe1, pipe2) await gather(pipe1, pipe2)
@@ -239,11 +214,14 @@ class Utilities:
self.stop_rdt_proxy() self.stop_rdt_proxy()
ip = self.context.settings.getSetting("developer.rdt.ip", None) ip = self.context.settings.getSetting("developer.rdt.ip", None)
if ip != None: if ip is not None:
self.logger.info("Connecting to React DevTools at " + ip) self.logger.info("Connecting to React DevTools at " + ip)
async with ClientSession() as web: async with ClientSession() as web:
res = await web.request("GET", "http://" + ip + ":8097", ssl=helpers.get_ssl_context()) res = await web.request(
script = """ "GET", "http://" + ip + ":8097", ssl=helpers.get_ssl_context()
)
script = (
"""
if (!window.deckyHasConnectedRDT) { if (!window.deckyHasConnectedRDT) {
window.deckyHasConnectedRDT = true; window.deckyHasConnectedRDT = true;
// This fixes the overlay when hovering over an element in RDT // This fixes the overlay when hovering over an element in RDT
@@ -251,10 +229,13 @@ class Utilities:
enumerable: true, enumerable: true,
configurable: true, configurable: true,
get: function() { get: function() {
return (GamepadNavTree?.m_context?.m_controller || FocusNavController)?.m_ActiveContext?.ActiveWindow || window; return FocusNavController?.m_ActiveContext?.ActiveWindow || window;
} }
}); });
""" + await res.text() + "\n}" """
+ await res.text()
+ "\n}"
)
if res.status != 200: if res.status != 200:
self.logger.error("Failed to connect to React DevTools at " + ip) self.logger.error("Failed to connect to React DevTools at " + ip)
return False return False
@@ -265,7 +246,7 @@ class Utilities:
await close_old_tabs() await close_old_tabs()
result = await tab.reload_and_evaluate(script) result = await tab.reload_and_evaluate(script)
self.logger.info(result) self.logger.info(result)
except Exception: except Exception:
self.logger.error("Failed to connect to React DevTools") self.logger.error("Failed to connect to React DevTools")
self.logger.error(format_exc()) self.logger.error(format_exc())
+1 -1
View File
@@ -42,7 +42,7 @@
} }
}, },
"dependencies": { "dependencies": {
"decky-frontend-lib": "^3.19.1", "decky-frontend-lib": "^3.18.10",
"react-file-icon": "^1.2.0", "react-file-icon": "^1.2.0",
"react-icons": "^4.4.0", "react-icons": "^4.4.0",
"react-markdown": "^8.0.3", "react-markdown": "^8.0.3",
+4 -4
View File
@@ -11,7 +11,7 @@ specifiers:
'@types/react-file-icon': ^1.0.1 '@types/react-file-icon': ^1.0.1
'@types/react-router': 5.1.18 '@types/react-router': 5.1.18
'@types/webpack': ^5.28.0 '@types/webpack': ^5.28.0
decky-frontend-lib: ^3.19.1 decky-frontend-lib: ^3.18.10
husky: ^8.0.1 husky: ^8.0.1
import-sort-style-module: ^6.0.0 import-sort-style-module: ^6.0.0
inquirer: ^8.2.4 inquirer: ^8.2.4
@@ -31,7 +31,7 @@ specifiers:
typescript: ^4.7.4 typescript: ^4.7.4
dependencies: dependencies:
decky-frontend-lib: 3.19.1 decky-frontend-lib: 3.18.10
react-file-icon: 1.2.0_wcqkhtmu7mswc6yz4uyexck3ty react-file-icon: 1.2.0_wcqkhtmu7mswc6yz4uyexck3ty
react-icons: 4.4.0_react@16.14.0 react-icons: 4.4.0_react@16.14.0
react-markdown: 8.0.3_vshvapmxg47tngu7tvrsqpq55u react-markdown: 8.0.3_vshvapmxg47tngu7tvrsqpq55u
@@ -975,8 +975,8 @@ packages:
dependencies: dependencies:
ms: 2.1.2 ms: 2.1.2
/decky-frontend-lib/3.19.1: /decky-frontend-lib/3.18.10:
resolution: {integrity: sha512-hU4+EFs74MGzUCv8l1AO2+EBj9RRbnpU19Crm4u+3lbLu6d63U2GsUeQ9ssmNRcOMY1OuVZkRoZBE58soOBJ3A==} resolution: {integrity: sha512-2mgbA3sSkuwQR/FnmhXVrcW6LyTS95IuL6muJAmQCruhBvXapDtjk1TcgxqMZxFZwGD1IPnemPYxHZll6IgnZw==}
dev: false dev: false
/decode-named-character-reference/1.0.2: /decode-named-character-reference/1.0.2:
+2 -2
View File
@@ -1,4 +1,4 @@
import { Focusable, Navigation } from 'decky-frontend-lib'; import { Focusable, Router } from 'decky-frontend-lib';
import { FunctionComponent, useRef } from 'react'; import { FunctionComponent, useRef } from 'react';
import ReactMarkdown, { Options as ReactMarkdownOptions } from 'react-markdown'; import ReactMarkdown, { Options as ReactMarkdownOptions } from 'react-markdown';
import remarkGfm from 'remark-gfm'; import remarkGfm from 'remark-gfm';
@@ -22,7 +22,7 @@ const Markdown: FunctionComponent<MarkdownProps> = (props) => {
onActivate={() => {}} onActivate={() => {}}
onOKButton={() => { onOKButton={() => {
props.onDismiss?.(); props.onDismiss?.();
Navigation.NavigateToExternalWeb(aRef.current!.href); Router.NavigateToExternalWeb(aRef.current!.href);
}} }}
style={{ display: 'inline' }} style={{ display: 'inline' }}
> >
@@ -4,11 +4,15 @@ const QuickAccessVisibleState = createContext<boolean>(true);
export const useQuickAccessVisible = () => useContext(QuickAccessVisibleState); export const useQuickAccessVisible = () => useContext(QuickAccessVisibleState);
export const QuickAccessVisibleStateProvider: FC<{ initial: boolean; tab: any }> = ({ children, initial, tab }) => { export const QuickAccessVisibleStateProvider: FC<{ initial: boolean; setter: ((val: boolean) => {}[]) | never[] }> = ({
children,
initial,
setter,
}) => {
const [visible, setVisible] = useState<boolean>(initial); const [visible, setVisible] = useState<boolean>(initial);
const [prev, setPrev] = useState<boolean>(initial); const [prev, setPrev] = useState<boolean>(initial);
// HACK but i can't think of a better way to do this // hack to use an array as a "pointer" to pass the setter up the tree
tab.qAMVisibilitySetter = setVisible; setter[0] = setVisible;
if (initial != prev) { if (initial != prev) {
setPrev(initial); setPrev(initial);
setVisible(initial); setVisible(initial);
@@ -4,6 +4,13 @@ import Logger from '../../../../logger';
const logger = new Logger('LibraryPatch'); const logger = new Logger('LibraryPatch');
declare global {
interface Window {
SteamClient: any;
appDetailsStore: any;
}
}
let patch: Patch; let patch: Patch;
function rePatch() { function rePatch() {
@@ -13,9 +20,7 @@ function rePatch() {
const details = window.appDetailsStore.GetAppDetails(appid); const details = window.appDetailsStore.GetAppDetails(appid);
logger.debug('game details', details); logger.debug('game details', details);
// strShortcutStartDir // strShortcutStartDir
const file = await window.DeckyPluginLoader.openFilePicker( const file = await window.DeckyPluginLoader.openFilePicker(details.strShortcutStartDir.replaceAll('"', ''));
details?.strShortcutStartDir.replaceAll('"', '') || '/',
);
logger.debug('user selected', file); logger.debug('user selected', file);
window.SteamClient.Apps.SetShortcutExe(appid, JSON.stringify(file.path)); window.SteamClient.Apps.SetShortcutExe(appid, JSON.stringify(file.path));
const pathArr = file.path.split('/'); const pathArr = file.path.split('/');
@@ -52,7 +52,7 @@ export default function DeveloperSettings() {
> >
<Toggle <Toggle
value={reactDevtoolsEnabled} value={reactDevtoolsEnabled}
// disabled={reactDevtoolsIP == ''} disabled={reactDevtoolsIP == ''}
onChange={(toggleValue) => { onChange={(toggleValue) => {
setReactDevtoolsEnabled(toggleValue); setReactDevtoolsEnabled(toggleValue);
setShouldConnectToReactDevTools(toggleValue); setShouldConnectToReactDevTools(toggleValue);
@@ -6,7 +6,6 @@ import {
Focusable, Focusable,
ProgressBarWithInfo, ProgressBarWithInfo,
Spinner, Spinner,
findSP,
showModal, showModal,
} from 'decky-frontend-lib'; } from 'decky-frontend-lib';
import { useCallback } from 'react'; import { useCallback } from 'react';
@@ -15,6 +14,7 @@ import { useEffect, useState } from 'react';
import { FaExclamation } from 'react-icons/fa'; import { FaExclamation } from 'react-icons/fa';
import { VerInfo, callUpdaterMethod, finishUpdate } from '../../../../updater'; import { VerInfo, callUpdaterMethod, finishUpdate } from '../../../../updater';
import { findSP } from '../../../../utils/windows';
import { useDeckyState } from '../../../DeckyState'; import { useDeckyState } from '../../../DeckyState';
import InlinePatchNotes from '../../../patchnotes/InlinePatchNotes'; import InlinePatchNotes from '../../../patchnotes/InlinePatchNotes';
import WithSuspense from '../../../WithSuspense'; import WithSuspense from '../../../WithSuspense';
+1 -1
View File
@@ -15,7 +15,7 @@ import Logger from '../../logger';
import { StorePlugin, getPluginList } from '../../store'; import { StorePlugin, getPluginList } from '../../store';
import PluginCard from './PluginCard'; import PluginCard from './PluginCard';
const logger = new Logger('Store'); const logger = new Logger('FilePicker');
const StorePage: FC<{}> = () => { const StorePage: FC<{}> = () => {
const [currentTabRoute, setCurrentTabRoute] = useState<string>('browse'); const [currentTabRoute, setCurrentTabRoute] = useState<string>('browse');
+5 -3
View File
@@ -123,9 +123,11 @@ class RouterHook extends Logger {
this.wrapperPatch = afterPatch(this.gamepadWrapper, 'render', (_: any, ret: any) => { this.wrapperPatch = afterPatch(this.gamepadWrapper, 'render', (_: any, ret: any) => {
if (ret?.props?.children?.props?.children?.length == 5 || ret?.props?.children?.props?.children?.length == 4) { if (ret?.props?.children?.props?.children?.length == 5 || ret?.props?.children?.props?.children?.length == 4) {
const idx = ret?.props?.children?.props?.children?.length == 4 ? 1 : 2; const idx = ret?.props?.children?.props?.children?.length == 4 ? 1 : 2;
const potentialSettingsRootString = if (
ret.props.children.props.children[idx]?.props?.children?.[0]?.type?.type?.toString() || ''; ret.props.children.props.children[idx]?.props?.children?.[0]?.type?.type
if (potentialSettingsRootString?.includes('Settings.Root()')) { ?.toString()
?.includes('GamepadUI.Settings.Root()')
) {
if (!this.router) { if (!this.router) {
this.router = ret.props.children.props.children[idx]?.props?.children?.[0]?.type; this.router = ret.props.children.props.children[idx]?.props?.children?.[0]?.type;
this.routerPatch = afterPatch(this.router, 'type', (_: any, ret: any) => { this.routerPatch = afterPatch(this.router, 'type', (_: any, ret: any) => {
+1 -1
View File
@@ -7,6 +7,6 @@ export function deinitSteamFixes() {
} }
export async function initSteamFixes() { export async function initSteamFixes() {
fixes.push(await reloadFix()); fixes.push(reloadFix());
fixes.push(await restartFix()); fixes.push(await restartFix());
} }
+2 -9
View File
@@ -1,17 +1,10 @@
import { getFocusNavController, sleep } from 'decky-frontend-lib';
import Logger from '../logger'; import Logger from '../logger';
const logger = new Logger('ReloadSteamFix'); const logger = new Logger('ReloadSteamFix');
declare global { export default function reloadFix() {
var GamepadNavTree: any;
}
export default async function reloadFix() {
// Hack to unbreak the ui when reloading it // Hack to unbreak the ui when reloading it
await sleep(4000); if (window.FocusNavController?.m_rgAllContexts?.length == 0) {
if (getFocusNavController()?.m_rgAllContexts?.length == 0) {
SteamClient.URL.ExecuteSteamURL('steam://open/settings'); SteamClient.URL.ExecuteSteamURL('steam://open/settings');
logger.log('Applied UI reload fix.'); logger.log('Applied UI reload fix.');
} }
+7
View File
@@ -4,6 +4,13 @@ import Logger from '../logger';
const logger = new Logger('RestartSteamFix'); const logger = new Logger('RestartSteamFix');
declare global {
interface Window {
SteamClient: any;
appDetailsStore: any;
}
}
let patch: Patch; let patch: Patch;
function rePatch() { function rePatch() {
+9 -10
View File
@@ -35,7 +35,7 @@ class TabsHook extends Logger {
const tree = (document.getElementById('root') as any)._reactRootContainer._internalRoot.current; const tree = (document.getElementById('root') as any)._reactRootContainer._internalRoot.current;
let qAMRoot: any; let qAMRoot: any;
const findQAMRoot = (currentNode: any, iters: number): any => { const findQAMRoot = (currentNode: any, iters: number): any => {
if (iters >= 65) { if (iters >= 55) {
// currently 45 // currently 45
return null; return null;
} }
@@ -128,23 +128,22 @@ class TabsHook extends Logger {
let deckyTabAmount = existingTabs.reduce((prev: any, cur: any) => (cur.decky ? prev + 1 : prev), 0); let deckyTabAmount = existingTabs.reduce((prev: any, cur: any) => (cur.decky ? prev + 1 : prev), 0);
if (deckyTabAmount == this.tabs.length) { if (deckyTabAmount == this.tabs.length) {
for (let tab of existingTabs) { for (let tab of existingTabs) {
if (tab?.decky && tab?.qAMVisibilitySetter) tab?.qAMVisibilitySetter(visible); if (tab?.decky) tab.panel.props.setter[0](visible);
} }
return; return;
} }
for (const { title, icon, content, id } of this.tabs) { for (const { title, icon, content, id } of this.tabs) {
const tab: any = { existingTabs.push({
key: id, key: id,
title, title,
tab: icon, tab: icon,
decky: true, decky: true,
}; panel: (
tab.panel = ( <QuickAccessVisibleStateProvider initial={visible} setter={[]}>
<QuickAccessVisibleStateProvider initial={visible} tab={tab}> {content}
{content} </QuickAccessVisibleStateProvider>
</QuickAccessVisibleStateProvider> ),
); });
existingTabs.push(tab);
} }
} }
} }
+3 -7
View File
@@ -40,15 +40,11 @@ class Toaster extends Logger {
let instance: any; let instance: any;
const tree = (document.getElementById('root') as any)._reactRootContainer._internalRoot.current; const tree = (document.getElementById('root') as any)._reactRootContainer._internalRoot.current;
const findToasterRoot = (currentNode: any, iters: number): any => { const findToasterRoot = (currentNode: any, iters: number): any => {
if (iters >= 65) { if (iters >= 50) {
// currently 65 // currently 40
return null; return null;
} }
if ( if (currentNode?.memoizedProps?.className?.startsWith?.('toastmanager_ToastPlaceholder')) {
currentNode?.memoizedProps?.className?.startsWith?.('gamepadtoasts_GamepadToastPlaceholder') ||
currentNode?.memoizedProps?.className?.startsWith?.('toastmanager_ToastPlaceholder') ||
currentNode?.memoizedProps?.className?.startsWith?.('toastmanager_ToastPopup')
) {
this.log(`Toaster root was found in ${iters} recursion cycles`); this.log(`Toaster root was found in ${iters} recursion cycles`);
return currentNode; return currentNode;
} }
+7
View File
@@ -0,0 +1,7 @@
export function findSP(): Window {
// old (SP as host)
if (document.title == 'SP') return window;
// new (SP as popup)
return FocusNavController.m_ActiveContext.m_rgGamepadNavigationTrees.find((x: any) => x.m_ID == 'root_1_').Root
.Element.ownerDocument.defaultView;
}
+1 -1
View File
@@ -1,7 +1,7 @@
{ {
"compilerOptions": { "compilerOptions": {
"module": "ESNext", "module": "ESNext",
"target": "ES2021", "target": "ES2020",
"jsx": "react", "jsx": "react",
"jsxFactory": "window.SP_REACT.createElement", "jsxFactory": "window.SP_REACT.createElement",
"jsxFragmentFactory": "window.SP_REACT.Fragment", "jsxFragmentFactory": "window.SP_REACT.Fragment",
-201
View File
@@ -1,201 +0,0 @@
"""
This module exposes various constants and helpers useful for decky plugins.
* Plugin's settings and configurations should be stored under `DECKY_PLUGIN_SETTINGS_DIR`.
* Plugin's runtime data should be stored under `DECKY_PLUGIN_RUNTIME_DIR`.
* Plugin's persistent log files should be stored under `DECKY_PLUGIN_LOG_DIR`.
Avoid writing outside of `DECKY_HOME`, storing under the suggested paths is strongly recommended.
Some basic migration helpers are available: `migrate_any`, `migrate_settings`, `migrate_runtime`, `migrate_logs`.
A logging facility `logger` is available which writes to the recommended location.
"""
__version__ = '0.1.0'
import os
import subprocess
import logging
"""
Constants
"""
HOME: str = os.getenv("HOME", default="")
"""
The home directory of the effective user running the process.
Environment variable: `HOME`.
If `root` was specified in the plugin's flags it will be `/root` otherwise the user whose home decky resides in.
e.g.: `/home/deck`
"""
USER: str = os.getenv("USER", default="")
"""
The effective username running the process.
Environment variable: `USER`.
It would be `root` if `root` was specified in the plugin's flags otherwise the user whose home decky resides in.
e.g.: `deck`
"""
DECKY_VERSION: str = os.getenv("DECKY_VERSION", default="")
"""
The version of the decky loader.
Environment variable: `DECKY_VERSION`.
e.g.: `v2.5.0-pre1`
"""
DECKY_USER: str = os.getenv("DECKY_USER", default="")
"""
The user whose home decky resides in.
Environment variable: `DECKY_USER`.
e.g.: `deck`
"""
DECKY_USER_HOME: str = os.getenv("DECKY_USER_HOME", default="")
"""
The home of the user where decky resides in.
Environment variable: `DECKY_USER_HOME`.
e.g.: `/home/deck`
"""
DECKY_HOME: str = os.getenv("DECKY_HOME", default="")
"""
The root of the decky folder.
Environment variable: `DECKY_HOME`.
e.g.: `/home/deck/homebrew`
"""
DECKY_PLUGIN_SETTINGS_DIR: str = os.getenv(
"DECKY_PLUGIN_SETTINGS_DIR", default="")
"""
The recommended path in which to store configuration files (created automatically).
Environment variable: `DECKY_PLUGIN_SETTINGS_DIR`.
e.g.: `/home/deck/homebrew/settings/decky-plugin-template`
"""
DECKY_PLUGIN_RUNTIME_DIR: str = os.getenv(
"DECKY_PLUGIN_RUNTIME_DIR", default="")
"""
The recommended path in which to store runtime data (created automatically).
Environment variable: `DECKY_PLUGIN_RUNTIME_DIR`.
e.g.: `/home/deck/homebrew/data/decky-plugin-template`
"""
DECKY_PLUGIN_LOG_DIR: str = os.getenv("DECKY_PLUGIN_LOG_DIR", default="")
"""
The recommended path in which to store persistent logs (created automatically).
Environment variable: `DECKY_PLUGIN_LOG_DIR`.
e.g.: `/home/deck/homebrew/logs/decky-plugin-template`
"""
DECKY_PLUGIN_DIR: str = os.getenv("DECKY_PLUGIN_DIR", default="")
"""
The root of the plugin's directory.
Environment variable: `DECKY_PLUGIN_DIR`.
e.g.: `/home/deck/homebrew/plugins/decky-plugin-template`
"""
DECKY_PLUGIN_NAME: str = os.getenv("DECKY_PLUGIN_NAME", default="")
"""
The name of the plugin as specified in the 'plugin.json'.
Environment variable: `DECKY_PLUGIN_NAME`.
e.g.: `Example Plugin`
"""
DECKY_PLUGIN_VERSION: str = os.getenv("DECKY_PLUGIN_VERSION", default="")
"""
The version of the plugin as specified in the 'package.json'.
Environment variable: `DECKY_PLUGIN_VERSION`.
e.g.: `0.0.1`
"""
DECKY_PLUGIN_AUTHOR: str = os.getenv("DECKY_PLUGIN_AUTHOR", default="")
"""
The author of the plugin as specified in the 'plugin.json'.
Environment variable: `DECKY_PLUGIN_AUTHOR`.
e.g.: `John Doe`
"""
DECKY_PLUGIN_LOG: str = os.path.join(DECKY_PLUGIN_LOG_DIR, "plugin.log")
"""
The path to the plugin's main logfile.
Environment variable: `DECKY_PLUGIN_LOG`.
e.g.: `/home/deck/homebrew/logs/decky-plugin-template/plugin.log`
"""
"""
Migration helpers
"""
def migrate_any(target_dir: str, *files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories to a new location and remove old locations.
Specified files will be migrated to `target_dir`.
Specified directories will have their contents recursively migrated to `target_dir`.
Returns the mapping of old -> new location.
"""
file_map: dict[str, str] = {}
for f in files_or_directories:
if not os.path.exists(f):
file_map[f] = ""
continue
if os.path.isdir(f):
src_dir = f
src_file = "."
file_map[f] = target_dir
else:
src_dir = os.path.dirname(f)
src_file = os.path.basename(f)
file_map[f] = os.path.join(target_dir, src_file)
subprocess.run(["sh", "-c", "mkdir -p \"$3\"; tar -cf - -C \"$1\" \"$2\" | tar -xf - -C \"$3\" && rm -rf \"$4\"",
"_", src_dir, src_file, target_dir, f])
return file_map
def migrate_settings(*files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories relating to plugin settings to the recommended location and remove old locations.
Specified files will be migrated to `DECKY_PLUGIN_SETTINGS_DIR`.
Specified directories will have their contents recursively migrated to `DECKY_PLUGIN_SETTINGS_DIR`.
Returns the mapping of old -> new location.
"""
return migrate_any(DECKY_PLUGIN_SETTINGS_DIR, *files_or_directories)
def migrate_runtime(*files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories relating to plugin runtime data to the recommended location and remove old locations
Specified files will be migrated to `DECKY_PLUGIN_RUNTIME_DIR`.
Specified directories will have their contents recursively migrated to `DECKY_PLUGIN_RUNTIME_DIR`.
Returns the mapping of old -> new location.
"""
return migrate_any(DECKY_PLUGIN_RUNTIME_DIR, *files_or_directories)
def migrate_logs(*files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories relating to plugin logs to the recommended location and remove old locations.
Specified files will be migrated to `DECKY_PLUGIN_LOG_DIR`.
Specified directories will have their contents recursively migrated to `DECKY_PLUGIN_LOG_DIR`.
Returns the mapping of old -> new location.
"""
return migrate_any(DECKY_PLUGIN_LOG_DIR, *files_or_directories)
"""
Logging
"""
logging.basicConfig(filename=DECKY_PLUGIN_LOG,
format='[%(asctime)s][%(levelname)s]: %(message)s',
force=True)
logger: logging.Logger = logging.getLogger()
"""The main plugin logger writing to `DECKY_PLUGIN_LOG`."""
logger.setLevel(logging.INFO)
-173
View File
@@ -1,173 +0,0 @@
"""
This module exposes various constants and helpers useful for decky plugins.
* Plugin's settings and configurations should be stored under `DECKY_PLUGIN_SETTINGS_DIR`.
* Plugin's runtime data should be stored under `DECKY_PLUGIN_RUNTIME_DIR`.
* Plugin's persistent log files should be stored under `DECKY_PLUGIN_LOG_DIR`.
Avoid writing outside of `DECKY_HOME`, storing under the suggested paths is strongly recommended.
Some basic migration helpers are available: `migrate_any`, `migrate_settings`, `migrate_runtime`, `migrate_logs`.
A logging facility `logger` is available which writes to the recommended location.
"""
__version__ = '0.1.0'
import logging
"""
Constants
"""
HOME: str
"""
The home directory of the effective user running the process.
Environment variable: `HOME`.
If `root` was specified in the plugin's flags it will be `/root` otherwise the user whose home decky resides in.
e.g.: `/home/deck`
"""
USER: str
"""
The effective username running the process.
Environment variable: `USER`.
It would be `root` if `root` was specified in the plugin's flags otherwise the user whose home decky resides in.
e.g.: `deck`
"""
DECKY_VERSION: str
"""
The version of the decky loader.
Environment variable: `DECKY_VERSION`.
e.g.: `v2.5.0-pre1`
"""
DECKY_USER: str
"""
The user whose home decky resides in.
Environment variable: `DECKY_USER`.
e.g.: `deck`
"""
DECKY_USER_HOME: str
"""
The home of the user where decky resides in.
Environment variable: `DECKY_USER_HOME`.
e.g.: `/home/deck`
"""
DECKY_HOME: str
"""
The root of the decky folder.
Environment variable: `DECKY_HOME`.
e.g.: `/home/deck/homebrew`
"""
DECKY_PLUGIN_SETTINGS_DIR: str
"""
The recommended path in which to store configuration files (created automatically).
Environment variable: `DECKY_PLUGIN_SETTINGS_DIR`.
e.g.: `/home/deck/homebrew/settings/decky-plugin-template`
"""
DECKY_PLUGIN_RUNTIME_DIR: str
"""
The recommended path in which to store runtime data (created automatically).
Environment variable: `DECKY_PLUGIN_RUNTIME_DIR`.
e.g.: `/home/deck/homebrew/data/decky-plugin-template`
"""
DECKY_PLUGIN_LOG_DIR: str
"""
The recommended path in which to store persistent logs (created automatically).
Environment variable: `DECKY_PLUGIN_LOG_DIR`.
e.g.: `/home/deck/homebrew/logs/decky-plugin-template`
"""
DECKY_PLUGIN_DIR: str
"""
The root of the plugin's directory.
Environment variable: `DECKY_PLUGIN_DIR`.
e.g.: `/home/deck/homebrew/plugins/decky-plugin-template`
"""
DECKY_PLUGIN_NAME: str
"""
The name of the plugin as specified in the 'plugin.json'.
Environment variable: `DECKY_PLUGIN_NAME`.
e.g.: `Example Plugin`
"""
DECKY_PLUGIN_VERSION: str
"""
The version of the plugin as specified in the 'package.json'.
Environment variable: `DECKY_PLUGIN_VERSION`.
e.g.: `0.0.1`
"""
DECKY_PLUGIN_AUTHOR: str
"""
The author of the plugin as specified in the 'plugin.json'.
Environment variable: `DECKY_PLUGIN_AUTHOR`.
e.g.: `John Doe`
"""
DECKY_PLUGIN_LOG: str
"""
The path to the plugin's main logfile.
Environment variable: `DECKY_PLUGIN_LOG`.
e.g.: `/home/deck/homebrew/logs/decky-plugin-template/plugin.log`
"""
"""
Migration helpers
"""
def migrate_any(target_dir: str, *files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories to a new location and remove old locations.
Specified files will be migrated to `target_dir`.
Specified directories will have their contents recursively migrated to `target_dir`.
Returns the mapping of old -> new location.
"""
def migrate_settings(*files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories relating to plugin settings to the recommended location and remove old locations.
Specified files will be migrated to `DECKY_PLUGIN_SETTINGS_DIR`.
Specified directories will have their contents recursively migrated to `DECKY_PLUGIN_SETTINGS_DIR`.
Returns the mapping of old -> new location.
"""
def migrate_runtime(*files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories relating to plugin runtime data to the recommended location and remove old locations
Specified files will be migrated to `DECKY_PLUGIN_RUNTIME_DIR`.
Specified directories will have their contents recursively migrated to `DECKY_PLUGIN_RUNTIME_DIR`.
Returns the mapping of old -> new location.
"""
def migrate_logs(*files_or_directories: str) -> dict[str, str]:
"""
Migrate files and directories relating to plugin logs to the recommended location and remove old locations.
Specified files will be migrated to `DECKY_PLUGIN_LOG_DIR`.
Specified directories will have their contents recursively migrated to `DECKY_PLUGIN_LOG_DIR`.
Returns the mapping of old -> new location.
"""
"""
Logging
"""
logger: logging.Logger
"""The main plugin logger writing to `DECKY_PLUGIN_LOG`."""