NebulaLauncher/Nebula.Launcher/ViewModels/Pages/ServerListViewModel.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Linq;
using System.Threading.Tasks;
using CommunityToolkit.Mvvm.ComponentModel;
using Nebula.Launcher.Services;
using Nebula.Launcher.ViewModels.Popup;
using Nebula.Launcher.Views;
using Nebula.Launcher.Views.Pages;
using Nebula.Shared.Models;
using Nebula.Shared.Services;
using Nebula.Shared.Utils;

namespace Nebula.Launcher.ViewModels.Pages;

[ViewModelRegister(typeof(ServerListView))]
[ConstructGenerator]
public partial class ServerListViewModel : ViewModelBase, IViewModelPage
{
    [ObservableProperty] private string _searchText = string.Empty;

    [ObservableProperty] private bool _isFavoriteMode;
    
    public ObservableCollection<ServerEntryModelView> Servers { get; }= new();
    public ObservableCollection<Exception> HubErrors { get; } = new();
    
    public Action? OnSearchChange;
    [GenerateProperty] private HubService HubService { get; }
    [GenerateProperty] private PopupMessageService PopupMessageService { get; }
    [GenerateProperty] private CancellationService CancellationService { get; }
    [GenerateProperty] private DebugService DebugService { get; }
    [GenerateProperty, DesignConstruct] private ViewHelperService ViewHelperService { get; }
    
    private ServerViewContainer ServerViewContainer { get; set; } 
    
    private List<ServerHubInfo> UnsortedServers { get; } = new();

    //Design think
    protected override void InitialiseInDesignMode()
    {
        ServerViewContainer = new ServerViewContainer(this, ViewHelperService);
        HubErrors.Add(new Exception("UVI"));
    }

    //real think
    protected override void Initialise()
    {
        ServerViewContainer = new ServerViewContainer(this, ViewHelperService);
        
        foreach (var info in HubService.ServerList) UnsortedServers.Add(info);

        HubService.HubServerChangedEventArgs += HubServerChangedEventArgs;
        HubService.HubServerLoaded += UpdateServerEntries;
        HubService.HubServerLoadingError += HubServerLoadingError;
        OnSearchChange += OnChangeSearch;

        if (!HubService.IsUpdating) UpdateServerEntries();
        UpdateFavoriteEntries();
    }

    private void HubServerLoadingError(Exception obj)
    {
        HubErrors.Add(obj);
    }

    private void UpdateServerEntries()
    {
        Servers.Clear();
        Task.Run(() =>
        {
            UnsortedServers.Sort(new ServerComparer());
            foreach (var info in UnsortedServers.Where(a => CheckServerThink(a.StatusData)))
            {
                var view = ServerViewContainer.Get(info.Address.ToRobustUrl(), info.StatusData);
                Servers.Add(view);
            }
        });
    }

    private void OnChangeSearch()
    {
        if(string.IsNullOrEmpty(SearchText)) return;
        
        if(IsFavoriteMode)
        {
            UpdateFavoriteEntries();
        }
        else
        {
            UpdateServerEntries();
        }
    }

    private void HubServerChangedEventArgs(HubServerChangedEventArgs obj)
    {
        if (obj.Action == HubServerChangeAction.Add)
            foreach (var info in obj.Items)
                UnsortedServers.Add(info);
        if (obj.Action == HubServerChangeAction.Remove)
            foreach (var info in obj.Items)
                UnsortedServers.Remove(info);
        if (obj.Action == HubServerChangeAction.Clear)
        {
            UnsortedServers.Clear();
            ServerViewContainer.Clear();
            Servers.Clear();
            UpdateFavoriteEntries();
        }
    }

    private bool CheckServerThink(ServerStatus hubInfo)
    {
        if (string.IsNullOrEmpty(SearchText)) return true;
        return hubInfo.Name.ToLower().Contains(SearchText.ToLower());
    }

    public void FilterRequired()
    {
    }

    public void AddFavoriteRequired()
    {
        var p = ViewHelperService.GetViewModel<AddFavoriteViewModel>();
        PopupMessageService.Popup(p);
    }

    public void UpdateRequired()
    {
        HubErrors.Clear();
        Task.Run(HubService.UpdateHub);
    }

    public void OnPageOpen(object? args)
    {
        if (args is bool fav)
        {
            IsFavoriteMode = fav;
        }
    }
}

public class ServerViewContainer(
    ServerListViewModel serverListViewModel,
    ViewHelperService viewHelperService
    )
{
    private readonly Dictionary<string, ServerEntryModelView> _entries = new();

    public void Clear()
    {
        _entries.Clear();
    }

    public ServerEntryModelView Get(RobustUrl url, ServerStatus? serverStatus = null)
    {
        ServerEntryModelView? entry;
        
        lock (_entries)
        {
            if (_entries.TryGetValue(url.ToString(), out entry))
            {
                return entry;
            }

            entry = viewHelperService.GetViewModel<ServerEntryModelView>().WithData(url, serverStatus);
            
            _entries.Add(url.ToString(), entry);
        }
        
        entry.OnFavoriteToggle += () =>
        {
            if (entry.IsFavorite) serverListViewModel.RemoveFavorite(entry);
            else serverListViewModel.AddFavorite(entry);
        };
        
        return entry;
    }
}

public class ServerComparer : IComparer<ServerHubInfo>, IComparer<ServerStatus>, IComparer<(RobustUrl,ServerStatus)>
{
    public int Compare(ServerHubInfo? x, ServerHubInfo? y)
    {
        if (ReferenceEquals(x, y))
            return 0;
        if (ReferenceEquals(null, y))
            return 1;
        if (ReferenceEquals(null, x))
            return -1;

        return Compare(x.StatusData, y.StatusData);
    }

    public int Compare(ServerStatus? x, ServerStatus? y)
    {
        if (ReferenceEquals(x, y))
            return 0;
        if (ReferenceEquals(null, y))
            return 1;
        if (ReferenceEquals(null, x))
            return -1;

        return y.Players.CompareTo(x.Players);
    }

    public int Compare((RobustUrl, ServerStatus) x, (RobustUrl, ServerStatus) y)
    {
        return Compare(x.Item2, y.Item2);
    }
}