Merge pull request #673 from TheTrackerCouncil/add-custom-tracker-sprites

Add custom tracker sprite functionality

Author: MattEqualsCoder

src/TrackerCouncil.Smz3.Data/RandomizerDirectories.cs

@@ -100,6 +100,9 @@ public static string TrackerSpritePath
         }
     }
 
+    public static string UserTrackerSpritePath => Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
+        "SMZ3CasRandomizer", "TrackerSprites");
+
 #if DEBUG
     public static string TrackerSpriteHashYamlFilePath => Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "SMZ3CasRandomizer", "tracker-sprite-hashes-debug.yml");
 #else

src/TrackerCouncil.Smz3.Data/Services/TrackerSpriteService.cs

@@ -17,46 +17,8 @@ public class TrackerSpriteService(OptionsFactory optionsFactory)
     public void LoadSprites()
     {
         _packs = [];
-        var path = RandomizerDirectories.TrackerSpritePath;
-        var packFolders = Directory.EnumerateDirectories(RandomizerDirectories.TrackerSpritePath);
-        foreach (var folder in packFolders)
-        {
-            var packName = Path.GetFileName(folder);
-
-            if (string.IsNullOrEmpty(packName) || !File.Exists(Path.Combine(folder, "default_idle.png")) ||
-                !File.Exists(Path.Combine(folder, "default_talk.png")))
-            {
-                continue;
-            }
-
-            Dictionary<string, TrackerSpeechReactionImages> _reactions = [];
-
-            foreach (var idleImage in Directory.GetFiles(folder, "*_idle.png"))
-            {
-                var talkingImage = idleImage.Replace("_idle.png", "_talk.png");
-                if (!File.Exists(talkingImage))
-                {
-                    continue;
-                }
-
-                var reactionName = Path.GetFileName(idleImage).Replace("_idle.png", "").ToLower();
-                _reactions[reactionName] = new TrackerSpeechReactionImages
-                {
-                    IdleImage = idleImage, TalkingImage = talkingImage,
-                };
-            }
-
-            _packs.Add(new TrackerSpeechImagePack
-            {
-                Name = packName,
-                Default = new TrackerSpeechReactionImages
-                {
-                    IdleImage = Path.Combine(folder, "default_idle.png"),
-                    TalkingImage = Path.Combine(folder, "default_talk.png"),
-                },
-                Reactions = _reactions
-            });
-        }
+        LoadFolderPacks(RandomizerDirectories.TrackerSpritePath, false);
+        LoadFolderPacks(RandomizerDirectories.UserTrackerSpritePath, true);
     }
 
     public Dictionary<string, string> GetPackOptions()
@@ -73,4 +35,62 @@ public TrackerSpeechImagePack GetPack(string? packName = null)
         return _packs.FirstOrDefault(x => x.Name == packName) ??
                _packs.FirstOrDefault(x => x.Name == "Default") ?? _packs.First();
     }
+
+    private void LoadFolderPacks(string folder, bool isUserPack)
+    {
+        if (!Directory.Exists(folder))
+        {
+            return;
+        }
+
+        var packFolders = Directory.EnumerateDirectories(folder);
+        foreach (var packFolder in packFolders)
+        {
+            LoadPack(packFolder, isUserPack);
+        }
+    }
+
+    private void LoadPack(string folder, bool isUserPack)
+    {
+        var packName = Path.GetFileName(folder);
+
+        if (string.IsNullOrEmpty(packName) || !File.Exists(Path.Combine(folder, "default_idle.png")) ||
+            !File.Exists(Path.Combine(folder, "default_talk.png")))
+        {
+            return;
+        }
+
+        Dictionary<string, TrackerSpeechReactionImages> reactions = [];
+
+        foreach (var idleImage in Directory.GetFiles(folder, "*_idle.png"))
+        {
+            var talkingImage = idleImage.Replace("_idle.png", "_talk.png");
+            if (!File.Exists(talkingImage))
+            {
+                continue;
+            }
+
+            var reactionName = Path.GetFileName(idleImage).Replace("_idle.png", "").ToLower();
+            reactions[reactionName] = new TrackerSpeechReactionImages
+            {
+                IdleImage = idleImage, TalkingImage = talkingImage,
+            };
+        }
+
+        if (_packs.Any(x => x.Name == packName) && isUserPack)
+        {
+            packName += " (Custom)";
+        }
+
+        _packs.Add(new TrackerSpeechImagePack
+        {
+            Name = packName,
+            Default = new TrackerSpeechReactionImages
+            {
+                IdleImage = Path.Combine(folder, "default_idle.png"),
+                TalkingImage = Path.Combine(folder, "default_talk.png"),
+            },
+            Reactions = reactions
+        });
+    }
 }