using System; using System.Collections.Generic; using System.Collections.ObjectModel; using System.Linq; using System.Reactive.Disposables; using System.Reactive.Linq; using System.Threading; using System.Threading.Tasks; using System.Windows.Input; using DynamicData; using Microsoft.Extensions.Logging; using ReactiveUI; using ReactiveUI.Fody.Helpers; using Wabbajack.Common; using Wabbajack.Downloaders.GameFile; using Wabbajack.DTOs; using Wabbajack.Messages; using Wabbajack.Networking.WabbajackClientApi; using Wabbajack.Services.OSIntegrated; using Wabbajack.Services.OSIntegrated.Services; namespace Wabbajack { public class ModListGalleryVM : BackNavigatingVM { public MainWindowVM MWVM { get; } private readonly SourceCache _modLists = new(x => x.Metadata.NamespacedName); public ReadOnlyObservableCollection _filteredModLists; public ReadOnlyObservableCollection ModLists => _filteredModLists; private const string ALL_GAME_TYPE = "All"; [Reactive] public IErrorResponse Error { get; set; } [Reactive] public string Search { get; set; } [Reactive] public bool OnlyInstalled { get; set; } [Reactive] public bool ShowNSFW { get; set; } [Reactive] public bool ShowUnofficialLists { get; set; } [Reactive] public string GameType { get; set; } public class GameTypeEntry { public GameTypeEntry(string humanFriendlyName, int amount) { HumanFriendlyName = humanFriendlyName; Amount = amount; FormattedName = $"{HumanFriendlyName} ({Amount})"; } public string HumanFriendlyName { get; set; } public int Amount { get; set; } public string FormattedName { get; set; } } [Reactive] public List GameTypeEntries { get; set; } private bool _filteringOnGame; private GameTypeEntry _selectedGameTypeEntry = null; public GameTypeEntry SelectedGameTypeEntry { get => _selectedGameTypeEntry; set { RaiseAndSetIfChanged(ref _selectedGameTypeEntry, value == null ? GameTypeEntries?.FirstOrDefault(gte => gte.HumanFriendlyName == ALL_GAME_TYPE) : value); GameType = _selectedGameTypeEntry?.HumanFriendlyName; } } private readonly Client _wjClient; private readonly ILogger _logger; private readonly GameLocator _locator; private readonly ModListDownloadMaintainer _maintainer; private readonly SettingsManager _settingsManager; private readonly CancellationToken _cancellationToken; public ICommand ClearFiltersCommand { get; set; } public ModListGalleryVM(ILogger logger, Client wjClient, GameLocator locator, SettingsManager settingsManager, ModListDownloadMaintainer maintainer, CancellationToken cancellationToken) : base(logger) { _wjClient = wjClient; _logger = logger; _locator = locator; _maintainer = maintainer; _settingsManager = settingsManager; _cancellationToken = cancellationToken; ClearFiltersCommand = ReactiveCommand.Create( () => { OnlyInstalled = false; ShowNSFW = false; ShowUnofficialLists = false; Search = string.Empty; SelectedGameTypeEntry = GameTypeEntries.FirstOrDefault(); }); BackCommand = ReactiveCommand.Create( () => { NavigateToGlobal.Send(NavigateToGlobal.ScreenType.ModeSelectionView); }); this.WhenActivated(disposables => { LoadModLists().FireAndForget(); LoadSettings().FireAndForget(); Disposable.Create(() => SaveSettings().FireAndForget()) .DisposeWith(disposables); var searchTextPredicates = this.ObservableForProperty(vm => vm.Search) .Select(change => change.Value) .StartWith(Search) .Select>(txt => { if (string.IsNullOrWhiteSpace(txt)) return _ => true; return item => item.Metadata.Title.ContainsCaseInsensitive(txt) || item.Metadata.Description.ContainsCaseInsensitive(txt); }); var onlyInstalledGamesFilter = this.ObservableForProperty(vm => vm.OnlyInstalled) .Select(v => v.Value) .Select>(onlyInstalled => { if (onlyInstalled == false) return _ => true; return item => _locator.IsInstalled(item.Metadata.Game); }) .StartWith(_ => true); var showUnofficial = this.ObservableForProperty(vm => vm.ShowUnofficialLists) .Select(v => v.Value) .StartWith(false) .Select>(unoffical => { if (unoffical) return x => true; return x => x.Metadata.Official; }); var showNSFWFilter = this.ObservableForProperty(vm => vm.ShowNSFW) .Select(v => v.Value) .Select>(showNsfw => { return item => item.Metadata.NSFW == showNsfw; }) .StartWith(item => item.Metadata.NSFW == false); var gameFilter = this.ObservableForProperty(vm => vm.GameType) .Select(v => v.Value) .Select>(selected => { _filteringOnGame = true; if (selected is null or ALL_GAME_TYPE) return _ => true; return item => item.Metadata.Game.MetaData().HumanFriendlyGameName == selected; }) .StartWith(_ => true); _modLists.Connect() .ObserveOn(RxApp.MainThreadScheduler) .Filter(searchTextPredicates) .Filter(onlyInstalledGamesFilter) .Filter(showUnofficial) .Filter(showNSFWFilter) .Filter(gameFilter) .Bind(out _filteredModLists) .Subscribe((_) => { if (!_filteringOnGame) { var previousGameType = GameType; SelectedGameTypeEntry = null; GameTypeEntries = new(GetGameTypeEntries()); var nextEntry = GameTypeEntries.FirstOrDefault(gte => previousGameType == gte.HumanFriendlyName); SelectedGameTypeEntry = nextEntry != default ? nextEntry : GameTypeEntries.FirstOrDefault(gte => GameType == ALL_GAME_TYPE); } _filteringOnGame = false; }) .DisposeWith(disposables); }); } private class FilterSettings { public string GameType { get; set; } public bool ShowNSFW { get; set; } public bool ShowUnofficialLists { get; set; } public bool OnlyInstalled { get; set; } public string Search { get; set; } } public override void Unload() { Error = null; } private async Task SaveSettings() { await _settingsManager.Save("modlist_gallery", new FilterSettings { GameType = GameType, ShowNSFW = ShowNSFW, ShowUnofficialLists = ShowUnofficialLists, Search = Search, OnlyInstalled = OnlyInstalled, }); } private async Task LoadSettings() { using var ll = LoadingLock.WithLoading(); RxApp.MainThreadScheduler.Schedule(await _settingsManager.Load("modlist_gallery"), (_, s) => { SelectedGameTypeEntry = GameTypeEntries?.FirstOrDefault(gte => gte.HumanFriendlyName.Equals(s.GameType)); ShowNSFW = s.ShowNSFW; ShowUnofficialLists = s.ShowUnofficialLists; Search = s.Search; OnlyInstalled = s.OnlyInstalled; return Disposable.Empty; }); } private async Task LoadModLists() { using var ll = LoadingLock.WithLoading(); try { var modLists = await _wjClient.LoadLists(); _modLists.Edit(e => { e.Clear(); e.AddOrUpdate(modLists.Select(m => new ModListMetadataVM(_logger, this, m, _maintainer, _wjClient, _cancellationToken))); }); } catch (Exception ex) { _logger.LogError(ex, "While loading lists"); ll.Fail(); } ll.Succeed(); } private List GetGameTypeEntries() { return ModLists.Select(fm => fm.Metadata) .GroupBy(m => m.Game) .Select(g => new GameTypeEntry(g.Key.MetaData().HumanFriendlyGameName, g.Count())) .OrderBy(gte => gte.HumanFriendlyName) .Prepend(new GameTypeEntry(ALL_GAME_TYPE, ModLists.Count)) .ToList(); } } }