2024-01-17 12:31:04 +01:00
#!/usr/bin/env python3
2024-02-12 06:27:02 +01:00
import os , re
2024-01-17 12:31:04 +01:00
import json
import shutil
import binascii
2024-01-25 18:41:04 +01:00
import ctypes
2024-04-25 12:33:02 +02:00
import gzip
2024-01-17 12:31:04 +01:00
import zipfile
import time
import sys
import subprocess
2024-02-03 05:21:54 +01:00
import sqlite3
2024-01-17 12:31:04 +01:00
from urllib . request import urlopen
from urllib . request import urlretrieve
2024-02-28 09:54:25 +01:00
from base64 import b64encode
2024-01-24 12:39:03 +01:00
import xml . etree . ElementTree as ET
2024-02-29 14:25:01 +01:00
2024-01-17 12:31:04 +01:00
2024-02-29 14:44:17 +01:00
# Check the value of the DBUS_SESSION_BUS_ADDRESS environment variable
dbus_address = os . environ . get ( ' DBUS_SESSION_BUS_ADDRESS ' )
if not dbus_address or not dbus_address . startswith ( ' unix:path= ' ) :
# Set the value of the DBUS_SESSION_BUS_ADDRESS environment variable
dbus_address = f ' unix:path=/run/user/ { os . getuid ( ) } /bus '
os . environ [ ' DBUS_SESSION_BUS_ADDRESS ' ] = dbus_address
2024-02-29 14:32:16 +01:00
2024-01-29 14:51:56 +01:00
# Path to the env_vars file
env_vars_path = f " { os . environ [ ' HOME ' ] } /.config/systemd/user/env_vars "
# Check if the env_vars file exists
if not os . path . exists ( env_vars_path ) :
print ( f " Error: { env_vars_path } does not exist. " )
sys . exit ( 1 )
# Read variables from the file
with open ( env_vars_path , ' r ' ) as f :
2024-01-17 12:31:04 +01:00
lines = f . readlines ( )
for line in lines :
if line . startswith ( ' export ' ) :
line = line [ 7 : ] # Remove 'export '
2024-02-06 10:20:00 +01:00
name , value = line . strip ( ) . split ( ' = ' , 1 )
2024-01-17 12:31:04 +01:00
os . environ [ name ] = value
2024-02-09 15:23:36 +01:00
# Delete env_vars entries for Chrome shortcuts so that they're only added once
with open ( env_vars_path , ' w ' ) as f :
for line in lines :
if line . find ( ' chromelaunchoptions ' ) == - 1 and line . find ( ' websites_str ' ) == - 1 :
f . write ( line )
2024-01-17 12:31:04 +01:00
# Variables from NonSteamLaunchers.sh
steamid3 = os . environ [ ' steamid3 ' ]
logged_in_home = os . environ [ ' logged_in_home ' ]
compat_tool_name = os . environ [ ' compat_tool_name ' ]
2024-02-06 10:20:00 +01:00
controller_config_path = os . environ [ ' controller_config_path ' ]
python_version = os . environ [ ' python_version ' ]
#Scanner Variables
2024-01-17 12:31:04 +01:00
epic_games_launcher = os . environ . get ( ' epic_games_launcher ' , ' ' )
ubisoft_connect_launcher = os . environ . get ( ' ubisoft_connect_launcher ' , ' ' )
2024-01-23 23:30:55 +01:00
ea_app_launcher = os . environ . get ( ' ea_app_launcher ' , ' ' )
2024-01-26 10:18:26 +01:00
gog_galaxy_launcher = os . environ . get ( ' gog_galaxy_launcher ' , ' ' )
2024-02-02 05:51:08 +01:00
bnet_launcher = os . environ . get ( ' bnet_launcher ' , ' ' )
2024-02-03 05:21:54 +01:00
amazon_launcher = os . environ . get ( ' amazon_launcher ' , ' ' )
2024-04-25 12:33:02 +02:00
itchio_launcher = os . environ . get ( ' itchio_launcher ' , ' ' )
2024-01-17 12:31:04 +01:00
2024-02-06 10:20:00 +01:00
#Variables of the Launchers
# Define the path of the Launchers
epicshortcutdirectory = os . environ . get ( ' epicshortcutdirectory ' )
gogshortcutdirectory = os . environ . get ( ' gogshortcutdirectory ' )
uplayshortcutdirectory = os . environ . get ( ' uplayshortcutdirectory ' )
battlenetshortcutdirectory = os . environ . get ( ' battlenetshortcutdirectory ' )
eaappshortcutdirectory = os . environ . get ( ' eaappshortcutdirectory ' )
amazonshortcutdirectory = os . environ . get ( ' amazonshortcutdirectory ' )
itchioshortcutdirectory = os . environ . get ( ' itchioshortcutdirectory ' )
legacyshortcutdirectory = os . environ . get ( ' legacyshortcutdirectory ' )
humbleshortcutdirectory = os . environ . get ( ' humbleshortcutdirectory ' )
indieshortcutdirectory = os . environ . get ( ' indieshortcutdirectory ' )
rockstarshortcutdirectory = os . environ . get ( ' rockstarshortcutdirectory ' )
glyphshortcutdirectory = os . environ . get ( ' glyphshortcutdirectory ' )
minecraftshortcutdirectory = os . environ . get ( ' minecraftshortcutdirectory ' )
psplusshortcutdirectory = os . environ . get ( ' psplusshortcutdirectory ' )
vkplayhortcutdirectory = os . environ . get ( ' vkplayhortcutdirectory ' )
#Streaming
chromedirectory = os . environ . get ( ' chromedirectory ' )
websites_str = os . environ . get ( ' custom_websites_str ' )
custom_websites = websites_str . split ( ' , ' ) if websites_str else [ ]
2024-01-17 12:31:04 +01:00
# Define the parent folder
2024-01-26 04:18:58 +01:00
parent_folder = f " { logged_in_home } /.config/systemd/user/Modules "
2024-02-12 06:16:50 +01:00
sys . path . insert ( 0 , os . path . expanduser ( f " { logged_in_home } /.config/systemd/user/Modules " ) )
2024-02-11 02:18:13 +01:00
print ( sys . path )
2024-01-17 12:31:04 +01:00
# Now that the requests module has been downloaded, you can import it
2024-01-26 04:18:58 +01:00
sys . path . insert ( 0 , parent_folder )
2024-01-17 12:31:04 +01:00
import requests
2024-02-10 23:28:17 +01:00
import vdf
2024-01-17 12:31:04 +01:00
from steamgrid import SteamGridDB
2024-01-26 04:18:58 +01:00
print ( sys . path )
2024-01-17 12:31:04 +01:00
2024-02-11 02:18:13 +01:00
2024-01-17 12:31:04 +01:00
#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 ) :
2024-01-25 18:41:04 +01:00
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
2024-01-17 12:31:04 +01:00
# Initialize an empty dictionary to serve as the cache
api_cache = { }
2024-01-23 23:30:55 +01:00
#API KEYS FOR NONSTEAMLAUNCHER USE ONLY
2024-02-17 04:26:30 +01:00
sgdb = SteamGridDB ( ' 36e4bedbfdda27f42f9ef4a44f80955c ' )
api_key = ' 36e4bedbfdda27f42f9ef4a44f80955c '
2024-01-23 23:30:55 +01:00
#GLOBAL VARS
created_shortcuts = [ ]
new_shortcuts_added = False
2024-01-25 18:41:04 +01:00
shortcuts_updated = False
2024-01-23 23:30:55 +01:00
shortcut_id = None # Initialize shortcut_id
2024-02-28 09:54:25 +01:00
decky_shortcuts = { }
gridp64 = " "
grid64 = " "
logo64 = " "
hero64 = " "
2024-01-23 23:30:55 +01:00
2024-05-02 10:43:22 +02:00
# Define the path to the shortcuts file
shortcuts_file = f " { logged_in_home } /.steam/root/userdata/ { steamid3 } /config/shortcuts.vdf "
# Check if the shortcuts file exists
if not os . path . exists ( shortcuts_file ) :
# If the file does not exist, create a new file with an empty "shortcuts" section
with open ( shortcuts_file , ' wb ' ) as file :
vdf . binary_dumps ( { ' shortcuts ' : { } } , file )
else :
# If the file exists, try to load it
try :
with open ( shortcuts_file , ' rb ' ) as file :
shortcuts = vdf . binary_loads ( file . read ( ) )
except Exception as e :
print ( f " Error reading shortcuts file: { e } " )
# If an error occurs when reading the file, create a new file with an empty "shortcuts" section
with open ( shortcuts_file , ' wb ' ) as file :
vdf . binary_dumps ( { ' shortcuts ' : { } } , file )
2024-01-23 23:30:55 +01:00
# 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 ) :
2024-02-28 09:54:25 +01:00
global grid64
global gridp64
global logo64
global hero64
print ( f " Downloading icons artwork... " )
download_artwork ( game_id , api_key , " icons " , app_id )
print ( f " Downloading logos artwork... " )
logo64 = download_artwork ( game_id , api_key , " logos " , app_id )
print ( f " Downloading heroes artwork... " )
hero64 = download_artwork ( game_id , api_key , " heroes " , app_id )
print ( " Downloading grids artwork of size 600x900... " )
gridp64 = download_artwork ( game_id , api_key , " grids " , app_id , " 600x900 " )
print ( " Downloading grids artwork of size 920x430... " )
grid64 = download_artwork ( game_id , api_key , " grids " , app_id , " 920x430 " )
2024-01-23 23:30:55 +01:00
2024-02-17 04:26:30 +01:00
2024-01-17 12:31:04 +01:00
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 } "
2024-02-17 04:26:30 +01:00
2024-02-01 13:49:45 +01:00
directory = os . path . dirname ( file_path )
if not os . path . exists ( directory ) :
os . makedirs ( directory )
2024-02-17 04:26:30 +01:00
2024-01-17 12:31:04 +01:00
if os . path . exists ( file_path ) :
print ( f " Artwork for { game_id } already exists. Skipping download. " )
2024-02-28 09:54:25 +01:00
with open ( file_path , ' rb ' ) as image_file :
return b64encode ( image_file . read ( ) ) . decode ( ' utf-8 ' )
2024-01-17 12:31:04 +01:00
# 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
2024-02-17 04:26:30 +01:00
else :
print ( f " Error making API call: { response . status_code } " )
# Store the failed status in the cache
api_cache [ cache_key ] = None
return
2024-01-17 12:31:04 +01:00
# 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 )
2024-02-28 09:54:25 +01:00
return b64encode ( response . content ) . decode ( ' utf-8 ' )
2024-01-17 12:31:04 +01:00
except requests . exceptions . RequestException as e :
2024-01-19 03:53:00 +01:00
print ( f " Error downloading image: { e } " )
2024-01-17 12:31:04 +01:00
if art_type == ' icons ' :
download_artwork ( game_id , api_key , ' icons_ico ' , shortcut_id )
2024-02-17 04:26:30 +01:00
2024-01-17 12:31:04 +01:00
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 :
2024-01-21 22:59:38 +01:00
print ( f " No exact match found. Using game ID of the first result: { games [ 0 ] . name } : { games [ 0 ] . id } " )
2024-01-17 12:31:04 +01:00
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
2024-01-24 03:14:33 +01:00
2024-02-09 01:24:46 +01:00
# Add or update the proton compatibility settings
2024-02-09 10:39:43 +01:00
def add_compat_tool ( app_id , launchoptions ) :
2024-01-25 18:41:04 +01:00
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. " )
2024-02-09 10:48:35 +01:00
if ' chrome ' in launchoptions :
2024-02-09 10:39:43 +01:00
return False
2024-01-25 18:41:04 +01:00
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 } " )
2024-02-28 09:54:25 +01:00
return compat_tool_name
2024-01-23 23:30:55 +01:00
else :
2024-01-25 18:41:04 +01:00
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 } " )
2024-02-28 09:54:25 +01:00
return compat_tool_name
2024-01-17 12:31:04 +01:00
2024-01-24 10:51:05 +01:00
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
2024-01-25 18:41:04 +01:00
if any ( s . get ( ' appid ' ) == shortcut_id for s in shortcuts [ ' shortcuts ' ] . values ( ) ) :
2024-02-09 01:24:46 +01:00
print ( f " Existing shortcut found based on shortcut ID for game { display_name } . Skipping creation. " )
2024-01-24 12:39:03 +01:00
return True
2024-01-24 10:51:05 +01:00
# 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 ( ) ) :
2024-02-09 01:24:46 +01:00
print ( f " Existing shortcut found based on matching fields for game { display_name } . Skipping creation. " )
2024-01-24 12:39:03 +01:00
return True
2024-02-27 17:19:44 +01:00
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 creation. " )
return True
2024-01-24 03:14:33 +01:00
#End of Code
2024-01-17 12:31:04 +01:00
2024-02-06 10:20:00 +01:00
#Start of Refactoring code from the .sh file
sys . path . insert ( 0 , os . path . expanduser ( f " { logged_in_home } /Downloads/NonSteamLaunchersInstallation/lib/python { python_version } /site-packages " ) )
print ( sys . path )
# Create an empty dictionary to store the app IDs
app_ids = { }
2024-02-09 01:24:46 +01:00
2024-02-28 09:54:25 +01:00
#Create Shortcuts
2024-02-06 10:20:00 +01:00
def create_new_entry ( shortcutdirectory , appname , launchoptions , startingdir ) :
2024-02-09 01:24:46 +01:00
global new_shortcuts_added
global shortcuts_updated
global created_shortcuts
2024-02-28 09:54:25 +01:00
global decky_shortcuts
global grid64
global gridp64
global logo64
global hero64
2024-02-06 10:20:00 +01:00
# Check if the launcher is installed
2024-02-09 10:21:00 +01:00
if not shortcutdirectory or not appname or not launchoptions or not startingdir :
2024-02-06 10:20:00 +01:00
print ( f " { appname } is not installed. Skipping. " )
return
2024-02-09 01:24:46 +01:00
exe_path = f " { shortcutdirectory } "
2024-02-06 10:20:00 +01:00
signed_shortcut_id = get_steam_shortcut_id ( exe_path , appname )
2024-02-09 12:00:01 +01:00
unsigned_shortcut_id = get_unsigned_shortcut_id ( signed_shortcut_id )
2024-02-06 10:20:00 +01:00
# Only store the app ID for specific launchers
if appname in [ ' Epic Games ' , ' Gog Galaxy ' , ' Ubisoft Connect ' , ' Battle.net ' , ' EA App ' , ' Amazon Games ' , ' itch.io ' , ' Legacy Games ' , ' Humble Bundle ' , ' IndieGala Client ' , ' Rockstar Games Launcher ' , ' Glyph ' , ' Minecraft: Java Edition ' , ' Playstation Plus ' , ' VK Play ' ] :
2024-02-09 12:00:01 +01:00
app_ids [ appname ] = unsigned_shortcut_id
2024-02-09 01:24:46 +01:00
# Check if the game already exists in the shortcuts
2024-02-06 10:20:00 +01:00
if check_if_shortcut_exists ( signed_shortcut_id , appname , exe_path , startingdir , launchoptions ) :
2024-02-09 01:24:46 +01:00
# Check if proton needs applying or updating
2024-02-09 10:39:43 +01:00
if add_compat_tool ( unsigned_shortcut_id , launchoptions ) :
2024-02-06 10:20:00 +01:00
shortcuts_updated = True
return
2024-02-09 01:24:46 +01:00
#Get artwork
game_id = get_game_id ( appname )
if game_id is not None :
get_sgdb_art ( game_id , unsigned_shortcut_id )
2024-02-06 10:20:00 +01:00
# Create a new entry for the Steam shortcut
2024-02-28 09:54:25 +01:00
compatTool = add_compat_tool ( unsigned_shortcut_id , launchoptions )
2024-02-06 10:20:00 +01:00
new_entry = {
' appid ' : str ( signed_shortcut_id ) ,
' appname ' : appname ,
' exe ' : exe_path ,
' StartDir ' : startingdir ,
' icon ' : f " { logged_in_home } /.steam/root/userdata/ { steamid3 } /config/grid/ { get_file_name ( ' icons ' , unsigned_shortcut_id ) } " ,
' LaunchOptions ' : launchoptions ,
}
2024-02-28 09:54:25 +01:00
decky_entry = {
' appname ' : appname ,
' exe ' : exe_path ,
' StartDir ' : startingdir ,
' icon ' : f " { logged_in_home } /.steam/root/userdata/ { steamid3 } /config/grid/ { get_file_name ( ' icons ' , unsigned_shortcut_id ) } " ,
' LaunchOptions ' : launchoptions ,
' CompatTool ' : compatTool ,
2024-02-28 12:27:24 +01:00
' WideGrid ' : grid64 ,
' Grid ' : gridp64 ,
2024-02-28 09:54:25 +01:00
' Hero ' : hero64 ,
' Logo ' : logo64 ,
}
2024-02-09 01:24:46 +01:00
# Add the new entry to the shortcuts dictionary and add proton
2024-02-06 10:20:00 +01:00
shortcuts [ ' shortcuts ' ] [ str ( signed_shortcut_id ) ] = new_entry
2024-02-28 09:54:25 +01:00
decky_shortcuts [ appname ] = decky_entry
2024-02-06 10:20:00 +01:00
print ( f " Added new entry for { appname } to shortcuts. " )
new_shortcuts_added = True
2024-02-09 01:24:46 +01:00
created_shortcuts . append ( appname )
2024-02-06 10:20:00 +01:00
create_new_entry ( os . environ . get ( ' epicshortcutdirectory ' ) , ' Epic Games ' , os . environ . get ( ' epiclaunchoptions ' ) , os . environ . get ( ' epicstartingdir ' ) )
create_new_entry ( os . environ . get ( ' gogshortcutdirectory ' ) , ' Gog Galaxy ' , os . environ . get ( ' goglaunchoptions ' ) , os . environ . get ( ' gogstartingdir ' ) )
create_new_entry ( os . environ . get ( ' uplayshortcutdirectory ' ) , ' Ubisoft Connect ' , os . environ . get ( ' uplaylaunchoptions ' ) , os . environ . get ( ' uplaystartingdir ' ) )
create_new_entry ( os . environ . get ( ' battlenetshortcutdirectory ' ) , ' Battle.net ' , os . environ . get ( ' battlenetlaunchoptions ' ) , os . environ . get ( ' battlenetstartingdir ' ) )
create_new_entry ( os . environ . get ( ' eaappshortcutdirectory ' ) , ' EA App ' , os . environ . get ( ' eaapplaunchoptions ' ) , os . environ . get ( ' eaappstartingdir ' ) )
create_new_entry ( os . environ . get ( ' amazonshortcutdirectory ' ) , ' Amazon Games ' , os . environ . get ( ' amazonlaunchoptions ' ) , os . environ . get ( ' amazonstartingdir ' ) )
create_new_entry ( os . environ . get ( ' itchioshortcutdirectory ' ) , ' itch.io ' , os . environ . get ( ' itchiolaunchoptions ' ) , os . environ . get ( ' itchiostartingdir ' ) )
create_new_entry ( os . environ . get ( ' legacyshortcutdirectory ' ) , ' Legacy Games ' , os . environ . get ( ' legacylaunchoptions ' ) , os . environ . get ( ' legacystartingdir ' ) )
create_new_entry ( os . environ . get ( ' humbleshortcutdirectory ' ) , ' Humble Bundle ' , os . environ . get ( ' humblelaunchoptions ' ) , os . environ . get ( ' humblestartingdir ' ) )
create_new_entry ( os . environ . get ( ' indieshortcutdirectory ' ) , ' IndieGala Client ' , os . environ . get ( ' indielaunchoptions ' ) , os . environ . get ( ' indiestartingdir ' ) )
create_new_entry ( os . environ . get ( ' rockstarshortcutdirectory ' ) , ' Rockstar Games Launcher ' , os . environ . get ( ' rockstarlaunchoptions ' ) , os . environ . get ( ' rockstarstartingdir ' ) )
create_new_entry ( os . environ . get ( ' glyphshortcutdirectory ' ) , ' Glyph ' , os . environ . get ( ' glyphlaunchoptions ' ) , os . environ . get ( ' glyphstartingdir ' ) )
create_new_entry ( os . environ . get ( ' minecraftshortcutdirectory ' ) , ' Minecraft: Java Edition ' , os . environ . get ( ' minecraftlaunchoptions ' ) , os . environ . get ( ' minecraftstartingdir ' ) )
create_new_entry ( os . environ . get ( ' psplusshortcutdirectory ' ) , ' Playstation Plus ' , os . environ . get ( ' pspluslaunchoptions ' ) , os . environ . get ( ' psplusstartingdir ' ) )
create_new_entry ( os . environ . get ( ' vkplayhortcutdirectory ' ) , ' VK Play ' , os . environ . get ( ' vkplaylaunchoptions ' ) , os . environ . get ( ' vkplaystartingdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Xbox Game Pass ' , os . environ . get ( ' xboxchromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' GeForce Now ' , os . environ . get ( ' geforcechromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
2024-03-06 18:02:19 +01:00
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Netflix ' , os . environ . get ( ' netflixchromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
2024-02-06 10:20:00 +01:00
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Hulu ' , os . environ . get ( ' huluchromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Disney+ ' , os . environ . get ( ' disneychromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Amazon Prime Video ' , os . environ . get ( ' amazonchromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Youtube ' , os . environ . get ( ' youtubechromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Amazon Luna ' , os . environ . get ( ' lunachromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Twitch ' , os . environ . get ( ' twitchchromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' movie-web ' , os . environ . get ( ' moviewebchromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
2024-03-06 19:40:58 +01:00
create_new_entry ( os . environ . get ( ' chromedirectory ' ) , ' Fortnite ' , os . environ . get ( ' fortnitechromelaunchoptions ' ) , os . environ . get ( ' chrome_startdir ' ) )
2024-02-06 10:20:00 +01:00
# 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 regular expression 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 ( os . environ [ ' chromedirectory ' ] , game_name , chromelaunch_options , os . environ [ ' chrome_startdir ' ] )
#End of Creating Launcher Shortcuts
# Iterate over each launcher in the app_ids dictionary
for launcher_name , appid in app_ids . items ( ) :
print ( f " The app ID for { launcher_name } is { appid } " )
2024-02-09 13:02:55 +01:00
# Get the app ID for the first launcher that the user chose to install
2024-02-09 15:57:42 +01:00
if app_ids :
appid = app_ids . get ( launcher_name )
2024-02-09 12:27:03 +01:00
2024-02-06 10:20:00 +01:00
#Create User Friendly Symlinks for the launchers
# Define the path to the compatdata directory
compatdata_dir = f ' { logged_in_home } /.local/share/Steam/steamapps/compatdata '
# Define a dictionary of original folder names
folder_names = {
' Epic Games ' : ' EpicGamesLauncher ' ,
' Gog Galaxy ' : ' GogGalaxyLauncher ' ,
' Ubisoft Connect ' : ' UplayLauncher ' ,
' 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 ' ,
' Minecraft: Java Edition ' : ' MinecraftLauncher ' ,
' Playstation Plus ' : ' PlaystationPlusLauncher ' ,
' VK Play ' : ' VKPlayLauncher ' ,
2024-01-17 12:31:04 +01:00
}
2024-02-06 10:20:00 +01:00
# 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 = app_ids . get ( launcher_name )
2024-01-29 05:52:48 +01:00
2024-02-06 10:20:00 +01:00
# Define the new path of the folder
new_path = os . path . join ( compatdata_dir , str ( appid ) )
2024-01-24 03:14:33 +01:00
2024-02-06 10:20:00 +01:00
# 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
2024-02-09 15:57:42 +01:00
if app_ids and os . path . exists ( os . path . join ( compatdata_dir , ' NonSteamLaunchers ' ) ) :
2024-02-06 10:20:00 +01:00
# Get the first app ID from the app_ids list
first_app_id = next ( iter ( app_ids . values ( ) ) )
# 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 )
#End of Refactoring python code from .sh file
2024-01-24 03:14:33 +01:00
2024-01-23 23:30:55 +01:00
# Print the existing shortcuts
print ( " Existing Shortcuts: " )
for shortcut in shortcuts [ ' shortcuts ' ] . values ( ) :
2024-02-09 01:24:46 +01:00
if shortcut . get ( ' appname ' ) is None :
print ( f " AppID for { shortcut . get ( ' AppName ' ) } : { shortcut . get ( ' appid ' ) } " )
else :
print ( f " AppID for { shortcut . get ( ' appname ' ) } : { shortcut . get ( ' appid ' ) } " )
2024-01-17 12:31:04 +01:00
2024-01-26 07:24:04 +01:00
2024-02-06 10:20:00 +01:00
#Scanners
2024-01-17 12:31:04 +01:00
# 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 "
2024-01-26 07:24:04 +01:00
if os . path . exists ( dat_file_path ) and os . path . exists ( item_dir ) :
2024-01-17 12:31:04 +01:00
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 )
2024-01-21 07:43:23 +01:00
# 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 ' "
2024-01-17 12:31:04 +01:00
# Check if the game is still installed
for game in dat_data [ ' InstallationList ' ] :
if game [ ' AppName ' ] == item_data [ ' AppName ' ] :
2024-02-09 01:24:46 +01:00
create_new_entry ( exe_path , display_name , launch_options , start_dir )
2024-01-17 12:31:04 +01:00
else :
print ( " Epic Games Launcher data not found. Skipping Epic Games Scanner. " )
#End of the Epic Games Scanner
2024-01-21 07:43:23 +01:00
# Ubisoft Connect Scanner
2024-01-21 02:18:03 +01:00
def getUplayGameInfo ( folderPath , filePath ) :
# Get the game IDs from the folder
listOfFiles = os . listdir ( folderPath )
2024-01-25 18:41:04 +01:00
uplay_ids = [ re . findall ( r ' \ d+ ' , str ( entry ) ) [ 0 ] for entry in listOfFiles if re . findall ( r ' \ d+ ' , str ( entry ) ) ]
2024-01-17 12:31:04 +01:00
2024-01-21 02:18:03 +01:00
# Parse the registry file
2024-01-17 12:31:04 +01:00
game_dict = { }
2024-01-21 02:18:03 +01:00
with open ( filePath , ' r ' ) as file :
2024-01-25 18:41:04 +01:00
uplay_id = None
2024-01-21 02:18:03 +01:00
game_name = None
uplay_install_found = False
for line in file :
if " Uplay Install " in line :
2024-01-25 18:41:04 +01:00
uplay_id = re . findall ( r ' Uplay Install ( \ d+) ' , line )
if uplay_id :
uplay_id = uplay_id [ 0 ]
2024-01-21 02:18:03 +01:00
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
2024-01-25 18:41:04 +01:00
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
2024-01-21 02:18:03 +01:00
game_name = None # Reset game_name
2024-01-17 12:31:04 +01:00
return game_dict
# Define your paths
2024-01-19 13:21:02 +01:00
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/ "
2024-01-21 02:18:03 +01:00
registry_file_path = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { ubisoft_connect_launcher } /pfx/system.reg "
2024-01-17 12:31:04 +01:00
# Check if the paths exist
2024-01-21 02:37:57 +01:00
if not os . path . exists ( data_folder_path ) or not os . path . exists ( registry_file_path ) :
2024-01-17 12:31:04 +01:00
print ( " One or more paths do not exist. " )
2024-01-24 03:14:33 +01:00
print ( " Ubisoft Connect game data not found. Skipping Ubisoft Games Scanner. " )
2024-01-17 12:31:04 +01:00
else :
2024-01-21 02:18:03 +01:00
game_dict = getUplayGameInfo ( data_folder_path , registry_file_path )
2024-01-17 12:31:04 +01:00
2024-01-25 18:41:04 +01:00
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 \" "
2024-01-26 22:36:59 +01:00
exe_path = f " \" { logged_in_home } /.local/share/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/steamapps/compatdata/ { ubisoft_connect_launcher } /pfx/drive_c/Program Files (x86)/Ubisoft/Ubisoft Game Launcher/ \" "
2024-02-09 01:24:46 +01:00
create_new_entry ( exe_path , game , launch_options , start_dir )
2024-01-17 12:31:04 +01:00
2024-01-21 07:43:23 +01:00
# End of Ubisoft Game Scanner
2024-01-17 12:31:04 +01:00
2024-01-23 23:30:55 +01:00
# EA App Game Scanner
2024-01-24 12:39:03 +01:00
def get_ea_app_game_info ( installed_games , game_directory_path ) :
2024-01-23 23:30:55 +01:00
game_dict = { }
2024-01-24 12:39:03 +01:00
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
2024-01-23 23:30:55 +01:00
game_name = None
2024-01-24 12:39:03 +01:00
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
2024-01-23 23:30:55 +01:00
return game_dict
2024-01-24 03:14:33 +01:00
game_directory_path = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { ea_app_launcher } /pfx/drive_c/Program Files/EA Games/ "
2024-01-23 23:30:55 +01:00
2024-01-24 12:39:03 +01:00
if not os . path . isdir ( game_directory_path ) :
2024-01-24 05:01:36 +01:00
print ( " EA App game data not found. Skipping EA App Scanner. " )
2024-01-23 23:30:55 +01:00
else :
2024-01-24 12:39:03 +01:00
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 } \" "
2024-01-26 22:07:44 +01:00
exe_path = f " \" { logged_in_home } /.local/share/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/steamapps/compatdata/ { ea_app_launcher } /pfx/drive_c/Program Files/Electronic Arts/EA Desktop/EA Desktop/ \" "
2024-02-09 01:24:46 +01:00
create_new_entry ( exe_path , game , launch_options , start_dir )
2024-01-24 12:39:03 +01:00
#End of EA App Scanner
2024-01-23 23:30:55 +01:00
2024-01-24 03:14:33 +01:00
2024-01-26 10:18:26 +01:00
2024-05-05 19:12:28 +02:00
#Gog Galaxy Scanner
2024-01-26 10:18:26 +01:00
def getGogGameInfo ( filePath ) :
# Check if the file contains any GOG entries
with open ( filePath , ' r ' ) as file :
if " GOG.com " not in file . read ( ) :
print ( " No GOG entries found in the registry file. Skipping GOG Galaxy Games Scanner. " )
return { }
# If GOG entries exist, parse the registry file
game_dict = { }
with open ( filePath , ' r ' ) as file :
game_id = None
game_name = None
2024-02-16 07:54:58 +01:00
exe_path = None
2024-05-05 19:12:28 +02:00
depends_on = None
launch_command = None
2024-01-26 10:18:26 +01:00
for line in file :
2024-05-08 08:20:09 +02:00
if " [Software \\ Wow6432Node \\ GOG.com \\ Games \\ " in line :
split_line = line . split ( " = " )
if len ( split_line ) > 1 :
if " gameID " in line :
game_id = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if game_id :
game_id = game_id [ 0 ]
if " gameName " in line :
game_name = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if game_name :
game_name = bytes ( game_name [ 0 ] , ' utf-8 ' ) . decode ( ' unicode_escape ' )
game_name = game_name . replace ( ' !22 ' , ' ™ ' )
if " exe " in line and " GOG Galaxy " in line and not " unins000.exe " in line :
exe_path = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if exe_path :
exe_path = exe_path [ 0 ] . replace ( ' \\ \\ ' , ' \\ ' )
if " dependsOn " in line :
depends_on = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if depends_on :
depends_on = depends_on [ 0 ]
if " launchCommand " in line :
launch_command = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if launch_command :
launch_command = launch_command [ 0 ]
if game_id and game_name and launch_command :
game_dict [ game_name ] = { ' id ' : game_id , ' exe ' : exe_path }
game_id = None
game_name = None
exe_path = None
depends_on = None
launch_command = None
2024-01-26 10:18:26 +01:00
return game_dict
2024-05-05 19:12:28 +02:00
2024-05-05 19:17:19 +02:00
2024-01-26 10:18:26 +01:00
# Define your paths
gog_games_directory = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { gog_galaxy_launcher } /pfx/drive_c/Program Files (x86)/GOG Galaxy/Games "
registry_file_path = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { gog_galaxy_launcher } /pfx/system.reg "
# Check if the paths exist
if not os . path . exists ( gog_games_directory ) or not os . path . exists ( registry_file_path ) :
print ( " One or more paths do not exist. " )
print ( " GOG Galaxy game data not found. Skipping GOG Galaxy Games Scanner. " )
else :
game_dict = getGogGameInfo ( registry_file_path )
for game , game_info in game_dict . items ( ) :
2024-05-08 07:15:15 +02:00
if game_info [ ' id ' ] :
2024-02-16 07:54:58 +01:00
launch_options = f " STEAM_COMPAT_DATA_PATH= \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { gog_galaxy_launcher } / \" %command% /command=runGame /gameId= { game_info [ ' id ' ] } /path= \" { game_info [ ' exe ' ] } \" "
2024-05-08 07:15:15 +02:00
exe_path = f " \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { gog_galaxy_launcher } /pfx/drive_c/Program Files (x86)/GOG Galaxy/GalaxyClient.exe \" "
start_dir = f " \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { gog_galaxy_launcher } /pfx/drive_c/Program Files (x86)/GOG Galaxy/ \" "
2024-02-09 01:24:46 +01:00
create_new_entry ( exe_path , game , launch_options , start_dir )
2024-01-26 10:18:26 +01:00
2024-05-08 07:15:15 +02:00
# End of Gog Galaxy Scanner
2024-05-08 07:05:13 +02:00
2024-01-26 10:18:26 +01:00
# End of Gog Galaxy Scanner
2024-02-02 05:51:08 +01:00
#Battle.net Scanner
# Define your mapping
flavor_mapping = {
" Blizzard Arcade Collection " : " RTRO " ,
" Diablo " : " D1 " ,
" Diablo II: Resurrected " : " OSI " ,
" Diablo III " : " D3 " ,
" Diablo IV " : " Fen " ,
" Diablo Immortal (PC) " : " ANBS " ,
" Hearthstone " : " WTCG " ,
" Heroes of the Storm " : " Hero " ,
" Overwatch " : " Pro " ,
" Overwatch 2 " : " Pro " ,
" StarCraft " : " S1 " ,
" StarCraft 2 " : " S2 " ,
" Warcraft: Orcs & Humans " : " W1 " ,
" Warcraft II: Battle.net Edition " : " W2 " ,
" Warcraft III: Reforged " : " W3 " ,
" World of Warcraft " : " WoW " ,
" World of Warcraft Classic " : " WoWC " ,
" Warcraft Arclight Rumble " : " GRY " ,
" Call of Duty: Black Ops - Cold War " : " ZEUS " ,
" Call of Duty: Black Ops 4 " : " VIPR " ,
" Call of Duty: Modern Warfare " : " ODIN " ,
" Call of Duty " : " AUKS " ,
" Call of Duty: MW 2 Campaign Remastered " : " LAZR " ,
" Call of Duty: Vanguard " : " FORE " ,
" Call of Duty: Modern Warfare III " : " SPOT " ,
" Crash Bandicoot 4: It ' s About Time " : " WLBY " ,
# Add more games here...
}
def get_flavor_from_file ( game_path ) :
game_path = game_path . replace ( ' \\ ' , ' / ' )
flavor_file = os . path . join ( game_path , ' _retail_ ' , ' .flavor.info ' )
if os . path . exists ( flavor_file ) :
with open ( flavor_file , ' r ' ) as file :
for line in file :
if ' STRING ' in line :
return line . split ( ' : ' ) [ - 1 ] . strip ( ) . capitalize ( )
else :
print ( f " Flavor file not found: { flavor_file } " )
# Use the mapping as a fallback
game_name = os . path . basename ( game_path )
print ( f " Game name from file path: { game_name } " )
return flavor_mapping . get ( game_name , ' unknown ' )
def getBnetGameInfo ( filePath ) :
# Check if the file contains any Battle.net entries
with open ( filePath , ' r ' ) as file :
if " Battle.net " not in file . read ( ) :
print ( " No Battle.net entries found in the registry file. Skipping Battle.net Games Scanner. " )
return None
# If Battle.net entries exist, parse the registry file
game_dict = { }
with open ( filePath , ' r ' ) as file :
game_name = None
exe_path = None
publisher = None
contact = None
for line in file :
split_line = line . split ( " = " )
if len ( split_line ) > 1 :
if " Publisher " in line :
publisher = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if publisher :
publisher = publisher [ 0 ]
# Skip if the publisher is not Blizzard Entertainment
if publisher != " Blizzard Entertainment " :
game_name = None
exe_path = None
publisher = None
continue
if " Contact " in line :
contact = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if contact :
contact = contact [ 0 ]
if " DisplayName " in line :
game_name = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if game_name :
game_name = game_name [ 0 ]
if " InstallLocation " in line :
exe_path = re . findall ( r ' \ " (.+?) \ " ' , split_line [ 1 ] )
if exe_path :
exe_path = exe_path [ 0 ] . replace ( ' \\ \\ ' , ' \\ ' )
# Skip if the install location is for the Battle.net launcher
if " Battle.net " in exe_path :
game_name = None
exe_path = None
publisher = None
continue
if game_name and exe_path and publisher == " Blizzard Entertainment " and contact == " Blizzard Support " :
game_dict [ game_name ] = { ' exe ' : exe_path }
print ( f " Game added to dictionary: { game_name } " )
game_name = None
exe_path = None
publisher = None
contact = None
# If no games were found, return None
if not game_dict :
print ( " No Battle.net games found. Skipping Battle.net Games Scanner. " )
return None
return game_dict
# Define your paths
registry_file_path = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { bnet_launcher } /pfx/system.reg "
game_dict = { }
# Check if the paths exist
if not os . path . exists ( registry_file_path ) :
print ( " One or more paths do not exist. " )
print ( " Battle.net game data not found. Skipping Battle.net Games Scanner. " )
else :
game_dict = getBnetGameInfo ( registry_file_path )
2024-02-23 21:32:44 +01:00
if game_dict is None :
2024-02-02 05:51:08 +01:00
# Skip the rest of the Battle.net scanner
pass
else :
# Extract the flavor for each game and create the launch options
for game , game_info in game_dict . items ( ) :
game_info [ ' flavor ' ] = get_flavor_from_file ( game_info [ ' exe ' ] )
print ( f " Flavor inferred: { game_info [ ' flavor ' ] } " )
# Check if the game name is "Overwatch" and update it to "Overwatch 2"
if game == " Overwatch " :
game = " Overwatch 2 "
2024-02-23 21:32:44 +01:00
if game_info [ ' flavor ' ] == " unknown " :
pass
elif game_info [ ' flavor ' ] :
2024-02-02 05:51:08 +01:00
launch_options = f " STEAM_COMPAT_DATA_PATH= \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { bnet_launcher } / \" %command% \" battlenet:// { game_info [ ' flavor ' ] } \" "
exe_path = f " \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { bnet_launcher } /pfx/drive_c/Program Files (x86)/Battle.net/Battle.net.exe \" --exec= \" launch { game_info [ ' flavor ' ] } \" "
start_dir = f " \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { bnet_launcher } /pfx/drive_c/Program Files (x86)/Battle.net/ \" "
2024-02-09 01:24:46 +01:00
create_new_entry ( exe_path , game , launch_options , start_dir )
2024-02-02 05:51:08 +01:00
# End of Battle.net Scanner
2024-02-03 05:21:54 +01:00
# Amazon Games Scanner
def get_sqlite_path ( ) :
# Specify the full path to the SQLite file
path = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { amazon_launcher } /pfx/drive_c/users/steamuser/AppData/Local/Amazon Games/Data/Games/Sql/GameInstallInfo.sqlite "
if os . path . exists ( path ) :
return path
else :
print ( f " Amazon GameInstallInfo.sqlite not found at { path } " )
return None
def get_launcher_path ( ) :
# Specify the full path to the Amazon Games launcher executable
path = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { amazon_launcher } /pfx/drive_c/users/steamuser/AppData/Local/Amazon Games/App/Amazon Games.exe "
if os . path . exists ( path ) :
return path
else :
print ( f " Could not find Amazon Games.exe at { path } " )
return None
def get_amazon_games ( ) :
sqllite_path = get_sqlite_path ( )
launcher_path = get_launcher_path ( )
if sqllite_path is None or launcher_path is None :
print ( " Skipping Amazon Games Scanner due to missing paths. " )
return [ ]
result = [ ]
connection = sqlite3 . connect ( sqllite_path )
cursor = connection . cursor ( )
cursor . execute ( " SELECT Id, ProductTitle FROM DbSet WHERE Installed = 1 " )
for row in cursor . fetchall ( ) :
id , title = row
result . append ( { " id " : id , " title " : title , " launcher_path " : launcher_path } )
return result
amazon_games = get_amazon_games ( )
if amazon_games :
for game in amazon_games :
# Initialize variables
display_name = game [ ' title ' ]
exe_path = f " \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { amazon_launcher } /pfx/drive_c/users/steamuser/AppData/Local/Amazon Games/App/Amazon Games.exe \" "
start_dir = f " \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { amazon_launcher } /pfx/drive_c/users/steamuser/AppData/Local/Amazon Games/App/ \" "
launch_options = f " STEAM_COMPAT_DATA_PATH= \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { amazon_launcher } \" %command% - ' amazon-games://play/ { game [ ' id ' ] } ' "
2024-02-09 01:24:46 +01:00
create_new_entry ( exe_path , display_name , launch_options , start_dir )
2024-02-03 05:21:54 +01:00
#End of Amazon Games Scanner
2024-02-09 01:24:46 +01:00
2024-02-28 09:54:25 +01:00
2024-04-25 12:33:02 +02:00
#Itchio Scanner
def get_itch_games ( itch_db_location ) :
print ( f " Checking if { itch_db_location } exists... " )
if not os . path . exists ( itch_db_location ) :
print ( f " Path not found: { itch_db_location } . Continuing with the rest of the code... " )
return [ ]
print ( " Opening and reading the database file... " )
with open ( itch_db_location , ' rb ' ) as f :
shortcut_bytes = f . read ( )
print ( " Parsing the database file... " )
paths = parse_butler_db ( shortcut_bytes )
print ( " Converting paths to games... " )
games = [ dbpath_to_game ( path ) for path in paths if dbpath_to_game ( path ) is not None ]
# Remove duplicates
games = list ( set ( games ) )
print ( f " Found { len ( games ) } unique games. " )
return games
def parse_butler_db ( content ) :
print ( " Finding matches in the database content... " )
pattern = rb ' \ { " basePath " : " (.*?) " , " totalSize " .*? " candidates " : \ [(.*?) \ ] \ } '
matches = re . findall ( pattern , content )
print ( f " Found { len ( matches ) } matches. " )
print ( " Converting matches to database paths... " )
db_paths = [ ]
for match in matches :
base_path = match [ 0 ] . decode ( errors = ' ignore ' )
candidates_json = b ' [ ' + match [ 1 ] + b ' ] '
candidates = json . loads ( candidates_json . decode ( errors = ' ignore ' ) )
paths = [ candidate [ ' path ' ] for candidate in candidates ]
db_paths . append ( ( base_path , paths ) )
print ( f " Converted { len ( matches ) } matches to { len ( db_paths ) } database paths. " )
return db_paths
def dbpath_to_game ( paths ) :
# Convert the Windows-style path from the database to a Unix-style path
db_path = paths [ 0 ] . replace ( " \\ \\ " , " / " ) . replace ( " C: " , " " )
linux_path = " /home/deck/.local/share/Steam/steamapps/compatdata/NonSteamLaunchers/pfx/drive_c " + db_path
receipt_path = os . path . join ( linux_path , " .itch " , " receipt.json.gz " )
if not os . path . exists ( receipt_path ) :
return None
for executable in paths [ 1 ] :
exe_path = os . path . join ( linux_path , executable )
if os . access ( exe_path , os . X_OK ) : # check if file is executable
with gzip . open ( receipt_path , ' rb ' ) as f :
receipt_str = f . read ( ) . decode ( )
receipt = json . loads ( receipt_str )
return ( linux_path , executable , receipt [ ' game ' ] [ ' title ' ] )
# Usage:
itch_db_location = f " { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { itchio_launcher } /pfx/drive_c/users/steamuser/AppData/Roaming/itch/db/butler.db-wal "
print ( f " Getting games from { itch_db_location } ... " )
games = get_itch_games ( itch_db_location )
print ( " Printing games... " )
for game in games :
print ( game )
for game in games :
linux_path , executable , game_title = game
exe_path = f " \" { os . path . join ( linux_path , executable ) } \" "
start_dir = f " \" { linux_path } \" "
launchoptions = f " STEAM_COMPAT_DATA_PATH= \" { logged_in_home } /.local/share/Steam/steamapps/compatdata/ { itchio_launcher } / \" %command% "
create_new_entry ( exe_path , game_title , launchoptions , start_dir )
#End of Itchio Scanner
2024-02-28 09:54:25 +01:00
2024-02-12 06:44:26 +01:00
# 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 )
2024-05-02 10:43:22 +02:00
try :
with open ( f " { logged_in_home } /.steam/root/config/config.vdf " , ' w ' ) as file :
file . write ( conf )
except IOError as e :
print ( f " Error writing to config.vdf: { e } " )
try :
with open ( f " { logged_in_home } /.steam/root/userdata/ { steamid3 } /config/shortcuts.vdf " , ' wb ' ) as file :
file . write ( vdf . binary_dumps ( shortcuts ) )
except IOError as e :
print ( f " Error writing to shortcuts.vdf: { e } " )
2024-02-12 06:44:26 +01:00
# Print the created shortcuts
if created_shortcuts :
print ( " Created Shortcuts: " )
for name in created_shortcuts :
print ( name )
2024-02-28 05:12:24 +01:00
# Create the path to the output file
output_file_path = f " { logged_in_home } /.config/systemd/user/NSLGameScanner_output.log "
# Open the output file in write mode
2024-05-02 10:43:22 +02:00
try :
with open ( output_file_path , ' w ' ) as output_file :
for game in decky_shortcuts . values ( ) :
# Skip if 'appname' or 'exe' is None
if game . get ( ' appname ' ) is None or game . get ( ' exe ' ) is None :
continue
# Create a dictionary to hold the shortcut information
shortcut_info = {
' appname ' : game . get ( ' appname ' ) ,
' exe ' : game . get ( ' exe ' ) ,
' StartDir ' : game . get ( ' StartDir ' ) ,
' icon ' : game . get ( ' icon ' ) ,
' LaunchOptions ' : game . get ( ' LaunchOptions ' ) ,
' CompatTool ' : game . get ( ' CompatTool ' ) ,
' WideGrid ' : game . get ( ' WideGrid ' ) ,
' Grid ' : game . get ( ' Grid ' ) ,
' Hero ' : game . get ( ' Hero ' ) ,
' Logo ' : game . get ( ' Logo ' ) ,
}
# Print the shortcut information in JSON format
message = json . dumps ( shortcut_info )
print ( message , flush = True ) # Print to stdout
# Print the shortcut information to the output file
print ( message , file = output_file , flush = True )
except IOError as e :
print ( f " Error writing to output file: { e } " )
2024-02-17 04:40:51 +01:00
2024-02-18 15:23:09 +01:00
print ( " All finished! " )
2024-05-02 10:43:22 +02:00