using PrivaPub.Infrastructure.Statistics;
using PrivaPub.Models.Statistics;

using System.Collections.Concurrent;

namespace PrivaPub.Tests.Support
{
	public sealed class MemoryLedger : IInteractionLedger
	{
		public ConcurrentQueue<(InteractionEvent Event, string ActorUri, bool HostClaimed)> Events { get; } = new();
		public ConcurrentDictionary<(string Host, string Key), long> Counts { get; } = new();
		public ConcurrentDictionary<(string Section, string Key), long> ServerCounts { get; } = new();

		public long Dropped => 0;

		public void Record(InteractionEvent interaction, string actorUri = default, bool hostClaimed = false) =>
			Events.Enqueue((Interactions.Sanitize(interaction), actorUri, hostClaimed));

		public void Count(string host, string key, long bytes = 0) =>
			Counts.AddOrUpdate((Interactions.Host(host), key), 1, (_, count) => count + 1);

		public void CountServer(string section, string key, int? latencyMs = default) =>
			ServerCounts.AddOrUpdate((section, key), 1, (_, count) => count + 1);

		public IReadOnlyList<InteractionEvent> Of(string channel) => Events.Select(e => e.Event).Where(e => e.Channel == channel).ToList();
	}
}