using System.Net.WebSockets;
using DysonNetwork.Sphere.Chat;
using DysonNetwork.Sphere.Storage;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Caching.Memory;
using Microsoft.Extensions.Internal;
using SystemClock = NodaTime.SystemClock;

namespace DysonNetwork.Sphere.Connection.Handlers;

public class MessageReadHandler(
    AppDatabase db,
    IMemoryCache cache,
    ChatRoomService crs,
    FlushBufferService buffer
)
    : IWebSocketPacketHandler
{
    public string PacketType => "messages.read";

    public const string ChatMemberCacheKey = "ChatMember_{0}_{1}";

    public async Task HandleAsync(
        Account.Account currentUser,
        string deviceId,
        WebSocketPacket packet,
        WebSocket socket,
        WebSocketService srv
    )
    {
        var request = packet.GetData<ChatController.MarkMessageReadRequest>();
        if (request is null)
        {
            await socket.SendAsync(
                new ArraySegment<byte>(new WebSocketPacket
                {
                    Type = WebSocketPacketType.Error,
                    ErrorMessage = "Mark message as read requires you provide the ChatRoomId and MessageId"
                }.ToBytes()),
                WebSocketMessageType.Binary,
                true,
                CancellationToken.None
            );
            return;
        }

        ChatMember? sender = null;
        var cacheKey = string.Format(ChatMemberCacheKey, currentUser.Id, request.ChatRoomId);
        if (cache.TryGetValue(cacheKey, out ChatMember? cachedMember))
            sender = cachedMember;
        else
        {
            sender = await db.ChatMembers
                .Where(m => m.AccountId == currentUser.Id && m.ChatRoomId == request.ChatRoomId)
                .FirstOrDefaultAsync();

            if (sender != null)
            {
                var cacheOptions = new MemoryCacheEntryOptions()
                    .SetAbsoluteExpiration(TimeSpan.FromMinutes(5));
                cache.Set(cacheKey, sender, cacheOptions);
            }
        }

        if (sender is null)
        {
            await socket.SendAsync(
                new ArraySegment<byte>(new WebSocketPacket
                {
                    Type = WebSocketPacketType.Error,
                    ErrorMessage = "User is not a member of the chat room."
                }.ToBytes()),
                WebSocketMessageType.Binary,
                true,
                CancellationToken.None
            );
            return;
        }

        var readReceipt = new MessageReadReceipt
        {
            MessageId = request.MessageId,
            SenderId = sender.Id,
        };

        buffer.Enqueue(readReceipt);

        var otherMembers = (await crs.ListRoomMembers(request.ChatRoomId)).Select(m => m.AccountId).ToList();
        foreach (var member in otherMembers)
            srv.SendPacketToAccount(member, packet);
    }
}