using System.Collections.Concurrent;
using AyCode.Core;
using AyCode.Core.Extensions;
using AyCode.Core.Helpers;
using AyCode.Core.Loggers;
using AyCode.Services.Loggers;
using AyCode.Services.SignalRs;
using MessagePack.Resolvers;
using Microsoft.AspNetCore.SignalR.Client;

namespace AyCode.Blazor.Components.Services
{
    public abstract class AcSignalRClientBase : IAcSignalRHubClient
    {
        private readonly ConcurrentDictionary<int, object?> _responseByRequestId = new();

        protected readonly HubConnection HubConnection;
        protected readonly AcLoggerBase Logger;

        public event Action<int, byte[], int?> OnMessageReceived = null!;
        //public event Action<int, int> OnMessageRequested;

        public int Timeout = 10000;

        protected AcSignalRClientBase(string fullHubName, AcLoggerBase logger)
        {
            Logger = logger;

            HubConnection = new HubConnectionBuilder()
                .WithUrl(fullHubName)
                //.AddMessagePackProtocol(options => {
                //        options.SerializerOptions = MessagePackSerializerOptions.Standard
                //            .WithResolver(MessagePack.Resolvers.StandardResolver.Instance)
                //            .WithSecurity(MessagePackSecurity.UntrustedData)
                //            .WithCompression(MessagePackCompression.Lz4Block)
                //            .WithCompressionMinLength(256);})
                .Build();

            HubConnection.Closed += HubConnection_Closed;

            _ = HubConnection.On<int, byte[], int?>(nameof(IAcSignalRHubClient.OnReceiveMessage), OnReceiveMessage);
            //_ = HubConnection.On<int, int>(nameof(IAcSignalRHubClient.OnRequestMessage), OnRequestMessage);

            HubConnection.StartAsync().Forget();

        }

        private Task HubConnection_Closed(Exception? arg)
        {
            if (_responseByRequestId.IsEmpty) Logger.DebugConditional($"Client HubConnection_Closed");
            else Logger.Warning($"Client HubConnection_Closed; {nameof(_responseByRequestId)} count: {_responseByRequestId.Count}");

            _responseByRequestId.Clear();
            return Task.CompletedTask;
        }

        public async Task StartConnection()
        {
            if (HubConnection.State == HubConnectionState.Disconnected)
                await HubConnection.StartAsync();

            if (HubConnection.State != HubConnectionState.Connected)
                await TaskHelper.WaitToAsync(() => HubConnection.State == HubConnectionState.Connected, Timeout, 25);
        }

        public async Task StopConnection()
        {
            await HubConnection.StopAsync();
            await HubConnection.DisposeAsync();
        }

        public virtual Task SendMessageToServerAsync(int messageTag)
            => SendMessageToServerAsync(messageTag, null, AcDomain.NextUniqueInt32);

        public virtual Task SendMessageToServerAsync(int messageTag, ISignalRMessage? message, int? requestId)
        {
            Logger.DebugConditional($"Client SendMessageToServerAsync; {nameof(messageTag)}: {messageTag}; {nameof(requestId)}: {requestId};");

            return StartConnection().ContinueWith(x =>
            {
                var msgp = message?.ToMessagePack(ContractlessStandardResolver.Options);
                return HubConnection.SendAsync(nameof(IAcSignalRHubClient.OnReceiveMessage), messageTag, msgp, requestId);
            });
        }

        #region CRUD
        public virtual Task<TResponseData?> GetByIdAsync<TResponseData>(int messageTag, params Guid[] ids) where TResponseData : class
            => SendMessageToServerAsync<TResponseData>(messageTag, new SignalPostJsonDataMessage<IdMessage>(new IdMessage(ids)), AcDomain.NextUniqueInt32);
        public virtual Task GetByIdAsync<TResponseData>(int messageTag, Func<ISignalResponseMessage<TResponseData?>, Task> responseCallback, params Guid[] ids)
            => SendMessageToServerAsync(messageTag, new SignalPostJsonDataMessage<IdMessage>(new IdMessage(ids)), responseCallback);

