using Syroot.Windows.IO; using System; using ReactiveUI; using System.Collections.Generic; using System.Collections.ObjectModel; using System.ComponentModel; using System.Diagnostics; using System.IO; using System.IO.Compression; using System.Linq; using System.Net.Http; using System.Reactive.Subjects; using System.Reactive.Disposables; using System.Reactive.Linq; using System.Reflection; using System.Threading; using System.Windows; using System.Windows.Input; using System.Windows.Media.Imaging; using System.Windows.Threading; using Wabbajack.Common; using Wabbajack.Lib.Downloaders; using Wabbajack.Lib.NexusApi; using DynamicData; using DynamicData.Binding; using System.Reactive; using System.Text; using Wabbajack.Lib; using Splat; using ReactiveUI.Fody.Helpers; namespace Wabbajack { public class InstallerVM : ViewModel { public SlideShow Slideshow { get; } public MainWindowVM MWVM { get; } public BitmapImage WabbajackLogo { get; } = UIUtils.BitmapImageFromResource("Wabbajack.Resources.Wabba_Mouth.png"); private readonly ObservableAsPropertyHelper _ModList; public ModListVM ModList => _ModList.Value; [Reactive] public string ModListPath { get; set; } [Reactive] public bool UIReady { get; set; } private readonly ObservableAsPropertyHelper _HTMLReport; public string HTMLReport => _HTMLReport.Value; /// /// Tracks whether an install is currently in progress /// [Reactive] public bool Installing { get; set; } /// /// Tracks whether to show the installing pane /// [Reactive] public bool InstallingMode { get; set; } [Reactive] public string Location { get; set; } private readonly ObservableAsPropertyHelper _LocationError; public IErrorResponse LocationError => _LocationError.Value; [Reactive] public string DownloadLocation { get; set; } private readonly ObservableAsPropertyHelper _DownloadLocationError; public IErrorResponse DownloadLocationError => _DownloadLocationError.Value; private readonly ObservableAsPropertyHelper _ProgressPercent; public float ProgressPercent => _ProgressPercent.Value; private readonly ObservableAsPropertyHelper _Image; public BitmapImage Image => _Image.Value; private readonly ObservableAsPropertyHelper _TitleText; public string TitleText => _TitleText.Value; private readonly ObservableAsPropertyHelper _AuthorText; public string AuthorText => _AuthorText.Value; private readonly ObservableAsPropertyHelper _Description; public string Description => _Description.Value; // Command properties public IReactiveCommand BeginCommand { get; } public IReactiveCommand ShowReportCommand { get; } public IReactiveCommand OpenReadmeCommand { get; } public IReactiveCommand VisitWebsiteCommand { get; } public InstallerVM(MainWindowVM mainWindowVM) { if (Path.GetDirectoryName(Assembly.GetEntryAssembly().Location.ToLower()) == KnownFolders.Downloads.Path.ToLower()) { MessageBox.Show( "Wabbajack is running inside your Downloads folder. This folder is often highly monitored by antivirus software and these can often " + "conflict with the operations Wabbajack needs to perform. Please move this executable outside of your Downloads folder and then restart the app.", "Cannot run inside Downloads", MessageBoxButton.OK, MessageBoxImage.Error); Environment.Exit(1); } this.MWVM = mainWindowVM; this._ModList = this.WhenAny(x => x.ModListPath) .ObserveOn(RxApp.TaskpoolScheduler) .Select(source => { if (source == null) return default(ModListVM); var modList = Installer.LoadFromFile(source); if (modList == null) { MessageBox.Show("Invalid Modlist, or file not found.", "Invalid Modlist", MessageBoxButton.OK, MessageBoxImage.Error); Application.Current.Dispatcher.Invoke(() => { this.MWVM.MainWindow.ExitWhenClosing = false; var window = new ModeSelectionWindow { ShowActivated = true }; window.Show(); this.MWVM.MainWindow.Close(); }); return default(ModListVM); } return new ModListVM(modList, source); }) .ObserveOnGuiThread() .StartWith(default(ModListVM)) .ToProperty(this, nameof(this.ModList)); this._HTMLReport = this.WhenAny(x => x.ModList) .Select(modList => modList?.ReportHTML) .ToProperty(this, nameof(this.HTMLReport)); this._ProgressPercent = Observable.CombineLatest( this.WhenAny(x => x.Installing), this.WhenAny(x => x.InstallingMode), resultSelector: (installing, mode) => !installing && mode) .Select(show => show ? 1f : 0f) // Disable for now, until more reliable //this.WhenAny(x => x.MWVM.QueueProgress) // .Select(i => i / 100f) .ToProperty(this, nameof(this.ProgressPercent)); this.Slideshow = new SlideShow(this); // Set display items to modlist if configuring or complete, // or to the current slideshow data if installing this._Image = Observable.CombineLatest( this.WhenAny(x => x.ModList) .SelectMany(x => x?.ImageObservable ?? Observable.Empty()) .NotNull() .StartWith(WabbajackLogo), this.WhenAny(x => x.Slideshow.Image) .StartWith(default(BitmapImage)), this.WhenAny(x => x.Installing), resultSelector: (modList, slideshow, installing) => installing ? slideshow : modList) .ToProperty(this, nameof(this.Image)); this._TitleText = Observable.CombineLatest( this.WhenAny(x => x.ModList.Name), this.WhenAny(x => x.Slideshow.TargetMod.ModName) .StartWith(default(string)), this.WhenAny(x => x.Installing), resultSelector: (modList, mod, installing) => installing ? mod : modList) .ToProperty(this, nameof(this.TitleText)); this._AuthorText = Observable.CombineLatest( this.WhenAny(x => x.ModList.Author), this.WhenAny(x => x.Slideshow.TargetMod.ModAuthor) .StartWith(default(string)), this.WhenAny(x => x.Installing), resultSelector: (modList, mod, installing) => installing ? mod : modList) .ToProperty(this, nameof(this.AuthorText)); this._Description = Observable.CombineLatest( this.WhenAny(x => x.ModList.Description), this.WhenAny(x => x.Slideshow.TargetMod.ModDescription) .StartWith(default(string)), this.WhenAny(x => x.Installing), resultSelector: (modList, mod, installing) => installing ? mod : modList) .ToProperty(this, nameof(this.Description)); this._LocationError = this.WhenAny(x => x.Location) .Select(x => Utils.IsDirectoryPathValid(x)) .ToProperty(this, nameof(this.LocationError)); this._DownloadLocationError = this.WhenAny(x => x.DownloadLocation) .Select(x => Utils.IsDirectoryPathValid(x)) .ToProperty(this, nameof(this.DownloadLocationError)); // Define commands this.ShowReportCommand = ReactiveCommand.Create(ShowReport); this.OpenReadmeCommand = ReactiveCommand.Create( execute: this.OpenReadmeWindow, canExecute: this.WhenAny(x => x.ModList) .Select(modList => !string.IsNullOrEmpty(modList?.Readme)) .ObserveOnGuiThread()); this.BeginCommand = ReactiveCommand.Create( execute: this.ExecuteBegin, canExecute: Observable.CombineLatest( this.WhenAny(x => x.Installing), this.WhenAny(x => x.LocationError), this.WhenAny(x => x.DownloadLocationError), resultSelector: (installing, loc, download) => { if (installing) return false; return (loc?.Succeeded ?? false) && (download?.Succeeded ?? false); }) .ObserveOnGuiThread()); this.VisitWebsiteCommand = ReactiveCommand.Create( execute: () => Process.Start(this.ModList.Website), canExecute: this.WhenAny(x => x.ModList.Website) .Select(x => x?.StartsWith("https://") ?? false) .ObserveOnGuiThread()); // Have Installation location updates modify the downloads location if empty this.WhenAny(x => x.Location) .Skip(1) // Don't do it initially .Subscribe(installPath => { if (string.IsNullOrWhiteSpace(this.DownloadLocation)) { this.DownloadLocation = Path.Combine(installPath, "downloads"); } }) .DisposeWith(this.CompositeDisposable); } private void ShowReport() { var file = Path.GetTempFileName() + ".html"; File.WriteAllText(file, HTMLReport); Process.Start(file); } private void OpenReadmeWindow() { if (string.IsNullOrEmpty(this.ModList.Readme)) return; using (var fs = new FileStream(this.ModListPath, FileMode.Open, FileAccess.Read, FileShare.Read)) using (var ar = new ZipArchive(fs, ZipArchiveMode.Read)) using (var ms = new MemoryStream()) { var entry = ar.GetEntry(this.ModList.Readme); if (entry == null) { Utils.Log($"Tried to open a non-existant readme: {this.ModList.Readme}"); return; } using (var e = entry.Open()) { e.CopyTo(ms); } ms.Seek(0, SeekOrigin.Begin); using (var reader = new StreamReader(ms)) { var viewer = new TextViewer(reader.ReadToEnd(), this.ModList.Name); viewer.Show(); } } } private void ExecuteBegin() { this.Installing = true; this.InstallingMode = true; var installer = new Installer(this.ModListPath, this.ModList.SourceModList, Location) { DownloadFolder = DownloadLocation }; var th = new Thread(() => { try { installer.Install(); } catch (Exception ex) { while (ex.InnerException != null) ex = ex.InnerException; Utils.Log(ex.StackTrace); Utils.Log(ex.ToString()); Utils.Log($"{ex.Message} - Can't continue"); } finally { this.Installing = false; } }) { Priority = ThreadPriority.BelowNormal }; th.Start(); } } }