[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);

View File

@@ -318,13 +318,13 @@ namespace SabreTools
/// Wrap extracting and replacing headers
/// </summary>
/// <param name="inputs">Input file or folder names</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>
private static void InitHeaderer(List<string> inputs, bool extract, Logger logger)
private static void InitHeaderer(List<string> inputs, bool restore, Logger logger)
{
foreach (string input in inputs)
{
Headerer headerer = new Headerer(input, extract, logger);
Headerer headerer = new Headerer(input, restore, logger);
headerer.Process();
}
}

View File

@@ -93,7 +93,6 @@ namespace SabreTools
datprefix = false,
dedup = false,
enableGzip = false,
extract = true,
extsplit = false,
forceunpack = false,
generate = false,
@@ -113,6 +112,7 @@ namespace SabreTools
quotes = false,
rem = false,
remext = false,
restore = true,
romba = false,
single = false,
softlist = false,
@@ -377,7 +377,7 @@ namespace SabreTools
break;
case "-re":
case "--restore":
extract = false;
restore = true;
break;
case "-rm":
case "--remove":
@@ -713,7 +713,7 @@ namespace SabreTools
// If we're in headerer mode
else if (headerer)
{
InitHeaderer(inputs, extract, _logger);
InitHeaderer(inputs, restore, _logger);
}
// Import a file or folder