From 6204392c3dfdceca21a17d1285b93b6a0cc759ed Mon Sep 17 00:00:00 2001 From: pythoninthegrass <4097471+pythoninthegrass@users.noreply.github.com> Date: Wed, 21 Feb 2024 00:44:29 -0600 Subject: [PATCH] refactor: split config into file - Lots of LOC that make the main script hard to read - Remove `main.py` as that can come later --- _config.py => config.py | 87 +++++++++- main.py | 375 ---------------------------------------- 2 files changed, 78 insertions(+), 384 deletions(-) rename _config.py => config.py (59%) delete mode 100644 main.py diff --git a/_config.py b/config.py similarity index 59% rename from _config.py rename to config.py index 7f93ee8..21622c8 100644 --- a/_config.py +++ b/config.py @@ -1,5 +1,6 @@ #!/usr/bin/env python3 +from decouple import config from pathlib import Path # paths @@ -48,19 +49,87 @@ chromedirectory = "/usr/bin/flatpak" # Define a dictionary of original folder names folder_names = { + 'Amazon Games': 'AmazonGamesLauncher', + 'Battle.net': 'Battle.netLauncher', + 'DMM Games': 'DMMGameLauncher', + 'EA App': 'TheEAappLauncher', 'Epic Games': 'EpicGamesLauncher', 'Gog Galaxy': 'GogGalaxyLauncher', - 'Ubisoft Connect': 'UplayLauncher', - 'Origin': 'OriginLauncher', - 'Battle.net': 'Battle.netLauncher', - 'EA App': 'TheEAappLauncher', - 'Amazon Games': 'AmazonGamesLauncher', - 'itch.io': 'itchioLauncher', - 'Legacy Games': 'LegacyGamesLauncher', 'Humble Bundle': 'HumbleGamesLauncher', 'IndieGala Client': 'IndieGalaLauncher', - 'Rockstar Games Launcher': 'RockstarGamesLauncher', + 'itch.io': 'itchioLauncher', + 'Legacy Games': 'LegacyGamesLauncher', 'Minecraft: Java Edition': 'MinecraftLauncher', + 'Origin': 'OriginLauncher', 'Playstation Plus': 'PlaystationPlusLauncher', - 'DMM Games': 'DMMGameLauncher', + 'Rockstar Games Launcher': 'RockstarGamesLauncher', + 'Ubisoft Connect': 'UplayLauncher', + 'VK Play': 'VKPlayLauncher', +} + +# Variables from NonSteamLaunchers.sh +steamid3 = config('steamid3', default='') +logged_in_home = config('logged_in_home') +compat_tool_name = config('compat_tool_name') +controller_config_path = config('controller_config_path') +python_version = config('python_version') +#Scanner Variables +epic_games_launcher = config('epic_games_launcher') +ubisoft_connect_launcher = config('ubisoft_connect_launcher') +ea_app_launcher = config('ea_app_launcher') +gog_galaxy_launcher = config('gog_galaxy_launcher') +bnet_launcher = config('bnet_launcher') +amazon_launcher = config('amazon_launcher') + +# Variables of the Launchers +# Define the path of the Launchers +epicshortcutdirectory = config('epicshortcutdirectory') +gogshortcutdirectory = config('gogshortcutdirectory') +uplayshortcutdirectory = config('uplayshortcutdirectory') +battlenetshortcutdirectory = config('battlenetshortcutdirectory') +eaappshortcutdirectory = config('eaappshortcutdirectory') +amazonshortcutdirectory = config('amazonshortcutdirectory') +itchioshortcutdirectory = config('itchioshortcutdirectory') +legacyshortcutdirectory = config('legacyshortcutdirectory') +humbleshortcutdirectory = config('humbleshortcutdirectory') +indieshortcutdirectory = config('indieshortcutdirectory') +rockstarshortcutdirectory = config('rockstarshortcutdirectory') +glyphshortcutdirectory = config('glyphshortcutdirectory') +minecraftshortcutdirectory = config('minecraftshortcutdirectory') +psplusshortcutdirectory = config('psplusshortcutdirectory') +vkplayhortcutdirectory = config('vkplayhortcutdirectory') + +# Streaming +chromedirectory = config('chromedirectory') +websites_str = config('custom_websites_str') +custom_websites = websites_str.split(', ') if websites_str else [] + +# Define your mapping +flavor_mapping = { + "Blizzard Arcade Collection": "RTRO", + "Call of Duty: Black Ops - Cold War": "ZEUS", + "Call of Duty: Black Ops 4": "VIPR", + "Call of Duty: Modern Warfare III": "SPOT", + "Call of Duty: Modern Warfare": "ODIN", + "Call of Duty: MW 2 Campaign Remastered": "LAZR", + "Call of Duty: Vanguard": "FORE", + "Call of Duty": "AUKS", + "Crash Bandicoot 4: It's About Time": "WLBY", + "Diablo II: Resurrected": "OSI", + "Diablo III": "D3", + "Diablo Immortal (PC)": "ANBS", + "Diablo IV": "Fen", + "Diablo": "D1", + "Hearthstone": "WTCG", + "Heroes of the Storm": "Hero", + "Overwatch 2": "Pro", + "Overwatch": "Pro", + "StarCraft 2": "S2", + "StarCraft": "S1", + "Warcraft Arclight Rumble": "GRY", + "Warcraft II: Battle.net Edition": "W2", + "Warcraft III: Reforged": "W3", + "Warcraft: Orcs & Humans": "W1", + "World of Warcraft Classic": "WoWC", + "World of Warcraft": "WoW", } diff --git a/main.py b/main.py deleted file mode 100644 index d9a4a27..0000000 --- a/main.py +++ /dev/null @@ -1,375 +0,0 @@ -#!/usr/bin/env python3 - -import binascii -import _config -import os -import re -import shutil -import sys -import vdf -from _config import folder_names -from pathlib import Path - -# Append the path to the vdf module to the system path -python_version = f"{sys.version_info.major}.{sys.version_info.minor}" -sys.path.insert(0, os.path.expanduser( - f"~/Downloads/NonSteamLaunchersInstallation/lib/python{python_version}/site-packages") -) -print(sys.path) - -# $HOME -logged_in_home = str(Path.home()) - -# TODO: test in holoiso -# Define the path of the shortcuts.vdf file -userdata_parent = Path(logged_in_home) / '.steam/root/userdata' -userdata_folder = userdata_folder.glob('*') -shortcuts_vdf_path = None -for file in userdata_folder: - if file.name == 'shortcuts.vdf': - shortcuts_vdf_path = file - break - -# TODO: ^^ -# Load the shortcuts.vdf file -with open(shortcuts_vdf_path, 'rb') as f: - shortcuts = vdf.binary_load(f) - -# Check if the 'shortcuts' key exists in the dictionary -if 'shortcuts' not in shortcuts: - # Create an empty 'shortcuts' entry - shortcuts['shortcuts'] = {} - -# Check the format of the 'shortcuts' entry -if isinstance(shortcuts['shortcuts'], dict): - # The 'shortcuts' entry is a dictionary - for key, value in shortcuts['shortcuts'].items(): - # Check the type of the value - if not isinstance(value, (str, int, dict)): - pass - -# Define the path of the Launchers -# ! Moved to config.py - -# Streaming -chromedirectory = '$chromedirectory' -websites_str = '$custom_websites_str' -custom_websites = websites_str.split(', ') - -app_ids = [] - - -def get_steam_shortcut_id(exe, appname): - unique_id = ''.join([exe, appname]) - id_int = binascii.crc32(str.encode(unique_id)) | 0x80000000 - return id_int - - -app_id_to_name = {} - - -def create_new_entry(shortcutdirectory, appname, launchoptions, startingdir): - if shortcutdirectory != '' and launchoptions != '': - exe = f'"{shortcutdirectory}"' - if shortcutdirectory != chromedirectory: - appid = get_steam_shortcut_id(exe, appname) - app_ids.append(appid) - app_id_to_name[appid] = appname - else: - appid = None - - # Create a new entry for the Steam shortcut - new_entry = { - 'appid': f'{str(appid)}' if appid is not None else '', - 'appname': appname, - 'exe': shortcutdirectory, - 'StartDir': startingdir, - 'icon': '', - 'ShortcutPath': '', - 'LaunchOptions': launchoptions, - 'IsHidden': 0, - 'AllowDesktopConfig': 1, - 'AllowOverlay': 1, - 'OpenVR': 0, - 'Devkit': 0, - 'DevkitGameID': '', - 'LastPlayTime': 0, - 'tags': { - '0': 'favorite' - } - } - - # Add the new entry to the shortcuts dictionary - entry_exists = False - if type(shortcuts['shortcuts']) == list: - for entry in shortcuts['shortcuts']: - entry.setdefault('appname', '') - entry.setdefault('exe', '') - if entry['appname'] == new_entry['appname'] and entry['exe'] == new_entry['exe']: - entry_exists = True - break - if not entry_exists: - shortcuts['shortcuts'].append(new_entry) - elif type(shortcuts['shortcuts']) == dict: - for key in shortcuts['shortcuts'].keys(): - shortcuts['shortcuts'][key].setdefault('appname', '') - shortcuts['shortcuts'][key].setdefault('exe', '') - if shortcuts['shortcuts'][key]['appname'] == new_entry['appname'] and shortcuts['shortcuts'][key]['exe'] == new_entry['exe']: - entry_exists = True - break - if not entry_exists: - # Check if the shortcuts['shortcuts'] dictionary is empty - if not shortcuts['shortcuts']: - max_key = -1 - else: - # Find the highest key value - max_key = max(int(key) for key in shortcuts['shortcuts'].keys()) - # Add the new entry with a key value one higher than the current maximum - shortcuts['shortcuts'][str(max_key + 1)] = new_entry - - -# TODO: extract logic from shell script and move either here or config.py -create_new_entry('$epicshortcutdirectory', 'Epic Games', '$epiclaunchoptions', '$epicstartingdir') -create_new_entry('$gogshortcutdirectory', 'Gog Galaxy', '$goglaunchoptions', '$gogstartingdir') -create_new_entry('$uplayshortcutdirectory', 'Ubisoft Connect', '$uplaylaunchoptions', '$uplaystartingdir') -create_new_entry('$originshortcutdirectory', 'Origin', '$originlaunchoptions', '$originstartingdir') -create_new_entry('$battlenetshortcutdirectory', 'Battle.net', '$battlenetlaunchoptions', '$battlenetstartingdir') -create_new_entry('$eaappshortcutdirectory', 'EA App', '$eaapplaunchoptions', '$eaappstartingdir') -create_new_entry('$amazonshortcutdirectory', 'Amazon Games', '$amazonlaunchoptions', '$amazonstartingdir') -create_new_entry('$itchioshortcutdirectory', 'itch.io', '$itchiolaunchoptions', '$itchiostartingdir') -create_new_entry('$legacyshortcutdirectory', 'Legacy Games', '$legacylaunchoptions', '$legacystartingdir') -create_new_entry('$humbleshortcutdirectory', 'Humble Bundle', '$humblelaunchoptions', '$humblestartingdir') -create_new_entry('$indieshortcutdirectory', 'IndieGala Client', '$indielaunchoptions', '$indiestartingdir') -create_new_entry('$rockstarshortcutdirectory', 'Rockstar Games Launcher', '$rockstarlaunchoptions', '$rockstarstartingdir') -create_new_entry('$glyphshortcutdirectory', 'Glyph', '$glyphlaunchoptions', '$glyphstartingdir') -create_new_entry('$minecraftshortcutdirectory', 'Minecraft: Java Edition', '$minecraftlaunchoptions', '$minecraftstartingdir') -create_new_entry('$psplusshortcutdirectory', 'Playstation Plus', '$pspluslaunchoptions', '$psplusstartingdir') -create_new_entry('$dmmshortcutdirectory', 'DMM Games', '$dmmlaunchoptions', '$dmmstartingdir') -create_new_entry('$chromedirectory', 'Xbox Games Pass', '$xboxchromelaunchoptions', '$chrome_startdir') -create_new_entry('$chromedirectory', 'GeForce Now', '$geforcechromelaunchoptions', '$chrome_startdir') -create_new_entry('$chromedirectory', 'Netflix', '$netlfixchromelaunchoptions', '$chrome_startdir') -create_new_entry('$chromedirectory', 'Hulu', '$huluchromelaunchoptions', '$chrome_startdir') -create_new_entry('$chromedirectory', 'Disney+', '$disneychromelaunchoptions', '$chrome_startdir') -create_new_entry('$chromedirectory', 'Amazon Prime Video', '$amazonchromelaunchoptions', '$chrome_startdir') -create_new_entry('$chromedirectory', 'Youtube', '$youtubechromelaunchoptions', '$chrome_startdir') -create_new_entry('$chromedirectory', 'Amazon Luna', '$lunachromelaunchoptions', '$chrome_startdir') - -# Iterate over each custom website -for custom_website in custom_websites: - # Check if the custom website is not an empty string - if custom_website: - # Remove any leading or trailing spaces from the custom website URL - custom_website = custom_website.strip() - - # Remove the 'http://' or 'https://' prefix and the 'www.' prefix, if present - clean_website = custom_website.replace( - 'http://', '').replace('https://', '').replace('www.', '') - - # Define a regular expression pattern to extract the game name from the URL - pattern = r'/games/([\w-]+)' - - # Use the regex to search for the game name in the custom website URL - match = re.search(pattern, custom_website) - - # Check if a match was found - if match: - # Extract the game name from the match object - game_name = match.group(1) - - # Replace hyphens with spaces - game_name = game_name.replace('-', ' ') - - # Capitalize the first letter of each word in the game name - game_name = game_name.title() - else: - # Use the entire URL as the entry name - game_name = clean_website - - # Define the launch options for this website - chromelaunch_options = f"""run - --branch=stable - --arch=x86_64 - --command=/app/bin/chrome - --file-forwarding com.google.Chrome @@u @@ - - -window-size=1280,800 - --force-device-scale-factor=1.00 - --device-scale-factor=1.00 - --kiosk https://{clean_website}/ - --chrome-kiosk-type=fullscreen - --no-first-run - --enable-features=OverlayScrollbar""" - - # Call the create_new_entry function for this website - create_new_entry('$chromedirectory', game_name, chromelaunch_options, '$chrome_startdir') - -print(f'app_id_to_name: {app_id_to_name}') - -# Save the updated shortcuts dictionary to the shortcuts.vdf file -with open('$shortcuts_vdf_path', 'wb') as f: - vdf.binary_dump(shortcuts, f) - -# Writes to the config.vdf File - -excluded_appids = [] - -# Update the config.vdf file -with open(config_vdf_path, 'r') as f: - config = vdf.load(f) - -# Check if the CompatToolMapping key exists -if 'CompatToolMapping' not in config['InstallConfigStore']['Software']['Valve']['Steam']: - # Create the CompatToolMapping key and set its value to an empty dictionary - config['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'] = {} - -for app_id in app_ids: - # Check if the app_id is in the list of excluded appids - if app_id not in excluded_appids: - # Update the CompatToolMapping for this app_id - if str(app_id) in config['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping']: - config['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)]['name'] = '$compat_tool_name' - config['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)]['config'] = '' - config['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)]['priority'] = '250' - else: - config['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)] = {'name': '$compat_tool_name', 'config': '', 'priority': '250'} - -# Save the updated config dictionary to the config.vdf file -with open('$config_vdf_path', 'w') as f: - vdf.dump(config, f) - -# Load the configset_controller_neptune.vdf file -with open('$controller_config_path', 'r') as f: - config = vdf.load(f) - -# Add new entries for the games -for app_id in app_ids: - config['controller_config'][str(app_id)] = { - 'workshop': 'workshop_id' - } - -# TODO: convert to function -# Add new entries for the installed launchers and games -config['controller_config']['epic games'] = { - 'workshop': '2800178806' -} -config['controller_config']['gog galaxy'] = { - 'workshop': '2877189386' -} -config['controller_config']['ubisoft connect'] = { - 'workshop': '2804140248' -} -config['controller_config']['amazon games'] = { - 'workshop': '2871935783' -} -config['controller_config']['battlenet'] = { - 'workshop': '2887894308' -} -config['controller_config']['Origin'] = { - 'workshop': '2856043168' -} -config['controller_config']['rockstar games launcher'] = { - 'workshop': '1892570391' -} -config['controller_config']['indiegala'] = { - 'template': 'controller_neptune_webbrowser.vdf' -} -config['controller_config']['legacy games'] = { - 'template': 'controller_neptune_webbrowser.vdf' -} -config['controller_config']['ea app'] = { - 'workshop': '2899822740' -} -config['controller_config']['itchio'] = { - 'workshop': '2845891813' -} -config['controller_config']['humble games collection'] = { - 'workshop': '2883791560' -} -config['controller_config']['minecraft java edition'] = { - 'workshop': '2980553929' -} -config['controller_config']['playstation plus'] = { - 'workshop': 'controller_neptune_webbrowser.vdf' -} -config['controller_config']['glyph'] = { - 'template': 'controller_neptune_webbrowser.vdf' -} -config['controller_config']['dmm games'] = { - 'template': 'controller_neptune_webbrowser.vdf' -} -config['controller_config']['amazon prime video'] = { - 'workshop': '2970669392' -} -config['controller_config']['hulu'] = { - 'workshop': '2970669392' -} -config['controller_config']['netflix'] = { - 'workshop': '2970669392' -} -config['controller_config']['disney+'] = { - 'workshop': '2970669392' -} -config['controller_config']['youtube'] = { - 'workshop': '2970669392' -} -config['controller_config']['geforce now'] = { - 'template': 'controller_neptune_gamepad+mouse.vdf' -} -config['controller_config']['amazon luna'] = { - 'template': 'controller_neptune_gamepad+mouse.vdf' -} - -# Save the updated config dictionary to the configset_controller_neptune.vdf file -with open('$controller_config_path', 'w') as f: - vdf.dump(config, f) - -# Define the path to the compatdata directory -compatdata_dir = '${logged_in_home}/.local/share/Steam/steamapps/compatdata' - -# Iterate over each launcher in the folder_names dictionary -for launcher_name, folder in folder_names.items(): - # Define the current path of the folder - current_path = os.path.join(compatdata_dir, folder) - - # Check if the folder exists - if os.path.exists(current_path): - print(f'{launcher_name}: {folder} exists') - # Get the app ID for this launcher from the app_id_to_name dictionary - appid = next(key for key, value in app_id_to_name.items() if value == launcher_name) - - # Define the new path of the folder - new_path = os.path.join(compatdata_dir, str(appid)) - - # Rename the folder - os.rename(current_path, new_path) - - # Define the path of the symbolic link - symlink_path = os.path.join(compatdata_dir, folder) - - # Create a symbolic link to the renamed folder - os.symlink(new_path, symlink_path) - else: - print(f'{launcher_name}: {folder} does not exist') - -# Check if the NonSteamLaunchers folder exists -if os.path.exists(os.path.join(compatdata_dir, 'NonSteamLaunchers')): - # Get the first app ID from the app_ids list - first_app_id = app_ids[0] - - # Define the current path of the NonSteamLaunchers folder - current_path = os.path.join(compatdata_dir, 'NonSteamLaunchers') - - # Check if NonSteamLaunchers is already a symbolic link - if os.path.islink(current_path): - print('NonSteamLaunchers is already a symbolic link') - else: - # Define the new path of the NonSteamLaunchers folder - new_path = os.path.join(compatdata_dir, str(first_app_id)) - - # Move the NonSteamLaunchers folder to the new path - shutil.move(current_path, new_path) - - # Define the path of the symbolic link - symlink_path = os.path.join(compatdata_dir, 'NonSteamLaunchers') - - # Create a symbolic link to the renamed NonSteamLaunchers folder - os.symlink(new_path, symlink_path)