using System; using System.Collections.Generic; using System.Text; using SabreTools.Library.Data; using SabreTools.Library.DatItems; using SabreTools.Library.Tools; #if MONO using System.IO; #else using Alphaleonis.Win32.Filesystem; using FileStream = System.IO.FileStream; using StreamReader = System.IO.StreamReader; using StreamWriter = System.IO.StreamWriter; #endif using NaturalSort; namespace SabreTools.Library.DatFiles { /// /// Represents parsing and writing of an AttractMode DAT /// internal class AttractMode : DatFile { /// /// Constructor designed for casting a base DatFile /// /// Parent DatFile to copy from public AttractMode(DatFile datFile) : base(datFile, cloneHeader: false) { } /// /// Parse an AttractMode DAT and return all found games within /// /// Name of the file to be parsed /// System ID for the DAT /// Source ID for the DAT /// True if full pathnames are to be kept, false otherwise (default) /// True if game names are sanitized, false otherwise (default) /// True if we should remove non-ASCII characters from output, false otherwise (default) public override void ParseFile( // Standard Dat parsing string filename, int sysid, int srcid, // Miscellaneous bool keep, bool clean, bool remUnicode) { // Open a file reader Encoding enc = Utilities.GetEncoding(filename); StreamReader sr = new StreamReader(Utilities.TryOpenRead(filename), enc); sr.ReadLine(); // Skip the first line since it's the header while (!sr.EndOfStream) { string line = sr.ReadLine(); /* The gameinfo order is as follows 0 - game name 1 - game description 2 - emulator name (filename) 3 - cloneof 4 - year 5 - manufacturer 6 - category 7 - players 8 - rotation 9 - control 10 - status 11 - displaycount 12 - displaytype 13 - alt romname 14 - alt title 15 - extra 16 - buttons */ string[] gameinfo = line.Split(';'); Rom rom = new Rom { Name = "-", Size = Constants.SizeZero, CRC = Constants.CRCZero, MD5 = Constants.MD5Zero, RIPEMD160 = Constants.RIPEMD160Zero, SHA1 = Constants.SHA1Zero, ItemStatus = ItemStatus.None, MachineName = gameinfo[0], MachineDescription = gameinfo[1], CloneOf = gameinfo[3], Year = gameinfo[4], Manufacturer = gameinfo[5], Comment = gameinfo[15], }; // Now process and add the rom ParseAddHelper(rom, clean, remUnicode); } 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: {0}", outfile); FileStream fs = Utilities.TryCreate(outfile); // If we get back null for some reason, just log and return if (fs == null) { Globals.Logger.Warning("File '{0}' could not be created for writing! Please check to see if the file is writable", outfile); return false; } StreamWriter sw = new StreamWriter(fs, new UTF8Encoding(false)); // Write out the header WriteHeader(sw); // Write out each of the machines and roms string lastgame = null; // Get a properly sorted set of keys List keys = Keys; keys.Sort(new NaturalComparer()); foreach (string key in keys) { List roms = this[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 new game, output the beginning of the new item if (lastgame == null || lastgame.ToLowerInvariant() != item.MachineName.ToLowerInvariant()) { WriteDatItem(sw, item, ignoreblanks); } // 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: {0}", item.MachineName); item.Name = (item.Name == "null" ? "-" : item.Name); ((Rom)item).Size = Constants.SizeZero; } // Set the new data to compare against lastgame = item.MachineName; } } Globals.Logger.Verbose("File written!" + Environment.NewLine); sw.Dispose(); fs.Dispose(); } catch (Exception ex) { Globals.Logger.Error(ex.ToString()); return false; } return true; } /// /// Write out DAT header using the supplied StreamWriter /// /// StreamWriter to output to /// True if the data was written, false on error private bool WriteHeader(StreamWriter sw) { try { string header = "#Title;Name;Emulator;CloneOf;Year;Manufacturer;Category;Players;Rotation;Control;Status;DisplayCount;DisplayType;AltRomname;AltTitle;Extra;Buttons\n"; // Write the header out sw.Write(header); sw.Flush(); } catch (Exception ex) { Globals.Logger.Error(ex.ToString()); return false; } return true; } /// /// Write out Game start using the supplied StreamWriter /// /// StreamWriter to output to /// DatItem object to be output /// True if blank roms should be skipped on output, false otherwise (default) /// True if the data was written, false on error private bool WriteDatItem(StreamWriter sw, DatItem rom, bool ignoreblanks = false) { // If we are in ignore blanks mode AND we have a blank (0-size) rom, skip if (ignoreblanks && (rom.ItemType == ItemType.Rom && (((Rom)rom).Size == 0 || ((Rom)rom).Size == -1))) { return true; } try { // No game should start with a path separator if (rom.MachineName.StartsWith(Path.DirectorySeparatorChar.ToString())) { rom.MachineName = rom.MachineName.Substring(1); } string state = (!ExcludeFields[(int)Field.MachineName] ? rom.MachineName : "") + ";" + (!ExcludeFields[(int)Field.Description] ? rom.MachineDescription : "") + ";" + FileName + ";" + (!ExcludeFields[(int)Field.CloneOf] ? rom.CloneOf : "") + ";" + (!ExcludeFields[(int)Field.Year] ? rom.Year : "") + ";" + (!ExcludeFields[(int)Field.Manufacturer] ? rom.Manufacturer : "") + ";" /* + rom.Category */ + ";" /* + rom.Players */ + ";" /* + rom.Rotation */ + ";" /* + rom.Control */ + ";" /* + rom.Status */ + ";" /* + rom.DisplayCount */ + ";" /* + rom.DisplayType */ + ";" /* + rom.AltRomname */ + ";" /* + rom.AltTitle */ + ";" + (!ExcludeFields[(int)Field.Comment] ? rom.Comment : "") + ";" /* + rom.Buttons */ + "\n"; sw.Write(state); sw.Flush(); } catch (Exception ex) { Globals.Logger.Error(ex.ToString()); return false; } return true; } } }