CallerOptionsExtensions.cs

// Copyright (c) MASA Stack All rights reserved.
// Licensed under the MIT License. See LICENSE.txt in the project root for license information.

namespace Masa.Contrib.Service.Caller.DaprClient;

public static class CallerOptionsExtensions
{
    public static DefaultDaprClientBuilder UseDapr(this CallerOptions callerOptions,
        Func<MasaDaprClientBuilder> clientBuilder)
        => callerOptions.UseDapr(Microsoft.Extensions.Options.Options.DefaultName, clientBuilder);

    public static DefaultDaprClientBuilder UseDapr(this CallerOptions callerOptions,
        string name,
        Func<MasaDaprClientBuilder> clientBuilder)
    {
        ArgumentNullException.ThrowIfNull(clientBuilder, nameof(ArgumentNullException));

        MasaDaprClientBuilder builder = clientBuilder.Invoke();

        callerOptions.Services.AddDaprClient(daprClientBuilder =>
        {
            if (callerOptions.JsonSerializerOptions != null)
                daprClientBuilder.UseJsonSerializationOptions(callerOptions.JsonSerializerOptions);

            builder.Configure?.Invoke(daprClientBuilder);
        });

        callerOptions.Services.TryAddSingleton<ICallerProvider, DefaultCallerProvider>();
        callerOptions.Services.AddOptions();
        AddCallerExtensions.AddCaller(callerOptions, name,
            serviceProvider =>
            {
                string appId = builder.AppId;
                var daprCaller = new DaprCaller(serviceProvider,
                    name,
                    serviceProvider.GetRequiredService<ICallerProvider>().CompletionAppId(appId));
                return daprCaller;
            });
        return new DefaultDaprClientBuilder(callerOptions.Services, name);
    }

    public static DefaultDaprClientBuilder UseDapr(this CallerOptions callerOptions,
        Action<MasaDaprClientBuilder> clientBuilder)
        => callerOptions.UseDapr(Microsoft.Extensions.Options.Options.DefaultName, clientBuilder);

    public static DefaultDaprClientBuilder UseDapr(this CallerOptions callerOptions,
        string name,
        Action<MasaDaprClientBuilder> clientBuilder)
    {
        ArgumentNullException.ThrowIfNull(clientBuilder, nameof(clientBuilder));

        MasaDaprClientBuilder builder = new MasaDaprClientBuilder();
        clientBuilder.Invoke(builder);

        return callerOptions.UseDapr(name, () => builder);
    }
}