Added Battle.net to the NSLGameScanner

This commit is contained in:
Roy 2024-02-01 20:51:08 -08:00 committed by GitHub
parent e1c53dbc95
commit ecf585622a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -38,6 +38,7 @@ 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', '')
gog_galaxy_launcher = os.environ.get('gog_galaxy_launcher', '')
bnet_launcher = os.environ.get('bnet_launcher', '')
# Define the parent folder
parent_folder = f"{logged_in_home}/.config/systemd/user/Modules"
@ -631,6 +632,176 @@ else:
# End of Gog Galaxy Scanner
#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)
if game_dict is None:
# 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"
if game_info['flavor']:
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/\""
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 Battle.net 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: