添加 UserDataPanel添加解锁音乐功能

WYH2004-MC · Oct 8, 2024 · e37152d · e37152d
1 parent 47dc279
commit e37152d
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 6 deletions.
diff --git a/BuildInfo.cs b/BuildInfo.cs
@@ -1,6 +1,6 @@
 namespace SinmaiAssist { 
     public static partial class BuildInfo { 
-        public const string CommitHash = "c23d5c7"; 
-        public const string BuildDate = "2024-10-08T11:55:00.8502037+08:00"; 
+        public const string CommitHash = "47dc279"; 
+        public const string BuildDate = "2024-10-08T18:37:12.4228763+08:00"; 
     } 
 } 
diff --git a/GUI/UserDataPanel.cs b/GUI/UserDataPanel.cs
@@ -1,4 +1,6 @@
 using System;
+using System.IO;
+using System.Linq;
 using MAI2.Util;
 using Manager;
 using MelonLoader;
@@ -21,7 +23,7 @@ private enum CollectionType
         Frame = UserData.Collection.Frame
     }
 
-    private static string[] _collectionId = new string[6] { "", "", "", "", "", ""};
+    private static string[] _userInputId = new string[6] { "", "", "", "", "", ""};
 
     public static void OnGUI()
     {
@@ -44,15 +46,26 @@ public static void OnGUI()
             GUILayout.BeginHorizontal();
             int typeId = (int)type;
             GUILayout.Label(type.ToString(), new GUIStyle(UnityEngine.GUI.skin.label){fixedWidth = 50});
-            _collectionId[typeId] = GUILayout.TextField(_collectionId[typeId]);
+            _userInputId[typeId] = GUILayout.TextField(_userInputId[typeId]);
             if (GUILayout.Button("Add", new GUIStyle(UnityEngine.GUI.skin.button){ fixedWidth = 50}))
             {
-                AddCollections(0, type, _collectionId[typeId]);
-                AddCollections(1, type, _collectionId[typeId]);
+                AddCollections(0, type, _userInputId[typeId]);
+                AddCollections(1, type, _userInputId[typeId]);
             }
             GUILayout.EndHorizontal();
         }
 
+        GUILayout.Label("Unlock Music", MainGUI.Style.Title);
+        GUILayout.BeginHorizontal();
+        GUILayout.Label("Music", new GUIStyle(UnityEngine.GUI.skin.label){fixedWidth = 50});
+        _userInputId[0] = GUILayout.TextField(_userInputId[0]);
+        if (GUILayout.Button("Add", new GUIStyle(UnityEngine.GUI.skin.button){ fixedWidth = 50}))
+        {
+            UnlockMusic(0, _userInputId[0]);
+            UnlockMusic(1, _userInputId[0]);
+        }
+        GUILayout.EndHorizontal();
+
         GUILayout.Label("User Data Backup", MainGUI.Style.Title);
         GUILayout.BeginHorizontal();
         if (GUILayout.Button("1P")) User.ExportBackupData(0);
@@ -93,4 +106,50 @@ private static void AddCollections(long index, CollectionType type, string input
             MelonLogger.Error(e);
         }
     }
+
+    private static void UnlockMusic(long index, string input)
+    {
+        UserData userData = Singleton<UserDataManager>.Instance.GetUserData(index);
+        if (userData.IsGuest())
+        {
+            GameMessageManager.SendGameMessage("Guest Account\nUnable to unlock music", (int)index);
+            return;
+        }
+        try
+        {
+            if (int.TryParse(input, out int id))
+            {
+                if (!userData.IsUnlockMusic(UserData.MusicUnlock.Base, id))
+                {
+                    if (userData.AddUnlockMusic(UserData.MusicUnlock.Base, id))
+                    {
+                        GameMessageManager.SendGameMessage($"Unlock Music \n{id}", (int)index);
+                    }
+                    else
+                    {
+                        GameMessageManager.SendGameMessage($"Failed to unlock music \n{id}", (int)index);
+                    }
+                }
+                else if(!userData.IsUnlockMusic(UserData.MusicUnlock.Master, id))
+                {
+                    userData.AddUnlockMusic(UserData.MusicUnlock.Master, id);
+                    userData.AddUnlockMusic(UserData.MusicUnlock.ReMaster, id);
+                    GameMessageManager.SendGameMessage($"Unlock Master \n{id}", (int)index);
+                }
+                else
+                {
+                    GameMessageManager.SendGameMessage($"Music not found or already unlocked\n{id}", (int)index);
+                }
+            }
+            else
+            {
+                GameMessageManager.SendGameMessage($"Invalid ID\n {input}", (int)index);
+            }
+        }
+        catch (Exception e)
+        {
+            GameMessageManager.SendGameMessage($"Unknown error", (int)index);
+            MelonLogger.Error(e);
+        }
+    }
 }
diff --git a/Utils/User.cs b/Utils/User.cs
@@ -91,4 +91,6 @@ public static UserData GetUserData(long index)
     {
         return Singleton<UserDataManager>.Instance.GetUserData(index);
     }
+
+
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -91,4 +91,6 @@ public static UserData GetUserData(long index) @@
         {
             return Singleton<UserDataManager>.Instance.GetUserData(index);
         }
     }