using IW4MAdmin.Application.Factories; using IW4MAdmin.Application.Helpers; using IW4MAdmin.Application.IO; using IW4MAdmin.Application.Migration; using IW4MAdmin.Application.Misc; using Microsoft.Extensions.DependencyInjection; using SharedLibraryCore; using SharedLibraryCore.Configuration; using SharedLibraryCore.Exceptions; using SharedLibraryCore.Helpers; using SharedLibraryCore.Interfaces; using System; using System.Linq; using System.Text; using System.Threading; using System.Threading.Tasks; namespace IW4MAdmin.Application { public class Program { public static BuildNumber Version { get; private set; } = BuildNumber.Parse(Utilities.GetVersionAsString()); public static ApplicationManager ServerManager; private static Task ApplicationTask; private static readonly BuildNumber _fallbackVersion = BuildNumber.Parse("99.99.99.99"); private static ServiceProvider serviceProvider; /// /// entrypoint of the application /// /// public static async Task Main() { AppDomain.CurrentDomain.SetData("DataDirectory", Utilities.OperatingDirectory); Console.OutputEncoding = Encoding.UTF8; Console.ForegroundColor = ConsoleColor.Gray; Console.CancelKeyPress += new ConsoleCancelEventHandler(OnCancelKey); Console.WriteLine("====================================================="); Console.WriteLine(" IW4MAdmin"); Console.WriteLine(" by RaidMax "); Console.WriteLine($" Version {Utilities.GetVersionAsString()}"); Console.WriteLine("====================================================="); await LaunchAsync(); } /// /// event callback executed when the control + c combination is detected /// gracefully stops the server manager and waits for all tasks to finish /// /// /// private static async void OnCancelKey(object sender, ConsoleCancelEventArgs e) { ServerManager?.Stop(); await ApplicationTask; } /// /// task that initializes application and starts the application monitoring and runtime tasks /// /// private static async Task LaunchAsync() { restart: ITranslationLookup translationLookup = null; try { // do any needed housekeeping file/folder migrations ConfigurationMigration.MoveConfigFolder10518(null); ConfigurationMigration.CheckDirectories(); var services = ConfigureServices(); serviceProvider = services.BuildServiceProvider(); ServerManager = (ApplicationManager)serviceProvider.GetRequiredService(); translationLookup = serviceProvider.GetRequiredService(); ServerManager.Logger.WriteInfo(Utilities.CurrentLocalization.LocalizationIndex["MANAGER_VERSION"].FormatExt(Version)); await CheckVersion(translationLookup); await ServerManager.Init(); } catch (Exception e) { string failMessage = translationLookup == null ? "Failed to initalize IW4MAdmin" : translationLookup["MANAGER_INIT_FAIL"]; string exitMessage = translationLookup == null ? "Press any key to exit..." : translationLookup["MANAGER_EXIT"]; Console.WriteLine(failMessage); while (e.InnerException != null) { e = e.InnerException; } if (e is ConfigurationException configException) { if (translationLookup != null) { Console.WriteLine(translationLookup[configException.Message].FormatExt(configException.ConfigurationFileName)); } foreach (string error in configException.Errors) { Console.WriteLine(error); } } else { Console.WriteLine(e.Message); } Console.WriteLine(exitMessage); Console.ReadKey(); return; } try { ApplicationTask = RunApplicationTasksAsync(); await ApplicationTask; } catch { } if (ServerManager.IsRestartRequested) { goto restart; } serviceProvider.Dispose(); } /// /// runs the core application tasks /// /// private static async Task RunApplicationTasksAsync() { var webfrontTask = ServerManager.GetApplicationSettings().Configuration().EnableWebFront ? WebfrontCore.Program.Init(ServerManager, serviceProvider, ServerManager.CancellationToken) : Task.CompletedTask; // we want to run this one on a manual thread instead of letting the thread pool handle it, // because we can't exit early from waiting on console input, and it prevents us from restarting var inputThread = new Thread(async () => await ReadConsoleInput()); inputThread.Start(); var tasks = new[] { ServerManager.Start(), webfrontTask, }; await Task.WhenAll(tasks); ServerManager.Logger.WriteVerbose(Utilities.CurrentLocalization.LocalizationIndex["MANAGER_SHUTDOWN_SUCCESS"]); } /// /// checks for latest version of the application /// notifies user if an update is available /// /// private static async Task CheckVersion(ITranslationLookup translationLookup) { var api = API.Master.Endpoint.Get(); var loc = translationLookup; var version = new API.Master.VersionInfo() { CurrentVersionStable = _fallbackVersion }; try { version = await api.GetVersion(1); } catch (Exception e) { ServerManager.Logger.WriteWarning(loc["MANAGER_VERSION_FAIL"]); while (e.InnerException != null) { e = e.InnerException; } ServerManager.Logger.WriteDebug(e.Message); } if (version.CurrentVersionStable == _fallbackVersion) { Console.ForegroundColor = ConsoleColor.Red; Console.WriteLine(loc["MANAGER_VERSION_FAIL"]); Console.ForegroundColor = ConsoleColor.Gray; } #if !PRERELEASE else if (version.CurrentVersionStable > Version) { Console.ForegroundColor = ConsoleColor.DarkYellow; Console.WriteLine($"IW4MAdmin {loc["MANAGER_VERSION_UPDATE"]} [v{version.CurrentVersionStable.ToString()}]"); Console.WriteLine(loc["MANAGER_VERSION_CURRENT"].FormatExt($"[v{Version.ToString()}]")); Console.ForegroundColor = ConsoleColor.Gray; } #else else if (version.CurrentVersionPrerelease > Version) { Console.ForegroundColor = ConsoleColor.DarkYellow; Console.WriteLine($"IW4MAdmin-Prerelease {loc["MANAGER_VERSION_UPDATE"]} [v{version.CurrentVersionPrerelease.ToString()}-pr]"); Console.WriteLine(loc["MANAGER_VERSION_CURRENT"].FormatExt($"[v{Version.ToString()}-pr]")); Console.ForegroundColor = ConsoleColor.Gray; } #endif else { Console.ForegroundColor = ConsoleColor.Green; Console.WriteLine(loc["MANAGER_VERSION_SUCCESS"]); Console.ForegroundColor = ConsoleColor.Gray; } } /// /// reads input from the console and executes entered commands on the default server /// /// private static async Task ReadConsoleInput() { string lastCommand; var Origin = Utilities.IW4MAdminClient(ServerManager.Servers[0]); try { while (!ServerManager.CancellationToken.IsCancellationRequested) { lastCommand = Console.ReadLine(); if (lastCommand?.Length > 0) { if (lastCommand?.Length > 0) { GameEvent E = new GameEvent() { Type = GameEvent.EventType.Command, Data = lastCommand, Origin = Origin, Owner = ServerManager.Servers[0] }; ServerManager.GetEventHandler().AddEvent(E); await E.WaitAsync(Utilities.DefaultCommandTimeout, ServerManager.CancellationToken); Console.Write('>'); } } } } catch (OperationCanceledException) { } } /// /// Configures the dependency injection services /// private static IServiceCollection ConfigureServices() { var defaultLogger = new Logger("IW4MAdmin-Manager"); var pluginImporter = new PluginImporter(defaultLogger); var serviceCollection = new ServiceCollection(); serviceCollection.AddSingleton(_serviceProvider => serviceCollection) .AddSingleton(new BaseConfigurationHandler("IW4MAdminSettings") as IConfigurationHandler) .AddSingleton(new BaseConfigurationHandler("CommandConfiguration") as IConfigurationHandler) .AddSingleton(_serviceProvider => _serviceProvider.GetRequiredService>().Configuration()) .AddSingleton(_serviceProvider => _serviceProvider.GetRequiredService>().Configuration() ?? new CommandConfiguration()) .AddSingleton(_serviceProvider => defaultLogger) .AddSingleton() .AddSingleton() .AddSingleton() .AddSingleton() .AddSingleton() .AddSingleton(_serviceProvider => { var config = _serviceProvider.GetRequiredService>().Configuration(); return Localization.Configure.Initialize(useLocalTranslation: config?.UseLocalTranslations ?? false, customLocale: config?.EnableCustomLocale ?? false ? (config.CustomLocale ?? "en-US") : "en-US"); }) .AddSingleton(); // register the native commands foreach (var commandType in typeof(SharedLibraryCore.Commands.QuitCommand).Assembly.GetTypes() .Where(_command => _command.BaseType == typeof(Command))) { defaultLogger.WriteInfo($"Registered native command type {commandType.Name}"); serviceCollection.AddSingleton(typeof(IManagerCommand), commandType); } // register the plugin implementations var pluginImplementations = pluginImporter.DiscoverAssemblyPluginImplementations(); foreach (var pluginType in pluginImplementations.Item1) { defaultLogger.WriteInfo($"Registered plugin type {pluginType.FullName}"); serviceCollection.AddSingleton(typeof(IPlugin), pluginType); } // register the plugin commands foreach (var commandType in pluginImplementations.Item2) { defaultLogger.WriteInfo($"Registered plugin command type {commandType.FullName}"); serviceCollection.AddSingleton(typeof(IManagerCommand), commandType); } // register any script plugins foreach (var scriptPlugin in pluginImporter.DiscoverScriptPlugins()) { serviceCollection.AddSingleton(scriptPlugin); } return serviceCollection; } } }