mirror of
https://github.com/claunia/SabreTools.git
synced 2025-12-16 19:14:27 +00:00
Add new tool: HashSplit
This tool will simply split a DAT by the "best" available hash in order of SHA-1, MD5, and CRC/No-Hash. The files are output to the same directory as the original file. Requested by Obiwantje.
This commit is contained in:
Matt Nadareski
234
HashSplit/HashSplit.cs
Normal file
234
HashSplit/HashSplit.cs
Normal file
@@ -0,0 +1,234 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
using SabreTools.Helper;
|
||||
|
||||
namespace SabreTools
|
||||
{
|
||||
public class HashSplit
|
||||
{
|
||||
// Internal variables
|
||||
string _filename;
|
||||
Logger _logger;
|
||||
|
||||
/// <summary>
|
||||
/// Create a HashSplit object
|
||||
/// </summary>
|
||||
/// <param name="filename">The name of the file to be split</param>
|
||||
/// <param name="logger">Logger object for file and console output</param>
|
||||
public HashSplit(string filename, Logger logger)
|
||||
{
|
||||
_filename = filename;
|
||||
_logger = logger;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Start help or use supplied parameters
|
||||
/// </summary>
|
||||
/// <param name="args">String array representing command line parameters</param>
|
||||
public static void Main(string[] args)
|
||||
{
|
||||
Console.Clear();
|
||||
|
||||
// Credits take precidence over all
|
||||
if ((new List<string>(args)).Contains("--credits"))
|
||||
{
|
||||
Build.Credits();
|
||||
return;
|
||||
}
|
||||
|
||||
Logger logger = new Logger(true, "hashsplit.log");
|
||||
logger.Start();
|
||||
|
||||
// First things first, take care of all of the arguments that this could have
|
||||
string filename = "";
|
||||
foreach (string arg in args)
|
||||
{
|
||||
switch (arg)
|
||||
{
|
||||
case "-h":
|
||||
case "-?":
|
||||
case "--help":
|
||||
Build.Help();
|
||||
logger.Close();
|
||||
return;
|
||||
default:
|
||||
filename = arg;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// If there's no inputs, show the help
|
||||
if (filename == "")
|
||||
{
|
||||
Build.Help();
|
||||
logger.Close();
|
||||
return;
|
||||
}
|
||||
|
||||
// Output the title
|
||||
Build.Start("HashSplit");
|
||||
|
||||
// Verify the input file
|
||||
filename = filename.Replace("\"", "");
|
||||
if (!File.Exists(filename))
|
||||
{
|
||||
logger.Error(filename + " is not a valid file!");
|
||||
Console.WriteLine();
|
||||
Build.Help();
|
||||
return;
|
||||
}
|
||||
|
||||
// If so, run the program
|
||||
filename = Path.GetFullPath(filename);
|
||||
HashSplit hs = new HashSplit(filename, logger);
|
||||
hs.Split();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Split the DAT into parts by best-available hash data
|
||||
/// </summary>
|
||||
/// <returns>True if the DATs were output, false otherwise</returns>
|
||||
public bool Split()
|
||||
{
|
||||
// Get the file data to be split
|
||||
OutputFormat outputFormat = RomManipulation.GetOutputFormat(_filename);
|
||||
DatData datdata = new DatData
|
||||
{
|
||||
Roms = new Dictionary<string, List<RomData>>(),
|
||||
};
|
||||
datdata = RomManipulation.Parse(_filename, 0, 0, datdata, _logger, true);
|
||||
|
||||
// Create each of the respective output DATs
|
||||
DatData sha1 = new DatData
|
||||
{
|
||||
Name = datdata.Name + " (SHA-1)",
|
||||
Description = datdata.Description + " (SHA-1)",
|
||||
Category = datdata.Category,
|
||||
Version = datdata.Version,
|
||||
Date = datdata.Date,
|
||||
Author = datdata.Author,
|
||||
Email = datdata.Email,
|
||||
Homepage = datdata.Homepage,
|
||||
Url = datdata.Url,
|
||||
Comment = datdata.Comment,
|
||||
Header = datdata.Header,
|
||||
Type = datdata.Type,
|
||||
ForceMerging = datdata.ForceMerging,
|
||||
ForceNodump = datdata.ForceNodump,
|
||||
ForcePacking = datdata.ForcePacking,
|
||||
OutputFormat = outputFormat,
|
||||
MergeRoms = datdata.MergeRoms,
|
||||
Roms = new Dictionary<string, List<RomData>>(),
|
||||
};
|
||||
DatData md5 = new DatData
|
||||
{
|
||||
Name = datdata.Name + " (MD5)",
|
||||
Description = datdata.Description + " (MD5)",
|
||||
Category = datdata.Category,
|
||||
Version = datdata.Version,
|
||||
Date = datdata.Date,
|
||||
Author = datdata.Author,
|
||||
Email = datdata.Email,
|
||||
Homepage = datdata.Homepage,
|
||||
Url = datdata.Url,
|
||||
Comment = datdata.Comment,
|
||||
Header = datdata.Header,
|
||||
Type = datdata.Type,
|
||||
ForceMerging = datdata.ForceMerging,
|
||||
ForceNodump = datdata.ForceNodump,
|
||||
ForcePacking = datdata.ForcePacking,
|
||||
OutputFormat = outputFormat,
|
||||
MergeRoms = datdata.MergeRoms,
|
||||
Roms = new Dictionary<string, List<RomData>>(),
|
||||
};
|
||||
DatData crc = new DatData
|
||||
{
|
||||
Name = datdata.Name + " (CRC and None)",
|
||||
Description = datdata.Description + " (CRC and None)",
|
||||
Category = datdata.Category,
|
||||
Version = datdata.Version,
|
||||
Date = datdata.Date,
|
||||
Author = datdata.Author,
|
||||
Email = datdata.Email,
|
||||
Homepage = datdata.Homepage,
|
||||
Url = datdata.Url,
|
||||
Comment = datdata.Comment,
|
||||
Header = datdata.Header,
|
||||
Type = datdata.Type,
|
||||
ForceMerging = datdata.ForceMerging,
|
||||
ForceNodump = datdata.ForceNodump,
|
||||
ForcePacking = datdata.ForcePacking,
|
||||
OutputFormat = outputFormat,
|
||||
MergeRoms = datdata.MergeRoms,
|
||||
Roms = new Dictionary<string, List<RomData>>(),
|
||||
};
|
||||
|
||||
// Now populate each of the DAT objects in turn
|
||||
List<string> keys = datdata.Roms.Keys.ToList();
|
||||
foreach (string key in keys)
|
||||
{
|
||||
List<RomData> roms = datdata.Roms[key];
|
||||
foreach (RomData rom in roms)
|
||||
{
|
||||
// If the file has a SHA-1
|
||||
if (rom.SHA1 != null && rom.SHA1 != "")
|
||||
{
|
||||
if (sha1.Roms.ContainsKey(key))
|
||||
{
|
||||
sha1.Roms[key].Add(rom);
|
||||
}
|
||||
else
|
||||
{
|
||||
List<RomData> temp = new List<RomData>();
|
||||
temp.Add(rom);
|
||||
sha1.Roms.Add(key, temp);
|
||||
}
|
||||
}
|
||||
// If the file has no SHA-1 but has an MD5
|
||||
else if (rom.MD5 != null && rom.MD5 != "")
|
||||
{
|
||||
if (md5.Roms.ContainsKey(key))
|
||||
{
|
||||
md5.Roms[key].Add(rom);
|
||||
}
|
||||
else
|
||||
{
|
||||
List<RomData> temp = new List<RomData>();
|
||||
temp.Add(rom);
|
||||
md5.Roms.Add(key, temp);
|
||||
}
|
||||
}
|
||||
// All other cases
|
||||
else
|
||||
{
|
||||
if (crc.Roms.ContainsKey(key))
|
||||
{
|
||||
crc.Roms[key].Add(rom);
|
||||
}
|
||||
else
|
||||
{
|
||||
List<RomData> temp = new List<RomData>();
|
||||
temp.Add(rom);
|
||||
crc.Roms.Add(key, temp);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool success = true;
|
||||
|
||||
// Now, output all of the files to the original location
|
||||
string outdir = Path.GetDirectoryName(_filename);
|
||||
success &= Output.WriteDatfile(sha1, outdir, _logger);
|
||||
success &= Output.WriteDatfile(md5, outdir, _logger);
|
||||
success &= Output.WriteDatfile(crc, outdir, _logger);
|
||||
|
||||
return success;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user