[+] Load Jacket from A???\AssetBundleImages\jacket\*.png

AquaMai/UX/LoadJacketPng.cs

@@ -1,15 +1,21 @@
 using System;
 using System.Collections.Generic;
+using System.Collections.ObjectModel;
 using System.IO;
+using System.Linq;
 using System.Reflection;
 using HarmonyLib;
 using UnityEngine;
 using System.Text.RegularExpressions;
+using Manager;
+using MelonLoader;
 
 namespace AquaMai.UX
 {
-    [HarmonyPatch]
     public class LoadJacketPng
+    {
+        [HarmonyPatch]
+        public static class Loader
         {
             public static IEnumerable<MethodBase> TargetMethods()
             {
@@ -28,29 +34,44 @@ namespace AquaMai.UX
                 var id = matches[0].Groups[1].Value;
 
                 var texture = GetJacketTexture2D(id);
-            if (texture is null)
-            {
-                __result = __instance.LoadAsset<Texture2D>($"Jacket/UI_Jacket_{id}.png");
-            }
-            else
-            {
-                __result = texture;
-            }
+                __result = texture ?? __instance.LoadAsset<Texture2D>($"Jacket/UI_Jacket_{id}.png");
 
                 return false;
             }
+        }
+
+        private static string[] imageExts = [".jpg", ".png", ".jpeg"];
+        private static Regex localAssetsJacketExt = new(@"(\d{6})\.(png|jpg|jpeg)");
+        private static Dictionary<string, string> jacketPaths = new();
+
+        [HarmonyPrefix]
+        [HarmonyPatch(typeof(DataManager), "LoadMusicBase")]
+        public static void LoadMusicPostfix(List<string> ____targetDirs)
+        {
+            foreach (var aDir in ____targetDirs)
+            {
+                if (!Directory.Exists(Path.Combine(aDir, @"AssetBundleImages\jacket"))) continue;
+                foreach (var file in Directory.GetFiles(Path.Combine(aDir, @"AssetBundleImages\jacket")))
+                {
+                    if (!imageExts.Contains(Path.GetExtension(file))) continue;
+                    var idStr = Path.GetFileName(file).Substring("ui_jacket_".Length, 6);
+                    jacketPaths[idStr] = file;
+                }
+            }
+
+            foreach (var laFile in Directory.EnumerateFiles(Path.Combine(Environment.CurrentDirectory, "LocalAssets")))
+            {
+                var match = localAssetsJacketExt.Match(Path.GetFileName(laFile));
+                if (!match.Success) continue;
+                jacketPaths[match.Groups[1].Value] = laFile;
+            }
+
+            MelonLogger.Msg($"Loaded {jacketPaths.Count} custom jacket images.");
+        }
 
         private static string GetJacketPath(string id)
         {
-            foreach (var ext in new[] { ".jpg", ".png", ".webp", ".bmp", ".gif" })
-            {
-                if (File.Exists(Path.Combine(Environment.CurrentDirectory, "LocalAssets", id + ext)))
-                {
-                    return Path.Combine(Environment.CurrentDirectory, "LocalAssets", id + ext);
-                }
-            }
-
-            return null;
+            return jacketPaths.GetValueOrDefault(id);
         }
 
         public static Texture2D GetJacketTexture2D(string id)