using System.Diagnostics.CodeAnalysis;
using Autofac.Features.Indexed;

namespace Recyclarr.TestLibrary.Autofac;

public class StubAutofacIndex<TKey, TValue> : IIndex<TKey, TValue>
    where TKey : notnull
{
    private Dictionary<TKey, TValue> _values = new();

    public StubAutofacIndex()
    {
    }

    public StubAutofacIndex(Dictionary<TKey, TValue> values)
    {
        _values = values;
    }

    public void Add(TKey key, TValue value)
    {
        _values.Add(key, value);
    }

    public void AddRange(IEnumerable<(TKey, TValue)> pairs)
    {
        _values = _values.Union(pairs.ToDictionary(x => x.Item1, x => x.Item2)).ToDictionary();
    }

    public bool TryGetValue(TKey key, [UnscopedRef] out TValue value)
    {
        return _values.TryGetValue(key, out value!);
    }

    public TValue this[TKey key] => _values[key];
    public IReadOnlyCollection<TKey> Keys => _values.Keys;
    public IReadOnlyCollection<TValue> Values => _values.Values;
}