legendary/legendary/lfs/egl.py

101 lines
3.4 KiB
Python

# coding: utf-8
import configparser
import json
import os
from typing import List
from legendary.models.egl import EGLManifest
class EPCLFS:
# Known encryption key(s) for JSON user data
# Data is encrypted using AES-256-ECB mode
data_keys = []
def __init__(self):
if os.name == 'nt':
self.appdata_path = os.path.expandvars(
r'%LOCALAPPDATA%\EpicGamesLauncher\Saved\Config\Windows'
)
self.programdata_path = os.path.expandvars(
r'%PROGRAMDATA%\Epic\EpicGamesLauncher\Data\Manifests'
)
else:
self.appdata_path = self.programdata_path = None
self.config = configparser.ConfigParser(strict=False)
self.config.optionxform = lambda option: option
self.manifests = dict()
def read_config(self):
if not self.appdata_path:
raise ValueError('EGS AppData path is not set')
if not os.path.isdir(self.appdata_path):
raise ValueError('EGS AppData path does not exist')
self.config.read(os.path.join(self.appdata_path, 'GameUserSettings.ini'), encoding='utf-8')
def save_config(self):
if not self.appdata_path:
raise ValueError('EGS AppData path is not set')
if not os.path.isdir(self.appdata_path):
raise ValueError('EGS AppData path does not exist')
with open(os.path.join(self.appdata_path, 'GameUserSettings.ini'), 'w', encoding='utf-8') as f:
self.config.write(f, space_around_delimiters=False)
def read_manifests(self):
if not self.programdata_path:
raise ValueError('EGS ProgramData path is not set')
if not os.path.isdir(self.programdata_path):
# Not sure if we should `raise` here as well
return
for f in os.listdir(self.programdata_path):
if f.endswith('.item'):
data = json.load(open(os.path.join(self.programdata_path, f), encoding='utf-8'))
self.manifests[data['AppName']] = data
def get_manifests(self) -> List[EGLManifest]:
if not self.manifests:
self.read_manifests()
return [EGLManifest.from_json(m) for m in self.manifests.values()]
def get_manifest(self, app_name) -> EGLManifest:
if not self.manifests:
self.read_manifests()
if app_name in self.manifests:
return EGLManifest.from_json(self.manifests[app_name])
else:
raise ValueError('Cannot find manifest')
def set_manifest(self, manifest: EGLManifest):
if not self.programdata_path:
raise ValueError('EGS ProgramData path is not set')
if not os.path.isdir(self.programdata_path):
raise ValueError('EGS ProgramData path does not exist')
manifest_data = manifest.to_json()
self.manifests[manifest.app_name] = manifest_data
_path = os.path.join(self.programdata_path, f'{manifest.installation_guid}.item')
with open(_path, 'w', encoding='utf-8') as f:
json.dump(manifest_data, f, indent=4, sort_keys=True)
def delete_manifest(self, app_name):
if not self.manifests:
self.read_manifests()
if app_name not in self.manifests:
raise ValueError('AppName is not in manifests!')
manifest = EGLManifest.from_json(self.manifests.pop(app_name))
os.remove(os.path.join(self.programdata_path, f'{manifest.installation_guid}.item'))