claunia/SabreTools
1
0
Fork 0
mirror of https://github.com/claunia/SabreTools.git synced 2025-12-16 19:14:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update argument handling

Browse Source
This commit is contained in:
Matt Nadareski
2016-03-24 13:23:25 -07:00
parent 18ab82846e
commit 7bc7749c32
1 changed files with 132 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

142
DATabase/Program.cs
View File

@@ -20,68 +20,61 @@ namespace DATabase
Remapping.CreateRemappings(); Remapping.CreateRemappings();
// If there's not enough arguments, show the help screen // If there's not enough arguments, show the help screen
if (args.Length == 0 || args[0] == "-h" || args[0] == "-?" || args[0] == "--help") if (args.Length == 0)
{ {
Help(); Help();
return; return;
} }
// Determine what mode we are in from the first argument // Determine which switches are enabled (with values if necessary)
switch (args[0]) bool help = false, import = false, generate = false, convert = false,
listsys = false, listsrc = false, norename = false, old = false; ;
string system = "", source = "", input = "";
foreach (string arg in args)
{ {
help = help || (arg == "-h" || arg == "-?" || arg == "--help");
import = import || (arg == "-i" || arg == "--import");
generate = generate || (arg == "-g" || arg == "--generate");
convert = convert || (arg == "-c" || arg == "--convert");
listsys = listsys || (arg == "-lso");
listsrc = listsrc || (arg == "-lsy");
norename = norename || (arg == "-nr" || arg == "--no-rename");
old = old || (arg == "-old" || arg == "--old");
system = (arg.Split('=')[0] == "system" && system == "" ? arg.Split('=')[1] : system);
source = (arg.Split('=')[0] == "source" && source == "" ? arg.Split('=')[1] : source);
input = (arg.Split('=')[0] == "input" ? arg.Split('=')[1] : input);
}
// If more than one switch is enabled, show the help screen
if (!(help ^ import ^ generate ^ listsys ^ listsrc) || help)
{
Help();
return;
}
// Now take care of each mode in succesion
// Import a file or folder // Import a file or folder
case "-i": if (import)
case "--import":
// Check if there are enough arguments
if (args.Length > 1)
{ {
// Check to see if the second argument is a file that exists // Check to see if the second argument is a file that exists
if (args.Length > 1 && File.Exists(args[1])) if (input != "" && File.Exists(input))
{ {
Console.WriteLine(args[1]); Console.WriteLine(input);
Import imp = new Import(args[1], _connectionString); Import imp = new Import(input, _connectionString);
imp.ImportData(); imp.ImportData();
} }
// Check to see if the second argument is a directory that exists // Check to see if the second argument is a directory that exists
else if (args.Length > 1 && Directory.Exists(args[1])) else if (input != "" && Directory.Exists(input))
{ {
foreach (string filename in Directory.GetFiles(args[1], "*", SearchOption.TopDirectoryOnly)) foreach (string filename in Directory.GetFiles(input, "*", SearchOption.TopDirectoryOnly))
{ {
Console.WriteLine(filename); Console.WriteLine(filename);
Import imp = new Import(filename, _connectionString); Import imp = new Import(filename, _connectionString);
imp.ImportData(); imp.ImportData();
} }
} }
// If it's invalid for either, show the help // Otherwise, show help
else
{
Help();
}
}
// If there aren't enough arguments
else
{
Help();
}
break;
// Generate a DAT
case "-g":
case "--generate":
Generate gen;
if (args.Length > 1)
{
int system = -1, source = -1;
for (int i = 1; i < args.Length; i++)
{
if (args[i].StartsWith("system=") && system == -1)
{
Int32.TryParse(args[i].Split('=')[1], out system);
}
else if (args[i].StartsWith("source=") && source == -1)
{
Int32.TryParse(args[i].Split('=')[1], out source);
}
else else
{ {
Help(); Help();
@@ -89,17 +82,21 @@ namespace DATabase
} }
} }
gen = new Generate(system, source, _connectionString); // Generate a DAT
} else if (generate)
else
{ {
gen = new Generate(-1, -1, _connectionString); int sysid = -1, srcid = -1;
}
Int32.TryParse(system, out sysid);
Int32.TryParse(source, out srcid);
Generate gen = new Generate(sysid, srcid, _connectionString, norename, old);
gen.Export(); gen.Export();
break; }
// List all available sources // List all available sources
case "-lso": else if (listsrc)
{
string query = @" string query = @"
SELECT DISTINCT sources.id, sources.name SELECT DISTINCT sources.id, sources.name
FROM sources JOIN games on sources.id=games.source FROM sources JOIN games on sources.id=games.source
@@ -126,12 +123,12 @@ ORDER BY sources.name COLLATE NOCASE";
} }
} }
} }
}
break;
// List all available systems // List all available systems
case "-lsy": else if (listsys)
query = @" {
string query = @"
SELECT DISTINCT systems.id, systems.manufacturer, systems.system SELECT DISTINCT systems.id, systems.manufacturer, systems.system
FROM systems JOIN games ON systems.id=games.system FROM systems JOIN games ON systems.id=games.system
ORDER BY systems.manufacturer, systems.system"; ORDER BY systems.manufacturer, systems.system";
@@ -157,44 +154,29 @@ ORDER BY systems.manufacturer, systems.system";
} }
} }
} }
}
break; // Convert RV DAT to XML DAT
else if (convert)
case "-c":
case "--convert":
// Check if there are enough arguments
if (args.Length > 1)
{ {
// Check to see if the second argument is a file that exists if (File.Exists(input))
if (args.Length > 1 && File.Exists(args[1]))
{ {
Console.WriteLine("Converting " + args[1]); Console.WriteLine("Converting " + input);
XElement conv = Converters.RomVaultToXML(File.ReadAllLines(args[1])); XElement conv = Converters.RomVaultToXML(File.ReadAllLines(input));
FileStream fs = File.OpenWrite(Path.GetFileNameWithoutExtension(args[1]) + ".new.xml"); FileStream fs = File.OpenWrite(Path.GetFileNameWithoutExtension(input) + ".new.xml");
StreamWriter sw = new StreamWriter(fs); StreamWriter sw = new StreamWriter(fs);
sw.Write(conv); sw.Write(conv);
sw.Close(); sw.Close();
fs.Close(); fs.Close();
Console.WriteLine("Converted file: " + Path.GetFileNameWithoutExtension(args[1]) + ".new.xml"); Console.WriteLine("Converted file: " + Path.GetFileNameWithoutExtension(input) + ".new.xml");
} }
// If it's invalid, show the help // Otherwise, show help
else else
{ {
Help(); Help();
return;
} }
} }
// If there aren't enough arguments
else
{
Help();
}
break;
// Invalid argument
default:
Help();
break;
}
return; return;
} }
@@ -288,7 +270,7 @@ CREATE TABLE IF NOT EXISTS systems (
Console.Write(@" Console.Write(@"
DATabase - Import and Generate DAT files DATabase - Import and Generate DAT files
----------------------------------------- -----------------------------------------
Usage: DATabase <option> (<filename>|<dirname>) | (system=sy) (source=so) Usage: DATabase <option> (input=<filename>|<dirname>) | (system=sy) (source=so)
<option> can be one of the following: <option> can be one of the following:
-h, -?, --help Show this help -h, -?, --help Show this help
@@ -297,6 +279,8 @@ Usage: DATabase <option> (<filename>|<dirname>) | (system=sy) (source=so)
-lso List all sources (id <= name) -lso List all sources (id <= name)
-lsy List all systems (id <= name) -lsy List all systems (id <= name)
-c, --convert Convert a RV DAT to XML -c, --convert Convert a RV DAT to XML
-nr, --no-rename Do not rename games according to source/system
-old Use RV datfile format
If started in import or convert mode, either a filename If started in import or convert mode, either a filename
or directory name is required in order to run. or directory name is required in order to run.