IW4M-Admin/WebfrontCore/ViewComponents/ServerListViewComponent.cs

using System;
using System.Collections.Generic;
using Microsoft.AspNetCore.Mvc;
using SharedLibraryCore;
using SharedLibraryCore.Dtos;
using System.Linq;
using System.Net;
using System.Threading;
using System.Threading.Tasks;
using Data.Models.Client.Stats;
using Microsoft.AspNetCore.Hosting.Server;
using SharedLibraryCore.Configuration;
using SharedLibraryCore.Interfaces;
using static SharedLibraryCore.Server;

namespace WebfrontCore.ViewComponents
{
    public class ServerListViewComponent : ViewComponent
    {
        private readonly IServerDataViewer _serverDataViewer;
        private readonly ApplicationConfiguration _appConfig;

        public ServerListViewComponent(IServerDataViewer serverDataViewer,
            ApplicationConfiguration applicationConfiguration)
        {
            _serverDataViewer = serverDataViewer;
            _appConfig = applicationConfiguration;
        }

        public IViewComponentResult Invoke(Game? game)
        {
            var servers = Program.Manager.GetServers().Where(server => !game.HasValue || server.GameName == game);

            var serverInfo = new List<ServerInfo>();

            foreach (var server in servers)
            {
                var serverId = server.GetIdForServer().Result;
                var clientHistory = _serverDataViewer.ClientHistoryAsync(_appConfig.MaxClientHistoryTime,
                                            CancellationToken.None).Result?
                                        .FirstOrDefault(history => history.ServerId == serverId) ??
                                    new ClientHistoryInfo
                                    {
                                        ServerId = serverId
                                    };

                var counts = clientHistory.ClientCounts?.AsEnumerable() ?? Enumerable.Empty<ClientCountSnapshot>();

                if (server.ClientHistory.Count > 0)
                {
                    counts = counts.Union(server.ClientHistory
                        .Select(history => history.ToClientCountSnapshot()).Where(history =>
                            history.Time > (clientHistory.ClientCounts.LastOrDefault()?.Time ?? DateTime.MinValue)));
                }

                serverInfo.Add(new ServerInfo()
                {
                    Name = server.Hostname,
                    ID = server.EndPoint,
                    Port = server.Port,
                    Map = server.CurrentMap.Alias,
                    ClientCount = server.Clients.Count(client => client != null),
                    MaxClients = server.MaxClients,
                    GameType = server.Gametype,
                    PlayerHistory = server.ClientHistory.ToArray(),
                    Players = server.GetClientsAsList()
                        .Select(p => new PlayerInfo()
                        {
                            Name = p.Name,
                            ClientId = p.ClientId,
                            Level = p.Level.ToLocalizedLevelName(),
                            LevelInt = (int) p.Level,
                            Tag = p.Tag,
                            ZScore = p.GetAdditionalProperty<EFClientStatistics>(IW4MAdmin.Plugins.Stats.Helpers
                                .StatManager
                                .CLIENT_STATS_KEY)?.ZScore
                        }).ToList(),
                    ChatHistory = server.ChatHistory.ToList(),
                    ClientCountHistory =
                        counts.Where(history => history.Time >= DateTime.UtcNow - _appConfig.MaxClientHistoryTime)
                            .ToList(),
                    Online = !server.Throttled,
                    IPAddress =
                        $"{(server.ResolvedIpEndPoint.Address.IsInternal() ? Program.Manager.ExternalIPAddress : server.IP)}:{server.Port}",
                    ConnectProtocolUrl = server.EventParser.URLProtocolFormat.FormatExt(
                        server.ResolvedIpEndPoint.Address.IsInternal() ? Program.Manager.ExternalIPAddress : server.IP,
                        server.Port)
                });
            }

            return View("_List", serverInfo);
        }
    }
}