mirror of
https://github.com/ppy/osu
synced 2024-12-24 15:53:37 +00:00
67 lines
2.0 KiB
C#
67 lines
2.0 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Security.Cryptography;
|
|
using Ionic.Zip;
|
|
using osu.Game.Beatmaps.Formats;
|
|
using osu.Game.Database;
|
|
|
|
namespace osu.Game.Beatmaps.IO
|
|
{
|
|
public sealed class OszArchiveReader : ArchiveReader
|
|
{
|
|
public static void Register()
|
|
{
|
|
AddReader<OszArchiveReader>((storage, path) =>
|
|
{
|
|
using (var stream = storage.GetStream(path))
|
|
return ZipFile.IsZipFile(stream, false);
|
|
});
|
|
OsuLegacyDecoder.Register();
|
|
}
|
|
|
|
private Stream archiveStream;
|
|
private ZipFile archive;
|
|
private string[] beatmaps;
|
|
private Beatmap firstMap;
|
|
|
|
public OszArchiveReader(Stream archiveStream)
|
|
{
|
|
this.archiveStream = archiveStream;
|
|
archive = ZipFile.Read(archiveStream);
|
|
beatmaps = archive.Entries.Where(e => e.FileName.EndsWith(@".osu"))
|
|
.Select(e => e.FileName).ToArray();
|
|
if (beatmaps.Length == 0)
|
|
throw new FileNotFoundException(@"This directory contains no beatmaps");
|
|
using (var stream = new StreamReader(GetStream(beatmaps[0])))
|
|
{
|
|
var decoder = BeatmapDecoder.GetDecoder(stream);
|
|
firstMap = decoder.Decode(stream);
|
|
}
|
|
}
|
|
|
|
public override string[] ReadBeatmaps()
|
|
{
|
|
return beatmaps;
|
|
}
|
|
|
|
public override Stream GetStream(string name)
|
|
{
|
|
ZipEntry entry = archive.Entries.SingleOrDefault(e => e.FileName == name);
|
|
if (entry == null)
|
|
throw new FileNotFoundException();
|
|
return entry.OpenReader();
|
|
}
|
|
|
|
public override BeatmapMetadata ReadMetadata()
|
|
{
|
|
return firstMap.BeatmapInfo.Metadata;
|
|
}
|
|
|
|
public override void Dispose()
|
|
{
|
|
archive.Dispose();
|
|
archiveStream.Dispose();
|
|
}
|
|
}
|
|
} |