using System; using System.Collections.Generic; using System.IO; using System.Text; using SabreTools.Library.Data; using SabreTools.Library.DatItems; using SabreTools.Library.IO; using SabreTools.Library.Tools; namespace SabreTools.Library.DatFiles { /// /// Represents parsing and writing of an Everdrive SMDB file /// internal class EverdriveSMDB : DatFile { /// /// Constructor designed for casting a base DatFile /// /// Parent DatFile to copy from public EverdriveSMDB(DatFile datFile) : base(datFile) { } /// /// Parse an Everdrive SMDB file and return all found games within /// /// Name of the file to be parsed /// Index ID for the DAT /// True if full pathnames are to be kept, false otherwise (default) protected override void ParseFile( // Standard Dat parsing string filename, int indexId, // Miscellaneous bool keep) { // Open a file reader Encoding enc = FileExtensions.GetEncoding(filename); StreamReader sr = new StreamReader(FileExtensions.TryOpenRead(filename), enc); while (!sr.EndOfStream) { string line = sr.ReadLine(); /* The gameinfo order is as follows 0 - SHA-256 1 - Machine Name/Filename 2 - SHA-1 3 - MD5 4 - CRC32 */ string[] gameinfo = line.Split('\t'); string[] fullname = gameinfo[1].Split('/'); Rom rom = new Rom { Name = gameinfo[1].Substring(fullname[0].Length + 1), Size = -1, // No size provided, but we don't want the size being 0 CRC = gameinfo[4], MD5 = gameinfo[3], SHA1 = gameinfo[2], SHA256 = gameinfo[0], ItemStatus = ItemStatus.None, MachineName = fullname[0], MachineDescription = fullname[0], IndexId = indexId, IndexSource = filename, }; // Now process and add the rom ParseAddHelper(rom); } sr.Dispose(); } /// /// Create and open an output file for writing direct from a dictionary /// /// Name of the file to write to /// True if blank roms should be skipped on output, false otherwise (default) /// True if the DAT was written correctly, false otherwise public override bool WriteToFile(string outfile, bool ignoreblanks = false) { try { Globals.Logger.User($"Opening file for writing: {outfile}"); FileStream fs = FileExtensions.TryCreate(outfile); // If we get back null for some reason, just log and return if (fs == null) { Globals.Logger.Warning($"File '{outfile}' could not be created for writing! Please check to see if the file is writable"); return false; } SeparatedValueWriter svw = new SeparatedValueWriter(fs, new UTF8Encoding(false)) { Quotes = false, Separator = '\t', VerifyFieldCount = true }; // Use a sorted list of games to output foreach (string key in Items.SortedKeys) { List roms = Items[key]; // Resolve the names in the block roms = DatItem.ResolveNames(roms); for (int index = 0; index < roms.Count; index++) { DatItem item = roms[index]; // There are apparently times when a null rom can skip by, skip them if (item.Name == null || item.MachineName == null) { Globals.Logger.Warning("Null rom found!"); continue; } // If we have a "null" game (created by DATFromDir or something similar), log it to file if (item.ItemType == ItemType.Rom && ((Rom)item).Size == -1 && ((Rom)item).CRC == "null") { Globals.Logger.Verbose($"Empty folder found: {item.MachineName}"); item.Name = (item.Name == "null" ? "-" : item.Name); ((Rom)item).Size = Constants.SizeZero; } WriteDatItem(svw, item, ignoreblanks); } } Globals.Logger.Verbose($"File written!{Environment.NewLine}"); svw.Dispose(); fs.Dispose(); } catch (Exception ex) { Globals.Logger.Error(ex.ToString()); return false; } return true; } /// /// Write out Game start using the supplied StreamWriter /// /// SeparatedValueWriter to output to /// DatItem object to be output /// True if the data was written, false on error private bool WriteDatItem(SeparatedValueWriter svw, DatItem datItem, bool ignoreblanks = false) { // If we are in ignore blanks mode AND we have a blank (0-size) rom, skip if (ignoreblanks && (datItem.ItemType == ItemType.Rom && ((datItem as Rom).Size == 0 || (datItem as Rom).Size == -1))) return true; try { // No game should start with a path separator datItem.MachineName = datItem.MachineName.TrimStart(Path.DirectorySeparatorChar); // Pre-process the item name ProcessItemName(datItem, true); // Build the state based on excluded fields switch (datItem.ItemType) { case ItemType.Rom: var rom = datItem as Rom; string[] fields = new string[] { rom.GetField(Field.SHA256, Header.ExcludeFields), $"{rom.GetField(Field.MachineName, Header.ExcludeFields)}/", rom.GetField(Field.Name, Header.ExcludeFields), rom.GetField(Field.SHA1, Header.ExcludeFields), rom.GetField(Field.MD5, Header.ExcludeFields), rom.GetField(Field.CRC, Header.ExcludeFields), }; svw.WriteValues(fields); break; } svw.Flush(); } catch (Exception ex) { Globals.Logger.Error(ex.ToString()); return false; } return true; } } }