diff --git a/AquaMai/UX/HideSelfMadeCharts.cs b/AquaMai/UX/HideSelfMadeCharts.cs index 20af2b93..fdca5b64 100644 --- a/AquaMai/UX/HideSelfMadeCharts.cs +++ b/AquaMai/UX/HideSelfMadeCharts.cs @@ -86,12 +86,6 @@ public class HideSelfMadeCharts [HarmonyPatch(typeof(MusicSelectProcess), "OnStart")] public static void MusicSelectProcessOnStart(ref MusicSelectProcess __instance) { - if (File.Exists(Path.Combine(Environment.CurrentDirectory, "LocalAssets", "DisableSelfMadeCharts.txt"))) - { - isForceDisable = true; - return; - } - if (File.Exists(Path.Combine(Environment.CurrentDirectory, "LocalAssets", "SelfMadeChartsDenyUsers.txt"))) { var userIds = File.ReadAllLines(Path.Combine(Environment.CurrentDirectory, "LocalAssets", "SelfMadeChartsDenyUsers.txt")); @@ -105,6 +99,19 @@ public class HideSelfMadeCharts } } + if (File.Exists(Path.Combine(Environment.CurrentDirectory, "LocalAssets", "SelfMadeChartsWhiteListUsers.txt"))) + { + var userIds = File.ReadAllLines(Path.Combine(Environment.CurrentDirectory, "LocalAssets", "SelfMadeChartsWhiteListUsers.txt")); + for (var i = 0; i < 2; i++) + { + var user = Singleton.Instance.GetUserData(i); + if (!user.IsEntry) continue; + if (userIds.Contains(user.Detail.UserID.ToString())) continue; + isForceDisable = true; + return; + } + } + isForceDisable = false; }