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, 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 the DAT was written correctly, false otherwise public bool WriteToFile(string outfile) { 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()) { WriteStartGame(sw, item); } // If we have a "null" game (created by DATFromDir or something similar), log it to file if (item.Type == 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 the data was written, false on error private bool WriteStartGame(StreamWriter sw, DatItem rom) { try { // No game should start with a path separator if (rom.MachineName.StartsWith(Path.DirectorySeparatorChar.ToString())) { rom.MachineName = rom.MachineName.Substring(1); } string state = rom.MachineName + ";" + rom.MachineDescription + ";" + FileName + ";" + rom.CloneOf + ";" + rom.Year + ";" + rom.Manufacturer + ";" /* + rom.Category */ + ";" /* + rom.Players */ + ";" /* + rom.Rotation */ + ";" /* + rom.Control */ + ";" /* + rom.Status */ + ";" /* + rom.DisplayCount */ + ";" /* + rom.DisplayType */ + ";" /* + rom.AltRomname */ + ";" /* + rom.AltTitle */ + ";" + rom.Comment + ";" /* + rom.Buttons */ + "\n"; sw.Write(state); sw.Flush(); } catch (Exception ex) { Globals.Logger.Error(ex.ToString()); return false; } return true; } } }