Files
SabreTools/SabreTools.DatFiles/DatFile.cs

885 lines
38 KiB
C#
Raw Normal View History

2020-12-10 10:58:00 -08:00
using System;
2022-11-03 16:46:53 -07:00
using System.Collections.Generic;
2020-12-10 10:58:00 -08:00
using System.IO;
2020-09-07 14:47:27 -07:00
using System.Xml.Serialization;
using Newtonsoft.Json;
2020-12-10 22:16:53 -08:00
using SabreTools.Core.Tools;
2020-12-10 10:58:00 -08:00
using SabreTools.DatItems;
2021-02-02 10:23:43 -08:00
using SabreTools.DatItems.Formats;
2024-03-04 23:56:05 -05:00
using SabreTools.Hashing;
2024-10-24 00:36:44 -04:00
using SabreTools.IO.Logging;
2016-10-26 22:10:47 -07:00
namespace SabreTools.DatFiles
{
/// <summary>
/// Represents a format-agnostic DAT
/// </summary>
2020-09-08 10:12:41 -07:00
[JsonObject("datfile"), XmlRoot("datfile")]
public abstract partial class DatFile
{
2020-07-31 14:04:10 -07:00
#region Fields
/// <summary>
/// Header values
/// </summary>
[JsonProperty("header"), XmlElement("header")]
public DatHeader Header { get; private set; } = new DatHeader();
/// <summary>
/// DatItems and related statistics
/// </summary>
[JsonProperty("items"), XmlElement("items")]
2024-03-19 18:40:24 -04:00
public ItemDictionary Items { get; private set; } = [];
2024-03-19 15:03:22 -04:00
/// <summary>
/// DatItems and related statistics
/// </summary>
[JsonProperty("items"), XmlElement("items")]
2024-03-19 18:40:24 -04:00
public ItemDictionaryDB ItemsDB { get; private set; } = new ItemDictionaryDB();
2024-03-19 15:03:22 -04:00
/// <summary>
/// DAT statistics
/// </summary>
[JsonIgnore, XmlIgnore]
public DatStatistics DatStatistics => Items.DatStatistics;
//public DatStatistics DatStatistics => ItemsDB.DatStatistics;
2025-01-09 06:14:01 -05:00
/// <summary>
/// List of supported types for writing
/// </summary>
public abstract ItemType[] SupportedTypes { get; }
#endregion
#region Logging
/// <summary>
/// Logging object
/// </summary>
[JsonIgnore, XmlIgnore]
2025-01-08 16:59:44 -05:00
protected Logger _logger;
#endregion
#region Constructors
/// <summary>
/// Create a new DatFile from an existing one
/// </summary>
/// <param name="datFile">DatFile to get the values from</param>
public DatFile(DatFile? datFile)
{
2025-01-08 16:59:44 -05:00
_logger = new Logger(this);
if (datFile != null)
{
2025-01-09 03:44:42 -05:00
Header = (DatHeader)datFile.Header.Clone();
2020-07-31 14:04:10 -07:00
Items = datFile.Items;
ItemsDB = datFile.ItemsDB;
}
}
/// <summary>
2020-08-27 20:56:50 -07:00
/// Fill the header values based on existing Header and path
/// </summary>
/// <param name="path">Path used for creating a name, if necessary</param>
/// <param name="bare">True if the date should be omitted from name and description, false otherwise</param>
public void FillHeaderFromPath(string path, bool bare)
{
2024-07-15 12:48:26 -04:00
// Get the header strings
string? name = Header.GetStringFieldValue(Models.Metadata.Header.NameKey);
string? description = Header.GetStringFieldValue(Models.Metadata.Header.DescriptionKey);
string? date = Header.GetStringFieldValue(Models.Metadata.Header.DateKey);
2020-08-27 20:56:50 -07:00
// If the description is defined but not the name, set the name from the description
2024-07-15 12:48:26 -04:00
if (string.IsNullOrEmpty(name) && !string.IsNullOrEmpty(description))
2020-08-27 20:56:50 -07:00
{
2025-01-09 08:44:33 -05:00
name = description + (bare ? string.Empty : $" ({date})");
2020-08-27 20:56:50 -07:00
}
// If the name is defined but not the description, set the description from the name
2024-07-15 12:48:26 -04:00
else if (!string.IsNullOrEmpty(name) && string.IsNullOrEmpty(description))
2020-08-27 20:56:50 -07:00
{
2025-01-09 08:44:33 -05:00
description = name + (bare ? string.Empty : $" ({date})");
2020-08-27 20:56:50 -07:00
}
// If neither the name or description are defined, set them from the automatic values
2024-07-15 12:48:26 -04:00
else if (string.IsNullOrEmpty(name) && string.IsNullOrEmpty(description))
2020-08-27 20:56:50 -07:00
{
string[] splitpath = path.TrimEnd(Path.DirectorySeparatorChar).Split(Path.DirectorySeparatorChar);
#if NETFRAMEWORK
2025-01-09 08:44:33 -05:00
name = splitpath[splitpath.Length - 1];
description = splitpath[splitpath.Length - 1] + (bare ? string.Empty : $" ({date})");
#else
2025-01-09 08:44:33 -05:00
name = splitpath[^1] + (bare ? string.Empty : $" ({date})");
description = splitpath[^1] + (bare ? string.Empty : $" ({date})");
#endif
2020-08-27 20:56:50 -07:00
}
2025-01-09 08:44:33 -05:00
// Trim both fields
name = name?.Trim();
description = description?.Trim();
// Set the fields back
Header.SetFieldValue<string?>(Models.Metadata.Header.NameKey, name);
Header.SetFieldValue<string?>(Models.Metadata.Header.DescriptionKey, description);
2020-08-27 20:56:50 -07:00
}
#endregion
#region Accessors
2024-03-19 18:40:24 -04:00
2025-01-08 17:11:52 -05:00
/// <summary>
/// Set the internal header
/// </summary>
/// <param name="datHeader">Replacement header to be used</param>
public void SetHeader(DatHeader datHeader)
{
// TODO: Figure out why clone loses data here
Header = datHeader;
2025-01-08 17:11:52 -05:00
}
2024-03-19 18:40:24 -04:00
#endregion
#region Item Dictionary Passthrough - Accessors
/// <summary>
/// Add a value to the file dictionary
/// </summary>
/// <param name="key">Key in the dictionary to add to</param>
/// <param name="value">Value to add to the dictionary</param>
public void Add(string key, DatItem value)
{
Items.Add(key, value);
}
/// <summary>
/// Add a range of values to the file dictionary
/// </summary>
/// <param name="key">Key in the dictionary to add to</param>
/// <param name="value">Value to add to the dictionary</param>
public void Add(string key, List<DatItem>? value)
{
Items.Add(key, value);
}
/// <summary>
/// Add a DatItem to the dictionary after checking
/// </summary>
/// <param name="item">Item data to check against</param>
/// <param name="statsOnly">True to only add item statistics while parsing, false otherwise</param>
/// <returns>The key for the item</returns>
public string AddItem(DatItem item, bool statsOnly)
{
return Items.AddItem(item, statsOnly);
}
/// <summary>
/// Add a DatItem to the dictionary after validation
/// </summary>
/// <param name="item">Item data to validate</param>
/// <param name="machineIndex">Index of the machine related to the item</param>
/// <param name="sourceIndex">Index of the source related to the item</param>
/// <param name="statsOnly">True to only add item statistics while parsing, false otherwise</param>
/// <returns>The index for the added item, -1 on error</returns>
public long AddItemDB(DatItem item, long machineIndex, long sourceIndex, bool statsOnly)
{
return ItemsDB.AddItem(item, machineIndex, sourceIndex, statsOnly);
}
/// <summary>
/// Add a machine, returning the insert index
/// </summary>
public long AddMachineDB(Machine machine)
{
return ItemsDB.AddMachine(machine);
}
/// <summary>
/// Add a source, returning the insert index
/// </summary>
public long AddSourceDB(Source source)
{
return ItemsDB.AddSource(source);
}
/// <summary>
/// Remove any keys that have null or empty values
/// </summary>
public void ClearEmpty()
{
Items.ClearEmpty();
ItemsDB.ClearEmpty();
}
/// <summary>
/// Remove all items marked for removal
/// </summary>
public void ClearMarked()
{
Items.ClearMarked();
ItemsDB.ClearMarked();
}
/// <summary>
/// Get the items associated with a bucket name
/// </summary>
public List<DatItem> GetItemsForBucket(string? bucketName, bool filter = false)
=> Items.GetItemsForBucket(bucketName, filter);
/// <summary>
/// Get the indices and items associated with a bucket name
/// </summary>
public Dictionary<long, DatItem> GetItemsForBucketDB(string? bucketName, bool filter = false)
=> ItemsDB.GetItemsForBucket(bucketName, filter);
/// <summary>
/// Get all machines and their indicies
/// </summary>
public IDictionary<long, Machine> GetMachinesDB()
=> ItemsDB.GetMachines();
/// <summary>
/// Remove a key from the file dictionary if it exists
/// </summary>
/// <param name="key">Key in the dictionary to remove</param>
public bool Remove(string key)
{
return Items.Remove(key);
}
/// <summary>
/// Reset the internal item dictionary
/// </summary>
public void ResetDictionary()
{
Items.Clear();
ItemsDB = new ItemDictionaryDB();
}
#endregion
#region Item Dictionary Passthrough - Bucketing
/// <summary>
/// Take the arbitrarily bucketed Files Dictionary and convert to one bucketed by a user-defined method
/// </summary>
/// <param name="bucketBy">ItemKey enum representing how to bucket the individual items</param>
/// <param name="dedupeType">Dedupe type that should be used</param>
/// <param name="lower">True if the key should be lowercased (default), false otherwise</param>
/// <param name="norename">True if games should only be compared on game and file name, false if system and source are counted</param>
public void BucketBy(ItemKey bucketBy, DedupeType dedupeType, bool lower = true, bool norename = true)
{
Items.BucketBy(bucketBy, dedupeType, lower, norename);
ItemsDB.BucketBy(bucketBy, dedupeType, lower, norename);
}
/// <summary>
/// List all duplicates found in a DAT based on a DatItem
/// </summary>
/// <param name="datItem">Item to try to match</param>
/// <param name="sorted">True if the DAT is already sorted accordingly, false otherwise (default)</param>
/// <returns>List of matched DatItem objects</returns>
public List<DatItem> GetDuplicates(DatItem datItem, bool sorted = false)
=> Items.GetDuplicates(datItem, sorted);
/// <summary>
/// List all duplicates found in a DAT based on a DatItem
/// </summary>
/// <param name="datItem">Item to try to match</param>
/// <param name="sorted">True if the DAT is already sorted accordingly, false otherwise (default)</param>
/// <returns>List of matched DatItem objects</returns>
public Dictionary<long, DatItem> GetDuplicatesDB(DatItem datItem, bool sorted = false)
=> ItemsDB.GetDuplicates(datItem, sorted);
/// <summary>
/// List all duplicates found in a DAT based on a DatItem
/// </summary>
/// <param name="datItem">Item to try to match</param>
/// <param name="sorted">True if the DAT is already sorted accordingly, false otherwise (default)</param>
/// <returns>List of matched DatItem objects</returns>
public Dictionary<long, DatItem> GetDuplicatesDB(KeyValuePair<long, DatItem> datItem, bool sorted = false)
=> ItemsDB.GetDuplicates(datItem, sorted);
/// <summary>
/// Check if a DAT contains the given DatItem
/// </summary>
/// <param name="datItem">Item to try to match</param>
/// <param name="sorted">True if the DAT is already sorted accordingly, false otherwise (default)</param>
/// <returns>True if it contains the rom, false otherwise</returns>
public bool HasDuplicates(DatItem datItem, bool sorted = false)
=> Items.HasDuplicates(datItem, sorted);
/// <summary>
/// Check if a DAT contains the given DatItem
/// </summary>
/// <param name="datItem">Item to try to match</param>
/// <param name="sorted">True if the DAT is already sorted accordingly, false otherwise (default)</param>
/// <returns>True if it contains the rom, false otherwise</returns>
public bool HasDuplicates(KeyValuePair<long, DatItem> datItem, bool sorted = false)
=> ItemsDB.HasDuplicates(datItem, sorted);
#endregion
2020-12-10 10:58:00 -08:00
#region Parsing
/// <summary>
/// Parse DatFile and return all found games and roms within
/// </summary>
/// <param name="filename">Name of the file to be parsed</param>
/// <param name="indexId">Index ID for the DAT</param>
/// <param name="keep">True if full pathnames are to be kept, false otherwise (default)</param>
/// <param name="statsOnly">True to only add item statistics while parsing, false otherwise</param>
2020-12-10 10:58:00 -08:00
/// <param name="throwOnError">True if the error that is thrown should be thrown back to the caller, false otherwise</param>
public abstract void ParseFile(string filename, int indexId, bool keep, bool statsOnly = false, bool throwOnError = false);
2020-12-10 10:58:00 -08:00
#endregion
2020-12-10 11:28:11 -08:00
#region Writing
/// <summary>
/// Create and open an output file for writing direct from a dictionary
/// </summary>
/// <param name="outfile">Name of the file to write to</param>
/// <param name="ignoreblanks">True if blank roms should be skipped on output, false otherwise (default)</param>
/// <param name="throwOnError">True if the error that is thrown should be thrown back to the caller, false otherwise</param>
/// <returns>True if the DAT was written correctly, false otherwise</returns>
public abstract bool WriteToFile(string outfile, bool ignoreblanks = false, bool throwOnError = false);
/// <summary>
/// Create and open an output file for writing direct from a dictionary
/// </summary>
/// <param name="outfile">Name of the file to write to</param>
/// <param name="ignoreblanks">True if blank roms should be skipped on output, false otherwise (default)</param>
/// <param name="throwOnError">True if the error that is thrown should be thrown back to the caller, false otherwise</param>
/// <returns>True if the DAT was written correctly, false otherwise</returns>
public abstract bool WriteToFileDB(string outfile, bool ignoreblanks = false, bool throwOnError = false);
2020-12-10 11:28:11 -08:00
/// <summary>
/// Process an item and correctly set the item name
/// </summary>
/// <param name="item">DatItem to update</param>
/// <param name="forceRemoveQuotes">True if the Quotes flag should be ignored, false otherwise</param>
2025-01-09 10:13:47 -05:00
/// <param name="forceRomName">True if the UseRomName should be always on, false otherwise</param>
protected internal void ProcessItemName(DatItem item, Machine? machine, bool forceRemoveQuotes, bool forceRomName)
2020-12-10 11:28:11 -08:00
{
2025-01-09 09:43:35 -05:00
// Get the relevant processing values
2025-01-09 09:52:47 -05:00
bool quotes = forceRemoveQuotes ? false : Header.GetBoolFieldValue(DatHeader.QuotesKey) ?? false;
bool useRomName = forceRomName ? true : Header.GetBoolFieldValue(DatHeader.UseRomNameKey) ?? false;
2020-12-10 11:28:11 -08:00
// Create the full Prefix
string pre = Header.GetStringFieldValue(DatHeader.PrefixKey) + (quotes ? "\"" : string.Empty);
pre = FormatPrefixPostfix(item, machine, pre);
// Create the full Postfix
string post = (quotes ? "\"" : string.Empty) + Header.GetStringFieldValue(DatHeader.PostfixKey);
post = FormatPrefixPostfix(item, machine, post);
// Get the name to update
2025-01-09 16:25:53 -05:00
string? name = (useRomName
? item.GetName()
2025-01-09 10:13:47 -05:00
: machine?.GetStringFieldValue(Models.Metadata.Machine.NameKey)) ?? string.Empty;
// If we're in Depot mode, take care of that instead
2024-07-15 12:48:26 -04:00
var outputDepot = Header.GetFieldValue<DepotInformation?>(DatHeader.OutputDepotKey);
if (outputDepot?.IsActive == true)
{
if (item is Disk disk)
{
// We can only write out if there's a SHA-1
2024-10-24 04:21:13 -04:00
string? sha1 = disk.GetStringFieldValue(Models.Metadata.Disk.SHA1Key);
if (!string.IsNullOrEmpty(sha1))
{
2024-10-24 04:21:13 -04:00
name = Utilities.GetDepotPath(sha1, outputDepot.Depth)?.Replace('\\', '/');
item.SetName($"{pre}{name}{post}");
}
}
2025-01-09 16:25:53 -05:00
else if (item is DatItems.Formats.File file)
{
// We can only write out if there's a SHA-1
string? sha1 = file.SHA1;
if (!string.IsNullOrEmpty(sha1))
{
name = Utilities.GetDepotPath(sha1, outputDepot.Depth)?.Replace('\\', '/');
item.SetName($"{pre}{name}{post}");
}
}
else if (item is Media media)
{
// We can only write out if there's a SHA-1
2024-10-24 04:21:13 -04:00
string? sha1 = media.GetStringFieldValue(Models.Metadata.Media.SHA1Key);
if (!string.IsNullOrEmpty(sha1))
{
2024-10-24 04:21:13 -04:00
name = Utilities.GetDepotPath(sha1, outputDepot.Depth)?.Replace('\\', '/');
item.SetName($"{pre}{name}{post}");
}
}
else if (item is Rom rom)
{
// We can only write out if there's a SHA-1
2024-10-24 04:21:13 -04:00
string? sha1 = rom.GetStringFieldValue(Models.Metadata.Rom.SHA1Key);
if (!string.IsNullOrEmpty(sha1))
{
2024-10-24 04:21:13 -04:00
name = Utilities.GetDepotPath(sha1, outputDepot.Depth)?.Replace('\\', '/');
item.SetName($"{pre}{name}{post}");
}
}
return;
}
2024-07-15 12:48:26 -04:00
string? replaceExtension = Header.GetStringFieldValue(DatHeader.ReplaceExtensionKey);
bool? removeExtension = Header.GetBoolFieldValue(DatHeader.RemoveExtensionKey);
if (!string.IsNullOrEmpty(replaceExtension) || removeExtension == true)
{
2024-07-15 12:48:26 -04:00
if (removeExtension == true)
Header.SetFieldValue<string?>(DatHeader.ReplaceExtensionKey, string.Empty);
string? dir = Path.GetDirectoryName(name);
if (dir != null)
{
dir = dir.TrimStart(Path.DirectorySeparatorChar);
2024-07-15 12:48:26 -04:00
name = Path.Combine(dir, Path.GetFileNameWithoutExtension(name) + replaceExtension);
}
}
2024-07-15 12:48:26 -04:00
string? addExtension = Header.GetStringFieldValue(DatHeader.AddExtensionKey);
if (!string.IsNullOrEmpty(addExtension))
name += addExtension;
2025-01-09 09:52:47 -05:00
if (useRomName && Header.GetBoolFieldValue(DatHeader.GameNameKey) == true)
2025-01-09 10:13:47 -05:00
name = Path.Combine(machine?.GetStringFieldValue(Models.Metadata.Machine.NameKey) ?? string.Empty, name);
// Now assign back the formatted name
name = $"{pre}{name}{post}";
2025-01-09 09:52:47 -05:00
if (useRomName)
item.SetName(name);
else if (machine != null)
machine.SetFieldValue<string?>(Models.Metadata.Machine.NameKey, name);
2025-01-09 09:52:47 -05:00
}
/// <summary>
/// Format a prefix or postfix string
/// </summary>
/// <param name="item">DatItem to create a prefix/postfix for</param>
/// <param name="machine">Machine to get information from</param>
/// <param name="fix">Prefix or postfix pattern to populate</param>
/// <returns>Sanitized string representing the postfix or prefix</returns>
protected internal static string FormatPrefixPostfix(DatItem item, Machine? machine, string fix)
2025-01-09 09:52:47 -05:00
{
// Initialize strings
string? type = item.GetStringFieldValue(Models.Metadata.DatItem.TypeKey);
string
game = machine?.GetStringFieldValue(Models.Metadata.Machine.NameKey) ?? string.Empty,
manufacturer = machine?.GetStringFieldValue(Models.Metadata.Machine.ManufacturerKey) ?? string.Empty,
publisher = machine?.GetStringFieldValue(Models.Metadata.Machine.PublisherKey) ?? string.Empty,
category = machine?.GetStringFieldValue(Models.Metadata.Machine.CategoryKey) ?? string.Empty,
name = item.GetName() ?? type.AsEnumValue<ItemType>().AsStringValue() ?? string.Empty,
crc = string.Empty,
md2 = string.Empty,
md4 = string.Empty,
md5 = string.Empty,
sha1 = string.Empty,
sha256 = string.Empty,
sha384 = string.Empty,
sha512 = string.Empty,
size = string.Empty,
spamsum = string.Empty;
// Ensure we have the proper values for replacement
if (item is Disk disk)
{
md5 = disk.GetStringFieldValue(Models.Metadata.Disk.MD5Key) ?? string.Empty;
sha1 = disk.GetStringFieldValue(Models.Metadata.Disk.SHA1Key) ?? string.Empty;
}
else if (item is DatItems.Formats.File file)
{
2025-01-09 16:25:53 -05:00
name = $"{file.Id}.{file.Extension}";
size = file.Size.ToString() ?? string.Empty;
2025-01-09 09:52:47 -05:00
crc = file.CRC ?? string.Empty;
md5 = file.MD5 ?? string.Empty;
sha1 = file.SHA1 ?? string.Empty;
sha256 = file.SHA256 ?? string.Empty;
}
else if (item is Media media)
{
md5 = media.GetStringFieldValue(Models.Metadata.Media.MD5Key) ?? string.Empty;
sha1 = media.GetStringFieldValue(Models.Metadata.Media.SHA1Key) ?? string.Empty;
sha256 = media.GetStringFieldValue(Models.Metadata.Media.SHA256Key) ?? string.Empty;
spamsum = media.GetStringFieldValue(Models.Metadata.Media.SpamSumKey) ?? string.Empty;
}
else if (item is Rom rom)
{
crc = rom.GetStringFieldValue(Models.Metadata.Rom.CRCKey) ?? string.Empty;
md2 = rom.GetStringFieldValue(Models.Metadata.Rom.MD2Key) ?? string.Empty;
md4 = rom.GetStringFieldValue(Models.Metadata.Rom.MD4Key) ?? string.Empty;
md5 = rom.GetStringFieldValue(Models.Metadata.Rom.MD5Key) ?? string.Empty;
sha1 = rom.GetStringFieldValue(Models.Metadata.Rom.SHA1Key) ?? string.Empty;
sha256 = rom.GetStringFieldValue(Models.Metadata.Rom.SHA256Key) ?? string.Empty;
sha384 = rom.GetStringFieldValue(Models.Metadata.Rom.SHA384Key) ?? string.Empty;
sha512 = rom.GetStringFieldValue(Models.Metadata.Rom.SHA512Key) ?? string.Empty;
size = rom.GetInt64FieldValue(Models.Metadata.Rom.SizeKey).ToString() ?? string.Empty;
spamsum = rom.GetStringFieldValue(Models.Metadata.Rom.SpamSumKey) ?? string.Empty;
}
// Now do bulk replacement where possible
fix = fix
.Replace("%game%", game)
.Replace("%machine%", game)
.Replace("%name%", name)
.Replace("%manufacturer%", manufacturer)
.Replace("%publisher%", publisher)
.Replace("%category%", category)
.Replace("%crc%", crc)
.Replace("%md2%", md2)
.Replace("%md4%", md4)
.Replace("%md5%", md5)
.Replace("%sha1%", sha1)
.Replace("%sha256%", sha256)
.Replace("%sha384%", sha384)
.Replace("%sha512%", sha512)
.Replace("%size%", size)
.Replace("%spamsum%", spamsum);
return fix;
}
2020-12-10 11:28:11 -08:00
/// <summary>
/// Process any DatItems that are "null", usually created from directory population
/// </summary>
/// <param name="item">DatItem to check for "null" status</param>
2025-01-09 16:53:55 -05:00
/// <returns>Cleaned DatItem, if possible</returns>
protected internal static DatItem ProcessNullifiedItem(DatItem item)
2020-12-10 11:28:11 -08:00
{
// If we don't have a Rom, we can ignore it
if (item is not Rom rom)
return item;
2020-12-10 11:28:11 -08:00
2025-01-09 09:21:16 -05:00
// If the item has a size
if (rom.GetInt64FieldValue(Models.Metadata.Rom.SizeKey) != null)
return rom;
// If the item CRC isn't "null"
if (rom.GetStringFieldValue(Models.Metadata.Rom.CRCKey) != "null")
return rom;
2024-03-19 15:03:22 -04:00
// If the Rom has "null" characteristics, ensure all fields
2025-01-09 09:21:16 -05:00
rom.SetName(rom.GetName() == "null" ? "-" : rom.GetName());
rom.SetFieldValue<string?>(Models.Metadata.Rom.SizeKey, Constants.SizeZero.ToString());
rom.SetFieldValue<string?>(Models.Metadata.Rom.CRCKey,
rom.GetStringFieldValue(Models.Metadata.Rom.CRCKey) == "null" ? ZeroHash.CRC32Str : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.MD2Key,
rom.GetStringFieldValue(Models.Metadata.Rom.MD2Key) == "null" ? ZeroHash.GetString(HashType.MD2) : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.MD4Key,
rom.GetStringFieldValue(Models.Metadata.Rom.MD4Key) == "null" ? ZeroHash.GetString(HashType.MD4) : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.MD5Key,
rom.GetStringFieldValue(Models.Metadata.Rom.MD5Key) == "null" ? ZeroHash.MD5Str : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.SHA1Key,
rom.GetStringFieldValue(Models.Metadata.Rom.SHA1Key) == "null" ? ZeroHash.SHA1Str : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.SHA256Key,
rom.GetStringFieldValue(Models.Metadata.Rom.SHA256Key) == "null" ? ZeroHash.SHA256Str : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.SHA384Key,
rom.GetStringFieldValue(Models.Metadata.Rom.SHA384Key) == "null" ? ZeroHash.SHA384Str : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.SHA512Key,
rom.GetStringFieldValue(Models.Metadata.Rom.SHA512Key) == "null" ? ZeroHash.SHA512Str : null);
rom.SetFieldValue<string?>(Models.Metadata.Rom.SpamSumKey,
rom.GetStringFieldValue(Models.Metadata.Rom.SpamSumKey) == "null" ? ZeroHash.SpamSumStr : null);
2024-03-19 15:03:22 -04:00
return rom;
2024-03-19 15:03:22 -04:00
}
/// <summary>
2022-11-03 16:46:53 -07:00
/// Return list of required fields missing from a DatItem
/// </summary>
2022-11-03 16:46:53 -07:00
/// <returns>List of missing required fields, null or empty if none were found</returns>
protected internal virtual List<string>? GetMissingRequiredFields(DatItem datItem) => null;
2020-12-10 11:28:11 -08:00
/// <summary>
2025-01-09 20:10:56 -05:00
/// Get if a list contains any writable items
2020-12-10 11:28:11 -08:00
/// </summary>
/// <param name="datItems">DatItems to check</param>
2025-01-09 20:10:56 -05:00
/// <returns>True if the list contains at least one writable item, false otherwise</returns>
/// <remarks>Empty list are kept with this</remarks>
protected internal bool ContainsWritable(List<DatItem> datItems)
2020-12-10 11:28:11 -08:00
{
2025-01-09 20:10:56 -05:00
// Empty list are considered writable
if (datItems.Count == 0)
2020-12-10 11:28:11 -08:00
return true;
foreach (DatItem datItem in datItems)
{
2024-07-15 12:48:26 -04:00
ItemType itemType = datItem.GetStringFieldValue(Models.Metadata.DatItem.TypeKey).AsEnumValue<ItemType>();
2025-01-09 06:14:01 -05:00
if (Array.Exists(SupportedTypes, t => t == itemType))
2020-12-10 11:28:11 -08:00
return true;
}
return false;
}
/// <summary>
/// Resolve name duplicates in an arbitrary set of DatItems based on the supplied information
/// </summary>
2025-01-10 10:30:49 -05:00
/// <param name="datItems">List of DatItem objects representing the items to be merged</param>
/// <returns>A List of DatItem objects representing the renamed items</returns>
2025-01-10 10:30:49 -05:00
protected internal List<DatItem> ResolveNames(List<DatItem> datItems)
{
2025-01-09 20:45:26 -05:00
// Ignore empty lists
2025-01-10 10:30:49 -05:00
if (datItems.Count == 0)
2025-01-09 20:45:26 -05:00
return [];
// Create the output list
List<DatItem> output = [];
// First we want to make sure the list is in alphabetical order
2025-01-10 10:30:49 -05:00
DatFileTool.Sort(ref datItems, true);
// Now we want to loop through and check names
DatItem? lastItem = null;
string? lastrenamed = null;
int lastid = 0;
2025-01-10 10:30:49 -05:00
for (int i = 0; i < datItems.Count; i++)
{
2025-01-10 10:30:49 -05:00
DatItem datItem = datItems[i];
// If we have the first item, we automatically add it
if (lastItem == null)
{
output.Add(datItem);
lastItem = datItem;
continue;
}
// Get the last item name, if applicable
string lastItemName = lastItem.GetName()
?? lastItem.GetStringFieldValue(Models.Metadata.DatItem.TypeKey).AsEnumValue<ItemType>().AsStringValue()
?? string.Empty;
// Get the current item name, if applicable
string datItemName = datItem.GetName()
?? datItem.GetStringFieldValue(Models.Metadata.DatItem.TypeKey).AsEnumValue<ItemType>().AsStringValue()
?? string.Empty;
// If the current item exactly matches the last item, then we don't add it
#if NET20 || NET35
if ((datItem.GetDuplicateStatus(lastItem) & DupeType.All) != 0)
#else
if (datItem.GetDuplicateStatus(lastItem).HasFlag(DupeType.All))
#endif
{
2025-01-08 16:59:44 -05:00
_logger.Verbose($"Exact duplicate found for '{datItemName}'");
continue;
}
// If the current name matches the previous name, rename the current item
else if (datItemName == lastItemName)
{
2025-01-08 16:59:44 -05:00
_logger.Verbose($"Name duplicate found for '{datItemName}'");
2025-01-09 05:52:33 -05:00
// Get the duplicate suffix
datItemName += datItem.GetDuplicateSuffix();
lastrenamed ??= datItemName;
// If we have a conflict with the last renamed item, do the right thing
if (datItemName == lastrenamed)
{
lastrenamed = datItemName;
datItemName += (lastid == 0 ? string.Empty : "_" + lastid);
lastid++;
}
// If we have no conflict, then we want to reset the lastrenamed and id
else
{
lastrenamed = null;
lastid = 0;
}
// Set the item name back to the datItem
datItem.SetName(datItemName);
output.Add(datItem);
}
// Otherwise, we say that we have a valid named file
else
{
output.Add(datItem);
lastItem = datItem;
lastrenamed = null;
lastid = 0;
}
}
// One last sort to make sure this is ordered
DatFileTool.Sort(ref output, true);
return output;
}
/// <summary>
/// Resolve name duplicates in an arbitrary set of DatItems based on the supplied information
/// </summary>
/// <param name="mappings">List of item ID to DatItem mappings representing the items to be merged</param>
/// <returns>A List of DatItem objects representing the renamed items</returns>
protected internal List<KeyValuePair<long, DatItem>> ResolveNamesDB(List<KeyValuePair<long, DatItem>> mappings)
{
2025-01-09 20:45:26 -05:00
// Ignore empty lists
if (mappings.Count == 0)
return [];
// Create the output dict
List<KeyValuePair<long, DatItem>> output = [];
// First we want to make sure the list is in alphabetical order
DatFileTool.SortDB(ref mappings, true);
// Now we want to loop through and check names
DatItem? lastItem = null;
string? lastrenamed = null;
int lastid = 0;
foreach (var datItem in mappings)
{
// If we have the first item, we automatically add it
if (lastItem == null)
{
output.Add(datItem);
lastItem = datItem.Value;
continue;
}
// Get the last item name, if applicable
string lastItemName = lastItem.GetName()
?? lastItem.GetStringFieldValue(Models.Metadata.DatItem.TypeKey).AsEnumValue<ItemType>().AsStringValue()
?? string.Empty;
// Get the current item name, if applicable
string datItemName = datItem.Value.GetName()
?? datItem.Value.GetStringFieldValue(Models.Metadata.DatItem.TypeKey).AsEnumValue<ItemType>().AsStringValue()
?? string.Empty;
// If the current item exactly matches the last item, then we don't add it
#if NET20 || NET35
if ((datItem.Value.GetDuplicateStatus(lastItem) & DupeType.All) != 0)
#else
if (datItem.Value.GetDuplicateStatus(lastItem).HasFlag(DupeType.All))
#endif
{
2025-01-08 16:59:44 -05:00
_logger.Verbose($"Exact duplicate found for '{datItemName}'");
continue;
}
// If the current name matches the previous name, rename the current item
else if (datItemName == lastItemName)
{
2025-01-08 16:59:44 -05:00
_logger.Verbose($"Name duplicate found for '{datItemName}'");
2025-01-09 05:52:33 -05:00
// Get the duplicate suffix
datItemName += datItem.Value.GetDuplicateSuffix();
lastrenamed ??= datItemName;
// If we have a conflict with the last renamed item, do the right thing
if (datItemName == lastrenamed)
{
lastrenamed = datItemName;
datItemName += (lastid == 0 ? string.Empty : "_" + lastid);
lastid++;
}
// If we have no conflict, then we want to reset the lastrenamed and id
else
{
lastrenamed = null;
lastid = 0;
}
// Set the item name back to the datItem
datItem.Value.SetName(datItemName);
output.Add(datItem);
}
// Otherwise, we say that we have a valid named file
else
{
output.Add(datItem);
lastItem = datItem.Value;
lastrenamed = null;
lastid = 0;
}
}
// One last sort to make sure this is ordered
DatFileTool.SortDB(ref output, true);
return output;
}
2020-12-10 11:28:11 -08:00
/// <summary>
/// Get if an item should be ignored on write
/// </summary>
/// <param name="datItem">DatItem to check</param>
/// <param name="ignoreBlanks">True if blank roms should be skipped on output, false otherwise</param>
2020-12-10 11:28:11 -08:00
/// <returns>True if the item should be skipped on write, false otherwise</returns>
protected internal bool ShouldIgnore(DatItem? datItem, bool ignoreBlanks)
2020-12-10 11:28:11 -08:00
{
2022-11-03 16:29:06 -07:00
// If this is invoked with a null DatItem, we ignore
if (datItem == null)
{
2025-01-09 10:22:28 -05:00
_logger.Verbose($"Item was skipped because it was null");
2022-11-03 16:29:06 -07:00
return true;
}
2020-12-10 11:28:11 -08:00
// If the item is supposed to be removed, we ignore
if (datItem.GetBoolFieldValue(DatItem.RemoveKey) == true)
{
string itemString = JsonConvert.SerializeObject(datItem, Formatting.None);
2025-01-09 10:22:28 -05:00
_logger.Verbose($"Item '{itemString}' was skipped because it was marked for removal");
2020-12-10 11:28:11 -08:00
return true;
}
2020-12-10 11:28:11 -08:00
// If we have the Blank dat item, we ignore
2024-03-10 16:49:07 -04:00
if (datItem is Blank)
{
string itemString = JsonConvert.SerializeObject(datItem, Formatting.None);
2025-01-09 10:22:28 -05:00
_logger.Verbose($"Item '{itemString}' was skipped because it was of type 'Blank'");
2020-12-10 11:28:11 -08:00
return true;
}
2020-12-10 11:28:11 -08:00
// If we're ignoring blanks and we have a Rom
2024-03-10 16:49:07 -04:00
if (ignoreBlanks && datItem is Rom rom)
2020-12-10 11:28:11 -08:00
{
// If we have a 0-size or blank rom, then we ignore
long? size = rom.GetInt64FieldValue(Models.Metadata.Rom.SizeKey);
if (size == 0 || size == null)
{
string itemString = JsonConvert.SerializeObject(datItem, Formatting.None);
2025-01-09 10:22:28 -05:00
_logger.Verbose($"Item '{itemString}' was skipped because it had an invalid size");
2020-12-10 11:28:11 -08:00
return true;
}
2020-12-10 11:28:11 -08:00
}
// If we have an item type not in the list of supported values
2024-12-06 13:57:48 -05:00
ItemType itemType = datItem.GetStringFieldValue(Models.Metadata.DatItem.TypeKey).AsEnumValue<ItemType>();
2025-01-09 06:14:01 -05:00
if (!Array.Exists(SupportedTypes, t => t == itemType))
{
string itemString = JsonConvert.SerializeObject(datItem, Formatting.None);
2025-01-09 10:22:28 -05:00
_logger.Verbose($"Item '{itemString}' was skipped because it was not supported for output");
2020-12-10 11:28:11 -08:00
return true;
}
2020-12-10 11:28:11 -08:00
// If we have an item with missing required fields
2024-03-05 23:41:00 -05:00
List<string>? missingFields = GetMissingRequiredFields(datItem);
2022-12-22 09:27:02 -08:00
if (missingFields != null && missingFields.Count != 0)
{
2024-03-19 15:03:22 -04:00
string itemString = JsonConvert.SerializeObject(datItem, Formatting.None);
#if NET20 || NET35
2025-01-09 10:22:28 -05:00
_logger.Verbose($"Item '{itemString}' was skipped because it was missing required fields: {string.Join(", ", [.. missingFields])}");
2024-03-19 15:03:22 -04:00
#else
2025-01-09 10:22:28 -05:00
_logger.Verbose($"Item '{itemString}' was skipped because it was missing required fields: {string.Join(", ", missingFields)}");
2024-03-19 15:03:22 -04:00
#endif
return true;
}
return false;
}
2020-12-10 11:28:11 -08:00
#endregion
}
}