view api/Exporters/WarFoundryHtmlExporter.cs @ 131:5145b7c61ae0

Re #68: Add "export army list" function * Add initial body of an export method (doesn't list costs, total or equipment)
author IBBoard <dev@ibboard.co.uk>
date Tue, 01 Sep 2009 16:06:39 +0000
parents e147a1e9399b
children e9c415839462
line wrap: on
line source

// This file (WarFoundryHtmlExporter.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2009 IBBoard
// 
// The file and the library/program it is in are licensed and distributed, without warranty, under the GNU Affero GPL license, either version 3 of the License or (at your option) any later version. Please see COPYING for more information and the full license.

using System;
using System.Collections.Generic;
using System.IO;
using System.Xml;
using System.Xml.Schema;
using IBBoard.WarFoundry.API.Objects;

namespace IBBoard.WarFoundry.API.Exporters
{
	/// <summary>
	/// Custom exporter that exports an army as a basic HTML file
	/// </summary>
	public class WarFoundryHtmlExporter : IWarFoundryExporter
	{
		private static WarFoundryHtmlExporter exporter;
		
		public static WarFoundryHtmlExporter GetDefault()
		{
			if (exporter == null)
			{
				exporter = new WarFoundryHtmlExporter();
			}
			
			return exporter;
		}
		
		private WarFoundryHtmlExporter()
		{
			//Hide constructor
		}
		
		public void ExportArmy(Army army, string path)
		{
			XmlDocument doc = new XmlDocument();
			doc.AppendChild(doc.CreateDocumentType("html", "-//W3C//DTD XHTML 1.0 Strict//EN", "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd", null));
			XmlElement html = doc.CreateElement("html");
			doc.AppendChild(html);
			XmlElement head = doc.CreateElement("head");
			html.AppendChild(head);
			XmlElement title = doc.CreateElement("title");
			head.AppendChild(title);
			title.InnerXml = army.Name;
			XmlElement body = doc.CreateElement("body");
			html.AppendChild(body);
			XmlElement header = doc.CreateElement("h1");
			header.InnerText = army.Name;
			body.AppendChild(header);
			
			foreach (XmlElement table in CreateTables(army, doc))
			{
				body.AppendChild(table);
			}
			
			StreamWriter writer = new StreamWriter(path, false);
			
			try
			{
				writer.Write(doc.OuterXml);
			}
			finally
			{
				writer.Close();
			}
		}
		
		private XmlElement[] CreateTables(Army army, XmlDocument doc)
		{
			Dictionary<string, XmlElement> tables = new Dictionary<string, XmlElement>();
			
			foreach (SystemStats statSets in army.GameSystem.SystemStats)
			{
				tables[statSets.ID] = CreateTable(statSets, doc);
			}
			
			foreach (Unit unit in army.GetUnits())
			{
				tables[unit.UnitType.StatsID].AppendChild(CreateUnitRow(unit, doc));
			}
			
			return DictionaryUtils.ToArray(tables);
		}
		
		private XmlElement CreateTable(SystemStats stats, XmlDocument doc)
		{
			XmlElement table = doc.CreateElement("table");
			XmlElement headerRow = doc.CreateElement("tr");
			table.AppendChild(headerRow);
			XmlElement name = doc.CreateElement("th");
			name.InnerText = "Name";
			headerRow.AppendChild(name);
			
			foreach (StatSlot stat in stats.StatSlots)
			{
				XmlElement statHeader = doc.CreateElement("th");
				statHeader.InnerText = stat.Name;
				headerRow.AppendChild(statHeader);
			}
			
			XmlElement notes = doc.CreateElement("th");
			notes.InnerText = "Notes";
			headerRow.AppendChild(notes);
			return table;
		}
		
		private XmlElement CreateUnitRow(Unit unit, XmlDocument doc)
		{
			XmlElement row = doc.CreateElement("tr");
			XmlElement name = doc.CreateElement("td");
			name.InnerText = unit.Name;
			row.AppendChild(name);
			
			foreach (Stat stat in unit.UnitStatsArray)
			{
				XmlElement statCell = doc.CreateElement("td");
				statCell.InnerText = stat.SlotValueString;
				row.AppendChild(statCell);
			}
			
			return row;
		}
	}
}