TourIAm/TIAMWebApp/Server/Services/DevAdminSignalRhub.cs

using System.Collections.Concurrent;
using System.Reflection;
using AyCode.Core.Extensions;
using AyCode.Core.Loggers;
using AyCode.Services.SignalRs;
using Microsoft.AspNetCore.SignalR;
using TIAM.Database.DataLayers.Admins;
using MessagePack.Resolvers;
using AyCode.Services.Server.SignalRs;
using TIAM.Services;
using TIAMWebApp.Server.Controllers;
using TIAM.Entities.ServiceProviders;
using System.Runtime.CompilerServices;
using MessagePack;
using TIAM.Entities.Addresses;
using TIAM.Entities.Profiles;

namespace TIAMWebApp.Server.Services;

public static class ExtensionMethods
{
    public static object? InvokeMethod(this MethodInfo methodInfo, object obj, params object[]? parameters)
    {
        if (methodInfo.GetCustomAttribute(typeof(AsyncStateMachineAttribute)) is AsyncStateMachineAttribute isAsyncTask)
        {
            dynamic awaitable = methodInfo.Invoke(obj, parameters)!;
            return awaitable.GetAwaiter().GetResult();
        }

        return methodInfo.Invoke(obj, parameters);
    }
}

public class MethodInfoModel<TAttribute> where TAttribute : TagAttribute
{
    public Type? ParameterType { get; init; } = null;
    public TAttribute Attribute { get; init; }
    public MethodInfo MethodInfo { get; init; }

    public MethodInfoModel(TAttribute attribute, MethodInfo methodInfo)
    {
        Attribute = attribute;
        MethodInfo = methodInfo;

        var parameters = methodInfo.GetParameters();

        if (parameters.Length > 1)
            throw new Exception("MethodInfoModel; parameters.Length > 1");

        if (parameters.Length == 1)
            ParameterType = parameters[0].ParameterType;
    }
}

public class DynamicMethodCallModel<TAttribute> where TAttribute : TagAttribute
{
    public object InstanceObject { get; init; }
    public ConcurrentDictionary<int, MethodInfoModel<TAttribute>> MethodsByMessageTag { get; init; } = new();

    public DynamicMethodCallModel(object instanceObject)
    {
        InstanceObject = instanceObject;

        foreach (var methodInfo in instanceObject.GetType().GetMethods())
        {
            if (methodInfo.GetCustomAttribute(typeof(TAttribute)) is not TAttribute attribute) continue;

            MethodsByMessageTag[attribute.MessageTag] = new MethodInfoModel<TAttribute>(attribute, methodInfo!);
        }
    }
}

public class DevAdminSignalRHub : Hub<ISignalRHubItemServer>, IAcSignalRHubServer
{
    private readonly List<DynamicMethodCallModel<SignalRAttribute>> _dynamicMethodCallModels = new();
    private readonly TIAM.Core.Loggers.Logger<DevAdminSignalRHub> _logger;

    private readonly AdminDal _adminDal;
    private readonly ServiceProviderAPIController _serviceProviderApiController;
    private readonly TransferDataAPIController _transferDataApiController;

    public DevAdminSignalRHub(AdminDal adminDal, ServiceProviderAPIController serviceProviderApiController, TransferDataAPIController transferDataApiController, IEnumerable<IAcLogWriterBase> logWriters)
    {
        _adminDal = adminDal;
        _serviceProviderApiController = serviceProviderApiController;
        _transferDataApiController = transferDataApiController;

        _logger = new(logWriters.ToArray());

        _dynamicMethodCallModels.Add(new DynamicMethodCallModel<SignalRAttribute>(serviceProviderApiController));
        _dynamicMethodCallModels.Add(new DynamicMethodCallModel<SignalRAttribute>(transferDataApiController));

    }

    // https://docs.microsoft.com/en-us/aspnet/core/signalr/hubs?view=aspnetcore-3.1#strongly-typed-hubs
    public override async Task OnConnectedAsync()
    {
        _logger.Debug($"Server OnConnectedAsync; ConnectionId: {Context.ConnectionId}; UserIdentifier: {Context.UserIdentifier}");

        //await Groups.AddToGroupAsync(Context.ConnectionId, "SignalR Users");
        await base.OnConnectedAsync();

        //Clients.Caller.ConnectionId = Context.ConnectionId;
        //Clients.Caller.UserIdentifier = Context.UserIdentifier;
    }

    public override async Task OnDisconnectedAsync(Exception? exception)
    {
        _logger.ErrorConditional($"Server OnDisconnectedAsync; ConnectionId: {Context.ConnectionId}; UserIdentifier: {Context.UserIdentifier}", exception);

        //await Groups.RemoveFromGroupAsync(Context.ConnectionId, "SignalR Users");
        await base.OnDisconnectedAsync(exception);
    }

