mirror of
https://github.com/Ryujinx/ryuko-ng.git
synced 2025-01-20 18:21:01 +00:00
9cd2d6550d
* Fix cheat_information regex * Add warning about default user profiles to notes * Fix disable_ro_section issues * Apply black formatting
142 lines
4.5 KiB
Python
142 lines
4.5 KiB
Python
import json
|
|
import os
|
|
from typing import Union
|
|
|
|
|
|
def get_disabled_ids_path(bot) -> str:
|
|
old_filepath = os.path.join(bot.state_dir, "data/disabled_tids.json")
|
|
new_filepath = os.path.join(bot.state_dir, "data/disabled_ids.json")
|
|
if os.path.isfile(old_filepath):
|
|
os.rename(old_filepath, new_filepath)
|
|
return new_filepath
|
|
|
|
|
|
def is_app_id_valid(app_id: str) -> bool:
|
|
return len(app_id) == 16 and app_id.isalnum()
|
|
|
|
|
|
def is_build_id_valid(build_id: str) -> bool:
|
|
return 32 <= len(build_id) <= 64 and build_id.isalnum()
|
|
|
|
|
|
def is_ro_section_valid(ro_section: dict[str, str]) -> bool:
|
|
return "module" in ro_section.keys() and "sdk_libraries" in ro_section.keys()
|
|
|
|
|
|
def get_disabled_ids(bot) -> dict[str, dict[str, Union[str, dict[str, str]]]]:
|
|
if os.path.isfile(get_disabled_ids_path(bot)):
|
|
with open(get_disabled_ids_path(bot), "r") as f:
|
|
disabled_ids = json.load(f)
|
|
# Migration code
|
|
if "app_id" not in disabled_ids.keys():
|
|
disabled_ids = {"app_id": disabled_ids, "build_id": {}, "ro_section": {}}
|
|
return disabled_ids
|
|
|
|
return {"app_id": {}, "build_id": {}, "ro_section": {}}
|
|
|
|
|
|
def set_disabled_ids(bot, contents: dict[str, dict[str, Union[str, dict[str, str]]]]):
|
|
with open(get_disabled_ids_path(bot), "w") as f:
|
|
json.dump(contents, f)
|
|
|
|
|
|
def is_app_id_disabled(bot, app_id: str) -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
app_id = app_id.lower()
|
|
return app_id in disabled_ids["app_id"].keys()
|
|
|
|
|
|
def is_build_id_disabled(bot, build_id: str) -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
build_id = build_id.lower()
|
|
if len(build_id) < 64:
|
|
build_id += "0" * (64 - len(build_id))
|
|
return build_id in disabled_ids["build_id"].keys()
|
|
|
|
|
|
def is_ro_section_disabled(bot, ro_section: dict[str, Union[str, list[str]]]) -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
matches = []
|
|
for note, entry in disabled_ids["ro_section"].items():
|
|
for key, content in entry.items():
|
|
if key == "module":
|
|
matches.append(ro_section[key].lower() == content.lower())
|
|
else:
|
|
matches.append(ro_section[key] == content)
|
|
if all(matches):
|
|
return True
|
|
else:
|
|
matches = []
|
|
return False
|
|
|
|
|
|
def add_disabled_app_id(bot, app_id: str, note="") -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
app_id = app_id.lower()
|
|
if app_id not in disabled_ids["app_id"].keys():
|
|
disabled_ids["app_id"][app_id] = note
|
|
set_disabled_ids(bot, disabled_ids)
|
|
return True
|
|
return False
|
|
|
|
|
|
def remove_disabled_app_id(bot, app_id: str) -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
app_id = app_id.lower()
|
|
if app_id in disabled_ids["app_id"].keys():
|
|
del disabled_ids["app_id"][app_id]
|
|
set_disabled_ids(bot, disabled_ids)
|
|
return True
|
|
return False
|
|
|
|
|
|
def add_disabled_build_id(bot, build_id: str, note="") -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
build_id = build_id.lower()
|
|
if len(build_id) < 64:
|
|
build_id += "0" * (64 - len(build_id))
|
|
if build_id not in disabled_ids["build_id"].keys():
|
|
disabled_ids["build_id"][build_id] = note
|
|
set_disabled_ids(bot, disabled_ids)
|
|
return True
|
|
return False
|
|
|
|
|
|
def remove_disabled_build_id(bot, build_id: str) -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
build_id = build_id.lower()
|
|
if len(build_id) < 64:
|
|
build_id += "0" * (64 - len(build_id))
|
|
if build_id in disabled_ids["build_id"].keys():
|
|
del disabled_ids["build_id"][build_id]
|
|
set_disabled_ids(bot, disabled_ids)
|
|
return True
|
|
return False
|
|
|
|
|
|
def add_disabled_ro_section(
|
|
bot, note: str, ro_section: dict[str, Union[str, list[str]]]
|
|
) -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
note = note.lower()
|
|
if note not in disabled_ids["ro_section"].keys():
|
|
disabled_ids["ro_section"][note] = {}
|
|
for key, content in ro_section.items():
|
|
if key == "module":
|
|
disabled_ids["ro_section"][note][key] = content.lower()
|
|
else:
|
|
disabled_ids["ro_section"][note][key] = content
|
|
set_disabled_ids(bot, disabled_ids)
|
|
return True
|
|
return False
|
|
|
|
|
|
def remove_disabled_ro_section(bot, note: str) -> bool:
|
|
disabled_ids = get_disabled_ids(bot)
|
|
note = note.lower()
|
|
if note in disabled_ids["ro_section"].keys():
|
|
del disabled_ids["ro_section"][note]
|
|
set_disabled_ids(bot, disabled_ids)
|
|
return True
|
|
return False
|