osu/osu.Game/Beatmaps/IO/OszArchiveReader.cs

64 lines
1.9 KiB
C#
Raw Normal View History

using System;
using System.IO;
2016-10-04 21:08:43 +00:00
using System.Linq;
using System.Security.Cryptography;
2016-10-04 21:08:43 +00:00
using Ionic.Zip;
using osu.Game.Beatmaps.Formats;
2016-10-18 17:35:01 +00:00
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 ZipFile archive { get; set; }
private string[] beatmaps { get; set; }
private Beatmap firstMap { get; set; }
public OszArchiveReader(Stream 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(ReadFile(beatmaps[0])))
{
var decoder = BeatmapDecoder.GetDecoder(stream);
2016-10-18 17:35:01 +00:00
firstMap = decoder.Decode(stream);
}
}
public override string[] ReadBeatmaps()
{
2016-10-10 18:00:33 +00:00
return beatmaps;
}
public override Stream ReadFile(string name)
{
ZipEntry entry = archive.Entries.SingleOrDefault(e => e.FileName == name);
if (entry == null)
throw new FileNotFoundException();
2016-10-04 21:08:43 +00:00
return entry.OpenReader();
}
public override BeatmapMetadata ReadMetadata()
{
2016-10-18 17:35:01 +00:00
return firstMap.BeatmapInfo.Metadata;
}
2016-10-18 17:35:01 +00:00
public override void Dispose()
{
archive.Dispose();
}
}
}