8ca4213bed* Include deck user application files in the list * Fix parsing of desktop files that are indented * Small optimization to the dedup process of the apps' names. * Use env var provided by decky to dynamically search the user's desktop files
137 lines
4.1 KiB
Python
137 lines
4.1 KiB
Python
import os, json, base64, ssl, certifi
|
|
from pathlib import Path
|
|
from json import dumps as jsonDumps
|
|
from itertools import chain
|
|
|
|
import decky_plugin
|
|
|
|
from subprocess import Popen, PIPE
|
|
|
|
from urllib.error import HTTPError
|
|
from urllib.request import urlopen, Request
|
|
|
|
|
|
confdir = os.environ["DECKY_PLUGIN_SETTINGS_DIR"]
|
|
|
|
send_buffer = []
|
|
|
|
|
|
def split_string(string):
|
|
return [
|
|
string[i : i + 1024 * 60] for i in range(0, len(string), 1024 * 60)
|
|
] # every 60KB
|
|
|
|
|
|
class Plugin:
|
|
async def get_flatpaks(self):
|
|
proc = Popen(
|
|
'flatpak list --app --columns="name,application" | awk \'BEGIN {FS="\\t"} {print "{\\"name\\":\\""$1"\\",\\"exec\\":\\"/usr/bin/flatpak run "$2"\\"},"}\\\'',
|
|
stdout=PIPE,
|
|
stderr=None,
|
|
shell=True,
|
|
)
|
|
packages = proc.communicate()[0]
|
|
packages = packages.decode("utf-8")
|
|
packages = packages[:-2]
|
|
return "[" + packages + "]"
|
|
|
|
async def get_desktops(self):
|
|
packages = []
|
|
for desktopFile in chain(
|
|
Path("/usr/share/applications").glob("*.desktop"),
|
|
Path(f"{decky_plugin.DECKY_USER_HOME}/.local/share/applications/").glob("*.desktop"),
|
|
):
|
|
if not desktopFile.is_file():
|
|
continue
|
|
with open(desktopFile) as f:
|
|
package = {}
|
|
foundName = foundExec = False
|
|
for line in f:
|
|
line = line.strip()
|
|
if line.startswith("Name="):
|
|
foundName = True
|
|
package["name"] = line[5:]
|
|
elif line.startswith("Exec="):
|
|
foundExec = True
|
|
package["exec"] = line[5:]
|
|
if foundName and foundExec:
|
|
packages.append(package)
|
|
break
|
|
|
|
return jsonDumps(packages)
|
|
|
|
async def get_config(self):
|
|
with open(os.path.join(confdir, "config.json"), "r") as f:
|
|
return json.load(f)
|
|
|
|
async def set_config_value(self, key, value):
|
|
config = json.load(open(os.path.join(confdir, "config.json")))
|
|
config[key] = value
|
|
with open(os.path.join(confdir, "config.json"), "w") as f:
|
|
json.dump(config, f)
|
|
|
|
return config
|
|
|
|
async def get_id(self):
|
|
with open(os.path.join(confdir, "scid.txt"), "r") as sc:
|
|
id = sc.read()
|
|
try:
|
|
id = int(id)
|
|
return id
|
|
except ValueError:
|
|
return -1
|
|
|
|
async def set_id(self, id):
|
|
with open(os.path.join(confdir, "scid.txt"), "w") as sc:
|
|
sc.write(str(id))
|
|
|
|
async def get_req_imgb64(self, url):
|
|
global send_buffer
|
|
if len(send_buffer) != 0:
|
|
return
|
|
|
|
req = Request(url)
|
|
|
|
req.add_header("User-Agent", "SDH-QuickLaunch")
|
|
|
|
try:
|
|
content = urlopen(
|
|
req, context=ssl.create_default_context(cafile=certifi.where())
|
|
).read()
|
|
img = base64.b64encode(content).decode("ascii")
|
|
send_buffer = split_string(img)
|
|
new_chunk = send_buffer.pop(0)
|
|
return {"data": new_chunk, "is_last": len(send_buffer) == 0}
|
|
except HTTPError:
|
|
decky_plugin.logger.error("HTTPError while requesting " + url)
|
|
pass
|
|
|
|
async def receive_next_chunk(self):
|
|
global send_buffer
|
|
new_chunk = send_buffer.pop(0)
|
|
return {"data": new_chunk, "is_last": len(send_buffer) == 0}
|
|
|
|
async def _main(self):
|
|
decky_plugin.logger.info("Loading plugin")
|
|
|
|
try:
|
|
os.mkdir(confdir)
|
|
except FileExistsError:
|
|
pass
|
|
|
|
try:
|
|
sc = open(os.path.join(confdir, "scid.txt"), "x")
|
|
sc.close()
|
|
except FileExistsError:
|
|
pass
|
|
try:
|
|
sc = open(os.path.join(confdir, "config.json"), "x")
|
|
sc.write("{}")
|
|
sc.close()
|
|
except FileExistsError:
|
|
pass
|
|
|
|
decky_plugin.logger.info("Plugin loaded")
|
|
|
|
async def _unload(self):
|
|
pass
|