mirror of
https://github.com/claunia/SabreTools.git
synced 2025-12-16 19:14:27 +00:00
684 lines
15 KiB
C#
684 lines
15 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
|
|
using SabreTools.Helper;
|
|
|
|
namespace SabreTools
|
|
{
|
|
/// <summary>
|
|
/// Entry class for the DATabase application
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// The following features are missing from DATabaseTwo with respect to the original DATabase:
|
|
/// - Source merging
|
|
/// - Custom DATs based on a system and a source
|
|
/// - Multi-source and multi-system DATs
|
|
///
|
|
/// The following features need to (want to) be implemented in DATabaseTwo for further stability
|
|
/// - Import updating file locations and names when SHA-1 hashes are matched
|
|
/// - True duplicate DATs being removed from the import folder (SHA-1 matches)
|
|
/// - Generate All only generating DATs that have been recently updated
|
|
/// + This requires implementing a "last updated" data point for all DATs and tracking for "last generate" somewhere
|
|
/// - Impelement a ToSort folder for DATs that will place DATs in the correct subfolder on Import
|
|
/// </remarks>
|
|
public partial class DATabase
|
|
{
|
|
// Private required variables
|
|
private static string _datroot = "DATS";
|
|
private static string _outroot = "Output";
|
|
private static string _dbName = "dats.sqlite";
|
|
private static string _connectionString = "Data Source=" + _dbName + ";Version = 3;";
|
|
|
|
private static Logger _logger;
|
|
|
|
/// <summary>
|
|
/// Start menu or use supplied parameters
|
|
/// </summary>
|
|
/// <param name="args">String array representing command line parameters</param>
|
|
public static void Main(string[] args)
|
|
{
|
|
// Perform initial setup and verification
|
|
_logger = new Logger(true, "database.log");
|
|
_logger.Start();
|
|
|
|
// If output is being redirected, don't allow clear screens
|
|
if (!Console.IsOutputRedirected)
|
|
{
|
|
Console.Clear();
|
|
}
|
|
Setup();
|
|
|
|
// Credits take precidence over all
|
|
if ((new List<string>(args)).Contains("--credits"))
|
|
{
|
|
Build.Credits();
|
|
_logger.Close();
|
|
return;
|
|
}
|
|
|
|
// If there's no arguments, show help
|
|
if (args.Length == 0)
|
|
{
|
|
/*
|
|
// If there are no arguments, show the menu
|
|
if (!Console.IsOutputRedirected)
|
|
{
|
|
ShowMainMenu();
|
|
}
|
|
else
|
|
{
|
|
Build.Help();
|
|
}
|
|
*/
|
|
|
|
Build.Help();
|
|
_logger.Close();
|
|
return;
|
|
}
|
|
|
|
// Set all default values
|
|
bool help = false,
|
|
add = false,
|
|
bare = false,
|
|
cascade = false,
|
|
clean = false,
|
|
datprefix = false,
|
|
dedup = false,
|
|
diff = false,
|
|
disableForce = false,
|
|
extsplit = false,
|
|
filter = false,
|
|
forceunpack = false,
|
|
generate = false,
|
|
genall = false,
|
|
hashsplit = false,
|
|
ignore = false,
|
|
import = false,
|
|
inplace = false,
|
|
listsrc = false,
|
|
listsys = false,
|
|
merge = false,
|
|
norename = false,
|
|
old = false,
|
|
outputCMP = false,
|
|
outputMiss = false,
|
|
outputRC = false,
|
|
outputSD = false,
|
|
outputXML = false,
|
|
quotes = false,
|
|
rem = false,
|
|
romba = false,
|
|
single = false,
|
|
stats = false,
|
|
superdat = false,
|
|
trim = false,
|
|
tsv = false,
|
|
skip = false,
|
|
update = false,
|
|
usegame = true;
|
|
bool? nodump = null;
|
|
long sgt = -1,
|
|
slt = -1,
|
|
seq = -1;
|
|
string addext = "",
|
|
author = "",
|
|
category = "",
|
|
comment = "",
|
|
crc = "",
|
|
date = "",
|
|
description = "",
|
|
email = "",
|
|
exta = "",
|
|
extb = "",
|
|
filename = "",
|
|
forcemerge = "",
|
|
forcend = "",
|
|
forcepack = "",
|
|
gamename = "",
|
|
header = "",
|
|
homepage = "",
|
|
name = "",
|
|
manu = "",
|
|
md5 = "",
|
|
outdir = "",
|
|
postfix = "",
|
|
prefix = "",
|
|
repext = "",
|
|
romname = "",
|
|
romtype = "",
|
|
sha1 = "",
|
|
sources = "",
|
|
systems = "",
|
|
root = "",
|
|
url = "",
|
|
version = "";
|
|
List<string> inputs = new List<string>();
|
|
|
|
// Determine which switches are enabled (with values if necessary)
|
|
foreach (string arg in args)
|
|
{
|
|
switch (arg)
|
|
{
|
|
case "-?":
|
|
case "-h":
|
|
case "--help":
|
|
help = true;
|
|
break;
|
|
case "-a":
|
|
case "--add":
|
|
add = true;
|
|
break;
|
|
case "-b":
|
|
case "--bare":
|
|
bare = true;
|
|
break;
|
|
case "-c":
|
|
case "--cascade":
|
|
cascade = true;
|
|
break;
|
|
case "-cc":
|
|
case "--convert-cmp":
|
|
outputCMP = true;
|
|
break;
|
|
case "-cm":
|
|
case "--convert-miss":
|
|
outputMiss = true;
|
|
break;
|
|
case "-cr":
|
|
case "--convert-rc":
|
|
outputRC = true;
|
|
break;
|
|
case "-cs":
|
|
case "--convert-sd":
|
|
outputSD = true;
|
|
break;
|
|
case "-cx":
|
|
case "--convert-xml":
|
|
outputXML = true;
|
|
break;
|
|
case "-clean":
|
|
case "--clean":
|
|
clean = true;
|
|
break;
|
|
case "-dd":
|
|
case "--dedup":
|
|
dedup = true;
|
|
break;
|
|
case "-df":
|
|
case "--disable-force":
|
|
disableForce = true;
|
|
break;
|
|
case "-di":
|
|
case "--diff":
|
|
diff = true;
|
|
break;
|
|
case "-es":
|
|
case "--ext-split":
|
|
extsplit = true;
|
|
break;
|
|
case "-f":
|
|
case "--filter":
|
|
filter = true;
|
|
break;
|
|
case "-g":
|
|
case "--generate":
|
|
generate = true;
|
|
break;
|
|
case "-ga":
|
|
case "--generate-all":
|
|
genall = true;
|
|
break;
|
|
case "-gp":
|
|
case "--game-prefix":
|
|
datprefix = true;
|
|
break;
|
|
case "-hs":
|
|
case "--hash-split":
|
|
hashsplit = true;
|
|
break;
|
|
case "-i":
|
|
case "--import":
|
|
import = true;
|
|
break;
|
|
case "-ig":
|
|
case "--ignore":
|
|
ignore = true;
|
|
break;
|
|
case "-ip":
|
|
case "--inplace":
|
|
inplace = true;
|
|
break;
|
|
case "-lso":
|
|
case "--list-sources":
|
|
listsrc = true;
|
|
break;
|
|
case "-lsy":
|
|
case "--list-systems":
|
|
listsys = true;
|
|
break;
|
|
case "-m":
|
|
case "--merge":
|
|
merge = true;
|
|
break;
|
|
case "-nd":
|
|
case "--nodump":
|
|
nodump = true;
|
|
break;
|
|
case "-nnd":
|
|
case "--not-nodump":
|
|
nodump = false;
|
|
break;
|
|
case "-nr":
|
|
case "--no-rename":
|
|
norename = true;
|
|
break;
|
|
case "-o":
|
|
case "--old":
|
|
old = true;
|
|
break;
|
|
case "-oc":
|
|
case "--output-cmp":
|
|
outputCMP = true;
|
|
break;
|
|
case "-om":
|
|
case "--output-miss":
|
|
outputMiss = true;
|
|
break;
|
|
case "-or":
|
|
case "--output-rc":
|
|
outputRC = true;
|
|
break;
|
|
case "-os":
|
|
case "--output-sd":
|
|
outputSD = true;
|
|
break;
|
|
case "-ox":
|
|
case "--output-xml":
|
|
outputXML = true;
|
|
break;
|
|
case "-q":
|
|
case "--quotes":
|
|
quotes = true;
|
|
break;
|
|
case "-r":
|
|
case "--roms":
|
|
usegame = false;
|
|
break;
|
|
case "-rm":
|
|
case "--remove":
|
|
rem = true;
|
|
break;
|
|
case "-ro":
|
|
case "--romba":
|
|
romba = true;
|
|
break;
|
|
case "-sd":
|
|
case "--superdat":
|
|
superdat = true;
|
|
break;
|
|
case "-si":
|
|
case "--single":
|
|
single = true;
|
|
break;
|
|
case "-st":
|
|
case "--stats":
|
|
stats = true;
|
|
break;
|
|
case "--skip":
|
|
skip = true;
|
|
break;
|
|
case "-tm":
|
|
case "--trim-merge":
|
|
trim = true;
|
|
break;
|
|
case "-tsv":
|
|
case " --tsv":
|
|
tsv = true;
|
|
break;
|
|
case "-u":
|
|
case "--unzip":
|
|
forceunpack = true;
|
|
break;
|
|
case "-ud":
|
|
case "--update":
|
|
update = true;
|
|
break;
|
|
default:
|
|
if (arg.StartsWith("-ae=") || arg.StartsWith("--add-ext="))
|
|
{
|
|
addext = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-au=") || arg.StartsWith("--author="))
|
|
{
|
|
author = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-ca=") || arg.StartsWith("--category="))
|
|
{
|
|
category = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-co=") || arg.StartsWith("--comment="))
|
|
{
|
|
comment = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-crc=") || arg.StartsWith("--crc="))
|
|
{
|
|
crc = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-da=") || arg.StartsWith("--date="))
|
|
{
|
|
date = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-de=") || arg.StartsWith("--desc="))
|
|
{
|
|
description = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-em=") || arg.StartsWith("--email="))
|
|
{
|
|
email = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-exta="))
|
|
{
|
|
exta = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-extb="))
|
|
{
|
|
extb = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-f=") || arg.StartsWith("--filename="))
|
|
{
|
|
filename = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-fm=") || arg.StartsWith("--forcemerge="))
|
|
{
|
|
forcemerge = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-fn=") || arg.StartsWith("--forcend="))
|
|
{
|
|
forcend = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-fp=") || arg.StartsWith("--forcepack="))
|
|
{
|
|
forcepack = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-gn=") || arg.StartsWith("--game-name="))
|
|
{
|
|
gamename = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-h=") || arg.StartsWith("--header="))
|
|
{
|
|
header = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-hp=") || arg.StartsWith("--homepage="))
|
|
{
|
|
homepage = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-input="))
|
|
{
|
|
inputs.Add(arg.Split('=')[1].Replace("\"", ""));
|
|
}
|
|
else if (arg.StartsWith("-manu=") && manu == "")
|
|
{
|
|
manu = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-md5=") || arg.StartsWith("--md5="))
|
|
{
|
|
md5 = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-n=") || arg.StartsWith("--name="))
|
|
{
|
|
name = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-out=") && outdir == "")
|
|
{
|
|
outdir = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-post=") || arg.StartsWith("--postfix="))
|
|
{
|
|
postfix = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-pre=") || arg.StartsWith("--prefix="))
|
|
{
|
|
prefix = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-rn=") || arg.StartsWith("--rom-name="))
|
|
{
|
|
romname = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-rt=") || arg.StartsWith("--rom-type="))
|
|
{
|
|
romtype = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-seq=") || arg.StartsWith("--equal="))
|
|
{
|
|
if (!Int64.TryParse(arg.Split('=')[1], out seq))
|
|
{
|
|
seq = -1;
|
|
}
|
|
}
|
|
else if (arg.StartsWith("-sgt=") || arg.StartsWith("--greater="))
|
|
{
|
|
if (!Int64.TryParse(arg.Split('=')[1], out sgt))
|
|
{
|
|
sgt = -1;
|
|
}
|
|
}
|
|
else if (arg.StartsWith("-sha1=") || arg.StartsWith("--sha1="))
|
|
{
|
|
sha1 = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-slt=") || arg.StartsWith("--less="))
|
|
{
|
|
if (!Int64.TryParse(arg.Split('=')[1], out slt))
|
|
{
|
|
slt = -1;
|
|
}
|
|
}
|
|
else if (arg.StartsWith("-source=") && sources == "")
|
|
{
|
|
sources = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-system=") && systems == "")
|
|
{
|
|
systems = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-rd=") || arg.StartsWith("--root-dir="))
|
|
{
|
|
root = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-re=") || arg.StartsWith("--rep-ext="))
|
|
{
|
|
repext = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-u=") || arg.StartsWith("--url="))
|
|
{
|
|
url = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-url=") && url == "")
|
|
{
|
|
url = arg.Split('=')[1];
|
|
}
|
|
else if (arg.StartsWith("-v=") || arg.StartsWith("--version="))
|
|
{
|
|
version = arg.Split('=')[1];
|
|
}
|
|
else if (File.Exists(arg.Replace("\"", "")) || Directory.Exists(arg.Replace("\"", "")))
|
|
{
|
|
inputs.Add(arg);
|
|
}
|
|
else
|
|
{
|
|
_logger.Error("Invalid input detected: " + arg);
|
|
Console.WriteLine();
|
|
Build.Help();
|
|
_logger.Close();
|
|
return;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
// If skip is set, it's being called from the UI so we just exit
|
|
if (skip)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// If help is set, show the help screen
|
|
if (help)
|
|
{
|
|
Build.Help();
|
|
_logger.Close();
|
|
return;
|
|
}
|
|
|
|
// If more than one switch is enabled, show the help screen
|
|
if (!(add ^ extsplit ^ filter ^ generate ^ genall ^ hashsplit ^ import ^ listsrc ^ listsys ^ (merge || diff) ^
|
|
(update || outputCMP || outputRC || outputSD || outputXML || outputMiss || romba) ^ rem ^ stats ^ trim))
|
|
{
|
|
_logger.Error("Only one feature switch is allowed at a time");
|
|
Build.Help();
|
|
_logger.Close();
|
|
return;
|
|
}
|
|
|
|
// If a switch that requires a filename is set and no file is, show the help screen
|
|
if (inputs.Count == 0 && (update || (outputMiss || romba) || outputCMP || outputRC || outputSD
|
|
|| outputXML || filter || extsplit || hashsplit || (merge || diff) || stats || trim))
|
|
{
|
|
_logger.Error("This feature requires at least one input");
|
|
Build.Help();
|
|
_logger.Close();
|
|
return;
|
|
}
|
|
|
|
// Now take care of each mode in succesion
|
|
|
|
// Import a file or folder
|
|
if (import)
|
|
{
|
|
InitImport(ignore);
|
|
}
|
|
|
|
// Generate a DAT
|
|
else if (generate)
|
|
{
|
|
InitImport(ignore);
|
|
InitGenerate(systems, norename, old);
|
|
}
|
|
|
|
// Generate all DATs
|
|
else if (genall)
|
|
{
|
|
InitImport(ignore);
|
|
InitGenerateAll(norename, old);
|
|
}
|
|
|
|
// List all available sources
|
|
else if (listsrc)
|
|
{
|
|
ListSources();
|
|
}
|
|
|
|
// List all available systems
|
|
else if (listsys)
|
|
{
|
|
ListSystems();
|
|
}
|
|
|
|
// Convert or update a DAT or folder of DATs
|
|
else if (update || outputCMP || outputMiss || outputRC || outputSD || outputXML || romba)
|
|
{
|
|
foreach (string input in inputs)
|
|
{
|
|
InitUpdate(input, filename, name, description, category, version, date, author, email, homepage, url, comment, header,
|
|
superdat, forcemerge, forcend, forcepack, outputCMP, outputMiss, outputRC, outputSD, outputXML, usegame, prefix,
|
|
postfix, quotes, repext, addext, datprefix, romba, tsv, outdir, clean, dedup);
|
|
}
|
|
}
|
|
|
|
// Add a source or system
|
|
else if (add)
|
|
{
|
|
if (manu != "" && systems != "")
|
|
{
|
|
InitAddSystem(manu, systems);
|
|
}
|
|
else if (sources != "" && url != "")
|
|
{
|
|
InitAddSource(manu, systems);
|
|
}
|
|
else
|
|
{
|
|
Build.Help();
|
|
}
|
|
}
|
|
|
|
// Remove a source or system
|
|
else if (rem)
|
|
{
|
|
if (systems != "")
|
|
{
|
|
InitRemoveSystem(systems);
|
|
}
|
|
else if (sources != "")
|
|
{
|
|
InitRemoveSource(sources);
|
|
}
|
|
else
|
|
{
|
|
Build.Help();
|
|
}
|
|
}
|
|
|
|
// Consolodate and trim DAT
|
|
else if (trim)
|
|
{
|
|
foreach (string input in inputs)
|
|
{
|
|
InitTrimMerge(input, root, !norename, !disableForce);
|
|
}
|
|
}
|
|
|
|
// Split a DAT by extension
|
|
else if (extsplit)
|
|
{
|
|
InitExtSplit(inputs, exta, extb, outdir);
|
|
}
|
|
|
|
// Merge, diff, and dedupe at least 2 DATs
|
|
else if (merge || diff)
|
|
{
|
|
InitMergeDiff(inputs, name, description, category, version, author, diff, dedup, bare, forceunpack, old, superdat, cascade, inplace, outdir, clean);
|
|
}
|
|
|
|
// Split a DAT by available hashes
|
|
else if (hashsplit)
|
|
{
|
|
InitHashSplit(inputs, outdir);
|
|
}
|
|
|
|
// Get statistics on input files
|
|
else if (stats)
|
|
{
|
|
InitStats(inputs, single);
|
|
}
|
|
|
|
// Filter input files
|
|
else if (filter)
|
|
{
|
|
InitFilter(inputs, outdir, gamename, romname, romtype, sgt, slt, seq, crc, md5, sha1, nodump, _logger);
|
|
}
|
|
|
|
// If nothing is set, show the help
|
|
else
|
|
{
|
|
Build.Help();
|
|
}
|
|
|
|
_logger.Close();
|
|
return;
|
|
}
|
|
}
|
|
}
|