RobustToolbox/Robust.Client/CommandLineArgs.cs

using System.Collections.Generic;
using Robust.Shared.Utility;
using C = System.Console;

namespace Robust.Client
{
    internal sealed class CommandLineArgs
    {
        public bool Headless { get; }
        public bool SelfContained { get; }
        public bool Connect { get; }
        public string ConnectAddress { get; }
        public bool Launcher { get; }
        public string Username { get; }
        public IReadOnlyCollection<(string key, string value)> CVars { get; }

        // Manual parser because C# has no good command line parsing libraries. Also dependencies bad.
        // Also I don't like spending 100ms parsing command line args. Do you?
        public static bool TryParse(IReadOnlyList<string> args, out CommandLineArgs parsed)
        {
            parsed = null;
            var headless = false;
            var selfContained = false;
            var connect = false;
            var connectAddress = "localhost";
            var launcher = false;
            string username = null;
            var cvars = new List<(string, string)>();

            using var enumerator = args.GetEnumerator();

            while (enumerator.MoveNext())
            {
                var arg = enumerator.Current;
                if (arg == "--connect")
                {
                    connect = true;
                }
                else if (arg == "--connect-address")
                {
                    if (!enumerator.MoveNext())
                    {
                        C.WriteLine("Missing connection address.");
                        return false;
                    }

                    connectAddress = enumerator.Current;
                }
                else if (arg == "--self-contained")
                {
                    selfContained = true;
                }
                else if (arg == "--launcher")
                {
                    launcher = true;
                }
                else if (arg == "--headless")
                {
                    headless = true;
                }
                else if (arg == "--username")
                {
                    if (!enumerator.MoveNext())
                    {
                        C.WriteLine("Missing username.");
                        return false;
                    }

                    username = enumerator.Current;
                }
                else if (arg == "--cvar")
                {
                    if (!enumerator.MoveNext())
                    {
                        C.WriteLine("Missing cvar value.");
                        return false;
                    }

                    var cvar = enumerator.Current;
                    DebugTools.AssertNotNull(cvar);
                    var split = cvar.Split("=");

                    if (split.Length < 2)
                    {
                        C.WriteLine("Expected = in cvar.");
                        return false;
                    }

                    cvars.Add((split[0], split[1]));
                }
                else if (arg == "--help")
                {
                    PrintHelp();
                    return false;
                }
                else
                {
                    C.WriteLine("Unknown argument: {0}", arg);
                }
            }

            parsed = new CommandLineArgs(headless, selfContained, connect, launcher, username, cvars, connectAddress);
            return true;
        }

        private static void PrintHelp()
        {
            C.WriteLine(@"
Options:
  --headless          Run without graphics/audio/input.
  --self-contained    Store data relative to executable instead of user-global locations.
  --connect           Automatically connect to connect-address.
  --connect-address   Address to automatically connect to.
                        Default: localhost
  --launcher          Run in launcher mode (no main menu, auto connect).
  --username          Override username.
  --cvar              Specifies an additional cvar overriding the config file. Syntax is <key>=<value>
  --help              Display this help text and exit.
");
        }

        private CommandLineArgs(
            bool headless,
            bool selfContained,
            bool connect,
            bool launcher,
            string username,
            IReadOnlyCollection<(string key, string value)> cVars,
            string connectAddress)
        {
            Headless = headless;
            SelfContained = selfContained;
            Connect = connect;
            Launcher = launcher;
            Username = username;
            CVars = cVars;
            ConnectAddress = connectAddress;
        }
    }
}