From c4d9add9192ea801a2c42c7a0f31cc795c9530b5 Mon Sep 17 00:00:00 2001
From: Roy <88516395+moraroy@users.noreply.github.com>
Date: Sat, 11 Jan 2025 00:41:13 -0800
Subject: [PATCH] fix for vk

---
 NSLGameScanner.py | 68 ++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 55 insertions(+), 13 deletions(-)

diff --git a/NSLGameScanner.py b/NSLGameScanner.py
index 938c7b2..855be81 100644
--- a/NSLGameScanner.py
+++ b/NSLGameScanner.py
@@ -1440,11 +1440,14 @@ else:
 gamecenter_ini_path = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{vkplay_launcher}/pfx/drive_c/users/steamuser/AppData/Local/GameCenter/GameCenter.ini"
 cache_folder_path = f"{logged_in_home}/.local/share/Steam/steamapps/compatdata/{vkplay_launcher}/pfx/drive_c/users/steamuser/AppData/Local/GameCenter/Cache/GameDescription/"
 
+# Check if the GameCenter.ini file exists
 if not os.path.exists(gamecenter_ini_path):
     print(f"VK Play scanner skipped: {gamecenter_ini_path} does not exist.")
 else:
+    print(f"Found file: {gamecenter_ini_path}")
     config = configparser.ConfigParser()
 
+    # Read the GameCenter.ini file
     try:
         with open(gamecenter_ini_path, 'r', encoding='utf-16') as file:
             config.read_file(file)
@@ -1453,37 +1456,75 @@ else:
         print(f"Error reading the file: {e}")
         exit(1)
 
-    downloaded_games = dict(config.items('StartDownloadingGames')) if 'StartDownloadingGames' in config else {}
-    installed_games = dict(config.items('DownloadFormSettings')) if 'DownloadFormSettings' in config else {}
+    # Collect game IDs from different sections
+    game_ids = set()
 
-    all_game_ids = set(downloaded_games.keys()).union(installed_games.keys())
+    # Parse game IDs from the 'StartDownloadingGames' section
+    if 'StartDownloadingGames' in config:
+        downloaded_games = dict(config.items('StartDownloadingGames'))
+        game_ids.update(downloaded_games.keys())
 
+    # Parse game IDs from the 'FirstOpeningGameIds' section
+    if 'FirstOpeningGameIds' in config:
+        first_opening_game_ids = config['FirstOpeningGameIds'].get('FirstOpeningGameIds', '').split(';')
+        game_ids.update(first_opening_game_ids)
+
+    # Parse game IDs from the 'GamePersIds' section
+    if 'GamePersIds' in config:
+        for key in config['GamePersIds']:
+            game_id = key.split('_')[0]
+            game_ids.add(game_id)
+
+    # Parse game IDs from the 'RunningGameClients' section
+    if 'RunningGameClients' in config:
+        running_game_clients = config['RunningGameClients'].get('RunningGameClients', '').split(';')
+        game_ids.update(running_game_clients)
+
+    # Parse game IDs from the 'LastAccessGames' section
+    if 'LastAccessGames' in config:
+        last_access_games = dict(config.items('LastAccessGames'))
+        game_ids.update(last_access_games.keys())
+
+    # Parse game IDs from the 'UndoList' section
+    if 'UndoList' in config:
+        for key in config['UndoList']:
+            if 'vkplay://show' in config['UndoList'][key]:
+                game_id = config['UndoList'][key].split('/')[1]
+                game_ids.add(game_id)
+
+    # Parse game IDs from the 'LeftBar' section
+    if 'LeftBar' in config:
+        left_bar_ids = config['LeftBar'].get('Ids', '').split(';')
+        game_ids.update(left_bar_ids)
+
+    # Parse game IDs from the 'Ad' section
     if 'Ad' in config:
         for key in config['Ad']:
-            key_lower = key.lower()
-            if key_lower.startswith('idmtlink0.') or key_lower.startswith('idmtlinkts0.'):
-                game_id = '0.' + key_lower.split('.')[1]
-                all_game_ids.add(game_id)
+            if 'IdMTLink' in key:
+                game_id = key.split('0.')[1]
+                game_ids.add(game_id)
 
     print("\nGame IDs found in GameCenter.ini file:")
-    print(f"All game IDs from INI: {all_game_ids}")
+    for game_id in game_ids:
+        print(f"ID: {game_id}")
 
+    # Handle the Cache folder
     if not os.path.exists(cache_folder_path):
         print(f"VK Play scanner skipped: Cache folder {cache_folder_path} does not exist.")
     else:
+        print(f"Found Cache folder: {cache_folder_path}")
         all_files = os.listdir(cache_folder_path)
         valid_xml_files = []
 
         for file_name in all_files:
             if file_name.endswith(".json"):
-                continue
-
+                continue  # Skip JSON files
             file_path = os.path.join(cache_folder_path, file_name)
             try:
                 tree = ET.parse(file_path)
                 valid_xml_files.append(file_path)
             except ET.ParseError:
-                continue
+                continue  # Skip invalid XML files
 
         processed_game_ids = set()
         found_games = []
@@ -1496,15 +1537,16 @@ else:
 
                 if game_item is not None:
                     game_id_xml = game_item.get('Name') or game_item.get('PackageName')
+
                     if game_id_xml:
                         game_id_in_ini = game_id_xml.replace('_', '.')
 
-                        if game_id_in_ini in all_game_ids and game_id_in_ini not in processed_game_ids:
+                        if game_id_in_ini in game_ids and game_id_in_ini not in processed_game_ids:
                             game_name = game_item.get('TitleEn', 'Unnamed Game')
                             found_games.append(f"{game_name} (ID: {game_id_in_ini})")
                             processed_game_ids.add(game_id_in_ini)
             except ET.ParseError:
-                continue
+                continue  # Skip invalid XML files
 
         if found_games:
             print("\nFound the following games:")