Mirrors/NonSteamLaunchers-On-Steam-Deck
2
0
Fork 0
mirror of https://github.com/moraroy/NonSteamLaunchers-On-Steam-Deck.git synced 2024-11-20 16:19:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #179 from sysmoon14/main

Browse Source 
Improved ShortcutID Generation and Allocation of Proton Compatibility
This commit is contained in:
Roy 2024-01-25 14:52:48 -08:00 committed by GitHub
commit 4fee1f8bdd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 86 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

139
NSLGameScanner.py
View File

@ -3,6 +3,7 @@ import os, re, vdf
import json import json
import shutil import shutil
import binascii import binascii
import ctypes
import zipfile import zipfile
import time import time
import sys import sys
@ -120,9 +121,16 @@ else:
#Code #Code
def get_steam_shortcut_id(exe_path, display_name): def get_steam_shortcut_id(exe_path, display_name):
crc_input = bytes(display_name + exe_path, 'utf-8') unique_id = "".join([exe_path, display_name])
crc = binascii.crc32(crc_input) | 0x80000000 id_int = binascii.crc32(str.encode(unique_id)) | 0x80000000
return crc 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 # Initialize an empty dictionary to serve as the cache
api_cache = {} api_cache = {}
@ -134,6 +142,7 @@ api_key = '239745e39536fdc12bfa84ce0056b036'
#GLOBAL VARS #GLOBAL VARS
created_shortcuts = [] created_shortcuts = []
new_shortcuts_added = False new_shortcuts_added = False
shortcuts_updated = False
shortcut_id = None # Initialize shortcut_id shortcut_id = None # Initialize shortcut_id
# Load the existing shortcuts # Load the existing shortcuts
@ -238,19 +247,26 @@ def is_match(name1, name2):
else: else:
return False return False
def add_compat_tool(shortcut_id): def add_compat_tool(app_id):
if str(shortcut_id) in config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping']: if 'CompatToolMapping' not in config_data['InstallConfigStore']['Software']['Valve']['Steam']:
config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(shortcut_id)]['name'] = f'{compat_tool_name}' config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'] = {}
print(f"Setting compat tool for {shortcut_id} to {compat_tool_name}") print(f"CompatToolMapping key not found in config.vdf, creating.")
config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(shortcut_id)]['config'] = '' 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}':
config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(shortcut_id)]['priority'] = '250' 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: else:
config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(shortcut_id)] = {'name': f'{compat_tool_name}', 'config': '', 'priority': '250'} config_data['InstallConfigStore']['Software']['Valve']['Steam']['CompatToolMapping'][str(app_id)] = {'name': f'{compat_tool_name}', 'config': '', 'priority': '250'}
print(f"Creating new entry for {shortcut_id} in CompatToolMapping") 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): 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 # Check if the game already exists in the shortcuts using the id
if any(s.get('appid') == str(shortcut_id) for s in shortcuts['shortcuts'].values()): 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.") print(f"Existing shortcut found based on shortcut ID for game {display_name}. Skipping.")
return True return True
# Check if the game already exists in the shortcuts using the fields (probably unnecessary) # Check if the game already exists in the shortcuts using the fields (probably unnecessary)
@ -314,9 +330,13 @@ for shortcut in shortcuts['shortcuts'].values():
print(f"Found game launcher: {app_name}") print(f"Found game launcher: {app_name}")
# Use the actual app ID instead of generating one # Use the actual app ID instead of generating one
app_id = shortcut.get('appid') app_id = shortcut.get('appid')
display_name = shortcut.get('appname')
exe_path = shortcut.get('exe')
signed_shortcut_id = get_steam_shortcut_id(exe_path, display_name)
unsigned_shortcut_id = get_unsigned_shortcut_id(signed_shortcut_id)
print(f"App ID for {app_name}: {app_id}") print(f"App ID for {app_name}: {app_id}")
# Check if the shortcut doesn't have artwork # Check if the shortcut doesn't have artwork
artwork_path = f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{app_id}.png" artwork_path = f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{unsigned_shortcut_id}.png"
if not os.path.exists(artwork_path): if not os.path.exists(artwork_path):
print(f"No artwork found for {app_name}, downloading...") print(f"No artwork found for {app_name}, downloading...")
# Get the game ID from SteamGridDB # Get the game ID from SteamGridDB
@ -325,9 +345,10 @@ for shortcut in shortcuts['shortcuts'].values():
if game_id is not None: if game_id is not None:
print(f"Got game ID from SteamGridDB: {game_id}") print(f"Got game ID from SteamGridDB: {game_id}")
# Download and apply artwork # Download and apply artwork
get_sgdb_art(game_id, app_id) get_sgdb_art(game_id, unsigned_shortcut_id)
add_compat_tool(app_id)
new_shortcuts_added = True new_shortcuts_added = True
if add_compat_tool(unsigned_shortcut_id):
shortcuts_updated = True
# End of finding the Launchers and applying artwork to already made shortcuts from NonSteamLaunchers.sh # End of finding the Launchers and applying artwork to already made shortcuts from NonSteamLaunchers.sh
@ -336,7 +357,7 @@ for shortcut in shortcuts['shortcuts'].values():
# Print the existing shortcuts # Print the existing shortcuts
print("Existing Shortcuts:") print("Existing Shortcuts:")
for shortcut in shortcuts['shortcuts'].values(): for shortcut in shortcuts['shortcuts'].values():
print(shortcut.get('appname')) print(f"AppID for {shortcut.get('appname')}: {shortcut.get('appid')}")
# Epic Games Scanner # 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/" item_dir = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{epic_games_launcher}/pfx/drive_c/ProgramData/Epic/EpicGamesLauncher/Data/Manifests/"
@ -358,9 +379,12 @@ if os.path.exists(dat_file_path):
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\"" 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/\"" 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'" 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'"
shortcut_id = get_steam_shortcut_id(exe_path, display_name) 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 # Check if the game already exists in the shortcuts
if check_if_shortcut_exists(shortcut_id, display_name, exe_path, start_dir, launch_options): 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 continue
@ -372,19 +396,19 @@ if os.path.exists(dat_file_path):
game_id = get_game_id(display_name) game_id = get_game_id(display_name)
print(f"No existing shortcut found for game {display_name}. Creating new shortcut.") print(f"No existing shortcut found for game {display_name}. Creating new shortcut.")
created_shortcuts.append(display_name) created_shortcuts.append(display_name)
shortcuts['shortcuts'][str(shortcut_id)] = { shortcuts['shortcuts'][str(signed_shortcut_id)] = {
'appid': str(shortcut_id), 'appid': str(signed_shortcut_id),
'appname': display_name, 'appname': display_name,
'exe': exe_path, 'exe': exe_path,
'StartDir': start_dir, 'StartDir': start_dir,
'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', shortcut_id)}", 'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', unsigned_shortcut_id)}",
'LaunchOptions': launch_options, 'LaunchOptions': launch_options,
'GameID': game_id if game_id is not None else "default_game_id" 'GameID': game_id if game_id is not None else "default_game_id"
} }
new_shortcuts_added = True new_shortcuts_added = True
if game_id is not None: if game_id is not None:
get_sgdb_art(game_id, shortcut_id) get_sgdb_art(game_id, unsigned_shortcut_id)
add_compat_tool(shortcut_id) add_compat_tool(unsigned_shortcut_id)
else: else:
print("Epic Games Launcher data not found. Skipping Epic Games Scanner.") print("Epic Games Launcher data not found. Skipping Epic Games Scanner.")
@ -397,19 +421,19 @@ else:
def getUplayGameInfo(folderPath, filePath): def getUplayGameInfo(folderPath, filePath):
# Get the game IDs from the folder # Get the game IDs from the folder
listOfFiles = os.listdir(folderPath) listOfFiles = os.listdir(folderPath)
game_ids = [re.findall(r'\d+', str(entry))[0] for entry in listOfFiles if re.findall(r'\d+', str(entry))] uplay_ids = [re.findall(r'\d+', str(entry))[0] for entry in listOfFiles if re.findall(r'\d+', str(entry))]
# Parse the registry file # Parse the registry file
game_dict = {} game_dict = {}
with open(filePath, 'r') as file: with open(filePath, 'r') as file:
game_id = None uplay_id = None
game_name = None game_name = None
uplay_install_found = False uplay_install_found = False
for line in file: for line in file:
if "Uplay Install" in line: if "Uplay Install" in line:
game_id = re.findall(r'Uplay Install (\d+)', line) uplay_id = re.findall(r'Uplay Install (\d+)', line)
if game_id: if uplay_id:
game_id = game_id[0] uplay_id = uplay_id[0]
game_name = None # Reset game_name game_name = None # Reset game_name
uplay_install_found = True uplay_install_found = True
if "DisplayName" in line and uplay_install_found: if "DisplayName" in line and uplay_install_found:
@ -417,9 +441,9 @@ def getUplayGameInfo(folderPath, filePath):
if game_name: if game_name:
game_name = game_name[0] game_name = game_name[0]
uplay_install_found = False uplay_install_found = False
if game_id and game_name and game_id in game_ids: # Add the game's info to the dictionary if its ID was found in the folder 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] = game_id game_dict[game_name] = uplay_id
game_id = None # Reset game_id uplay_id = None # Reset uplay_id
game_name = None # Reset game_name game_name = None # Reset game_name
return game_dict return game_dict
@ -435,30 +459,33 @@ if not os.path.exists(data_folder_path) or not os.path.exists(registry_file_path
else: else:
game_dict = getUplayGameInfo(data_folder_path, registry_file_path) game_dict = getUplayGameInfo(data_folder_path, registry_file_path)
for game, game_id in game_dict.items(): for game, uplay_id in game_dict.items():
if game_id: if uplay_id:
launch_options = f"STEAM_COMPAT_DATA_PATH=\"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{ubisoft_connect_launcher}/\" %command% \"uplay://launch/{game_id}/0\"" 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\"" 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/\"" 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/\""
shortcut_id = get_steam_shortcut_id(exe_path, game) 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 # Check if the game already exists in the shortcuts
if check_if_shortcut_exists(shortcut_id, game, exe_path, start_dir, launch_options): 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 continue
game_id = get_game_id(game) game_id = get_game_id(game)
if game_id is not None: if game_id is not None:
get_sgdb_art(game_id, shortcut_id) get_sgdb_art(game_id, unsigned_shortcut_id)
new_shortcuts_added = True new_shortcuts_added = True
created_shortcuts.append(game) created_shortcuts.append(game)
shortcuts['shortcuts'][str(len(shortcuts['shortcuts']))] = { shortcuts['shortcuts'][str(len(shortcuts['shortcuts']))] = {
'appid': str(shortcut_id), 'appid': str(signed_shortcut_id),
'appname': game, 'appname': game,
'exe': exe_path, 'exe': exe_path,
'StartDir': start_dir, 'StartDir': start_dir,
'LaunchOptions': launch_options, 'LaunchOptions': launch_options,
'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', shortcut_id)}" 'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', unsigned_shortcut_id)}"
} }
add_compat_tool(shortcut_id) add_compat_tool(unsigned_shortcut_id)
# End of Ubisoft Game Scanner # End of Ubisoft Game Scanner
@ -502,40 +529,46 @@ else:
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}\"" 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\"" 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/\"" 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/\""
shortcut_id = get_steam_shortcut_id(exe_path, game) 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 # Check if the game already exists in the shortcuts
if check_if_shortcut_exists(shortcut_id, game, exe_path, start_dir, launch_options): 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 continue
game_id = get_game_id(game) game_id = get_game_id(game)
if game_id is not None: if game_id is not None:
get_sgdb_art(game_id, shortcut_id) get_sgdb_art(game_id, unsigned_shortcut_id)
# Check if the game already exists in the shortcuts # Check if the game already exists in the shortcuts
new_shortcuts_added = True new_shortcuts_added = True
created_shortcuts.append(game) created_shortcuts.append(game)
shortcuts['shortcuts'][str(len(shortcuts['shortcuts']))] = { shortcuts['shortcuts'][str(len(shortcuts['shortcuts']))] = {
'appid': str(shortcut_id), 'appid': str(signed_shortcut_id),
'appname': game, 'appname': game,
'exe': exe_path, 'exe': exe_path,
'StartDir': start_dir, 'StartDir': start_dir,
'LaunchOptions': launch_options, 'LaunchOptions': launch_options,
'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', shortcut_id)}" 'icon': f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/grid/{get_file_name('icons', unsigned_shortcut_id)}"
} }
add_compat_tool(shortcut_id) add_compat_tool(unsigned_shortcut_id)
#End of EA App Scanner #End of EA App Scanner
#Push down when more scanners are added #Push down when more scanners are added
# Only write back to the shortcuts.vdf and config.vdf files if new shortcuts were added # Only write back to the shortcuts.vdf and config.vdf files if new shortcuts were added or compattools changed
if new_shortcuts_added: 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: with open(f"{logged_in_home}/.steam/root/userdata/{steamid3}/config/shortcuts.vdf", 'wb') as file:
file.write(vdf.binary_dumps(shortcuts)) file.write(vdf.binary_dumps(shortcuts))
with open(f"{logged_in_home}/.steam/root/config/config.vdf", 'w') as file:
vdf.dump(config_data, file)
# Print the created shortcuts # Print the created shortcuts
print("Created Shortcuts:") if created_shortcuts:
for name in created_shortcuts: print("Created Shortcuts:")
print(name) for name in created_shortcuts:
print(name)
print("All finished!") print("All finished!")