using System; using System.Collections.Generic; using SabreTools.Library.Data; using SabreTools.Library.Help; using SabreTools.Library.Tools; #if MONO using System.IO; #else using Alphaleonis.Win32.Filesystem; #endif namespace RombaSharp { /// /// Entry class for the RombaSharp application /// /// /// In the database, we want to enable "offline mode". That is, when a user does an operation /// that needs to read from the depot themselves, if the depot folder cannot be found, the /// user is prompted to reconnect the depot OR skip that depot entirely. /// public partial class RombaSharp { // General settings private static string _logdir; // Log folder location private static string _tmpdir; // Temp folder location private static string _webdir; // Web frontend location private static string _baddir; // Fail-to-unpack file folder location private static int _verbosity; // Verbosity of the output private static int _cores; // Forced CPU cores // DatRoot settings private static string _dats; // DatRoot folder location private static string _db; // Database name // Depot settings private static Dictionary> _depots; // Folder location, Max size // Server settings private static int _port; // Web server port // Other private variables private static string _config = "config.xml"; private static string _dbSchema = "rombasharp"; private static string _connectionString; private static Help _help; /// /// Entry class for the RombaSharp application /// public static void Main(string[] args) { // Perform initial setup and verification Globals.Logger = new Logger(true, "romba.log"); InitializeConfiguration(); DatabaseTools.EnsureDatabase(_dbSchema, _db, _connectionString); // Create a new Help object for this program _help = RombaSharp.RetrieveHelp(); // Get the location of the script tag, if it exists int scriptLocation = (new List(args)).IndexOf("--script"); // If output is being redirected or we are in script mode, don't allow clear screens if (!Console.IsOutputRedirected && scriptLocation == -1) { Console.Clear(); Build.PrepareConsole("RombaSharp"); } // Now we remove the script tag because it messes things up if (scriptLocation > -1) { List newargs = new List(args); newargs.RemoveAt(scriptLocation); args = newargs.ToArray(); } // Credits take precidence over all if ((new List(args)).Contains("--credits")) { _help.OutputCredits(); Globals.Logger.Close(); return; } // If there's no arguments, show help if (args.Length == 0) { _help.OutputGenericHelp(); Globals.Logger.Close(); return; } // User flags bool copy = false, fixdatOnly = false, logOnly = false, noDb = false, onlyNeeded = false, skipInitialScan = false, useGolangZip = false; // User inputs string backup = "", description = "", missingSha1s = "", name = "", newdat = "", old = "", outdat = "", resume = "", source = ""; int include7Zips = 0, includeGZips = 0, includeZips = 0, subworkers = 0, workers = 0; long size = -1; List dats = new List(); List depot = new List(); List inputs = new List(); // Get the first argument as a feature flag string feature = args[0]; // Verify that the flag is valid if (!_help.TopLevelFlag(feature)) { Globals.Logger.User("'{0}' is not valid feature flag", feature); _help.OutputIndividualFeature(feature); Globals.Logger.Close(); return; } // Now get the proper name for the feature feature = _help.GetFeatureName(feature); // If we had the help feature first if (feature == "Help") { // If we had something else after help if (args.Length > 1) { _help.OutputIndividualFeature(args[1]); Globals.Logger.Close(); return; } // Otherwise, show generic help else { _help.OutputGenericHelp(); Globals.Logger.Close(); return; } } // Now verify that all other flags are valid for (int i = 1; i < args.Length; i++) { // Verify that the current flag is proper for the feature if (!_help[feature].ValidateInput(args[i])) { Globals.Logger.Error("Invalid input detected: {0}", args[i]); _help.OutputIndividualFeature(feature); Globals.Logger.Close(); return; } // Special precautions for files and directories if (File.Exists(args[i]) || Directory.Exists(args[i])) { inputs.Add(args[i]); } } // Now loop through all inputs Dictionary features = _help.GetEnabledFeatures(); foreach (KeyValuePair feat in features) { // Check all of the flag names and translate to arguments switch (feat.Key) { #region User Flags case "copy": copy = true; break; case "fixdatOnly": fixdatOnly = true; break; case "log-only": logOnly = true; break; case "no-db": noDb = true; break; case "only-needed": onlyNeeded = true; break; case "skip-initial-scan": skipInitialScan = true; break; case "use-golang-zip": useGolangZip = true; break; #endregion #region User Int32 Inputs case "include-7zips": include7Zips = (int)feat.Value.GetValue() == Int32.MinValue ? (int)feat.Value.GetValue() : 0; break; case "include-gzips": includeGZips = (int)feat.Value.GetValue() == Int32.MinValue ? (int)feat.Value.GetValue() : 0; break; case "include-zips": includeZips = (int)feat.Value.GetValue() == Int32.MinValue ? (int)feat.Value.GetValue() : 0; break; case "subworkers": subworkers = (int)feat.Value.GetValue() == Int32.MinValue ? (int)feat.Value.GetValue() : _cores; break; case "workers": workers = (int)feat.Value.GetValue() == Int32.MinValue ? (int)feat.Value.GetValue() : _cores; break; #endregion #region User Int64 Inputs case "size": size = (long)feat.Value.GetValue() == Int64.MinValue ? (long)feat.Value.GetValue() : 0; break; #endregion #region User List Inputs case "dats": dats.AddRange((List)feat.Value.GetValue()); break; case "depot": depot.AddRange((List)feat.Value.GetValue()); break; #endregion #region User String Inputs case "backup": backup = (string)feat.Value.GetValue(); break; case "description": description = (string)feat.Value.GetValue(); break; case "missingSha1s": missingSha1s = (string)feat.Value.GetValue(); break; case "name": name = (string)feat.Value.GetValue(); break; case "new": newdat = (string)feat.Value.GetValue(); break; case "old": old = (string)feat.Value.GetValue(); break; case "out": outdat = (string)feat.Value.GetValue(); break; case "resume": resume = (string)feat.Value.GetValue(); break; case "source": source = (string)feat.Value.GetValue(); break; #endregion } } // Now take care of each mode in succesion switch(feature) { case "Help": // No-op as this should be caught break; // Adds ROM files from the specified directories to the ROM archive case "Archive": VerifyInputs(inputs, feature); InitArchive(inputs, onlyNeeded, resume, includeZips, workers, includeGZips, include7Zips, skipInitialScan, useGolangZip, noDb); break; // For each specified DAT file it creates the torrentzip files case "Build": VerifyInputs(inputs, feature); InitBuild(inputs, outdat, fixdatOnly, copy, workers, subworkers); break; // Cancels current long-running job case "Cancel": InitCancel(); break; // Prints dat stats case "DatStats": VerifyInputs(inputs, feature); InitDatStats(inputs); break; // Prints db stats case "DbStats": InitDbStats(); break; // Creates a DAT file with those entries that are in -new DAT case "Diffdat": InitDiffDat(outdat, old, newdat, name, description); break; // Creates a DAT file for the specified input directory and saves it to the -out filename case "Dir2Dat": InitDir2Dat(outdat, source, name, description); break; // Creates a DAT file with those entries that are in -new DAT case "EDiffdat": InitEDiffDat(outdat, old, newdat); break; // Exports db to export.csv case "Export": InitExport(); break; // For each specified DAT file it creates a fix DAT case "Fixdat": VerifyInputs(inputs, feature); InitFixdat(inputs, outdat, fixdatOnly, workers, subworkers); break; // Import a database from a formatted CSV file case "Import": VerifyInputs(inputs, feature); InitImport(inputs); break; // For each specified hash it looks up any available information case "Lookup": VerifyInputs(inputs, feature); InitLookup(inputs, size, outdat); break; // Prints memory stats case "Memstats": InitMemstats(); break; // Merges depot case "Merge": VerifyInputs(inputs, feature); InitMerge(inputs, onlyNeeded, resume, workers, skipInitialScan); break; // Create miss and have file case "Miss": VerifyInputs(inputs, feature); InitMiss(inputs); break; // Shows progress of the currently running command case "Progress": InitProgress(); break; // Moves DAT index entries for orphaned DATs case "Purge Backup": InitPurgeBackup(backup, workers, depot, dats, logOnly); break; // Deletes DAT index entries for orphaned DATs case "Purge Delete": InitPurgeDelete(workers, depot, dats, logOnly); break; // Refreshes the DAT index from the files in the DAT master directory tree case "Refresh DATs": InitRefreshDats(workers, missingSha1s); break; // Rescan a specific depot case "Rescan Depots": VerifyInputs(inputs, feature); InitRescanDepots(inputs); break; // Gracefully shuts down server case "Shutdown": InitShutdown(); break; // Prints version case "Version": InitVersion(); break; // If nothing is set, show the help default: _help.OutputGenericHelp(); break; } Globals.Logger.Close(); return; } private static void VerifyInputs(List inputs, string feature) { if (inputs.Count == 0) { Globals.Logger.Error("This feature requires at least one input"); _help.OutputIndividualFeature(feature); Environment.Exit(0); } } } }