using System; using System.Collections.Generic; using System.Linq; using System.Threading; using System.IO; using System.Threading.Tasks; using SharedLibrary; using SharedLibrary.Interfaces; using SharedLibrary.Commands; using SharedLibrary.Helpers; using SharedLibrary.Exceptions; namespace IW4MAdmin { class ApplicationManager : IManager { private List _servers; public List Servers => _servers.OrderByDescending(s => s.ClientNum).ToList(); public ILogger Logger { get; private set; } public bool Running { get; private set; } static ApplicationManager Instance; List TaskStatuses; Database ClientDatabase; Database AliasesDatabase; IPenaltyList ClientPenalties; List Commands; List MessageTokens; Kayak.IScheduler webServiceTask; Thread WebThread; List PrivilegedClients; #if FTP_LOG const int UPDATE_FREQUENCY = 15000; #else const int UPDATE_FREQUENCY = 300; #endif private ApplicationManager() { Logger = new Logger("Logs/IW4MAdmin.log"); _servers = new List(); Commands = new List(); TaskStatuses = new List(); MessageTokens = new List(); ClientDatabase = new ClientsDB("Database/clients.rm", Logger); AliasesDatabase = new AliasesDB("Database/aliases.rm", Logger); ClientPenalties = new PenaltyList(); } public IList GetServers() { return Servers; } public IList GetCommands() { return Commands; } public static ApplicationManager GetInstance() { return Instance ?? (Instance = new ApplicationManager()); } public void Init() { #region WEBSERVICE SharedLibrary.WebService.Init(); webServiceTask = WebService.GetScheduler(); WebThread = new Thread(webServiceTask.Start) { Name = "Web Thread" }; WebThread.Start(); #endregion #region PLUGINS SharedLibrary.Plugins.PluginImporter.Load(this); foreach (var Plugin in SharedLibrary.Plugins.PluginImporter.ActivePlugins) { try { Plugin.OnLoadAsync(this); } catch (Exception e) { Logger.WriteError($"An error occured loading plugin {Plugin.Name}"); Logger.WriteDebug($"Exception: {e.Message}"); Logger.WriteDebug($"Stack Trace: {e.StackTrace}"); } } #endregion #region CONFIG var Configs = Directory.EnumerateFiles("config/servers").Where(x => x.Contains(".cfg")); if (Configs.Count() == 0) ServerConfigurationGenerator.Generate(); foreach (var file in Configs) { var Conf = ServerConfiguration.Read(file); Task.Run(async () => { try { var ServerInstance = new IW4MServer(this, Conf); await ServerInstance.Initialize(); lock (_servers) { _servers.Add(ServerInstance); } Logger.WriteVerbose($"Now monitoring {ServerInstance.Hostname}"); // this way we can keep track of execution time and see if problems arise. var Status = new AsyncStatus(ServerInstance, UPDATE_FREQUENCY); lock (TaskStatuses) { TaskStatuses.Add(Status); } } catch (ServerException e) { Logger.WriteError($"Not monitoring server {Conf.IP}:{Conf.Port} due to uncorrectable errors"); if (e.GetType() == typeof(DvarException)) Logger.WriteDebug($"Could not get the dvar value for {(e as DvarException).Data["dvar_name"]} (ensure the server has a map loaded)"); else if (e.GetType() == typeof(NetworkException)) { Logger.WriteDebug(e.Message); Logger.WriteDebug($"Internal Exception: {e.Data["internal_exception"]}"); } } }); } #endregion #region COMMANDS if ((ClientDatabase as ClientsDB).GetOwner() == null) Commands.Add(new COwner()); Commands.Add(new CQuit()); Commands.Add(new CKick()); Commands.Add(new CSay()); Commands.Add(new CTempBan()); Commands.Add(new CBan()); Commands.Add(new CWhoAmI()); Commands.Add(new CList()); Commands.Add(new CHelp()); Commands.Add(new CFastRestart()); Commands.Add(new CMapRotate()); Commands.Add(new CSetLevel()); Commands.Add(new CUsage()); Commands.Add(new CUptime()); Commands.Add(new CWarn()); Commands.Add(new CWarnClear()); Commands.Add(new CUnban()); Commands.Add(new CListAdmins()); Commands.Add(new CLoadMap()); Commands.Add(new CFindPlayer()); Commands.Add(new CListRules()); Commands.Add(new CPrivateMessage()); Commands.Add(new CReload()); Commands.Add(new CFlag()); Commands.Add(new CReport()); Commands.Add(new CListReports()); Commands.Add(new CListBanInfo()); Commands.Add(new CListAlias()); Commands.Add(new CExecuteRCON()); Commands.Add(new CFindAllPlayers()); Commands.Add(new CPlugins()); Commands.Add(new CIP()); Commands.Add(new CMask()); foreach (Command C in SharedLibrary.Plugins.PluginImporter.ActiveCommands) Commands.Add(C); #endregion #region ADMINS PrivilegedClients = GetClientDatabase().GetAdmins(); #endregion Running = true; } public void Start() { while (Running) { for (int i = 0; i < TaskStatuses.Count; i++) { var Status = TaskStatuses[i]; if (Status.RequestedTask == null || Status.RequestedTask.IsCompleted) { Status.Update(new Task(() => (Status.Dependant as Server).ProcessUpdatesAsync(Status.GetToken()))); if (Status.RunAverage > 1000 + UPDATE_FREQUENCY) Logger.WriteWarning($"Update task average execution is longer than desired for {(Status.Dependant as Server).GetIP()}::{(Status.Dependant as Server).GetPort()} [{Status.RunAverage}ms]"); } } Thread.Sleep(UPDATE_FREQUENCY); } #if !DEBUG foreach (var S in Servers) S.Broadcast("^1IW4MAdmin going offline!"); #endif _servers.Clear(); WebThread.Abort(); webServiceTask.Stop(); } public void Stop() { Running = false; } public ClientsDB GetClientDatabase() { return ClientDatabase as ClientsDB; } public AliasesDB GetAliasesDatabase() { return AliasesDatabase as AliasesDB; } public IPenaltyList GetClientPenalties() { return ClientPenalties; } public ILogger GetLogger() { return Logger; } public IList GetMessageTokens() { return MessageTokens; } public IList GetActiveClients() { var ActiveClients = new List(); foreach (var server in _servers) ActiveClients.AddRange(server.Players.Where(p => p != null)); return ActiveClients; } public IList GetAliasClients(Player Origin) { List databaseIDs = new List(); foreach (Aliases A in GetAliases(Origin)) databaseIDs.Add(A.Number); return GetClientDatabase().GetPlayers(databaseIDs); } public IList GetAliases(Player Origin) { List allAliases = new List(); if (Origin == null) return allAliases; Aliases currentIdentityAliases = GetAliasesDatabase().GetPlayerAliases(Origin.DatabaseID); if (currentIdentityAliases == null) return allAliases; GetAliases(allAliases, currentIdentityAliases); if (Origin.Alias != null) allAliases.Add(Origin.Alias); allAliases.Add(currentIdentityAliases); return allAliases; } public IList GetPrivilegedClients() { return PrivilegedClients; } private void GetAliases(List returnAliases, Aliases currentAlias) { foreach (String IP in currentAlias.IPS) { List Matching = GetAliasesDatabase().GetPlayerAliases(IP); foreach (Aliases I in Matching) { if (!returnAliases.Contains(I) && returnAliases.Find(x => x.Number == I.Number) == null) { returnAliases.Add(I); GetAliases(returnAliases, I); } } } } } }