using System.Collections.Concurrent;
using System.Linq;
using System.Threading.Tasks;
using SequenceAuth.Lib;

namespace SequenceAuth.Example.Tests;

public class InMemorySequenceStore : ISequenceStore
{
    private readonly ConcurrentDictionary<string, SequenceData> _store = new();

    public Task<Option<SequenceData>> GetSequenceAsync(string sequenceId)
    {
        if (_store.TryGetValue(sequenceId, out var data))
        {
            return Task.FromResult(Option<SequenceData>.Some(data));
        }
        return Task.FromResult(Option<SequenceData>.None());
    }

    public Task<StoreOutcome> SaveSequenceAsync(string sequenceId, SequenceData data)
    {
        _store[sequenceId] = data;
        return Task.FromResult(StoreOutcome.Success);
    }

    public Task<StoreOutcome> InvalidateUserSessionsAsync(string userId)
    {
        var keysToRemove = _store.Where(kvp => kvp.Value.UserId == userId).Select(kvp => kvp.Key).ToList();
        foreach (var key in keysToRemove)
        {
            _store.TryRemove(key, out _);
        }
        return Task.FromResult(StoreOutcome.Success);
    }
}