using Fleet.Core;
using Fleet.Rpc;
using Fleet.Server;
using Microsoft.Extensions.DependencyInjection;
using Monitor.Backend.Application;
using Monitor.Protocol;

namespace Monitor.Backend.Gateway;

public static class GatewayServiceCollectionExtensions
{
    /// <summary>
    /// 在 Fleet.Server 之上接入 Monitor：期望状态（TaskSet）、RawChunk 收件箱、<c>monitor.*</c> 方法、注册挂钩和节点事件。
    /// 需要事先注册 <see cref="IFleetStore"/> 和 <see cref="TimeProvider"/>，并在 <c>AddMonitorApplication</c> 之后调用。
    /// </summary>
    public static IServiceCollection AddMonitorGateway(this IServiceCollection services)
    {
        services.AddSingleton<IAgentGateway, FleetGateway>();
        services.AddHostedService<ConnectionShutdown>();
        services.AddFleetEnrollment();
        services.AddFleetServer(options => options.NodeIdPrefix = AgentIds.Prefix)
            .AddState<TaskSetStateProvider>()
            .AddState<ProbeAgentsStateProvider>()
            .AddInbox<RawChunkInbox>()
            .MapMethods(MapMonitorMethods);
        return services;
    }

    private static void MapMonitorMethods(RpcRouter<NodeCallContext> router, IServiceProvider services)
    {
        router
            .OnNotify(
                MonitorMethods.Event,
                MonitorJsonContext.Default.AgentEventParams,
                (context, parameters, ct) => services.GetRequiredService<AgentEventService>().OnEventAsync(context.Node.Id, parameters, ct))
            .OnNotify(
                MonitorMethods.Host,
                MonitorJsonContext.Default.HostSnapshot,
                (context, snapshot, ct) => services.GetRequiredService<HostStatusService>().OnHostAsync(context.Node.Id, snapshot, ct))
            .OnNotify(
                MonitorMethods.TestStatus,
                MonitorJsonContext.Default.TestStatusParams,
                (context, parameters, ct) => services.GetRequiredService<TestService>().OnStatusAsync(context.Node.Id, parameters, ct));
    }
}