# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1299355721 0
# Node ID 2ba88d5fb881408727f621652b808e90e4bbbc4a
# Parent  4be289645c4fc2c0d562764fa03c5e376a1cff9d
Re #324: Add saving of Race and System data to files
 * Construct correct names for entries

diff -r 4be289645c4f -r 2ba88d5fb881 api/Savers/Xml/WarFoundryXmlFileSaver.cs
--- a/api/Savers/Xml/WarFoundryXmlFileSaver.cs	Sat Mar 05 15:51:24 2011 +0000
+++ b/api/Savers/Xml/WarFoundryXmlFileSaver.cs	Sat Mar 05 20:08:41 2011 +0000
@@ -30,13 +30,33 @@
 
 		public void AddFile(ZipOutputStream zipStream, WarFoundryLoadedObject obj)
 		{
-			ZipEntry entry = new ZipEntry(obj.ID);
+			ZipEntry entry = new ZipEntry(GetZipEntryName(obj));
 			zipStream.PutNextEntry(entry);
 		}
+		
+		private string GetZipEntryName(WarFoundryLoadedObject obj)
+		{
+			return obj.ID + GetFileExtension(obj);
+		}
 
 		public string GetFileExtension (WarFoundryLoadedObject obj)
 		{
-			return "";
+			string ext = "";
+			
+			if (obj is Army)
+			{
+				ext = ".armyx";
+			}
+			else if (obj is Race)
+			{
+				ext = ".racex";
+			}
+			else if (obj is GameSystem)
+			{
+				ext = ".systemx";
+			}
+			
+			return ext;
 		}
 	}
 }