using Microsoft.WindowsAPICodePack.Dialogs; using ReactiveUI; using ReactiveUI.Fody.Helpers; using System; using System.Reactive.Disposables; using System.Reactive.Linq; using System.Threading.Tasks; using DynamicData; using Wabbajack.Common; using Wabbajack.Compiler; using Wabbajack.DTOs; using Wabbajack.DTOs.GitHub; using Wabbajack; using Wabbajack.Extensions; using Wabbajack.Paths; using Wabbajack.Paths.IO; using Consts = Wabbajack.Consts; namespace Wabbajack { public class MO2CompilerVM : ViewModel, ISubCompilerVM { public CompilerVM Parent { get; } private readonly MO2CompilationSettings _settings; private readonly ObservableAsPropertyHelper _mo2Folder; public AbsolutePath Mo2Folder => _mo2Folder.Value; private readonly ObservableAsPropertyHelper _moProfile; public string MOProfile => _moProfile.Value; public FilePickerVM DownloadLocation { get; } public FilePickerVM ModListLocation { get; } [Reactive] public ACompiler ActiveCompilation { get; private set; } private readonly ObservableAsPropertyHelper _modlistSettings; private readonly IObservable> _authorKeys; public ModlistSettingsEditorVM ModlistSettings => _modlistSettings.Value; [Reactive] public object StatusTracker { get; private set; } public IObservable CanCompile { get; } public MO2CompilerVM(CompilerVM parent) { Parent = parent; ModListLocation = new FilePickerVM { ExistCheckOption = FilePickerVM.CheckOptions.On, PathType = FilePickerVM.PathTypeOptions.File, PromptTitle = "Select a Modlist" }; ModListLocation.Filters.Add(new CommonFileDialogFilter("MO2 Profile (modlist.txt) or Native Settings (native_compiler_settings.json)", "*.txt,*.json")); DownloadLocation = new FilePickerVM() { ExistCheckOption = FilePickerVM.CheckOptions.On, PathType = FilePickerVM.PathTypeOptions.Folder, PromptTitle = "Select a downloads location", }; _mo2Folder = this.WhenAny(x => x.ModListLocation.TargetPath) .Select(loc => { try { /* TODO if (loc.FileName == Consts.ModListTxt) { var profileFolder = loc.Parent; return profileFolder.Parent.Parent; } if (loc.FileName == Consts.NativeSettingsJson) { return loc.Parent; } */ return loc.Parent; return default; } catch (Exception) { return default; } }) .ToGuiProperty(this, nameof(Mo2Folder)); _moProfile = this.WhenAny(x => x.ModListLocation.TargetPath) .Select(loc => { /* TODO try { if (loc.FileName == Consts.NativeSettingsJson) { var settings = loc.FromJson(); return settings.ModListName; } return (string)loc.Parent.FileName; } catch (Exception) { return null; } */ return (string)loc.Parent.FileName; }) .ToGuiProperty(this, nameof(MOProfile)); // Wire missing Mo2Folder to signal error state for ModList Location ModListLocation.AdditionalError = this.WhenAny(x => x.Mo2Folder) .Select(moFolder => { if (moFolder.DirectoryExists()) return ErrorResponse.Success; return ErrorResponse.Fail($"MO2 folder could not be located from the given ModList location.{Environment.NewLine}Make sure your ModList is inside a valid MO2 distribution."); }); // Load custom ModList settings per MO2 profile _modlistSettings = Observable.CombineLatest( (this).WhenAny(x => x.ModListLocation.ErrorState), (this).WhenAny(x => x.ModListLocation.TargetPath), resultSelector: (state, path) => (State: state, Path: path)) // A short throttle is a quick hack to make the above changes "atomic" .Throttle(TimeSpan.FromMilliseconds(25), RxApp.MainThreadScheduler) .Select(u => { if (u.State.Failed) return null; /* TODO var modlistSettings = _settings.ModlistSettings.TryCreate(u.Path); return new ModlistSettingsEditorVM(modlistSettings) { ModListName = MOProfile }; */ return new ModlistSettingsEditorVM(null); }) // Interject and save old while loading new .Pairwise() .Do(pair => { pair.Previous?.Save(); pair.Current?.Init(); }) .Select(x => x.Current) .ToGuiProperty(this, nameof(ModlistSettings)); CanCompile = Observable.CombineLatest( this.WhenAny(x => x.ModListLocation.InError), this.WhenAny(x => x.DownloadLocation.InError), parent.WhenAny(x => x.OutputLocation.InError), this.WhenAny(x => x.ModlistSettings) .Select(x => x?.InError ?? Observable.Return(false)) .Switch(), resultSelector: (ml, down, output, modlistSettings) => !ml && !down && !output && !modlistSettings) .Publish() .RefCount(); // Load settings _settings = parent.MWVM.Settings.Compiler.MO2Compilation; ModListLocation.TargetPath = _settings.LastCompiledProfileLocation; if (_settings.DownloadLocation != default) { DownloadLocation.TargetPath = _settings.DownloadLocation; } parent.MWVM.Settings.SaveSignal .Subscribe(_ => Unload()) .DisposeWith(CompositeDisposable); // If Mo2 folder changes and download location is empty, set it for convenience this.WhenAny(x => x.Mo2Folder) .DelayInitial(TimeSpan.FromMilliseconds(100), RxApp.MainThreadScheduler) .Where(x => x.DirectoryExists()) .FlowSwitch( (this).WhenAny(x => x.DownloadLocation.Exists) .Invert()) // A skip is needed to ignore the initial signal when the FilterSwitch turns on .Skip(1) .Subscribe(_ => { DownloadLocation.TargetPath = MO2Compiler.GetTypicalDownloadsFolder(Mo2Folder); }) .DisposeWith(CompositeDisposable); } public void Unload() { _settings.DownloadLocation = DownloadLocation.TargetPath; _settings.LastCompiledProfileLocation = ModListLocation.TargetPath; ModlistSettings?.Save(); } public async Task> Compile() { AbsolutePath outputFile; var profileName = string.IsNullOrWhiteSpace(ModlistSettings.ModListName) ? MOProfile : ModlistSettings.ModListName; if (Parent.OutputLocation.TargetPath == default) { outputFile = (profileName.ToRelativePath().WithExtension(Ext.Wabbajack)).RelativeTo(KnownFolders.EntryPoint); } else { outputFile = Parent.OutputLocation.TargetPath.Combine(profileName).WithExtension(Ext.Wabbajack); } try { ACompiler compiler; UpdateRequest request = null; if (ModlistSettings.Publish) { request = new UpdateRequest { MachineUrl = ModlistSettings.MachineUrl.Trim(), Version = ModlistSettings.Version, }; } /* TODO if (ModListLocation.TargetPath.FileName == Consts.NativeSettingsJson) { var settings = ModListLocation.TargetPath.FromJson(); compiler = new NativeCompiler(settings, Mo2Folder, DownloadLocation.TargetPath, outputFile, request) { ModListName = ModlistSettings.ModListName, ModListAuthor = ModlistSettings.AuthorText, ModListDescription = ModlistSettings.Description, ModListImage = ModlistSettings.ImagePath.TargetPath, ModListWebsite = ModlistSettings.Website, ModlistReadme = ModlistSettings.Readme, ModlistVersion = ModlistSettings.Version, ModlistIsNSFW = ModlistSettings.IsNSFW }; } else { compiler = new MO2Compiler( sourcePath: Mo2Folder, downloadsPath: DownloadLocation.TargetPath, mo2Profile: MOProfile, outputFile: outputFile, publishData: request) { ModListName = ModlistSettings.ModListName, ModListAuthor = ModlistSettings.AuthorText, ModListDescription = ModlistSettings.Description, ModListImage = ModlistSettings.ImagePath.TargetPath, ModListWebsite = ModlistSettings.Website, ModlistReadme = ModlistSettings.Readme, ModlistVersion = ModlistSettings.Version, ModlistIsNSFW = ModlistSettings.IsNSFW }; } using (ActiveCompilation = compiler) { Parent.MWVM.Settings.Performance.SetProcessorSettings(ActiveCompilation); var success = await ActiveCompilation.Begin(); return GetResponse.Create(success, ActiveCompilation.ModList); } */ return GetResponse.Create(true, ActiveCompilation.ModList); } finally { StatusTracker = null; ActiveCompilation = null; } } } }