Update Song.cs

hmxmilohax · Jan 13, 2025 · 20e4a9d · 20e4a9d
1 parent 6082346
commit 20e4a9d
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/AmpHelper.Library/Song.cs b/AmpHelper.Library/Song.cs
@@ -758,9 +758,15 @@ public static void AddSong(DirectoryInfo unpackedPath, string[] songNames, Conso
 
                 ValidateSong(songPath, songName, true);
 
+                var moggSong = MoggSong.FromMoggsong(Path.Combine(songPath, $"{songName}.moggsong"));
+                moggSong.MoggPath = $"{songName}.mogg";
+                moggSong.MidiPath = $"{songName}.mid";
+
+                var donorSong = moggSong.DonorSong ?? "tut0";
+
                 foreach (var ext in new string[] { $"mid_{platform}", $"png.dta_dta_{platform}", $"png_{platform}" })
                 {
-                    var donorFile = Path.Combine(paths.Songs, "tut0", $"tut0.{ext}");
+                    var donorFile = Path.Combine(paths.Songs, donorSong, $"{donorSong}.{ext}");
                     var destFile = Path.Combine(paths.Songs, songName, $"{songName}.{ext}");
 
                     if (!File.Exists(destFile))
@@ -769,9 +775,6 @@ public static void AddSong(DirectoryInfo unpackedPath, string[] songNames, Conso
                     }
                 }
 
-                var moggSong = MoggSong.FromMoggsong(Path.Combine(songPath, $"{songName}.moggsong"));
-                moggSong.MoggPath = $"{songName}.mogg";
-                moggSong.MidiPath = $"{songName}.mid";
                 var dtx = moggSong.ToDtx();
 
                 using (var dtbStream = File.Create(Path.Combine(songPath, $"{songName}.moggsong_dta_{platform}")))