wabbajack/Wabbajack.App/Screens/CompilerConfigurationViewModel.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reactive;
using System.Text.Json;
using System.Threading.Tasks;
using Avalonia.Controls.Mixins;
using ReactiveUI;
using ReactiveUI.Fody.Helpers;
using Wabbajack.App.Messages;
using Wabbajack.App.Models;
using Wabbajack.App.ViewModels;
using Wabbajack.Common;
using Wabbajack.Compiler;
using Wabbajack.DTOs;
using Wabbajack.DTOs.JsonConverters;
using Wabbajack.Installer;
using Wabbajack.Paths;
using Wabbajack.Paths.IO;
using Consts = Wabbajack.Compiler.Consts;

namespace Wabbajack.App.Screens;

public class CompilerConfigurationViewModel : ViewModelBase, IReceiverMarker
{
    private readonly DTOSerializer _dtos;
    private readonly SettingsManager _settingsManager;

    [Reactive]
    public string Title { get; set; }
    
    [Reactive]
    public AbsolutePath SettingsFile { get; set; }
    
    [Reactive]
    public AbsolutePath Downloads { get; set; }
    
    [Reactive]
    public GameMetaData BaseGame { get; set; }
    
    [Reactive]
    public AbsolutePath Source { get; set; }
    
    [Reactive]
    public AbsolutePath GamePath { get; set; }
    
    [Reactive]
    public string SelectedProfile { get; set; }
    
    [Reactive]
    public AbsolutePath OutputFolder { get; set; }
    
    [Reactive]
    public IEnumerable<GameMetaData> AllGames { get; set; }
    
    [Reactive]
    public ReactiveCommand<Unit, Unit> StartCompilation { get; set; }

    [Reactive] 
    public IEnumerable<RelativePath> AlwaysEnabled { get; set; } = Array.Empty<RelativePath>();

    public AbsolutePath SettingsOutputLocation => Source.Combine(Title).WithExtension(IsMO2Compilation ? Ext.MO2CompilerSettings : Ext.CompilerSettings);
    
    [Reactive]
    public bool IsMO2Compilation { get; set; }


    public CompilerConfigurationViewModel(DTOSerializer dtos, SettingsManager settingsManager)
    {
        _settingsManager = settingsManager;
        _dtos = dtos;
        Activator = new ViewModelActivator();

        AllGames = GameRegistry.Games.Values.ToArray();
        
        StartCompilation = ReactiveCommand.Create(() => BeginCompilation().FireAndForget());

        OutputFolder = KnownFolders.EntryPoint;
        
        this.WhenActivated(disposables =>
        {
            LoadLastCompilation().FireAndForget();
            this.WhenAnyValue(v => v.SettingsFile)
                .Subscribe( location =>
                {
                    LoadNewSettingsFile(location).FireAndForget();
                })
                .DisposeWith(disposables);
        });
        
    }

    private async Task LoadNewSettingsFile(AbsolutePath location)
    {
        if (location == default) return;
        if (location.FileExists()) await LoadSettings(location);
    }

    private async Task LoadLastCompilation()
    {
        var location = await _settingsManager.Load<AbsolutePath>("last_compilation");
        SettingsFile = location;
    }

    private async Task BeginCompilation()
    {
        var settings = GetSettings();
        await SaveSettingsFile();
        await _settingsManager.Save("last_compilation", SettingsOutputLocation);
        
        MessageBus.Instance.Send(new StartCompilation(settings));
        MessageBus.Instance.Send(new NavigateTo(typeof(CompilationViewModel)));
    }

    private CompilerSettings GetSettings()
    {
        return new MO2CompilerSettings
        {
            Downloads = Downloads,
            Source = Source,
            Game = BaseGame.Game,
            Profile = SelectedProfile,
            UseGamePaths = true,
            OutputFile = OutputFolder.Combine(SelectedProfile).WithExtension(Ext.Wabbajack),
            AlwaysEnabled = AlwaysEnabled.ToArray()
        };
    }

    public bool AddAlwaysExcluded(AbsolutePath path)
    {
        if (!path.InFolder(Source)) return false;
        var relative = path.RelativeTo(Source);
        AlwaysEnabled = AlwaysEnabled.Append(relative).Distinct().ToArray();
        return true;
    }

    public void RemoveAlwaysExcluded(RelativePath path)
    {
        AlwaysEnabled = AlwaysEnabled.Where(p => p != path).ToArray();
    }

    public async Task InferSettingsFromModlistTxt(AbsolutePath settingsFile)
    {
        if (settingsFile.FileName == "modlist.txt".ToRelativePath() && settingsFile.Depth > 3)
        {
            var mo2Folder = settingsFile.Parent.Parent.Parent;
            var mo2Ini = mo2Folder.Combine(Consts.MO2IniName);
            if (mo2Ini.FileExists())
            {
                var iniData = mo2Ini.LoadIniFile();

                var general = iniData["General"];

                BaseGame = GameRegistry.GetByFuzzyName(general["gameName"].FromMO2Ini());
                Source = mo2Folder;
                
                SelectedProfile = general["selected_profile"].FromMO2Ini();
                GamePath = general["gamePath"].FromMO2Ini().ToAbsolutePath();
                Title = SelectedProfile;
                
                var settings = iniData["Settings"];
                Downloads = settings["download_directory"].FromMO2Ini().ToAbsolutePath();
                IsMO2Compilation = true;

                
                // Find Always Enabled mods
                foreach (var modFolder in mo2Folder.Combine("mods").EnumerateDirectories())
                {
                    var iniFile = modFolder.Combine("meta.ini");
                    if (!iniFile.FileExists()) continue;
                    
                    var data = iniFile.LoadIniFile();
                    var generalModData = data["General"];
                    if ((generalModData["notes"]?.Contains("WABBAJACK_ALWAYS_ENABLE") ?? false) || 
                        (generalModData["comments"]?.Contains("WABBAJACK_ALWAYS_ENABLE") ?? false))
                    {
                        AlwaysEnabled = AlwaysEnabled.Append(modFolder.RelativeTo(mo2Folder)).ToArray();
                    }

                }

                if (mo2Folder.Depth > 1)
                    OutputFolder = mo2Folder.Parent;
                
                await SaveSettingsFile();
                SettingsFile = SettingsOutputLocation;
            }

        }
    }

    private async Task SaveSettingsFile()
    {
        await using var st = SettingsOutputLocation.Open(FileMode.Create, FileAccess.Write, FileShare.None);
        if (IsMO2Compilation) 
            await JsonSerializer.SerializeAsync(st, (MO2CompilerSettings)GetSettings(), _dtos.Options);
        else 
            await JsonSerializer.SerializeAsync(st, GetSettings(), _dtos.Options);
    }

    private async Task LoadSettings(AbsolutePath path)
    {
        CompilerSettings s;
        if (path.Extension == Ext.MO2CompilerSettings)
        {
            var mo2 = await LoadSettingsFile<MO2CompilerSettings>(path);
            AlwaysEnabled = mo2.AlwaysEnabled;
            SelectedProfile = mo2.Profile;
            s = mo2;
        }
        else
        {
            throw new NotImplementedException();
        }

        Source = s.Source;
        Downloads = s.Downloads;
        OutputFolder = s.OutputFile.Depth > 1 ? s.OutputFile.Parent : s.OutputFile;
        BaseGame = s.Game.MetaData();
    }

    private async Task<T> LoadSettingsFile<T>(AbsolutePath path)
    {
        await using var st = path.Open(FileMode.Open);
        return (await JsonSerializer.DeserializeAsync<T>(st, _dtos.Options))!;
    }
}