        public virtual Task<TResponseData?> GetAllAsync<TResponseData>(int messageTag) where TResponseData : class
            => SendMessageToServerAsync<TResponseData>(messageTag);
        public virtual Task GetAllAsync<TResponseData>(int messageTag, Func<ISignalResponseMessage<TResponseData?>, Task> responseCallback)
            => SendMessageToServerAsync(messageTag, null, responseCallback);
        public virtual Task GetAllAsync<TResponseData>(int messageTag, Func<ISignalResponseMessage<TResponseData?>, Task> responseCallback, params Guid[]? contextIds)
            => SendMessageToServerAsync(messageTag, (contextIds == null || contextIds.Length == 0 ? null : new SignalPostJsonDataMessage<IdMessage>(new IdMessage(contextIds))), responseCallback);

        public virtual Task<TResponseData?> GetAllAsync<TResponseData>(int messageTag, params Guid[]? contextIds) where TResponseData : class
            => SendMessageToServerAsync<TResponseData>(messageTag, contextIds == null || contextIds.Length == 0 ? null : new SignalPostJsonDataMessage<IdMessage>(new IdMessage(contextIds)), AcDomain.NextUniqueInt32);

        public virtual Task<TPostData?> PostDataAsync<TPostData>(int messageTag, TPostData postData) where TPostData : class
            => SendMessageToServerAsync<TPostData>(messageTag, new SignalPostJsonDataMessage<TPostData>(postData), AcDomain.NextUniqueInt32);
        public virtual Task PostDataAsync<TPostData>(int messageTag, TPostData postData, Func<ISignalResponseMessage<TPostData?>, Task> responseCallback) where TPostData : class
            => SendMessageToServerAsync(messageTag, new SignalPostJsonDataMessage<TPostData>(postData), responseCallback);

        #endregion CRUD

        public virtual Task<TResponse?> SendMessageToServerAsync<TResponse>(int messageTag) where TResponse : class
            => SendMessageToServerAsync<TResponse>(messageTag, null, AcDomain.NextUniqueInt32);

        public virtual Task<TResponse?> SendMessageToServerAsync<TResponse>(int messageTag, ISignalRMessage? message) where TResponse : class
            => SendMessageToServerAsync<TResponse>(messageTag, message, AcDomain.NextUniqueInt32);

        protected virtual async Task<TResponse?> SendMessageToServerAsync<TResponse>(int messageTag, ISignalRMessage? message, int requestId) where TResponse : class
        {
            Logger.DebugConditional($"Client SendMessageToServerAsync<TResult>; {nameof(messageTag)}: {messageTag}; {nameof(requestId)}: {requestId};");

            _responseByRequestId[requestId] = null;
            await SendMessageToServerAsync(messageTag, message, requestId);

            try
            {
                if (await TaskHelper.WaitToAsync(() => _responseByRequestId[requestId] != null, Timeout, 25) &&
                    _responseByRequestId.TryRemove(requestId, out var obj) && obj is ISignalResponseMessage<string> responseMessage)
                {
                    if (responseMessage.Status == SignalResponseStatus.Error || responseMessage.ResponseData == null)
                    {
                        var errorText = $"Client SendMessageToServerAsync<TResponseData> response error; await; tag: {messageTag}; Status: {responseMessage.Status}; requestId: {requestId};";

                        Logger.Error(errorText);

                        //TODO: Ideiglenes, majd a ResponseMessage-et kell visszaadni a Status miatt! - J.
                        throw new Exception(errorText);
                        //return default;
                    }

                    return responseMessage.ResponseData.JsonTo<TResponse>();
                }
            }
            catch (Exception ex)
            {
                Logger.Error($"SendMessageToServerAsync; messageTag: {messageTag}; requestId: {requestId}; {ex.Message}", ex);
            }

            _responseByRequestId.TryRemove(requestId, out _);
            return default;
        }