    public async Task OnReceiveMessage(int messageTag, byte[]? message, int? requestId)
    {
        var logText = $"Server OnReceiveMessage; {nameof(messageTag)}: {messageTag}; {nameof(requestId)}: {requestId}; ConnectionId: {Context.ConnectionId}; UserIdentifier: {Context.UserIdentifier}";

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

        try
        {
            foreach (var methodsByDeclaringObject in _dynamicMethodCallModels)
            {
                if (!methodsByDeclaringObject.MethodsByMessageTag.TryGetValue(messageTag, out var methodInfoModel)) continue;

                object[]? paramValues = null;

                logText = $"Found dynamic method for the tag! tag: {messageTag}; method: {methodsByDeclaringObject.InstanceObject.GetType().Name}.{methodInfoModel.MethodInfo.Name}";

                if (methodInfoModel.ParameterType != null)
                {
                    _logger.Debug($"{logText}({methodInfoModel.ParameterType.Name})");

                    paramValues = new object[1];

                    if (methodInfoModel.ParameterType == typeof(Guid) || methodInfoModel.ParameterType == typeof(Guid?))
                    {
                        var msg = message!.MessagePackTo<SignalPostJsonDataMessage<IdMessage>>();
                        if (msg.PostData.Id.IsNullOrEmpty()) throw new NullReferenceException($"PostData.Id.IsNullOrEmpty(); Id: {msg.PostData.Id}");

                        paramValues[0] = msg.PostData.Id;
                    }
                    else paramValues[0] = message!.MessagePackTo<SignalPostJsonDataMessage<object>>(MessagePackSerializerOptions.Standard).PostDataJson.JsonTo(methodInfoModel.ParameterType)!;
                }
                else _logger.Debug($"{logText}()");

                await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, methodInfoModel.MethodInfo.InvokeMethod(methodsByDeclaringObject.InstanceObject, paramValues)), requestId);
                    
                return;
            }

            _logger.Debug($"Not found dynamic method for the tag! tag: {messageTag};");

            switch (messageTag)
            {
                case SignalRTags.GetAddress:
                    var id = message!.MessagePackTo<SignalPostJsonDataMessage<IdMessage>>().PostData.Id;

                    var address = await _adminDal.GetAddressByIdAsync(id);
                    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, address), requestId);

                    return;

                case SignalRTags.GetAddressesByContextId:
                    id = message!.MessagePackTo<SignalPostJsonDataMessage<IdMessage>>().PostData.Id;

                    address = await _adminDal.GetAddressByIdAsync(id);
                    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, new List<Address> { address! }), requestId);

                    return;

                case SignalRTags.UpdateAddress:
                    address = message!.MessagePackTo<SignalPostJsonDataMessage<Address>>().PostData;

                    await _adminDal.UpdateAddressAsync(address);
                    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, address), requestId);

                    return;

                case SignalRTags.UpdateProfile:
                    var profile = message!.MessagePackTo<SignalPostJsonDataMessage<Profile>>().PostData;

                    await _adminDal.UpdateProfileAsync(profile);
                    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, profile), requestId);

                    return;

                //case SignalRTags.GetTransfersAsync:
                //    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, await _transferDataApiController.GetTransfers()), requestId);

                //    return;

                //case SignalRTags.GetPropertiesByOwnerIdAsync:
                //    var ownerId = message!.MessagePackTo<SignalRequestByIdMessage>().Id;

                //    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, await _serviceProviderApiController.GetServiceProvidersByOwnerId(ownerId)), requestId);

                //    return;

                //case SignalRTags.UpdateTransferAsync:
                //    var transfer = message!.MessagePackTo<SignalPostJsonDataMessage<Transfer>>().PostData;

                //    await _transferDataApiController.UpdateTransfer(transfer);
                //    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, transfer), requestId);

                //    return;

                //case SignalRTags.GetCompaniesAsync:
                //    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, await _serviceProviderApiController.GetServiceProviders()), requestId);

                //    return;
                //case SignalRTags.UpdateCompanyAsync:

                //    var updateCompany = message!.MessagePackTo<SignalPostJsonDataMessage<Company>>().PostData;

                //    await _serviceProviderApiController.UpdateServiceProvider(updateCompany);
                //    await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Success, updateCompany), requestId);

                //    return; 

                default:
                    _logger.Error($"Server OnReceiveMessage; messageTag not found! messageTag: {messageTag}");
                    break;
            }
        }
        catch (Exception ex)
        {
            _logger.Error($"Server OnReceiveMessage; {ex.Message}", ex);
        }

        await ResponseToCaller(messageTag, new SignalResponseJsonMessage(SignalResponseStatus.Error), requestId);
    }

    protected async Task ResponseToCaller(int messageTag, ISignalRMessage message, int? requestId)
        => await SendMessageToClient(Clients.Caller, messageTag, message, requestId);

    protected async Task SendMessageToClient(ISignalRHubItemServer sendTo, int messageTag, ISignalRMessage message, int? requestId = null)
    {
        _logger.Info($"Server SendMessageToClient; {nameof(messageTag)}: {messageTag}; {nameof(requestId)}: {requestId}; ConnectionId: {Context.ConnectionId}; UserIdentifier: {Context.UserIdentifier}");

        await sendTo.OnReceiveMessage(messageTag, message.ToMessagePack(ContractlessStandardResolver.Options), requestId);
    }

    public async Task SendMessageToGroup(string groupId, int messageTag, string message)
    {
        //await Clients.Group(groupId).Post("", messageTag, message);
    }
}