inabox/InABox.Server/RPC/RPCServer.cs

using System.Reflection;
using H.Pipes.Args;
using InABox.API;
using InABox.Clients;
using InABox.Core;
using InABox.IPC;

namespace InABox.Rpc
{

    public class RpcServer<TTransport> : IDisposable, IRpcServer where TTransport : IRpcServerTransport
    {
        private TTransport _transport;

        public event LogEvent? OnLog;
        
        public bool IsSecure() => _transport?.IsSecure() ?? false;

        public RpcServer(TTransport transport)
        {
            _transport = transport;

            _transport.OnOpen += Transport_OnOpen;
            _transport.OnClose += Transport_OnClose;
            _transport.OnException += Transport_OnException;

            _transport.BeforeMessage += Transport_BeforeMessage;
            _transport.AfterMessage += Transport_AfterMessage;

            _transport.AddHandler<IRpcServer, RpcPingCommand, RpcPingParameters, RpcPingResult>(new RpcPingHandler(this));
            _transport.AddHandler<IRpcServer, RpcInfoCommand, RpcInfoParameters, RpcInfoResult>(new RpcInfoHandler(this));

            _transport.AddHandler<IRpcServer, RpcValidateCommand, RpcValidateParameters, RpcValidateResult>(new RpcValidateHandler(this));
            _transport.AddHandler<IRpcServer, RpcCheck2FACommand, RpcCheck2FAParameters, RpcCheck2FAResult>(new RpcCheck2FAHandler(this));

            _transport.AddHandler<IRpcServer, RpcQueryCommand, RpcQueryParameters, RpcQueryResult>(new RpcQueryHandler(this));
            _transport.AddHandler<IRpcServer, RpcSaveCommand, RpcSaveParameters, RpcSaveResult>(new RpcSaveHandler(this));
            _transport.AddHandler<IRpcServer, RpcDeleteCommand, RpcDeleteParameters, RpcDeleteResult>(new RpcDeleteHandler(this));

        }

        private void Transport_OnOpen(IRpcTransport transport, RpcTransportOpenArgs e)
        {
            OnLog?.Invoke(LogType.Information, "", $"Client Connected [{e.Session?.ID}]");
        }

        private void Transport_OnClose(IRpcTransport transport, RpcTransportCloseArgs e)
        {
            OnLog?.Invoke(LogType.Information, "", $"Client Disconnected({e.Type}) [{e.Session?.ID}]");
        }

        private void Transport_BeforeMessage(IRpcTransport transport, RpcTransportMessageArgs e)
        {
            OnLog?.Invoke(LogType.Information, "", $"Request Received [{e.Session?.ID}]: {e.Message?.Command ?? "Unknown"}");
        }

        private void Transport_AfterMessage(IRpcTransport transport, RpcTransportMessageArgs e)
        {
            OnLog?.Invoke(LogType.Information, "", $"Sending Response [{e.Session?.ID}]: {e.Message?.Command ?? "Unknown"}");
        }

        private void Transport_OnException(IRpcTransport transport, RpcTransportExceptionArgs e)
        {
            OnLog?.Invoke(LogType.Error, $"", $"Exception Occurred in {e.Session?.ID}: {e.Exception.Message}");
        }

        public void Start()
        {
            _transport.Start();
        }

        public void Stop()
        {
            _transport.Stop();
        }
        
        public void Dispose()
        {
            _transport?.Stop();
        }

        ~RpcServer()
        {
            Dispose();
        }
    }
}