using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Xml;
using System.Xml.Linq;

using SabreTools.Helper;

namespace DatSplit
{
	class DatSplit
	{
		private static string extA;
		private static string extB;
		private static string filename;

		public static void Main(string[] args)
		{
			// If we don't have arguments, show help
			if (args.Length == 0 && args.Length != 3)
			{
				Help();
				return;
			}

			// Set needed strings
			filename = args[0];
			extA = args[1];
			extB = args[2];

			// Take the filename, and load it as an XML document
			XmlDocument doc = new XmlDocument();
			try
			{
				doc.LoadXml(File.ReadAllText(filename));
			}
			catch (XmlException ex)
			{
				doc.LoadXml(Converters.RomVaultToXML(File.ReadAllLines(filename)).ToString());
			}

			// We all start the same
			XmlNode node = doc.FirstChild;
			if (node != null && node.Name == "xml")
			{
				// Skip over everything that's not an element
				while (node.NodeType != XmlNodeType.Element)
				{
					node = node.NextSibling;
				}
			}

			XmlDocument tempDoc = new XmlDocument();
			XmlNode outA = tempDoc.CreateNode(XmlNodeType.Element, node.Name, "");
			XmlNode outB = tempDoc.CreateNode(XmlNodeType.Element, node.Name, "");

			// Once we find the main body, enter it
			if (node != null && node.Name == "datafile")
			{
				node = node.FirstChild;
			}

			// Now here's where it differs from import
			while (node != null)
			{
				// If we're at a game node, add the parent node but not all the internals
				if (node.NodeType == XmlNodeType.Element && (node.Name == "machine" || node.Name == "game"))
				{
					bool inA = false;
					bool inB = false;

					// Get the roms from the machine
					if (node.HasChildNodes)
					{
						// If this node has children, traverse the children
						foreach (XmlNode child in node.ChildNodes)
						{
							// If we find a rom or disk, add it
							if (child.NodeType == XmlNodeType.Element && (child.Name == "rom" || child.Name == "disk"))
							{
								// Take care of hex-sized files
								long size = -1;
								if (child.Attributes["size"] != null && child.Attributes["size"].Value.Contains("0x"))
								{
									size = Convert.ToInt64(child.Attributes["size"].Value, 16);
								}
								else if (child.Attributes["size"] != null)
								{
									size = Int64.Parse(child.Attributes["size"].Value);
								}

								if (child.Attributes["name"].Value.EndsWith(extA))
								{
									if (!inA)
									{
										//XmlNode temp = tempDoc.CreateNode(XmlNodeType.Element, node.Name, "");
										XmlNode temp = tempDoc.ImportNode(node, false);
										outA.AppendChild(temp);
										outA = outA.LastChild;
										inA = true;
									}
									outA.AppendChild(tempDoc.ImportNode(child, true));
								}
								else if (child.Attributes["name"].Value.EndsWith(extB))
								{
									if (!inB)
									{
										//XmlNode temp = tempDoc.CreateNode(XmlNodeType.Element, node.Name, "");
										XmlNode temp = tempDoc.ImportNode(node, false);
										outB.AppendChild(temp);
										outB = outB.LastChild;
										inB = true;
									}
									outB.AppendChild(tempDoc.ImportNode(child, true));
								}
								else
								{
									outA.AppendChild(tempDoc.ImportNode(child, true));
									outB.AppendChild(tempDoc.ImportNode(child, true));
								}
							}
						}

						// Set the output node to the right one for both
						if (inA)
						{
							outA = outA.ParentNode;
						}
						if (inB)
						{
							outB = outB.ParentNode;
						}
					}
				}
				else
				{
					XmlNode tempNode = tempDoc.ImportNode(node, true);
					outA.AppendChild(tempNode);
					tempNode = tempDoc.ImportNode(node, true);
					outB.AppendChild(tempNode);
				}
				node = node.NextSibling;
			}

			XmlDocument outDocA = new XmlDocument();
			outDocA.AppendChild(outDocA.CreateDocumentType("datafile", "-//Logiqx//DTD ROM Management Datafile//EN", "http://www.logiqx.com/Dats/datafile.dtd", null));
			outDocA.AppendChild(outDocA.ImportNode(outA, true));
			string outPathA = Path.GetFileNameWithoutExtension(filename) + extA + Path.GetExtension(filename);
			File.WriteAllText(outPathA, Beautify(outDocA));

			XmlDocument outDocB = new XmlDocument();
			outDocB.AppendChild(outDocB.CreateDocumentType("datafile", "-//Logiqx//DTD ROM Management Datafile//EN", "http://www.logiqx.com/Dats/datafile.dtd", null));
			outDocB.AppendChild(outDocB.ImportNode(outB, true));
			string outPathB = Path.GetFileNameWithoutExtension(filename) + extB + Path.GetExtension(filename);
			File.WriteAllText(outPathB, Beautify(outDocB));
		}

		public static void Help()
		{
			Console.WriteLine("DatSplit.exe <filename> <ext> <ext>");
		}

		// http://stackoverflow.com/questions/203528/what-is-the-simplest-way-to-get-indented-xml-with-line-breaks-from-xmldocument
		static public string Beautify(XmlDocument doc)
		{
			StringBuilder sb = new StringBuilder();
			XmlWriterSettings settings = new XmlWriterSettings
			{
				Indent = true,
				IndentChars = "\t",
				NewLineChars = "\r\n",
				NewLineHandling = NewLineHandling.Replace
			};
			using (XmlWriter writer = XmlWriter.Create(sb, settings))
			{
				doc.Save(writer);
			}
			return sb.ToString();
		}
	}
}