Fixed an issue with Extended Folder Scanning not running without stri…

…ct definition of Folder Names
LavaGang · Dec 8, 2024 · e5a3b9f · e5a3b9f
1 parent 91d56f3
commit e5a3b9f
Showing 1 changed file with 36 additions and 2 deletions.
diff --git a/MelonLoader/Melons/MelonFolderHandler.cs b/MelonLoader/Melons/MelonFolderHandler.cs
@@ -159,6 +159,36 @@ private static void FindMelonFolders(
             }
         }
 
+        private static void ScanExtraFolders(eScanType scanType,
+            string path,
+            ref List<string> userLibDirectories,
+            ref List<string> pluginDirectories,
+            ref List<string> modDirectories)
+        {
+            string[] directories = Directory.GetDirectories(path, "*", SearchOption.TopDirectoryOnly);
+            if ((directories == null)
+                || (directories.Length <= 0))
+                return;
+
+            foreach (var dir in directories)
+            {
+                if (userLibDirectories.Contains(dir)
+                    || pluginDirectories.Contains(dir)
+                    || modDirectories.Contains(dir))
+                    continue;
+
+                ScanExtraFolders(scanType, dir, ref userLibDirectories, ref pluginDirectories, ref modDirectories);
+
+                // Add to Directories List
+                if (scanType == eScanType.UserLibs)
+                    userLibDirectories.Add(dir); 
+                else if (scanType == eScanType.Plugins)
+                    pluginDirectories.Add(dir);
+                else
+                    modDirectories.Add(dir);
+            }
+        }
+
         private static void ScanFolder(eScanType scanType,
             string path,
             ref List<string> userLibDirectories,
@@ -193,20 +223,22 @@ private static void ScanFolder(eScanType scanType,
 
                 // Is UserLibs Scan?
                 if (scanType == eScanType.UserLibs)
+                {
+                    ScanExtraFolders(scanType, dir, ref userLibDirectories, ref pluginDirectories, ref modDirectories);
                     userLibDirectories.Add(dir); // Add to Directories List
+                }
                 else
                 {
                     // Check for Deeper Melon Folder
                     string melonPath = Path.Combine(dir, "Plugins");
                     if (Directory.Exists(melonPath))
                     {
                         pluginDirectories.Add(melonPath);
-                        ScanFolder(scanType, melonPath, ref userLibDirectories, ref pluginDirectories, ref modDirectories);
+                        ScanFolder(eScanType.Plugins, melonPath, ref userLibDirectories, ref pluginDirectories, ref modDirectories);
                     }
 
                     if (scanType == eScanType.Mods)
                     {
-                        // Check for Deeper Melon Folder
                         melonPath = Path.Combine(dir, "Mods");
                         if (Directory.Exists(melonPath))
                         {
@@ -215,6 +247,8 @@ private static void ScanFolder(eScanType scanType,
                         }
                     }
 
+                    ScanExtraFolders(scanType, dir, ref userLibDirectories, ref pluginDirectories, ref modDirectories);
+
                     // Add to Directories List
                     if (scanType == eScanType.Mods)
                         modDirectories.Add(dir);