mirror of
https://github.com/rommapp/romm.git
synced 2026-04-23 06:54:40 +00:00
Georges-Antoine Assi
86 lines
3.0 KiB
Python
86 lines
3.0 KiB
Python
from pathlib import Path, PureWindowsPath
|
|
|
|
from defusedxml import ElementTree as ET
|
|
|
|
from logger.logger import log
|
|
|
|
from .platforms import get_platform
|
|
from .types import LAUNCHBOX_PLATFORMS_DIR
|
|
|
|
|
|
class LocalSource:
|
|
def __init__(self) -> None:
|
|
self._cache: dict[str, dict[str, dict[str, str]]] = {}
|
|
self._mtime: dict[str, int] = {}
|
|
|
|
async def get_rom(self, fs_name: str, platform_slug: str) -> dict[str, str] | None:
|
|
if not LAUNCHBOX_PLATFORMS_DIR.exists():
|
|
return None
|
|
|
|
platform_name = get_platform(platform_slug).get("name")
|
|
xml_path = (
|
|
LAUNCHBOX_PLATFORMS_DIR / f"{platform_name}.xml" if platform_name else None
|
|
)
|
|
if not xml_path or not xml_path.exists():
|
|
return None
|
|
|
|
current_mtime = xml_path.stat().st_mtime_ns
|
|
if (
|
|
platform_slug not in self._cache
|
|
or self._mtime.get(platform_slug) != current_mtime
|
|
):
|
|
try:
|
|
indexed_val: dict[str, dict[str, str]] = {}
|
|
root = ET.parse(str(xml_path.resolve())).getroot()
|
|
if root is not None:
|
|
for game_elem in root.findall(".//Game"):
|
|
entry: dict[str, str] = {}
|
|
for child_elem in game_elem:
|
|
if child_elem.tag and child_elem.text is not None:
|
|
entry[child_elem.tag] = child_elem.text
|
|
if not entry:
|
|
continue
|
|
|
|
app_path = (entry.get("ApplicationPath") or "").strip()
|
|
if app_path:
|
|
app_base = PureWindowsPath(app_path).name.strip().lower()
|
|
if app_base:
|
|
indexed_val.setdefault(app_base, entry)
|
|
|
|
title = (entry.get("Title") or "").strip().lower()
|
|
if title:
|
|
indexed_val.setdefault(f"title:{title}", entry)
|
|
except (ET.ParseError, FileNotFoundError, PermissionError) as e:
|
|
log.warning(f"Failed to parse local LaunchBox XML {xml_path}: {e}")
|
|
self._cache[platform_slug] = {}
|
|
self._mtime[platform_slug] = current_mtime
|
|
return None
|
|
|
|
self._cache[platform_slug] = indexed_val
|
|
self._mtime[platform_slug] = current_mtime
|
|
|
|
indexed_val = self._cache[platform_slug]
|
|
|
|
if not indexed_val:
|
|
return None
|
|
|
|
fs_key = fs_name.strip().lower()
|
|
if not fs_key:
|
|
return None
|
|
|
|
direct = indexed_val.get(fs_key)
|
|
if direct is not None:
|
|
return direct
|
|
|
|
try:
|
|
stem = Path(fs_name).stem.strip().lower()
|
|
except Exception:
|
|
stem = ""
|
|
|
|
if stem:
|
|
by_title = indexed_val.get(f"title:{stem}")
|
|
if by_title is not None:
|
|
return by_title
|
|
|
|
return indexed_val.get(f"title:{fs_key}")
|