[SabreTools, Headerer] Reverse flag

This commit is contained in:
Matt Nadareski
2016-09-12 16:56:47 -07:00
parent 15c20c26ec
commit 08520c5b67
3 changed files with 31 additions and 31 deletions

View File

@@ -12,7 +12,7 @@ namespace SabreTools
{
// Private instance variables
private string _input;
private bool _extract;
private bool _restore;
private Logger _logger;
// Private required variables
@@ -24,12 +24,12 @@ namespace SabreTools
/// Create a new Headerer object
/// </summary>
/// <param name="input">Input file or folder name</param>
/// <param name="extract">True if we're extracting headers (default), false if we're replacing them</param>
/// <param name="restore">False if we're extracting headers (default), true if we're restoring them</param>
/// <param name="logger">Logger object for file and console output</param>
public Headerer(string input, bool extract, Logger logger)
public Headerer(string input, bool restore, Logger logger)
{
_input = input;
_extract = extract;
_restore = restore;
_logger = logger;
}
@@ -39,7 +39,26 @@ namespace SabreTools
/// <returns>True if it succeeded, false otherwise</returns>
public bool Process()
{
if (_extract)
if (_restore)
{
// If it's a single file, just check it
if (File.Exists(_input))
{
RestoreHeader(_input);
}
// If it's a directory, recursively check all
else if (Directory.Exists(_input))
{
foreach (string sub in Directory.GetFiles(_input))
{
if (sub != ".." && sub != ".")
{
RestoreHeader(sub);
}
}
}
}
else
{
// If it's a single file, just check it
if (File.Exists(_input))
@@ -58,25 +77,6 @@ namespace SabreTools
}
}
}
else
{
// If it's a single file, just check it
if (File.Exists(_input))
{
ReplaceHeader(_input);
}
// If it's a directory, recursively check all
else if (Directory.Exists(_input))
{
foreach (string sub in Directory.GetFiles(_input))
{
if (sub != ".." && sub != ".")
{
ReplaceHeader(sub);
}
}
}
}
return true;
}
@@ -184,7 +184,7 @@ namespace SabreTools
/// </summary>
/// <param name="file">Name of the file to be parsed</param>
/// <returns>True if a header was found and appended, false otherwise</returns>
public bool ReplaceHeader(string file)
public bool RestoreHeader(string file)
{
// First, get the SHA-1 hash of the file
Rom rom = FileTools.GetSingleFileInfo(file);