using System; using System.Collections.Generic; using System.Collections.ObjectModel; using System.Linq; using System.Reactive.Disposables; using System.Reactive.Linq; 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 List GameTypeEntries => GetGameTypeEntries(); private ObservableAsPropertyHelper _Loaded; private readonly Client _wjClient; private readonly ILogger _logger; private readonly GameLocator _locator; private readonly ModListDownloadMaintainer _maintainer; private readonly SettingsManager _settingsManager; private FiltersSettings settings { get; set; } = new(); public ICommand ClearFiltersCommand { get; set; } public ModListGalleryVM(ILogger logger, Client wjClient, GameLocator locator, SettingsManager settingsManager, ModListDownloadMaintainer maintainer) : base(logger) { _wjClient = wjClient; _logger = logger; _locator = locator; _maintainer = maintainer; _settingsManager = settingsManager; ClearFiltersCommand = ReactiveCommand.Create( () => { OnlyInstalled = false; ShowNSFW = false; ShowUnofficialLists = false; Search = string.Empty; GameType = ALL_GAME_TYPE; }); 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("") .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 => { 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() .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) => { GameType = 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))); }); } catch (Exception ex) { _logger.LogError(ex, "While loading lists"); ll.Fail(); } ll.Succeed(); } private List GetGameTypeEntries() { List gameEntries = new List {ALL_GAME_TYPE}; gameEntries.AddRange(GameRegistry.Games.Values.Select(gameType => gameType.HumanFriendlyGameName)); gameEntries.Sort(); return gameEntries; } } }