wabbajack/Wabbajack.App/Models/InstallationStateManager.cs

using System;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.Extensions.Logging;
using Wabbajack.DTOs.JsonConverters;
using Wabbajack.DTOs.SavedSettings;
using Wabbajack.Paths;
using Wabbajack.Paths.IO;

namespace Wabbajack.App.Models;

public class InstallationStateManager
{
    private readonly DTOSerializer _dtos;
    private readonly ILogger<InstallationStateManager> _logger;

    public InstallationStateManager(ILogger<InstallationStateManager> logger, DTOSerializer dtos)
    {
        _dtos = dtos;
        _logger = logger;
    }

    private static AbsolutePath Path => KnownFolders.WabbajackAppLocal.Combine("install-configuration-state.json");

    public async Task<InstallationConfigurationSetting> GetLastState()
    {
        var state = await GetAll();
        var result = state.Settings.FirstOrDefault(s => s.ModList == state.LastModlist) ??
                     new InstallationConfigurationSetting();

        if (!result.ModList.FileExists())
            return new InstallationConfigurationSetting();
        return result;
    }

    public async Task SetLastState(InstallationConfigurationSetting setting)
    {
        if (!setting.ModList.FileExists())
        {
            _logger.LogCritical("ModList path doesn't exist, not saving settings");
            return;
        }

        var state = await GetAll();
        state.LastModlist = setting.ModList;
        state.Settings = state.Settings
            .Where(s => s.ModList != setting.ModList)
            .Append(setting)
            .ToArray();

        await using var fs = Path.Open(FileMode.Create, FileAccess.Write, FileShare.None);
        await _dtos.Serialize(state, fs, true);
    }

    public async Task<InstallConfigurationState> GetAll()
    {
        if (!Path.FileExists()) return new InstallConfigurationState();

        try
        {
            await using var fs = Path.Open(FileMode.Open);
            return (await _dtos.DeserializeAsync<InstallConfigurationState>(fs))!;
        }
        catch (Exception ex)
        {
            _logger.LogError(ex, "While loading json");
            return new InstallConfigurationState();
        }
    }

    public async Task<InstallationConfigurationSetting?> Get(AbsolutePath modListPath)
    {
        return (await GetAll()).Settings.FirstOrDefault(f => f.ModList == modListPath);
    }

    public async Task<InstallationConfigurationSetting?> GetByInstallFolder(AbsolutePath folder)
    {
        return (await GetAll()).Settings.FirstOrDefault(f => f.Install == folder);
    }
}