57 lines
2.1 KiB
C#
57 lines
2.1 KiB
C#
using Falcon.SugarApi.Encryption;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.Extensions.DependencyInjection.Extensions;
|
||
using System;
|
||
|
||
namespace Falcon.SugarApi.JWT
|
||
{
|
||
/// <summary>
|
||
/// 服务扩展
|
||
/// </summary>
|
||
public static class IServiceCollectionExtend
|
||
{
|
||
/// <summary>
|
||
/// 注册Falcon.SqlSugar.JWT
|
||
/// </summary>
|
||
/// <param name="services">服务集合</param>
|
||
/// <returns>服务集合</returns>
|
||
public static IServiceCollection AddFalconJWT(this IServiceCollection services)
|
||
=> services.AddFalconJWT(null);
|
||
|
||
/// <summary>
|
||
/// 注册Falcon.SqlSugar.JWT.JwtContext，并配置相关参数
|
||
/// <para>消费端通过JwtContext注入</para>
|
||
/// </summary>
|
||
/// <param name="services">服务集合</param>
|
||
/// <param name="OptionBuilder">参数创建器</param>
|
||
/// <returns>服务集合</returns>
|
||
public static IServiceCollection AddFalconJWT(this IServiceCollection services, Action<JwtContext>? OptionBuilder) {
|
||
services.AddSingleton<JwtContext>(sp => {
|
||
var option = new JwtContext();
|
||
var jtb = sp.GetService<IJwtTokenBuilder>();
|
||
if (jtb == null) {
|
||
var aesc = sp.GetService<AESConfig>() ?? new AESConfig();
|
||
jtb = new AESTokenbuilder(aesc, option);
|
||
}
|
||
option.JwtTokenBuilder = jtb;
|
||
|
||
var ulj = sp.GetService<IUserLogin>();
|
||
if (ulj != null) {
|
||
option.UserLogin = ulj;
|
||
}
|
||
|
||
OptionBuilder?.Invoke(option);
|
||
|
||
if (option.UserLogin == null) {
|
||
throw new Exception("必须为JwtOptions提供UserLogin属性！");
|
||
}
|
||
if (option.JwtTokenBuilder == null) {
|
||
throw new Exception("必须为JwtOptions提供JwtTokenBuilder属性！");
|
||
}
|
||
return option;
|
||
});
|
||
return services;
|
||
}
|
||
}
|
||
}
|