Implement mod-defined package loaders.

OpenRA.Game/FileSystem/D2kSoundResources.cs

@@ -9,73 +9,94 @@
  */
 #endregion
 
+using System;
 using System.Collections.Generic;
 using System.IO;
-using OpenRA.Primitives;
 
 namespace OpenRA.FileSystem
 {
-	public sealed class D2kSoundResources : IReadOnlyPackage
+	public class D2kSoundResourcesLoader : IPackageLoader
 	{
-		struct Entry
+		sealed class D2kSoundResources : IReadOnlyPackage
 		{
-			public readonly uint Offset;
-			public readonly uint Length;
-
-			public Entry(uint offset, uint length)
+			struct Entry
 			{
-				Offset = offset;
-				Length = length;
-			}
-		}
+				public readonly uint Offset;
+				public readonly uint Length;
 
-		public string Name { get; private set; }
-		public IEnumerable<string> Contents { get { return index.Keys; } }
-
-		readonly Stream s;
-		readonly Dictionary<string, Entry> index = new Dictionary<string, Entry>();
-
-		public D2kSoundResources(FileSystem context, string filename)
-		{
-			Name = filename;
-
-			s = context.Open(filename);
-			try
-			{
-				var headerLength = s.ReadUInt32();
-				while (s.Position < headerLength + 4)
+				public Entry(uint offset, uint length)
 				{
-					var name = s.ReadASCIIZ();
-					var offset = s.ReadUInt32();
-					var length = s.ReadUInt32();
-					index.Add(name, new Entry(offset, length));
+					Offset = offset;
+					Length = length;
 				}
 			}
-			catch
+
+			public string Name { get; private set; }
+			public IEnumerable<string> Contents { get { return index.Keys; } }
+
+			readonly Stream s;
+			readonly Dictionary<string, Entry> index = new Dictionary<string, Entry>();
+
+			public D2kSoundResources(Stream s, string filename)
 			{
-				Dispose();
-				throw;
+				Name = filename;
+				this.s = s;
+
+				try
+				{
+					var headerLength = s.ReadUInt32();
+					while (s.Position < headerLength + 4)
+					{
+						var name = s.ReadASCIIZ();
+						var offset = s.ReadUInt32();
+						var length = s.ReadUInt32();
+						index.Add(name, new Entry(offset, length));
+					}
+				}
+				catch
+				{
+					Dispose();
+					throw;
+				}
+			}
+
+			public Stream GetStream(string filename)
+			{
+				Entry e;
+				if (!index.TryGetValue(filename, out e))
+					return null;
+
+				s.Seek(e.Offset, SeekOrigin.Begin);
+				return new MemoryStream(s.ReadBytes((int)e.Length));
+			}
+
+			public IReadOnlyPackage OpenPackage(string filename, FileSystem context)
+			{
+				// Not implemented
+				return null;
+			}
+
+			public bool Contains(string filename)
+			{
+				return index.ContainsKey(filename);
+			}
+
+			public void Dispose()
+			{
+				s.Dispose();
 			}
 		}
 
-		public Stream GetStream(string filename)
+		bool IPackageLoader.TryParsePackage(Stream s, string filename, FileSystem context, out IReadOnlyPackage package)
 		{
-			Entry e;
-			if (!index.TryGetValue(filename, out e))
-				return null;
+			if (!filename.EndsWith(".rs", StringComparison.InvariantCultureIgnoreCase))
+			{
+				package = null;
+				return false;
+			}
 
-			s.Seek(e.Offset, SeekOrigin.Begin);
-			return new MemoryStream(s.ReadBytes((int)e.Length));
-		}
-
-		public bool Contains(string filename)
-		{
-			return index.ContainsKey(filename);
-		}
-
-		public void Dispose()
-		{
-			s.Dispose();
+			package = new D2kSoundResources(s, filename);
+			return true;
 		}
 	}
 }