using Serilog.Core;
using Serilog.Events;
using System.Collections.Concurrent;
using System.Globalization;

namespace ErsatzTV.Core;

public class InMemorySink : ILogEventSink
{
    private readonly ConcurrentDictionary<Guid, ConcurrentQueue<string>> _logs = new();

    public void Emit(LogEvent logEvent)
    {
        if (logEvent.Properties.TryGetValue(InMemoryLogService.CorrelationIdKey, out var correlationIdValue) &&
            correlationIdValue is ScalarValue { Value: Guid correlationId })
        {
            ConcurrentQueue<string> logQueue = _logs.GetOrAdd(correlationId, _ => new ConcurrentQueue<string>());

            string message = logEvent.RenderMessage(CultureInfo.CurrentCulture);
            logQueue.Enqueue($"[{logEvent.Timestamp:HH:mm:ss} {logEvent.Level}] {message}");

            while (logQueue.Count > 100)
            {
                logQueue.TryDequeue(out _);
            }
        }
    }

    public IEnumerable<string> GetLogs(Guid correlationId)
    {
        _logs.TryGetValue(correlationId, out ConcurrentQueue<string> logs);
        return logs ?? Enumerable.Empty<string>();
    }

    public void ClearLogs(Guid correlationId)
    {
        _logs.TryRemove(correlationId, out _);
    }
}

public class InMemoryLogService
{
    public InMemorySink Sink { get; } = new();

    public static readonly string CorrelationIdKey = "CorrelationId";
}