SetupWizard.cs

Go to the documentation of this file.

using Microsoft.Data.Sqlite;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using MySql.Data.MySqlClient;
using Newtonsoft.Json;
using Npgsql;
using System;
using System.Collections.Generic;
using System.Data.Common;
using System.Data.SqlClient;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Host.Configuration;
using Tgstation.Server.Host.Core;
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.IO;
using Tgstation.Server.Host.System;

namespace Tgstation.Server.Host.Setup
{
    sealed class SetupWizard : IHostedService
    {
        readonly IIOManager ioManager;

        readonly IConsole console;

        readonly IHostEnvironment hostingEnvironment;

        readonly IAssemblyInformationProvider assemblyInformationProvider;

        readonly IDatabaseConnectionFactory dbConnectionFactory;

        readonly IPlatformIdentifier platformIdentifier;

        readonly IAsyncDelayer asyncDelayer;

        readonly IHostApplicationLifetime applicationLifetime;

        readonly GeneralConfiguration generalConfiguration;

        public SetupWizard(
            IIOManager ioManager,
            IConsole console,
            IHostEnvironment hostingEnvironment,
            IAssemblyInformationProvider assemblyInformationProvider,
            IDatabaseConnectionFactory dbConnectionFactory,
            IPlatformIdentifier platformIdentifier,
            IAsyncDelayer asyncDelayer,
            IHostApplicationLifetime applicationLifetime,
            IOptions<GeneralConfiguration> generalConfigurationOptions)
        {
            this.ioManager = ioManager ?? throw new ArgumentNullException(nameof(ioManager));
            this.console = console ?? throw new ArgumentNullException(nameof(console));
            this.hostingEnvironment = hostingEnvironment ?? throw new ArgumentNullException(nameof(hostingEnvironment));
            this.assemblyInformationProvider = assemblyInformationProvider ?? throw new ArgumentNullException(nameof(assemblyInformationProvider));
            this.dbConnectionFactory = dbConnectionFactory ?? throw new ArgumentNullException(nameof(dbConnectionFactory));
            this.platformIdentifier = platformIdentifier ?? throw new ArgumentNullException(nameof(platformIdentifier));
            this.asyncDelayer = asyncDelayer ?? throw new ArgumentNullException(nameof(asyncDelayer));
            this.applicationLifetime = applicationLifetime ?? throw new ArgumentNullException(nameof(applicationLifetime));
            generalConfiguration = generalConfigurationOptions?.Value ?? throw new ArgumentNullException(nameof(generalConfigurationOptions));
        }

        async Task<bool> PromptYesNo(string question, CancellationToken cancellationToken)
        {
            do
            {
                await console.WriteAsync(question, false, cancellationToken).ConfigureAwait(false);
                var responseString = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                var upperResponse = responseString.ToUpperInvariant();
                if (upperResponse == "Y" || upperResponse == "YES")
                    return true;
                else if (upperResponse == "N" || upperResponse == "NO")
                    return false;
                await console.WriteAsync("Invalid response!", true, cancellationToken).ConfigureAwait(false);
            }
            while (true);
        }

        async Task<ushort?> PromptForHostingPort(CancellationToken cancellationToken)
        {
            await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
            await console.WriteAsync("What port would you like to connect to TGS on?", true, cancellationToken).ConfigureAwait(false);
            await console.WriteAsync("Note: If this is a docker container with the default port already mapped, use the default.", true, cancellationToken).ConfigureAwait(false);

            do
            {
                await console.WriteAsync(
                    $"API Port (leave blank for default of {GeneralConfiguration.DefaultApiPort}): ",
                    false,
                    cancellationToken)
                    .ConfigureAwait(false);
                var portString = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                if (String.IsNullOrWhiteSpace(portString))
                    return null;
                if (UInt16.TryParse(portString, out var port) && port != 0)
                    return port;
                await console.WriteAsync("Invalid port! Please enter a value between 1 and 65535", true, cancellationToken).ConfigureAwait(false);
            }
            while (true);
        }

