#!/usr/bin/env python3 import os, re, vdf import json import shutil import binascii import ctypes import zipfile import time import sys import subprocess from urllib.request import urlopen from urllib.request import urlretrieve import xml.etree.ElementTree as ET # Read variables from a file with open(f"{os.environ['HOME']}/.config/systemd/user/env_vars", 'r') as f: lines = f.readlines() for line in lines: if line.startswith('export '): line = line[7:] # Remove 'export ' name, value = line.strip().split('=') os.environ[name] = value # Variables from NonSteamLaunchers.sh steamid3 = os.environ['steamid3'] logged_in_home = os.environ['logged_in_home'] compat_tool_name = os.environ['compat_tool_name'] epic_games_launcher = os.environ.get('epic_games_launcher', '') ubisoft_connect_launcher = os.environ.get('ubisoft_connect_launcher', '') ea_app_launcher = os.environ.get('ea_app_launcher', '') # Create Folders # Define the repository and the folders to clone repo_url = 'https://github.com/moraroy/NonSteamLaunchers-On-Steam-Deck/archive/refs/heads/main.zip' folders_to_clone = ['requests', 'urllib3', 'steamgrid'] # Get the directory of the Python script current_dir = os.path.dirname(os.path.realpath(__file__)) # Define the parent folder parent_folder = os.path.join(current_dir, 'Modules') os.makedirs(parent_folder, exist_ok=True) # Check if the folders already exist folders_exist = all(os.path.exists(os.path.join(current_dir, parent_folder, folder)) for folder in folders_to_clone) if not folders_exist: # Download the repository as a zip file zip_file_path = os.path.join(current_dir, 'repo.zip') urlretrieve(repo_url, zip_file_path) # Extract the zip file with zipfile.ZipFile(zip_file_path, 'r') as zip_ref: zip_ref.extractall(current_dir) # Move the folders to the current directory and delete the unnecessary files for folder in folders_to_clone: destination_path = os.path.join(current_dir, parent_folder, folder) source_path = os.path.join(current_dir, 'NonSteamLaunchers-On-Steam-Deck-main', 'Modules', folder) if not os.path.exists(destination_path): shutil.move(source_path, destination_path) # Delete the downloaded zip file and the extracted repository folder os.remove(zip_file_path) shutil.rmtree(os.path.join(current_dir, 'NonSteamLaunchers-On-Steam-Deck-main')) # Now that the requests module has been downloaded, you can import it sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), parent_folder)) import requests from steamgrid import SteamGridDB #Set Up nslgamescanner.service # Define the paths service_path = f"{logged_in_home}/.config/systemd/user/nslgamescanner.service" # Define the service file content service_content = f""" [Unit] Description=NSL Game Scanner [Service] ExecStart=/usr/bin/python3 '{logged_in_home}/.config/systemd/user/NSLGameScanner.py' Restart=always RestartSec=10 StartLimitBurst=40 StartLimitInterval=240 [Install] WantedBy=default.target """ # Check if the service file already exists if not os.path.exists(service_path): # Create the service file with open(service_path, 'w') as f: f.write(service_content) print("Service file created.") # Check if the service is already running result = subprocess.run(['systemctl', '--user', 'is-active', 'nslgamescanner.service'], stdout=subprocess.PIPE) if result.stdout.decode('utf-8').strip() != 'active': # Reload the systemd manager configuration subprocess.run(['systemctl', '--user', 'daemon-reload']) # Enable the service to start on boot subprocess.run(['systemctl', '--user', 'enable', 'nslgamescanner.service']) # Start the service immediately subprocess.run(['systemctl', '--user', 'start', 'nslgamescanner.service']) print("Service started.") else: print("Service is already running.") #Code def get_steam_shortcut_id(exe_path, display_name): unique_id = "".join([exe_path, display_name]) id_int = binascii.crc32(str.encode(unique_id)) | 0x80000000 signed = ctypes.c_int(id_int) # print(f"Signed ID: {signed.value}") return signed.value def get_unsigned_shortcut_id(signed_shortcut_id): unsigned = ctypes.c_uint(signed_shortcut_id) # print(f"Unsigned ID: {unsigned.value}") return unsigned.value # Initialize an empty dictionary to serve as the cache api_cache = {} #API KEYS FOR NONSTEAMLAUNCHER USE ONLY sgdb = SteamGridDB('239745e39536fdc12bfa84ce0056b036') api_key = '239745e39536fdc12bfa84ce0056b036' #GLOBAL VARS created_shortcuts = [] new_shortcuts_added = False shortcuts_updated = False shortcut_id = None # Initialize shortcut_id # Load the existing shortcuts with open(f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/shortcuts.vdf", 'rb') as file: shortcuts = vdf.binary_loads(file.read()) # Open the config.vdf file with open(f"{logged_in_home}/.steam/root/config/config.vdf", 'r') as file: config_data = vdf.load(file) def get_sgdb_art(game_id, app_id): for art_type in ["icons", "logos", "heroes", "grids"]: print(f"Downloading {art_type} artwork...") download_artwork(game_id, api_key, art_type, app_id) print("Downloading grids artwork of size 600x900...") download_artwork(game_id, api_key, "grids", app_id, "600x900") print("Downloading grids artwork of size 920x430...") download_artwork(game_id, api_key, "grids", app_id, "920x430") def download_artwork(game_id, api_key, art_type, shortcut_id, dimensions=None): # Create a cache key based on the function's arguments cache_key = (game_id, art_type, dimensions) # Check if the artwork already exists if dimensions is not None: filename = get_file_name(art_type, shortcut_id, dimensions) else: filename = get_file_name(art_type, shortcut_id) file_path = f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{filename}" if os.path.exists(file_path): print(f"Artwork for {game_id} already exists. Skipping download.") return # If the result is in the cache, use it if cache_key in api_cache: data = api_cache[cache_key] else: # If the result is not in the cache, make the API call print(f"Game ID: {game_id}, API Key: {api_key}") url = f"https://www.steamgriddb.com/api/v2/{art_type}/game/{game_id}" if dimensions: url += f"?dimensions={dimensions}" headers = {'Authorization': f'Bearer {api_key}'} print(f"Sending request to: {url}") # Added print statement response = requests.get(url, headers=headers) if response.status_code == 200: data = response.json() # Store the result in the cache api_cache[cache_key] = data # Continue with the rest of your function using `data` for artwork in data['data']: image_url = artwork['thumb'] print(f"Downloading image from: {image_url}") # Added print statement try: response = requests.get(image_url, stream=True) response.raise_for_status() if response.status_code == 200: with open(file_path, 'wb') as file: file.write(response.content) break except requests.exceptions.RequestException as e: print(f"Error downloading image: {e}") if art_type == 'icons': download_artwork(game_id, api_key, 'icons_ico', shortcut_id) def get_game_id(game_name): print(f"Searching for game ID for: {game_name}") games = sgdb.search_game(game_name) for game in games: if game.name == game_name: # Case-sensitive comparison print(f"Found game ID: {game.id}") return game.id # Fallback: return the ID of the first game in the search results if games: print(f"No exact match found. Using game ID of the first result: {games[0].name}: {games[0].id}") return games[0].id print("No game ID found") return "default_game_id" # Return a default value when no games are found def get_file_name(art_type, shortcut_id, dimensions=None): singular_art_type = art_type.rstrip('s') if art_type == 'icons': if os.path.exists(f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{shortcut_id}-{singular_art_type}.png"): return f"{shortcut_id}-{singular_art_type}.png" else: return f"{shortcut_id}-{singular_art_type}.ico" elif art_type == 'grids': if dimensions == '600x900': return f"{shortcut_id}p.png" else: return f"{shortcut_id}.png" elif art_type == 'heroes': return f"{shortcut_id}_hero.png" elif art_type == 'logos': return f"{shortcut_id}_logo.png" else: return f"{shortcut_id}.png" def is_match(name1, name2): if name1 and name2: return name1.lower() in name2.lower() or name2.lower() in name1.lower() else: return False def add_compat_tool(app_id): if 'CompatToolMapping' not in config_data['InstallConfigStore']['Software']['Valve']['Steam']: config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'] = {} print(f"CompatToolMapping key not found in config.vdf, creating.") if str(app_id) in config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'] and config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)]['name'] == f'{compat_tool_name}': return False elif str(app_id) in config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping']: config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)]['name'] = f'{compat_tool_name}' config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)]['config'] = '' config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)]['priority'] = '250' print(f"Updated CompatToolMapping entry for appid: {app_id}") return True else: config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)] = {'name': f'{compat_tool_name}', 'config': '', 'priority': '250'} print(f"Created new CompatToolMapping entry for appid: {app_id}") return True def check_if_shortcut_exists(shortcut_id, display_name, exe_path, start_dir, launch_options): # Check if the game already exists in the shortcuts using the id if any(s.get('appid') == shortcut_id for s in shortcuts['shortcuts'].values()): print(f"Existing shortcut found based on shortcut ID for game {display_name}. Skipping.") return True # Check if the game already exists in the shortcuts using the fields (probably unnecessary) if any(s.get('appname') == display_name and s.get('exe') == exe_path and s.get('StartDir') == start_dir and s.get('LaunchOptions') == launch_options for s in shortcuts['shortcuts'].values()): print(f"Existing shortcut found based on matching fields for game {display_name}. Skipping.") return True #End of Code #Finding the Launchers and applying artwork to already made shortcuts from NonSteamLaunchers.sh # List of game launchers to look for game_launchers = { 'Epic Games', 'Gog Galaxy', 'Ubisoft Connect', 'Origin', 'Battle.net', 'EA App', 'Amazon Games', 'itch.io', 'Legacy Games', 'Humble Bundle', 'Glyph', 'IndieGala Client', 'Rockstar Games Launcher', 'Minecraft: Java Edition', 'Playstation Plus', 'DMM Games', 'VK Play', 'Hulu', 'Twitch', 'Amazon Luna', 'Youtube', 'Amazon Prime Video', 'Disney+', 'Netflix', 'GeForce Now', 'Xbox Game Pass' } # replace with your actual game launchers # Mapping between shortcut names and SteamGridDB names name_mapping = { 'Epic Games': 'Epic Games Store (Program)', 'Gog Galaxy': 'GOG Galaxy (Program)', 'Ubisoft Connect': 'Ubisoft Connect (Program)', 'Origin': 'Origin (Program)', 'Battle.net': 'Battle.net (Program)', 'Legacy Games': 'Legacy Games (Program)', 'Humble Bundle': 'Humble Bundle (Website)', 'VK Play': 'VK Play (Website)', 'Disney+': 'Disney+ (Website)' # Add more mappings as needed } # Iterate over the shortcuts for shortcut in shortcuts['shortcuts'].values(): # Check if the shortcut is a game launcher app_name = shortcut.get('appname') if app_name in game_launchers: print(f"Found game launcher: {app_name}") # Use the actual app ID instead of generating one app_id = shortcut.get('appid') unsigned_app_id = get_unsigned_shortcut_id(app_id) print(f"App ID for {app_name}: {app_id}") # Check if the shortcut doesn't have artwork artwork_path = f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{unsigned_app_id}.png" if not os.path.exists(artwork_path): print(f"No artwork found for {app_name}, downloading...") # Get the game ID from SteamGridDB steamgriddb_name = name_mapping.get(app_name, app_name) game_id = get_game_id(steamgriddb_name) if game_id is not None: print(f"Got game ID from SteamGridDB: {game_id}") # Download and apply artwork get_sgdb_art(game_id, unsigned_app_id) new_shortcuts_added = True if add_compat_tool(unsigned_app_id): shortcuts_updated = True # End of finding the Launchers and applying artwork to already made shortcuts from NonSteamLaunchers.sh # Print the existing shortcuts print("Existing Shortcuts:") for shortcut in shortcuts['shortcuts'].values(): print(f"AppID for {shortcut.get('appname')}: {shortcut.get('appid')}") # Epic Games Scanner item_dir = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{epic_games_launcher}/pfx/drive_c/ProgramData/Epic/EpicGamesLauncher/Data/Manifests/" dat_file_path = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{epic_games_launcher}/pfx/drive_c/ProgramData/Epic/UnrealEngineLauncher/LauncherInstalled.dat" if os.path.exists(dat_file_path): with open(dat_file_path, 'r') as file: dat_data = json.load(file) #Epic Game Scanner for item_file in os.listdir(item_dir): if item_file.endswith('.item'): with open(os.path.join(item_dir, item_file), 'r') as file: item_data = json.load(file) # Initialize variables display_name = item_data['DisplayName'] app_name = item_data['AppName'] exe_path = f"\"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{epic_games_launcher}/pfx/drive_c/Program Files (x86)/Epic Games/Launcher/Portal/Binaries/Win32/EpicGamesLauncher.exe\"" start_dir = f"\"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{epic_games_launcher}/pfx/drive_c/Program Files (x86)/Epic Games/Launcher/Portal/Binaries/Win32/\"" launch_options = f"STEAM_COMPAT_DATA_PATH=\"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{epic_games_launcher}\" %command% -'com.epicgames.launcher://apps/{app_name}?action=launch&silent=true'" signed_shortcut_id = get_steam_shortcut_id(exe_path, display_name) unsigned_shortcut_id = get_unsigned_shortcut_id(signed_shortcut_id) # Check if the game already exists in the shortcuts if check_if_shortcut_exists(signed_shortcut_id, display_name, exe_path, start_dir, launch_options): if add_compat_tool(unsigned_shortcut_id): shortcuts_updated = True continue # Check if the game is still installed for game in dat_data['InstallationList']: print(f"Checking game: {game['AppName']}") if game['AppName'] == item_data['AppName']: print(f"Match found: {game['AppName']}") game_id = get_game_id(display_name) print(f"No existing shortcut found for game {display_name}. Creating new shortcut.") created_shortcuts.append(display_name) shortcuts['shortcuts'][str(signed_shortcut_id)] = { 'appid': str(signed_shortcut_id), 'appname': display_name, 'exe': exe_path, 'StartDir': start_dir, 'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', unsigned_shortcut_id)}", 'LaunchOptions': launch_options, 'GameID': game_id if game_id is not None else "default_game_id" } new_shortcuts_added = True if game_id is not None: get_sgdb_art(game_id, unsigned_shortcut_id) add_compat_tool(unsigned_shortcut_id) else: print("Epic Games Launcher data not found. Skipping Epic Games Scanner.") #End of the Epic Games Scanner # Ubisoft Connect Scanner def getUplayGameInfo(folderPath, filePath): # Get the game IDs from the folder listOfFiles = os.listdir(folderPath) uplay_ids = [re.findall(r'\d+', str(entry))[0] for entry in listOfFiles if re.findall(r'\d+', str(entry))] # Parse the registry file game_dict = {} with open(filePath, 'r') as file: uplay_id = None game_name = None uplay_install_found = False for line in file: if "Uplay Install" in line: uplay_id = re.findall(r'Uplay Install (\d+)', line) if uplay_id: uplay_id = uplay_id[0] game_name = None # Reset game_name uplay_install_found = True if "DisplayName" in line and uplay_install_found: game_name = re.findall(r'\"(.+?)\"', line.split("=")[1]) if game_name: game_name = game_name[0] uplay_install_found = False if uplay_id and game_name and uplay_id in uplay_ids: # Add the game's info to the dictionary if its ID was found in the folder game_dict[game_name] = uplay_id uplay_id = None # Reset uplay_id game_name = None # Reset game_name return game_dict # Define your paths data_folder_path = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{ubisoft_connect_launcher}/pfx/drive_c/Program Files (x86)/Ubisoft/Ubisoft Game Launcher/data/" registry_file_path = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{ubisoft_connect_launcher}/pfx/system.reg" # Check if the paths exist if not os.path.exists(data_folder_path) or not os.path.exists(registry_file_path): print("One or more paths do not exist.") print("Ubisoft Connect game data not found. Skipping Ubisoft Games Scanner.") else: game_dict = getUplayGameInfo(data_folder_path, registry_file_path) for game, uplay_id in game_dict.items(): if uplay_id: launch_options = f"STEAM_COMPAT_DATA_PATH=\"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{ubisoft_connect_launcher}/\" %command% \"uplay://launch/{uplay_id}/0\"" exe_path = f"\"{logged_in_home}/.local/share/Steam/Steam/steamapps/compatdata/{ubisoft_connect_launcher}/pfx/drive_c/Program Files (x86)/Ubisoft/Ubisoft Game Launcher/upc.exe\"" start_dir = f"\"{logged_in_home}/.local/share/Steam/Steam/steamapps/compatdata/{ubisoft_connect_launcher}/pfx/drive_c/Program Files (x86)/Ubisoft/Ubisoft Game Launcher/\"" signed_shortcut_id = get_steam_shortcut_id(exe_path, game) unsigned_shortcut_id = get_unsigned_shortcut_id(signed_shortcut_id) # Check if the game already exists in the shortcuts if check_if_shortcut_exists(signed_shortcut_id, game, exe_path, start_dir, launch_options): if add_compat_tool(unsigned_shortcut_id): shortcuts_updated = True continue game_id = get_game_id(game) if game_id is not None: get_sgdb_art(game_id, unsigned_shortcut_id) new_shortcuts_added = True created_shortcuts.append(game) shortcuts['shortcuts'][str(len(shortcuts['shortcuts']))] = { 'appid': str(signed_shortcut_id), 'appname': game, 'exe': exe_path, 'StartDir': start_dir, 'LaunchOptions': launch_options, 'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', unsigned_shortcut_id)}" } add_compat_tool(unsigned_shortcut_id) # End of Ubisoft Game Scanner # EA App Game Scanner def get_ea_app_game_info(installed_games, game_directory_path): game_dict = {} for game in installed_games: xml_file = ET.parse(f"{game_directory_path}{game}/__Installer/installerdata.xml") xml_root = xml_file.getroot() ea_ids = None game_name = None for content_id in xml_root.iter('contentID'): if ea_ids is None: ea_ids = content_id.text else: ea_ids = ea_ids + ',' + content_id.text for game_title in xml_root.iter('gameTitle'): if game_name is None: game_name = game_title.text continue for game_title in xml_root.iter('title'): if game_name is None: game_name = game_title.text continue if game_name is None: game_name = game if ea_ids: # Add the game's info to the dictionary if its ID was found in the folder game_dict[game_name] = ea_ids return game_dict game_directory_path = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{ea_app_launcher}/pfx/drive_c/Program Files/EA Games/" if not os.path.isdir(game_directory_path): print("EA App game data not found. Skipping EA App Scanner.") else: installed_games = os.listdir(game_directory_path) # Get a list of game folders game_dict = get_ea_app_game_info(installed_games, game_directory_path) for game, ea_ids in game_dict.items(): launch_options = f"STEAM_COMPAT_DATA_PATH=\"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{ea_app_launcher}/\" %command% \"origin2://game/launch?offerIds={ea_ids}\"" exe_path = f"\"{logged_in_home}/.local/share/Steam/Steam/steamapps/compatdata/{ea_app_launcher}/pfx/drive_c/Program Files/Electronic Arts/EA Desktop/EA Desktop/EALaunchHelper.exe\"" start_dir = f"\"{logged_in_home}/.local/share/Steam/Steam/steamapps/compatdata/{ea_app_launcher}/pfx/drive_c/Program Files/Electronic Arts/EA Desktop/EA Desktop/\"" signed_shortcut_id = get_steam_shortcut_id(exe_path, game) unsigned_shortcut_id = get_unsigned_shortcut_id(signed_shortcut_id) # Check if the game already exists in the shortcuts if check_if_shortcut_exists(signed_shortcut_id, game, exe_path, start_dir, launch_options): if add_compat_tool(unsigned_shortcut_id): shortcuts_updated = True continue game_id = get_game_id(game) if game_id is not None: get_sgdb_art(game_id, unsigned_shortcut_id) # Check if the game already exists in the shortcuts new_shortcuts_added = True created_shortcuts.append(game) shortcuts['shortcuts'][str(len(shortcuts['shortcuts']))] = { 'appid': str(signed_shortcut_id), 'appname': game, 'exe': exe_path, 'StartDir': start_dir, 'LaunchOptions': launch_options, 'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', unsigned_shortcut_id)}" } add_compat_tool(unsigned_shortcut_id) #End of EA App Scanner #Push down when more scanners are added # Only write back to the shortcuts.vdf and config.vdf files if new shortcuts were added or compattools changed if new_shortcuts_added or shortcuts_updated: print(f"Saving new config and shortcuts files") conf = vdf.dumps(config_data, pretty=True) with open(f"{logged_in_home}/.steam/root/config/config.vdf", 'w') as file: file.write(conf) with open(f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/shortcuts.vdf", 'wb') as file: file.write(vdf.binary_dumps(shortcuts)) # Print the created shortcuts if created_shortcuts: print("Created Shortcuts:") for name in created_shortcuts: print(name) print("All finished!")