vrcauthproxy/VRCAuthProxy/Services/RedisService.cs

using System.Text.Json;
using StackExchange.Redis;

namespace VRCAuthProxy.Services;

public class RedisService
{
    private readonly IConnectionMultiplexer _redis;
    private readonly IDatabase _db;
    private const string AUTH_TOKEN_KEY = "vrcauthproxy:tokens";

    public RedisService(string connectionString)
    {
        _redis = ConnectionMultiplexer.Connect(connectionString);
        _db = _redis.GetDatabase();
    }

    // Constructor for testing with mocked dependencies
    internal RedisService(IConnectionMultiplexer redis, IDatabase database)
    {
        _redis = redis;
        _db = database;
    }

    public virtual async Task SaveAuthToken(string username, Dictionary<string, string> cookies)
    {
        var serializedCookies = JsonSerializer.Serialize(cookies);
        await _db.HashSetAsync(AUTH_TOKEN_KEY, username, serializedCookies);
    }

    public virtual async Task<Dictionary<string, string>?> GetAuthToken(string username)
    {
        var value = await _db.HashGetAsync(AUTH_TOKEN_KEY, username);
        if (!value.HasValue) return null;
        return JsonSerializer.Deserialize<Dictionary<string, string>>(value!);
    }

    public virtual async Task<Dictionary<string, Dictionary<string, string>>> GetAllAuthTokens()
    {
        var entries = await _db.HashGetAllAsync(AUTH_TOKEN_KEY);
        var result = new Dictionary<string, Dictionary<string, string>>();
        
        foreach (var entry in entries)
        {
            var cookies = JsonSerializer.Deserialize<Dictionary<string, string>>(entry.Value!);
            if (cookies != null)
            {
                result[entry.Name!] = cookies;
            }
        }
        
        return result;
    }

    public virtual async Task RemoveAuthToken(string username)
    {
        await _db.HashDeleteAsync(AUTH_TOKEN_KEY, username);
    }
}