BililiveRecorder/BililiveRecorder.Core/DependencyInjectionExtensions.cs

using BililiveRecorder.Core;
using BililiveRecorder.Core.Api;
using BililiveRecorder.Core.Api.Danmaku;
using BililiveRecorder.Core.Api.Http;
using BililiveRecorder.Core.Config.V2;
using BililiveRecorder.Core.Danmaku;
using BililiveRecorder.Core.Recording;
using BililiveRecorder.Flv;
using Microsoft.Extensions.DependencyInjection;
using Polly;
using Polly.Registry;

namespace BililiveRecorder.DependencyInjection
{
    public static class DependencyInjectionExtensions
    {
        public static IServiceCollection AddRecorderConfig(this IServiceCollection services, ConfigV2 config) => services
            .AddSingleton(config)
            .AddSingleton(sp => sp.GetRequiredService<ConfigV2>().Global)
            ;

        public static IServiceCollection AddRecorder(this IServiceCollection services) => services
            .AddSingleton<IMemoryStreamProvider, RecyclableMemoryStreamProvider>()
            .AddRecorderPollyPolicy()
            .AddRecorderApiClients()
            .AddRecorderRecording()
            .AddSingleton<IRecorder, Recorder>()
            .AddSingleton<IRoomFactory, RoomFactory>()
            .AddScoped<IRoom, Room>()
            .AddScoped<IBasicDanmakuWriter, BasicDanmakuWriter>()
            ;

        private static IServiceCollection AddRecorderPollyPolicy(this IServiceCollection services)
        {
            var registry = new PolicyRegistry // TODO
            {
                [PolicyNames.PolicyRoomInfoApiRequestAsync] = Policy.NoOpAsync(),
                [PolicyNames.PolicyDanmakuApiRequestAsync] = Policy.NoOpAsync(),
                [PolicyNames.PolicyStreamApiRequestAsync] = Policy.NoOpAsync(),
            };

            return services.AddSingleton<IReadOnlyPolicyRegistry<string>>(registry);
        }

        public static IServiceCollection AddRecorderApiClients(this IServiceCollection services) => services
            .AddSingleton<HttpApiClient>()
            .AddSingleton<PolicyWrappedApiClient<HttpApiClient>>()
            .AddSingleton<IApiClient>(sp => sp.GetRequiredService<PolicyWrappedApiClient<HttpApiClient>>())
            .AddSingleton<IDanmakuServerApiClient>(sp => sp.GetRequiredService<PolicyWrappedApiClient<HttpApiClient>>())
            .AddScoped<IDanmakuClient, DanmakuClient>()
            ;

        public static IServiceCollection AddRecorderRecording(this IServiceCollection services) => services
            .AddScoped<IRecordTaskFactory, RecordTaskFactory>()
            .AddScoped<IFlvProcessingContextWriterFactory, FlvProcessingContextWriterFactory>()
            .AddScoped<IFlvTagReaderFactory, FlvTagReaderFactory>()
            .AddScoped<ITagGroupReaderFactory, TagGroupReaderFactory>()
            ;
    }
}