325 lines
14 KiB
C#
325 lines
14 KiB
C#
using IW4MAdmin.Application.API.Master;
|
|
using IW4MAdmin.Application.EventParsers;
|
|
using IW4MAdmin.Application.Factories;
|
|
using IW4MAdmin.Application.Helpers;
|
|
using IW4MAdmin.Application.Meta;
|
|
using IW4MAdmin.Application.Migration;
|
|
using IW4MAdmin.Application.Misc;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using RestEase;
|
|
using SharedLibraryCore;
|
|
using SharedLibraryCore.Configuration;
|
|
using SharedLibraryCore.Database.Models;
|
|
using SharedLibraryCore.Dtos.Meta.Responses;
|
|
using SharedLibraryCore.Exceptions;
|
|
using SharedLibraryCore.Helpers;
|
|
using SharedLibraryCore.Interfaces;
|
|
using SharedLibraryCore.QueryHelper;
|
|
using SharedLibraryCore.Repositories;
|
|
using SharedLibraryCore.Services;
|
|
using Stats.Dtos;
|
|
using StatsWeb;
|
|
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 ServiceProvider serviceProvider;
|
|
|
|
/// <summary>
|
|
/// entrypoint of the application
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public static async Task Main(string[] args)
|
|
{
|
|
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(args);
|
|
}
|
|
|
|
/// <summary>
|
|
/// event callback executed when the control + c combination is detected
|
|
/// gracefully stops the server manager and waits for all tasks to finish
|
|
/// </summary>
|
|
/// <param name="sender"></param>
|
|
/// <param name="e"></param>
|
|
private static async void OnCancelKey(object sender, ConsoleCancelEventArgs e)
|
|
{
|
|
ServerManager?.Stop();
|
|
await ApplicationTask;
|
|
}
|
|
|
|
/// <summary>
|
|
/// task that initializes application and starts the application monitoring and runtime tasks
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
private static async Task LaunchAsync(string[] args)
|
|
{
|
|
restart:
|
|
ITranslationLookup translationLookup = null;
|
|
try
|
|
{
|
|
// do any needed housekeeping file/folder migrations
|
|
ConfigurationMigration.MoveConfigFolder10518(null);
|
|
ConfigurationMigration.CheckDirectories();
|
|
|
|
var services = ConfigureServices(args);
|
|
serviceProvider = services.BuildServiceProvider();
|
|
var versionChecker = serviceProvider.GetRequiredService<IMasterCommunication>();
|
|
ServerManager = (ApplicationManager)serviceProvider.GetRequiredService<IManager>();
|
|
translationLookup = serviceProvider.GetRequiredService<ITranslationLookup>();
|
|
|
|
ServerManager.Logger.WriteInfo(Utilities.CurrentLocalization.LocalizationIndex["MANAGER_VERSION"].FormatExt(Version));
|
|
|
|
await versionChecker.CheckVersion();
|
|
await ServerManager.Init();
|
|
}
|
|
|
|
catch (Exception e)
|
|
{
|
|
string failMessage = translationLookup == null ? "Failed to initalize IW4MAdmin" : translationLookup["MANAGER_INIT_FAIL"];
|
|
string exitMessage = translationLookup == null ? "Press enter 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);
|
|
await Console.In.ReadAsync(new char[1], 0, 1);
|
|
return;
|
|
}
|
|
|
|
try
|
|
{
|
|
ApplicationTask = RunApplicationTasksAsync();
|
|
await ApplicationTask;
|
|
}
|
|
|
|
catch (Exception e)
|
|
{
|
|
string failMessage = translationLookup == null ? "Failed to initalize IW4MAdmin" : translationLookup["MANAGER_INIT_FAIL"];
|
|
Console.WriteLine($"{failMessage}: {e.GetExceptionInfo()}");
|
|
}
|
|
|
|
if (ServerManager.IsRestartRequested)
|
|
{
|
|
goto restart;
|
|
}
|
|
|
|
serviceProvider.Dispose();
|
|
}
|
|
|
|
/// <summary>
|
|
/// runs the core application tasks
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
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,
|
|
serviceProvider.GetRequiredService<IMasterCommunication>().RunUploadStatus(ServerManager.CancellationToken)
|
|
};
|
|
|
|
await Task.WhenAll(tasks);
|
|
|
|
ServerManager.Logger.WriteVerbose(Utilities.CurrentLocalization.LocalizationIndex["MANAGER_SHUTDOWN_SUCCESS"]);
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// reads input from the console and executes entered commands on the default server
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
private static async Task ReadConsoleInput()
|
|
{
|
|
if (Console.IsInputRedirected)
|
|
{
|
|
ServerManager.Logger.WriteInfo("Disabling console input as it has been redirected");
|
|
return;
|
|
}
|
|
|
|
string lastCommand;
|
|
var Origin = Utilities.IW4MAdminClient(ServerManager.Servers[0]);
|
|
|
|
try
|
|
{
|
|
while (!ServerManager.CancellationToken.IsCancellationRequested)
|
|
{
|
|
lastCommand = await Console.In.ReadLineAsync();
|
|
|
|
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.AddEvent(E);
|
|
await E.WaitAsync(Utilities.DefaultCommandTimeout, ServerManager.CancellationToken);
|
|
Console.Write('>');
|
|
}
|
|
}
|
|
}
|
|
}
|
|
catch (OperationCanceledException)
|
|
{ }
|
|
}
|
|
|
|
/// <summary>
|
|
/// Configures the dependency injection services
|
|
/// </summary>
|
|
private static IServiceCollection ConfigureServices(string[] args)
|
|
{
|
|
var defaultLogger = new Logger("IW4MAdmin-Manager");
|
|
var pluginImporter = new PluginImporter(defaultLogger);
|
|
|
|
var serviceCollection = new ServiceCollection();
|
|
serviceCollection.AddSingleton<IServiceCollection>(_serviceProvider => serviceCollection)
|
|
.AddSingleton(new BaseConfigurationHandler<ApplicationConfiguration>("IW4MAdminSettings") as IConfigurationHandler<ApplicationConfiguration>)
|
|
.AddSingleton(new BaseConfigurationHandler<CommandConfiguration>("CommandConfiguration") as IConfigurationHandler<CommandConfiguration>)
|
|
.AddSingleton(_serviceProvider => _serviceProvider.GetRequiredService<IConfigurationHandler<ApplicationConfiguration>>().Configuration())
|
|
.AddSingleton(_serviceProvider => _serviceProvider.GetRequiredService<IConfigurationHandler<CommandConfiguration>>().Configuration() ?? new CommandConfiguration())
|
|
.AddSingleton<ILogger>(_serviceProvider => defaultLogger)
|
|
.AddSingleton<IPluginImporter, PluginImporter>()
|
|
.AddSingleton<IMiddlewareActionHandler, MiddlewareActionHandler>()
|
|
.AddSingleton<IRConConnectionFactory, RConConnectionFactory>()
|
|
.AddSingleton<IGameServerInstanceFactory, GameServerInstanceFactory>()
|
|
.AddSingleton<IConfigurationHandlerFactory, ConfigurationHandlerFactory>()
|
|
.AddSingleton<IParserRegexFactory, ParserRegexFactory>()
|
|
.AddSingleton<IDatabaseContextFactory, DatabaseContextFactory>()
|
|
.AddSingleton<IGameLogReaderFactory, GameLogReaderFactory>()
|
|
.AddSingleton<IScriptCommandFactory, ScriptCommandFactory>()
|
|
.AddSingleton<IAuditInformationRepository, AuditInformationRepository>()
|
|
.AddSingleton<IEntityService<EFClient>, ClientService>()
|
|
.AddSingleton<IMetaService, MetaService>()
|
|
.AddSingleton<IMetaRegistration, MetaRegistration>()
|
|
.AddSingleton<IResourceQueryHelper<ClientPaginationRequest, ReceivedPenaltyResponse>, ReceivedPenaltyResourceQueryHelper>()
|
|
.AddSingleton<IResourceQueryHelper<ClientPaginationRequest, AdministeredPenaltyResponse>, AdministeredPenaltyResourceQueryHelper>()
|
|
.AddSingleton<IResourceQueryHelper<ClientPaginationRequest, UpdatedAliasResponse>, UpdatedAliasResourceQueryHelper>()
|
|
.AddSingleton<IResourceQueryHelper<ChatSearchQuery, MessageResponse>, ChatResourceQueryHelper>()
|
|
.AddTransient<IParserPatternMatcher, ParserPatternMatcher>()
|
|
.AddSingleton(_serviceProvider =>
|
|
{
|
|
var config = _serviceProvider.GetRequiredService<IConfigurationHandler<ApplicationConfiguration>>().Configuration();
|
|
return Localization.Configure.Initialize(useLocalTranslation: config?.UseLocalTranslations ?? false,
|
|
apiInstance: _serviceProvider.GetRequiredService<IMasterApi>(),
|
|
customLocale: config?.EnableCustomLocale ?? false ? (config.CustomLocale ?? "en-US") : "en-US");
|
|
})
|
|
.AddSingleton<IManager, ApplicationManager>()
|
|
.AddSingleton(_serviceProvider => RestClient
|
|
.For<IMasterApi>(Utilities.IsDevelopment ? new Uri("http://127.0.0.1:8080") : _serviceProvider
|
|
.GetRequiredService<IConfigurationHandler<ApplicationConfiguration>>().Configuration()?.MasterUrl ??
|
|
new ApplicationConfiguration().MasterUrl))
|
|
.AddSingleton<IMasterCommunication, MasterCommunication>();
|
|
|
|
if (args.Contains("serialevents"))
|
|
{
|
|
serviceCollection.AddSingleton<IEventHandler, SerialGameEventHandler>();
|
|
}
|
|
else
|
|
{
|
|
serviceCollection.AddSingleton<IEventHandler, GameEventHandler>();
|
|
}
|
|
|
|
// 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);
|
|
}
|
|
|
|
// register any eventable types
|
|
foreach (var assemblyType in typeof(Program).Assembly.GetTypes()
|
|
.Where(_asmType => typeof(IRegisterEvent).IsAssignableFrom(_asmType))
|
|
.Union(pluginImplementations
|
|
.Item1.SelectMany(_asm => _asm.Assembly.GetTypes())
|
|
.Distinct()
|
|
.Where(_asmType => typeof(IRegisterEvent).IsAssignableFrom(_asmType))))
|
|
{
|
|
var instance = Activator.CreateInstance(assemblyType) as IRegisterEvent;
|
|
serviceCollection.AddSingleton(instance);
|
|
}
|
|
|
|
return serviceCollection;
|
|
}
|
|
}
|
|
}
|