        public virtual Task SendMessageToServerAsync<TResponseData>(int messageTag, Func<ISignalResponseMessage<TResponseData?>, Task> responseCallback)
            => SendMessageToServerAsync(messageTag, null, responseCallback);

        public virtual Task SendMessageToServerAsync<TResponseData>(int messageTag, ISignalRMessage? message, Func<ISignalResponseMessage<TResponseData?>, Task> responseCallback)
        {
            if (messageTag == 0)
                Logger.Error($"SendMessageToServerAsync; messageTag == 0");

            var requestId = AcDomain.NextUniqueInt32;

            _responseByRequestId[requestId] = new Action<ISignalResponseMessage<string>>(responseMessage =>
            {
                TResponseData? responseData = default;

                if (responseMessage.Status == SignalResponseStatus.Success)
                {
                    responseData = string.IsNullOrEmpty(responseMessage.ResponseData) ? default : responseMessage.ResponseData.JsonTo<TResponseData?>();
                }
                else Logger.Error($"Client SendMessageToServerAsync<TResponseData> response error; callback; tag: {messageTag}; Status: {responseMessage.Status}; requestId: {requestId};");

                responseCallback(new SignalResponseMessage<TResponseData?>(responseMessage.Status, responseData));
            });

            return SendMessageToServerAsync(messageTag, message, requestId);
        }

        public virtual Task OnReceiveMessage(int messageTag, byte[] message, int? requestId)
        {
            var logText = $"Client OnReceiveMessage; {nameof(messageTag)}: {messageTag}; {nameof(requestId)}: {requestId};";

            if (message.Length == 0) Logger.Warning($"message.Length == 0! {logText}");
            else Logger.Info(logText);

            try
            {
                if (requestId.HasValue && _responseByRequestId.ContainsKey(requestId.Value))
                {
                    var reqId = requestId.Value;

                    var responseMessage = message.MessagePackTo<SignalResponseJsonMessage>(ContractlessStandardResolver.Options);

                    switch (_responseByRequestId[reqId])
                    {
                        case null:
                            _responseByRequestId[reqId] = responseMessage;
                            return Task.CompletedTask;

                        case Action<ISignalResponseMessage<string>> messagePackCallback:
                            _responseByRequestId.TryRemove(reqId, out _);

                            messagePackCallback.Invoke(responseMessage);
                            return Task.CompletedTask;

                        //case Action<string> jsonCallback:
                        //    _responseByRequestId.TryRemove(reqId, out _);

                        //    jsonCallback.Invoke(responseMessage);
                        //    return Task.CompletedTask;

                        default:
                            Logger.Error($"Client OnReceiveMessage switch; unknown message type: {_responseByRequestId[reqId]?.GetType().Name}");
                            break;
                    }

                    _responseByRequestId.TryRemove(reqId, out _);
                }

                OnMessageReceived(messageTag, message, requestId);
            }
            catch (Exception ex)
            {
                if (requestId.HasValue)
                    _responseByRequestId.TryRemove(requestId.Value, out _);

                Logger.Error($"Client OnReceiveMessage; messageTag: {messageTag}; requestId: {requestId}; {ex.Message}", ex);
                throw;
            }

            return Task.CompletedTask;
        }
        //public virtual Task OnRequestMessage(int messageTag, int requestId)
        //{
        //    Logger.DebugConditional($"Client OnRequestMessage; {nameof(messageTag)}: {messageTag}; {nameof(requestId)}: {requestId};");

        //    try
        //    {
        //        OnMessageRequested(messageTag, requestId);
        //    }
        //    catch(Exception ex)
        //    {
        //        Logger.Error($"Client OnReceiveMessage; {nameof(messageTag)}: {messageTag}; {nameof(requestId)}: {requestId}; {ex.Message}", ex);
        //        throw;
        //    }

        //    return Task.CompletedTask;

        //}
    }
}