fix: 截取预览的持续时间错误

Author: clansty

MaiChartManager/Controllers/Music/CueConvertController.cs

@@ -63,9 +63,7 @@ public async Task SetAudioPreview(int id, [FromBody] SetAudioPreviewRequest requ
         var loopStart = TimeSpan.FromSeconds(request.StartTime);
         var loopEnd = TimeSpan.FromSeconds(request.EndTime);
 
-        using var sha1 = SHA1.Create();
-        var awbHash = await sha1.ComputeHashAsync(System.IO.File.OpenRead(StaticSettings.AcbAwb[$"music{id:000000}.awb"]));
-        var acbBytes = CriUtils.CreateAcbWithPreview(cachePath, awbHash, loopStart, loopEnd);
+        var acbBytes = await CriUtils.CreateAcbWithPreview(cachePath, await System.IO.File.ReadAllBytesAsync(StaticSettings.AcbAwb[$"music{id:000000}.awb"]), loopStart, loopEnd);
         await System.IO.File.WriteAllBytesAsync(targetAcbPath, acbBytes);
     }
 

MaiChartManager/Utils/CriUtils.cs

@@ -1,4 +1,5 @@
 using System.Reflection;
+using System.Security.Cryptography;
 using System.Text;
 using SonicAudioLib.CriMw;
 using VGAudio.Containers.Wave;
@@ -29,8 +30,11 @@ public static byte[] SetAcbPreview(string acbPath, TimeSpan loopStart, TimeSpan
         return criTable.Save();
     }
 
-    public static byte[] CreateAcbWithPreview(string wavPath, byte[] sha1HashOfAwb, TimeSpan loopStart, TimeSpan loopEnd)
+    public static async Task<byte[]> CreateAcbWithPreview(string wavPath, byte[] awbBytes, TimeSpan loopStart, TimeSpan loopEnd)
     {
+        using var sha1 = SHA1.Create();
+        var sha1HashOfAwb = await sha1.ComputeHashAsync(new MemoryStream(awbBytes));
+
         var waveReader = new WaveReader();
         var audioData = waveReader.Read(File.ReadAllBytes(wavPath));
         var format = audioData.GetFormat<Pcm16Format>();
@@ -56,6 +60,16 @@ public static byte[] CreateAcbWithPreview(string wavPath, byte[] sha1HashOfAwb,
         waveFormTable.Rows[0]["NumSamples"] = (byte)format.SampleCount;
         criTable.Rows[0]["WaveformTable"] = waveFormTable.Save();
 
+        // 修复歌曲持续时间
+        var count = BitConverter.ToInt32(awbBytes, 8) + 1;
+        var headSize = 16 + awbBytes[5] * count + awbBytes[6] * count + awbBytes[7] * count;
+        var header = new byte[headSize];
+        Array.Copy(awbBytes, header, headSize);
+        var streamAwbAfs2Header = new CriTable();
+        streamAwbAfs2Header.Load(criTable.Rows[0]["StreamAwbAfs2Header"] as byte[]);
+        streamAwbAfs2Header.Rows[0]["Header"] = header;
+        criTable.Rows[0]["StreamAwbAfs2Header"] = streamAwbAfs2Header.Save();
+
         criTable.WriterSettings = CriTableWriterSettings.Adx2Settings;
         return criTable.Save();
     }
@@ -105,4 +119,4 @@ private static CriTable GetTable(this CriRow row, string name)
         table.Load(row[name] as byte[]);
         return table;
     }
-}
+}