Create simple Dir2DAT program for internal use

2025-12-16 19:14:27 +00:00 · 2016-04-11 15:09:50 -07:00
parent 72c4a95ae3
commit eb02006c97
14 changed files with 547 additions and 0 deletions
--- a/DATFromDir/DATFromDir.cs
+++ b/DATFromDir/DATFromDir.cs
@@ -0,0 +1,177 @@
+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());
+			}
+		}
+	}
+}