MPF/DICUI.Check/Program.cs

using System;
using System.IO;
using DICUI.Data;
using DICUI.Utilities;

namespace DICUI.Check
{
    public class Program
    {
        public static void Main(string[] args)
        {
            // Help options
            if (args.Length == 0
                || args[0] == "/h" || args[0] == "/?"
                || args[0] == "-h" || args[0] == "-?")
            {
                DisplayHelp();
                return;
            }

            // List options
            if (args[0] == "/lm" || args[0] == "/listmedia"
                || args[0] == "-lm" || args[0] == "--listmedia")
            {
                ListMediaTypes();
                Console.ReadLine();
                return;
            }
            else if (args[0] == "/ls" || args[0] == "/listsystems"
                || args[0] == "-ls" || args[0] == "--listsystems")
            {
                ListKnownSystems();
                Console.ReadLine();
                return;
            }

            // Normal operation check
            if (args.Length < 3)
            {
                DisplayHelp("Invalid number of arguments");
                return;
            }

            // Check the MediaType
            var mediaType = Converters.ToMediaType(args[0].Trim('"'));
            if (mediaType == MediaType.NONE)
            {
                DisplayHelp($"{args[0]} is not a recognized media type");
                return;
            }

            // Check the KnownSystem
            var knownSystem = Converters.ToKnownSystem(args[1].Trim('"'));
            if (knownSystem == KnownSystem.NONE)
            {
                DisplayHelp($"{args[1]} is not a recognized system");
                return;
            }

            // Check for additional flags
            string username = null, password = null;
            string internalProgram = "DiscImageCreator";
            int startIndex = 2;
            for (; startIndex < args.Length; startIndex++)
            {
                // Redump login
                if (args[startIndex] == "-c" || args[startIndex] == "--credentials")
                {
                    username = args[startIndex + 1];
                    password = args[startIndex + 2];
                    startIndex += 2;
                }

                // Use specific program
                else if (args[startIndex].StartsWith("-u=") || args[startIndex].StartsWith("--use="))
                {
                    internalProgram = args[startIndex].Split('=')[1];
                }

                // Default, we fall out
                else
                {
                    break;
                }
            }

            // Make a new Progress object
            var progress = new Progress<Result>();
            progress.ProgressChanged += ProgressUpdated;

            // Loop through all the rest of the args
            for (int i = startIndex; i < args.Length; i++)
            {
                // Check for a file
                if (!File.Exists(args[i]))
                {
                    DisplayHelp($"{args[i]} does not exist");
                    return;
                }

                // Get the full file path
                string filepath = Path.GetFullPath(args[i]);

                // Now populate an environment
                var env = new DumpEnvironment
                {
                    OutputDirectory = "",
                    OutputFilename = filepath,
                    System = knownSystem,
                    Type = mediaType,
                    ScanForProtection = false,
                    PromptForDiscInformation = false,
                    InternalProgram = Converters.ToInternalProgram(internalProgram),

                    Username = username,
                    Password = password,
                };
                env.FixOutputPaths();

                // Finally, attempt to do the output dance
                var result = env.VerifyAndSaveDumpOutput(progress);
                Console.WriteLine(result.Message);
            }
        }

        /// <summary>
        /// Display help for DICUI.Check
        /// </summary>
        /// <param name="error">Error string to prefix the help text with</param>
        private static void DisplayHelp(string error = null)
        {
            if (error != null)
                Console.WriteLine(error);

            Console.WriteLine("Usage:");
            Console.WriteLine("DICUI.Check.exe <mediatype> <system> [options] </path/to/output.bin> ...");
            Console.WriteLine();
            Console.WriteLine(@"Common Media Types:\r\n
bd / bluray     - BD-ROM
cd / cdrom      - CD-ROM
dvd             - DVD-ROM
fd / floppy     - Floppy Disk
gd / gdrom      - GD-ROM
umd             - UMD");
            Console.WriteLine("Run 'DICUI.Check.exe [-lm|--listmedia' for more options");
            Console.WriteLine();
            Console.WriteLine(@"Common Systems:\r\n
apple / mac     - Apple Macintosh
cdi             - Philips CD-i
ibm / ibmpc     - IBM PC Compatible
psx / ps1       - Sony PlayStation
ps2             - Sony PlayStation 2
psp             - Sony PlayStation Portable
saturn          - Sega Saturn
xbox            - Microsoft XBOX
x360            - Microsoft XBOX 360");
            Console.WriteLine("Run 'DICUI.Check.exe [-ls|--listsystems' for more options");
            Console.WriteLine();
            Console.WriteLine(@"Common Options:\r\n
-c username password    - Redump credentials
-u                      - Check for Aaru");
        }

        /// <summary>
        /// List all media types with their short usable names
        /// </summary>
        private static void ListMediaTypes()
        {
            Console.WriteLine("Supported Media Types:");
            foreach (var val in Enum.GetValues(typeof(MediaType)))
            {
                if (((MediaType)val) == MediaType.NONE)
                    continue;

                Console.WriteLine($"{((MediaType?)val).ShortName()} - {((MediaType?)val).LongName()}");
            }
        }

        /// <summary>
        /// List all known systems with their short usable names
        /// </summary>
        private static void ListKnownSystems()
        {
            Console.WriteLine("Supported Known Systems:");
            foreach (var val in Enum.GetValues(typeof(KnownSystem)))
            {
                if (((KnownSystem)val) == KnownSystem.NONE)
                    continue;

                Console.WriteLine($"{((KnownSystem?)val).ShortName()} - {((KnownSystem?)val).LongName()}");
            }
        }

        /// <summary>
        /// Simple process counter to write to console
        /// </summary>
        private static void ProgressUpdated(object sender, Result value)
        {
            Console.WriteLine(value.Message);
        }
    }
}