wabbajack/Wabbajack.Services.OSIntegrated/EncryptedJsonTokenProvider.cs

using System;
using System.Threading.Tasks;
using Microsoft.Extensions.Logging;
using Wabbajack.DTOs.JsonConverters;
using Wabbajack.Networking.Http.Interfaces;
using Wabbajack.Paths;
using Wabbajack.Paths.IO;

namespace Wabbajack.Services.OSIntegrated;

public class EncryptedJsonTokenProvider<T> : ITokenProvider<T>
{
    private readonly DTOSerializer _dtos;
    private readonly string _key;
    private readonly ILogger _logger;

    public EncryptedJsonTokenProvider(ILogger logger, DTOSerializer dtos, string key)
    {
        _logger = logger;
        _key = key;
        _dtos = dtos;
    }

    private string? EnvValue => Environment.GetEnvironmentVariable(_key.ToUpperInvariant().Replace("-", "_"));

    private AbsolutePath KeyPath => KnownFolders.WabbajackAppLocal.Combine("encrypted", _key);

    public async ValueTask SetToken(T token)
    {
        _logger.LogInformation("Setting token {token}", _key);
        await token.AsEncryptedJsonFile(KeyPath);
    }

    public async ValueTask<bool> TryDelete(T val)
    {
        if (!KeyPath.FileExists()) return false;
        KeyPath.Delete();
        return true;
    }

    public async ValueTask<T?> Get()
    {
        var path = KeyPath;
        if (path.FileExists())
        {
            return await path.FromEncryptedJsonFile<T>();
        }

        var value = EnvValue;
        if (value == default)
        {
            _logger.LogCritical("No login data for {key}", _key);
            throw new Exception("No login data for {_key}");
        }

        return _dtos.Deserialize<T>(EnvValue!);
    }

    public bool HaveToken()
    {
        return KeyPath.FileExists() || EnvValue != null;
    }

    public void DeleteToken()
    {
        _logger.LogInformation("Deleting token {token}", _key);
        if (HaveToken())
            KeyPath.Delete();
    }
}