HyperbolicDownloader/HyperbolicDownloader/Networking/HostsManager.cs

using HyperbolicDownloader.Networking;

using Stone_Red_Utilities.ConsoleExtentions;

using System.Net.Sockets;
using System.Text.Json;

namespace HyperbolicDownloader;

internal class HostsManager
{
    private List<NetworkSocket> hosts = new List<NetworkSocket>();
    public int Count => hosts.Count;

    public void AddRange(IEnumerable<NetworkSocket> hosts)
    {
        foreach (NetworkSocket host in hosts)
        {
            if (!this.hosts.Any(x => x.IPAddress == host.IPAddress && x.Port == host.Port))
            {
                this.hosts.Add(host);
            }
        }

        SaveHosts();
    }

    public void Add(NetworkSocket host)
    {
        if (!hosts.Any(x => x.IPAddress == host.IPAddress && x.Port == host.Port))
        {
            hosts.Add(host);
        }
        SaveHosts();
    }

    public void Remove(NetworkSocket host, bool forceRemove = false)
    {
        if (DateTime.Now - host.LastActive >= new TimeSpan(24, 0, 0) || forceRemove)
        {
            hosts.RemoveAll(x => x.IPAddress == host.IPAddress && x.Port == host.Port);
        }
        SaveHosts();
    }

    public int CheckHostsActivity()
    {
        List<NetworkSocket> hostsToRemove = new List<NetworkSocket>();
        int activeHostsCount = 0;
        foreach (NetworkSocket host in hosts)
        {
            ConsoleExt.Write($"{host.IPAddress}:{host.Port} > ???", ConsoleColor.DarkYellow);
            using TcpClient tcpClient = new TcpClient();

            try
            {
                tcpClient.ConnectAsync(host.IPAddress, host.Port).Wait(1000);
                Console.CursorLeft = 0;
                if (tcpClient.Connected)
                {
                    ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Active", ConsoleColor.Green);
                    host.LastActive = DateTime.Now;
                    activeHostsCount++;
                }
                else
                {
                    if (DateTime.Now - host.LastActive >= new TimeSpan(24, 0, 0))
                    {
                        hostsToRemove.Add(host);
                    }
                    ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Inactive", ConsoleColor.Red);
                }
            }
            catch
            {
                if (DateTime.Now - host.LastActive >= new TimeSpan(24, 0, 0))
                {
                    hostsToRemove.Add(host);
                }
                Console.CursorLeft = 0;
                ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Inactive", ConsoleColor.Red);
            }
        }

        foreach (NetworkSocket host in hostsToRemove)
        {
            hosts.Remove(host);
        }

        SaveHosts();
        return activeHostsCount;
    }

    public List<NetworkSocket> ToList()
    {
        return hosts.ToList();
    }

    public void SaveHosts()
    {
        hosts = hosts.OrderByDescending(h => h.LastActive).ToList();
        File.WriteAllText(Program.HostsFilePath, JsonSerializer.Serialize(hosts));
    }
}