        async Task TestDatabaseConnection(
            DbConnection testConnection,
            DatabaseConfiguration databaseConfiguration,
            string databaseName,
            bool dbExists,
            CancellationToken cancellationToken)
        {
            bool isSqliteDB = databaseConfiguration.DatabaseType == DatabaseType.Sqlite;
            using (testConnection)
            {
                await console.WriteAsync("Testing connection...", true, cancellationToken).ConfigureAwait(false);
                await testConnection.OpenAsync(cancellationToken).ConfigureAwait(false);
                await console.WriteAsync("Connection successful!", true, cancellationToken).ConfigureAwait(false);

                if (databaseConfiguration.DatabaseType == DatabaseType.MariaDB
                    || databaseConfiguration.DatabaseType == DatabaseType.MySql
                    || databaseConfiguration.DatabaseType == DatabaseType.PostgresSql)
                {
                    await console.WriteAsync($"Checking {databaseConfiguration.DatabaseType} version...", true, cancellationToken).ConfigureAwait(false);
                    using var command = testConnection.CreateCommand();
                    command.CommandText = "SELECT VERSION()";
                    var fullVersion = (string)await command.ExecuteScalarAsync(cancellationToken).ConfigureAwait(false);
                    await console.WriteAsync(String.Format(CultureInfo.InvariantCulture, "Found {0}", fullVersion), true, cancellationToken).ConfigureAwait(false);

                    if (databaseConfiguration.DatabaseType == DatabaseType.PostgresSql)
                    {
                        var splits = fullVersion.Split(' ');
                        databaseConfiguration.ServerVersion = splits[1].TrimEnd(',');
                    }
                    else
                    {
                        var splits = fullVersion.Split('-');
                        databaseConfiguration.ServerVersion = splits.First();
                    }
                }

                if (!isSqliteDB && !dbExists)
                {
                    await console.WriteAsync("Testing create DB permission...", true, cancellationToken).ConfigureAwait(false);
                    using (var command = testConnection.CreateCommand())
                    {
                        // I really don't care about user sanitization here, they want to fuck their own DB? so be it
#pragma warning disable CA2100 // Review SQL queries for security vulnerabilities
                        command.CommandText = $"CREATE DATABASE {databaseName}";
#pragma warning restore CA2100 // Review SQL queries for security vulnerabilities
                        await command.ExecuteNonQueryAsync(cancellationToken).ConfigureAwait(false);
                    }

                    await console.WriteAsync("Success!", true, cancellationToken).ConfigureAwait(false);
                    await console.WriteAsync("Dropping test database...", true, cancellationToken).ConfigureAwait(false);
                    using (var command = testConnection.CreateCommand())
                    {
#pragma warning disable CA2100 // Review SQL queries for security vulnerabilities
                        command.CommandText = $"DROP DATABASE {databaseName}";
#pragma warning restore CA2100 // Review SQL queries for security vulnerabilities
                        try
                        {
                            await command.ExecuteNonQueryAsync(cancellationToken).ConfigureAwait(false);
                        }
                        catch (OperationCanceledException)
                        {
                            throw;
                        }
                        catch (Exception e)
                        {
                            await console.WriteAsync(e.Message, true, cancellationToken).ConfigureAwait(false);
                            await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                            await console.WriteAsync("This should be okay, but you may want to manually drop the database before continuing!", true, cancellationToken).ConfigureAwait(false);
                            await console.WriteAsync("Press any key to continue...", true, cancellationToken).ConfigureAwait(false);
                            await console.PressAnyKeyAsync(cancellationToken).ConfigureAwait(false);
                        }
                    }
                }
            }

            if (isSqliteDB && !dbExists)
                await Task.WhenAll(
                    console.WriteAsync("Deleting test database file...", true, cancellationToken),
                    ioManager.DeleteFile(databaseName, cancellationToken)).ConfigureAwait(false);
        }

