Added Battle.net to the NSLGameScanner

NSLGameScanner.py

@@ -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: