SabreTools/DATFromDir/DATFromDir.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.IO;
using System.Security.Cryptography;
using System.Text;
using System.Threading.Tasks;
using System.Web;

namespace SabreTools
{
	class DATFromDir
	{
		private static string _7zPath;
		private static string _basePath;
		private static string _tempDir;
		private static string _baseExtract;

		static void Main(string[] args)
		{
			// Set local paths
			_7zPath = Environment.CurrentDirectory.Replace('/', '\\') + "\\7z" + (Environment.Is64BitOperatingSystem ? "\\x64" : "") + "\\";
			_tempDir = Environment.CurrentDirectory.Replace('/', '\\') + "\\temp" + DateTime.Now.ToString("yyyyMMddHHmmss") + "\\";
			_basePath = (args.Length == 0 ? Environment.CurrentDirectory : args[0]).Replace('/', '\\') + "\\";

			// Set base arguments to be used
			_baseExtract = "x -o\"" + _tempDir + "\"";

			// Set the basic Process information for 7za
			ProcessStartInfo psi = new ProcessStartInfo
			{
				FileName = _7zPath + "7za.exe",
				RedirectStandardError = true,
				RedirectStandardOutput = true,
				UseShellExecute = false,
			};

			// Get an output array going that has the right mappings (parent, name, size, hash)
			List<Tuple<string, string, long, string>> roms = new List<Tuple<string, string, long, string>>();

			// This is where the main loop would go
			foreach (string item in Directory.EnumerateFileSystemEntries(_basePath))
			{
				// Create the temporary output directory
				Directory.CreateDirectory(_tempDir);

				psi.Arguments = _baseExtract + " " + item;
				Process zip = Process.Start(psi);
				zip.WaitForExit();

				bool encounteredErrors = zip.StandardError.ReadToEnd().Contains("ERROR");

				// Get a list of files including size and hash
				SHA1 sha1 = SHA1.Create();

				// If the file was an archive and was extracted successfully, check it
				if (!encounteredErrors)
				{
					foreach (string entry in Directory.GetFiles(_tempDir, "*", SearchOption.AllDirectories))
					{
						FileStream fs = File.OpenRead(entry);
						string tempHash = BitConverter.ToString(sha1.ComputeHash(fs)).Replace("-", "");
						fs.Close();

						roms.Add(new Tuple<string, string, long, string>(Path.GetFileNameWithoutExtension(item), entry.Remove(0, _tempDir.Length), (new FileInfo(entry)).Length, tempHash));
					}
				}
				// Otherwise, just get the info on the file itself
				else
				{
					FileStream fs = File.OpenRead(item);
					string tempHash = BitConverter.ToString(sha1.ComputeHash(fs)).Replace("-", "");
					fs.Close();

					roms.Add(new Tuple<string, string, long, string>("Default", item.Remove(0, _basePath.Length), (new FileInfo(item)).Length, tempHash));
				}

				// Delete the temp directory
				Directory.Delete(_tempDir, true);
			}

			// Order the roms by name of parent, then name of rom
			roms.Sort(delegate (Tuple<string, string, long, string> A, Tuple<string, string, long, string> B)
			{
				if (A.Item1 == B.Item1)
				{
					if (A.Item2 == B.Item2)
					{
						if (A.Item3 == B.Item3)
						{
							return String.Compare(A.Item4, B.Item4);
						}
						return (int)(A.Item3 - B.Item3);
					}
					return String.Compare(A.Item2, B.Item2);
				}
				return String.Compare(A.Item1, B.Item1);
			});

			// Write out all of the file information
			Console.WriteLine("Parent\t\tRom\t\tSize\t\tSHA-1");
			foreach (Tuple<string, string, long, string> rom in roms)
			{
				Console.WriteLine(rom.Item1 + "\t" + rom.Item2 + "\t" + rom.Item3 + "\t" + rom.Item4);
			}

			// Now write it all out as a DAT
			try
			{
				FileStream fs = File.Create("dirdat.xml");
				StreamWriter sw = new StreamWriter(fs, Encoding.UTF8);

				/*
				string header_old = "clrmamepro (\n" +
					"\tname \"dirdat\"\n" +
					"\tdescription \"dirdat\"\n" +
					"\tversion \"\"\n" +
					"\tcomment \"\"\n" +
					"\tauthor \"DATFromDir\"\n" +
					")\n";
				*/

				string header = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" +
					"<!DOCTYPE datafile PUBLIC \"-//Logiqx//DTD ROM Management Datafile//EN\" \"http://www.logiqx.com/Dats/datafile.dtd\">\n\n" +
					"\t<datafile>\n" +
					"\t\t<header>\n" +
					"\t\t\t<name>dirdat</name>\n" +
					"\t\t\t<description>dirdat</description>\n" +
					"\t\t\t<category>DATFromDir</category>\n" +
					"\t\t\t<version></version>\n" +
					"\t\t\t<date></date>\n" +
					"\t\t\t<author>DATFromDir</author>\n" +
					"\t\t\t<clrmamepro />\n" +
					"\t\t</header>\n";

				// Write the header out
				sw.Write(header);

				// Write out each of the machines and roms
				string lastgame = "";
				foreach (Tuple<string, string, long, string> rom in roms)
				{
					string state = "";
					if (lastgame != "" && lastgame != rom.Item1)
					{
						state += "\t</machine>\n";
					}

					if (lastgame != rom.Item1)
					{
						state += "\t<machine name=\"" + HttpUtility.HtmlEncode(rom.Item1) + "\">\n" +
							"\t\t<description>" + HttpUtility.HtmlEncode(rom.Item1) + "</description>\n";
					}

					state += "\t\t<rom name=\"" + HttpUtility.HtmlEncode(rom.Item2) + "\"" +
						(rom.Item3 != -1 ? " size=\"" + rom.Item3 + "\"" : "") +
						(rom.Item4 != "" ? " sha1=\"" + rom.Item4.ToLowerInvariant() + "\"" : "") +
						" />\n";

					lastgame = rom.Item1;

					sw.Write(state);
				}

				sw.Write("\t</machine>\n</datafile>");
				Console.Write("File written!");
				sw.Close();
				fs.Close();
			}
			catch (Exception ex)
			{
				Console.Write(ex.ToString());
			}
		}
	}
}