        async Task<string> ValidateNonExistantSqliteDBName(string databaseName, CancellationToken cancellationToken)
        {
            var resolvedPath = ioManager.ResolvePath(databaseName);
            try
            {
                var directoryName = ioManager.GetDirectoryName(resolvedPath);
                bool directoryExisted = await ioManager.DirectoryExists(directoryName, cancellationToken).ConfigureAwait(false);
                await ioManager.CreateDirectory(directoryName, cancellationToken).ConfigureAwait(false);
                try
                {
                    await ioManager.WriteAllBytes(resolvedPath, Array.Empty<byte>(), cancellationToken).ConfigureAwait(false);
                }
                catch
                {
                    if (!directoryExisted)
                        await ioManager.DeleteDirectory(directoryName, cancellationToken).ConfigureAwait(false);
                    throw;
                }
            }
            catch (IOException)
            {
                return null;
            }

            if (!Path.IsPathRooted(databaseName))
            {
                await console.WriteAsync("Note, this relative path (currently) resolves to the following:", true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(resolvedPath, true, cancellationToken).ConfigureAwait(false);
                bool writeResolved = await PromptYesNo(
                    "Would you like to save the relative path in the configuration? If not, the full path will be saved. (y/n): ",
                    cancellationToken)
                    .ConfigureAwait(false);

                if (writeResolved)
                    databaseName = resolvedPath;
            }

            await ioManager.DeleteFile(databaseName, cancellationToken).ConfigureAwait(false);
            return databaseName;
        }

        async Task<DatabaseType> PromptDatabaseType(bool firstTime, CancellationToken cancellationToken)
        {
            if (firstTime)
            {
                await console.WriteAsync(String.Empty, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(
                    "NOTE: It is HIGHLY reccommended that TGS runs on a complete relational database, specfically *NOT* Sqlite.",
                    true,
                    cancellationToken)
                    .ConfigureAwait(false);
                await console.WriteAsync(
                    "Sqlite, by nature cannot perform several DDL operations. Because of this future compatiblility cannot be guaranteed.",
                    true,
                    cancellationToken)
                    .ConfigureAwait(false);
                await console.WriteAsync(
                    "This means that you may not be able to update to the next minor version of TGS4 without a clean re-installation!",
                    true,
                    cancellationToken)
                    .ConfigureAwait(false);
                await console.WriteAsync(
                    "Please consider taking the time to set up a relational database if this is meant to be a long-standing server.",
                    true,
                    cancellationToken)
                    .ConfigureAwait(false);
                await console.WriteAsync(String.Empty, true, cancellationToken).ConfigureAwait(false);

                await asyncDelayer.Delay(TimeSpan.FromSeconds(3), cancellationToken).ConfigureAwait(false);
            }

            await console.WriteAsync("What SQL database type will you be using?", true, cancellationToken).ConfigureAwait(false);
            do
            {
                await console.WriteAsync(
                    String.Format(
                        CultureInfo.InvariantCulture,
                        "Please enter one of {0}, {1}, {2}, {3} or {4}: ",
                        DatabaseType.MariaDB,
                        DatabaseType.MySql,
                        DatabaseType.PostgresSql,
                        DatabaseType.SqlServer,
                        DatabaseType.Sqlite),
                    false,
                    cancellationToken)
                    .ConfigureAwait(false);
                var databaseTypeString = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                if (Enum.TryParse<DatabaseType>(databaseTypeString, out var databaseType))
                    return databaseType;

                await console.WriteAsync("Invalid database type!", true, cancellationToken).ConfigureAwait(false);
            }
            while (true);
        }

        #pragma warning disable CA1502 // TODO: Decomplexify
        async Task<DatabaseConfiguration> ConfigureDatabase(CancellationToken cancellationToken)
        {
            bool firstTime = true;
            do
            {
                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);

                var databaseConfiguration = new DatabaseConfiguration
                {
                    DatabaseType = await PromptDatabaseType(firstTime, cancellationToken).ConfigureAwait(false)
                };
                firstTime = false;

                string serverAddress = null;
                ushort? serverPort = null;

                bool isSqliteDB = databaseConfiguration.DatabaseType == DatabaseType.Sqlite;
                if (!isSqliteDB)
                    do
                    {
                        await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync("Enter the server's address and port [<server>:<port> or <server>] (blank for local): ", false, cancellationToken).ConfigureAwait(false);
                        serverAddress = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                        if (String.IsNullOrWhiteSpace(serverAddress))
                            serverAddress = null;
                        else if (databaseConfiguration.DatabaseType == DatabaseType.SqlServer)
                        {
                            var match = Regex.Match(serverAddress, @"^(?<server>.+):(?<port>.+)$");
                            if (match.Success)
                            {
                                serverAddress = match.Groups["server"].Value;
                                var portString = match.Groups["port"].Value;
                                if (UInt16.TryParse(portString, out var port))
                                    serverPort = port;
                                else
                                {
                                    await console.WriteAsync($"Failed to parse port \"{portString}\", please try again.", true, cancellationToken).ConfigureAwait(false);
                                    continue;
                                }
                            }
                        }

                        break;
                    }
                    while (true);

                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync($"Enter the database {(isSqliteDB ? "file path" : "name")} (Can be from previous installation. Otherwise, should not exist): ", false, cancellationToken).ConfigureAwait(false);

                string databaseName;
                bool dbExists = false;
                do
                {
                    databaseName = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                    if (!String.IsNullOrWhiteSpace(databaseName))
                    {
                        if (isSqliteDB)
                        {
                            dbExists = await ioManager.FileExists(databaseName, cancellationToken).ConfigureAwait(false);
                            if (!dbExists)
                                databaseName = await ValidateNonExistantSqliteDBName(databaseName, cancellationToken).ConfigureAwait(false);
                        }
                        else
                            dbExists = await PromptYesNo("Does this database already exist? If not, we will attempt to CREATE it. (y/n): ", cancellationToken).ConfigureAwait(false);
                    }

                    if (String.IsNullOrWhiteSpace(databaseName))
                        await console.WriteAsync("Invalid database name!", true, cancellationToken).ConfigureAwait(false);
                    else
                        break;
                }
                while (true);

                bool useWinAuth;
                if (databaseConfiguration.DatabaseType == DatabaseType.SqlServer && platformIdentifier.IsWindows)
                    useWinAuth = await PromptYesNo("Use Windows Authentication? (y/n): ", cancellationToken).ConfigureAwait(false);
                else
                    useWinAuth = false;

                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);

                string username = null;
                string password = null;
                if (!isSqliteDB)
                    if (!useWinAuth)
                    {
                        await console.WriteAsync("Enter username: ", false, cancellationToken).ConfigureAwait(false);
                        username = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync("Enter password: ", false, cancellationToken).ConfigureAwait(false);
                        password = await console.ReadLineAsync(true, cancellationToken).ConfigureAwait(false);
                    }
                    else
                    {
                        await console.WriteAsync("IMPORTANT: If using the service runner, ensure this computer's LocalSystem account has CREATE DATABASE permissions on the target server!", true, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync("The account it uses in MSSQL is usually \"NT AUTHORITY\\SYSTEM\" and the role it needs is usually \"dbcreator\".", true, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync("We'll run a sanity test here, but it won't be indicative of the service's permissions if that is the case", true, cancellationToken).ConfigureAwait(false);
                    }

                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);

                DbConnection testConnection;
                void CreateTestConnection(string connectionString) =>
                    testConnection = dbConnectionFactory.CreateConnection(
                        connectionString,
                        databaseConfiguration.DatabaseType);

                switch (databaseConfiguration.DatabaseType)
                {
                    case DatabaseType.SqlServer:
                        {
                            var csb = new SqlConnectionStringBuilder
                            {
                                ApplicationName = assemblyInformationProvider.VersionPrefix,
                                DataSource = serverAddress ?? "(local)"
                            };

                            if (useWinAuth)
                                csb.IntegratedSecurity = true;
                            else
                            {
                                csb.UserID = username;
                                csb.Password = password;
                            }

                            CreateTestConnection(csb.ConnectionString);
                            csb.InitialCatalog = databaseName;
                            databaseConfiguration.ConnectionString = csb.ConnectionString;
                        }

                        break;
                    case DatabaseType.MariaDB:
                    case DatabaseType.MySql:
                        {
                            // MySQL/MariaDB
                            var csb = new MySqlConnectionStringBuilder
                            {
                                Server = serverAddress ?? "127.0.0.1",
                                UserID = username,
                                Password = password
                            };

                            if (serverPort.HasValue)
                                csb.Port = serverPort.Value;

                            CreateTestConnection(csb.ConnectionString);
                            csb.Database = databaseName;
                            databaseConfiguration.ConnectionString = csb.ConnectionString;
                        }

                        break;
                    case DatabaseType.Sqlite:
                        {
                            var csb = new SqliteConnectionStringBuilder
                            {
                                DataSource = databaseName,
                                Mode = dbExists ? SqliteOpenMode.ReadOnly : SqliteOpenMode.ReadWriteCreate
                            };

                            CreateTestConnection(csb.ConnectionString);
                            databaseConfiguration.ConnectionString = csb.ConnectionString;
                        }

                        break;
                    case DatabaseType.PostgresSql:
                        {
                            var csb = new NpgsqlConnectionStringBuilder
                            {
                                ApplicationName = assemblyInformationProvider.VersionPrefix,
                                Host = serverAddress ?? "127.0.0.1",
                                Password = password,
                                Username = username
                            };

                            if (serverPort.HasValue)
                                csb.Port = serverPort.Value;

                            CreateTestConnection(csb.ConnectionString);
                            csb.Database = databaseName;
                            databaseConfiguration.ConnectionString = csb.ConnectionString;
                        }

                        break;
                    default:
                        throw new InvalidOperationException("Invalid DatabaseType!");
                }

                try
                {
                    await TestDatabaseConnection(testConnection, databaseConfiguration, databaseName, dbExists, cancellationToken).ConfigureAwait(false);

                    return databaseConfiguration;
                }
                catch (OperationCanceledException)
                {
                    throw;
                }
                catch (Exception e)
                {
                    await console.WriteAsync(e.Message, true, cancellationToken).ConfigureAwait(false);
                    await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                    await console.WriteAsync("Retrying database configuration...", true, cancellationToken).ConfigureAwait(false);
                }
            }
            while (true);
        }
        #pragma warning restore CA1502

        async Task<GeneralConfiguration> ConfigureGeneral(CancellationToken cancellationToken)
        {
            var newGeneralConfiguration = new GeneralConfiguration
            {
                SetupWizardMode = SetupWizardMode.Never
            };

            do
            {
                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(String.Format(CultureInfo.InvariantCulture, "Minimum database user password length (leave blank for default of {0}): ", newGeneralConfiguration.MinimumPasswordLength), false, cancellationToken).ConfigureAwait(false);
                var passwordLengthString = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                if (String.IsNullOrWhiteSpace(passwordLengthString))
                    break;
                if (UInt32.TryParse(passwordLengthString, out var passwordLength) && passwordLength >= 0)
                {
                    newGeneralConfiguration.MinimumPasswordLength = passwordLength;
                    break;
                }

                await console.WriteAsync("Please enter a positive integer!", true, cancellationToken).ConfigureAwait(false);
            }
            while (true);

            do
            {
                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(String.Format(CultureInfo.InvariantCulture, "Default timeout for sending and receiving BYOND topics (ms, 0 for infinite, leave blank for default of {0}): ", newGeneralConfiguration.ByondTopicTimeout), false, cancellationToken).ConfigureAwait(false);
                var topicTimeoutString = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                if (String.IsNullOrWhiteSpace(topicTimeoutString))
                    break;
                if (UInt32.TryParse(topicTimeoutString, out var topicTimeout) && topicTimeout >= 0)
                {
                    newGeneralConfiguration.ByondTopicTimeout = topicTimeout;
                    break;
                }

                await console.WriteAsync("Please enter a positive integer!", true, cancellationToken).ConfigureAwait(false);
            }
            while (true);

            await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
            await console.WriteAsync("Enter a GitHub personal access token to bypass some rate limits (this is optional and does not require any scopes)", true, cancellationToken).ConfigureAwait(false);
            await console.WriteAsync("GitHub personal access token: ", false, cancellationToken).ConfigureAwait(false);
            newGeneralConfiguration.GitHubAccessToken = await console.ReadLineAsync(true, cancellationToken).ConfigureAwait(false);
            if (String.IsNullOrWhiteSpace(newGeneralConfiguration.GitHubAccessToken))
                newGeneralConfiguration.GitHubAccessToken = null;

            return newGeneralConfiguration;
        }

        async Task<FileLoggingConfiguration> ConfigureLogging(CancellationToken cancellationToken)
        {
            var fileLoggingConfiguration = new FileLoggingConfiguration();
            await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
            fileLoggingConfiguration.Disable = !await PromptYesNo("Enable file logging? (y/n): ", cancellationToken).ConfigureAwait(false);

            if (!fileLoggingConfiguration.Disable)
            {
                do
                {
                    await console.WriteAsync("Log file directory path (leave blank for default): ", false, cancellationToken).ConfigureAwait(false);
                    fileLoggingConfiguration.Directory = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                    if (String.IsNullOrWhiteSpace(fileLoggingConfiguration.Directory))
                    {
                        fileLoggingConfiguration.Directory = null;
                        break;
                    }

                    // test a write of it
                    await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                    await console.WriteAsync("Testing directory access...", true, cancellationToken).ConfigureAwait(false);
                    try
                    {
                        await ioManager.CreateDirectory(fileLoggingConfiguration.Directory, cancellationToken).ConfigureAwait(false);
                        var testFile = ioManager.ConcatPath(fileLoggingConfiguration.Directory, String.Format(CultureInfo.InvariantCulture, "WizardAccesTest.{0}.deleteme", Guid.NewGuid()));
                        await ioManager.WriteAllBytes(testFile, Array.Empty<byte>(), cancellationToken).ConfigureAwait(false);
                        try
                        {
                            await ioManager.DeleteFile(testFile, cancellationToken).ConfigureAwait(false);
                        }
                        catch (OperationCanceledException)
                        {
                            throw;
                        }
                        catch (Exception e)
                        {
                            await console.WriteAsync(String.Format(CultureInfo.InvariantCulture, "Error deleting test log file: {0}", testFile), true, cancellationToken).ConfigureAwait(false);
                            await console.WriteAsync(e.Message, true, cancellationToken).ConfigureAwait(false);
                            await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                        }

                        break;
                    }
                    catch (OperationCanceledException)
                    {
                        throw;
                    }
                    catch (Exception e)
                    {
                        await console.WriteAsync(e.Message, true, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync("Please verify the path is valid and you have access to it!", true, cancellationToken).ConfigureAwait(false);
                    }
                }
                while (true);

                async Task<LogLevel?> PromptLogLevel(string question)
                {
                    do
                    {
                        await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync(question, true, cancellationToken).ConfigureAwait(false);
                        await console.WriteAsync(String.Format(CultureInfo.InvariantCulture, "Enter one of {0}/{1}/{2}/{3}/{4}/{5} (leave blank for default): ", nameof(LogLevel.Trace), nameof(LogLevel.Debug), nameof(LogLevel.Information), nameof(LogLevel.Warning), nameof(LogLevel.Error), nameof(LogLevel.Critical)), false, cancellationToken).ConfigureAwait(false);
                        var responseString = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                        if (String.IsNullOrWhiteSpace(responseString))
                            return null;
                        if (Enum.TryParse<LogLevel>(responseString, out var logLevel) && logLevel != LogLevel.None)
                            return logLevel;
                        await console.WriteAsync("Invalid log level!", true, cancellationToken).ConfigureAwait(false);
                    }
                    while (true);
                }

                fileLoggingConfiguration.LogLevel = await PromptLogLevel(String.Format(CultureInfo.InvariantCulture, "Enter the level limit for normal logs (default {0}).", fileLoggingConfiguration.LogLevel)).ConfigureAwait(false) ?? fileLoggingConfiguration.LogLevel;
                fileLoggingConfiguration.MicrosoftLogLevel = await PromptLogLevel(String.Format(CultureInfo.InvariantCulture, "Enter the level limit for Microsoft logs (VERY verbose, default {0}).", fileLoggingConfiguration.MicrosoftLogLevel)).ConfigureAwait(false) ?? fileLoggingConfiguration.MicrosoftLogLevel;
            }

            return fileLoggingConfiguration;
        }

        async Task<ControlPanelConfiguration> ConfigureControlPanel(CancellationToken cancellationToken)
        {
            var config = new ControlPanelConfiguration
            {
                Enable = await PromptYesNo("Enable the web control panel (Incomplete)? (y/n): ", cancellationToken).ConfigureAwait(false),
                AllowAnyOrigin = await PromptYesNo("Allow web control panels hosted elsewhere to access the server? (Access-Control-Allow-Origin: *) (y/n): ", cancellationToken).ConfigureAwait(false)
            };

            if (!config.AllowAnyOrigin)
            {
                await console.WriteAsync("Enter a comma seperated list of CORS allowed origins (optional): ", false, cancellationToken).ConfigureAwait(false);
                var commaSeperatedOrigins = await console.ReadLineAsync(false, cancellationToken).ConfigureAwait(false);
                if (!String.IsNullOrWhiteSpace(commaSeperatedOrigins))
                {
                    var splits = commaSeperatedOrigins.Split(',');
                    config.AllowedOrigins = new List<string>(splits.Select(x => x.Trim()));
                }
            }

            return config;
        }

        async Task SaveConfiguration(string userConfigFileName, ushort? hostingPort, DatabaseConfiguration databaseConfiguration, GeneralConfiguration newGeneralConfiguration, FileLoggingConfiguration fileLoggingConfiguration, ControlPanelConfiguration controlPanelConfiguration, CancellationToken cancellationToken)
        {
            await console.WriteAsync(String.Format(CultureInfo.InvariantCulture, "Configuration complete! Saving to {0}", userConfigFileName), true, cancellationToken).ConfigureAwait(false);

            newGeneralConfiguration.ApiPort = hostingPort ?? GeneralConfiguration.DefaultApiPort;
            newGeneralConfiguration.ConfigVersion = GeneralConfiguration.CurrentConfigVersion;
            var map = new Dictionary<string, object>()
            {
                { DatabaseConfiguration.Section, databaseConfiguration },
                { GeneralConfiguration.Section, newGeneralConfiguration },
                { FileLoggingConfiguration.Section, fileLoggingConfiguration },
                { ControlPanelConfiguration.Section, controlPanelConfiguration }
            };

            var json = JsonConvert.SerializeObject(map, Formatting.Indented);
            var configBytes = Encoding.UTF8.GetBytes(json);

            try
            {
                await ioManager.WriteAllBytes(userConfigFileName, configBytes, cancellationToken).ConfigureAwait(false);
            }
            catch (OperationCanceledException)
            {
                throw;
            }
            catch (Exception e)
            {
                await console.WriteAsync(e.Message, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync("For your convienence, here's the json we tried to write out:", true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(json, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);
                await console.WriteAsync("Press any key to exit...", true, cancellationToken).ConfigureAwait(false);
                await console.PressAnyKeyAsync(cancellationToken).ConfigureAwait(false);
                throw new OperationCanceledException();
            }
        }

        async Task RunWizard(string userConfigFileName, CancellationToken cancellationToken)
        {
            // welcome message
            await console.WriteAsync("Welcome to tgstation-server 4!", true, cancellationToken).ConfigureAwait(false);
            await console.WriteAsync("This wizard will help you configure your server.", true, cancellationToken).ConfigureAwait(false);

            var hostingPort = await PromptForHostingPort(cancellationToken).ConfigureAwait(false);

            var databaseConfiguration = await ConfigureDatabase(cancellationToken).ConfigureAwait(false);

            var newGeneralConfiguration = await ConfigureGeneral(cancellationToken).ConfigureAwait(false);

            var fileLoggingConfiguration = await ConfigureLogging(cancellationToken).ConfigureAwait(false);

            var controlPanelConfiguration = await ConfigureControlPanel(cancellationToken).ConfigureAwait(false);

            await console.WriteAsync(null, true, cancellationToken).ConfigureAwait(false);

            await SaveConfiguration(userConfigFileName, hostingPort, databaseConfiguration, newGeneralConfiguration, fileLoggingConfiguration, controlPanelConfiguration, cancellationToken).ConfigureAwait(false);
        }

        async Task CheckRunWizard(CancellationToken cancellationToken)
        {
            var setupWizardMode = generalConfiguration.SetupWizardMode;
            if (setupWizardMode == SetupWizardMode.Never)
                return;

            var forceRun = setupWizardMode == SetupWizardMode.Force || setupWizardMode == SetupWizardMode.Only;
            if (!console.Available)
            {
                if (forceRun)
                    throw new InvalidOperationException("Asked to run setup wizard with no console avaliable!");
                return;
            }

            var userConfigFileName = String.Format(CultureInfo.InvariantCulture, "appsettings.{0}.json", hostingEnvironment.EnvironmentName);

            async Task HandleSetupCancel()
            {
                await console.WriteAsync(String.Empty, true, default).ConfigureAwait(false);
                await console.WriteAsync("Aborting setup!", true, default).ConfigureAwait(false);
            }

            // Link passed cancellationToken with cancel key press
            Task finalTask = Task.CompletedTask;
            using (var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, console.CancelKeyPress))
            using ((cancellationToken = cts.Token).Register(() => finalTask = HandleSetupCancel()))
                try
                {
                    var exists = await ioManager.FileExists(userConfigFileName, cancellationToken).ConfigureAwait(false);

                    bool shouldRunBasedOnAutodetect;
                    if (exists)
                    {
                        var bytes = await ioManager.ReadAllBytes(userConfigFileName, cancellationToken).ConfigureAwait(false);
                        var contents = Encoding.UTF8.GetString(bytes);
                        var existingConfigIsEmpty = String.IsNullOrWhiteSpace(contents) || contents.Trim() == "{}";
                        shouldRunBasedOnAutodetect = existingConfigIsEmpty;
                    }
                    else
                        shouldRunBasedOnAutodetect = true;

                    if (!shouldRunBasedOnAutodetect)
                    {
                        if (forceRun)
                        {
                            await console.WriteAsync(String.Format(CultureInfo.InvariantCulture, "The configuration settings are requesting the setup wizard be run, but you already appear to have a configuration file ({0})!", userConfigFileName), true, cancellationToken).ConfigureAwait(false);

                            forceRun = await PromptYesNo("Continue running setup wizard? (y/n): ", cancellationToken).ConfigureAwait(false);
                        }

                        if (!forceRun)
                            return;
                    }

                    // flush the logs to prevent console conflicts
                    await asyncDelayer.Delay(TimeSpan.FromSeconds(1), cancellationToken).ConfigureAwait(false);

                    await RunWizard(userConfigFileName, cancellationToken).ConfigureAwait(false);
                }
                finally
                {
                    await finalTask.ConfigureAwait(false);
                }
        }

        public async Task StartAsync(CancellationToken cancellationToken)
        {
            await CheckRunWizard(cancellationToken).ConfigureAwait(false);
            applicationLifetime.StopApplication();
        }

        public Task StopAsync(CancellationToken cancellationToken) => Task.CompletedTask;
    }
}