| | | 1 | | using System.Reflection; |
| | | 2 | | using Elsa.Features.Services; |
| | | 3 | | using Elsa.Permissions; |
| | | 4 | | using FastEndpoints; |
| | | 5 | | using Microsoft.Extensions.DependencyInjection; |
| | | 6 | | |
| | | 7 | | // ReSharper disable once CheckNamespace |
| | | 8 | | namespace Elsa.Extensions; |
| | | 9 | | |
| | | 10 | | /// <summary> |
| | | 11 | | /// Provides extensions to <see cref="IModule"/>. |
| | | 12 | | /// </summary> |
| | | 13 | | public static class ModuleExtensions |
| | | 14 | | { |
| | 1 | 15 | | private static readonly object FastEndpointsAssembliesKey = new(); |
| | | 16 | | |
| | | 17 | | /// <summary> |
| | | 18 | | /// Registers the specified assembly for FastEndpoint assembly discovery. |
| | | 19 | | /// </summary> |
| | | 20 | | public static IModule AddFastEndpointsAssembly(this IModule module, Assembly assembly) |
| | | 21 | | { |
| | 21 | 22 | | var assemblies = module.Properties.GetOrAdd(FastEndpointsAssembliesKey, () => new HashSet<Assembly>()); |
| | 18 | 23 | | assemblies.Add(assembly); |
| | | 24 | | |
| | | 25 | | // Registering an endpoint assembly is what guarantees its permissions work, so the evaluator and |
| | | 26 | | // that assembly's descriptors are registered here rather than in AddFastEndpointsFromModule. A host |
| | | 27 | | // that wires FastEndpoints itself never calls that helper, and without the authorization handler |
| | | 28 | | // every RequirePermission endpoint would answer 403 -- fail-closed, but broken. |
| | 18 | 29 | | module.Services.AddElsaAuthorization(); |
| | 18 | 30 | | module.Services.AddPermissionDescriptorsFromAssembly(assembly); |
| | | 31 | | |
| | 18 | 32 | | return module; |
| | | 33 | | } |
| | | 34 | | |
| | | 35 | | /// <summary> |
| | | 36 | | /// Registers the assembly for FastEndpoint assembly discovery using the specified marker type. |
| | | 37 | | /// </summary> |
| | 12 | 38 | | public static IModule AddFastEndpointsAssembly<T>(this IModule module) => module.AddFastEndpointsAssembly(typeof(T)) |
| | | 39 | | |
| | | 40 | | /// <summary> |
| | | 41 | | /// Registers the assembly for FastEndpoint assembly discovery using the specified marker type. |
| | | 42 | | /// </summary> |
| | 18 | 43 | | public static IModule AddFastEndpointsAssembly(this IModule module, Type markerType) => module.AddFastEndpointsAssem |
| | | 44 | | |
| | | 45 | | /// <summary> |
| | | 46 | | /// Returns all collected assemblies for discovery of endpoints. |
| | | 47 | | /// </summary> |
| | 6 | 48 | | public static IEnumerable<Assembly> GetFastEndpointsAssembliesFromModule(this IModule module) => module.Properties.G |
| | | 49 | | |
| | | 50 | | /// <summary> |
| | | 51 | | /// Adds FastEndpoints to the service container and registers all collected assemblies for discovery of endpoints. |
| | | 52 | | /// </summary> |
| | | 53 | | public static IModule AddFastEndpointsFromModule(this IModule module) |
| | | 54 | | { |
| | 6 | 55 | | var assemblies = module.GetFastEndpointsAssembliesFromModule().ToList(); |
| | | 56 | | |
| | 6 | 57 | | module.Services.AddFastEndpoints(options => |
| | 6 | 58 | | { |
| | 6 | 59 | | options.DisableAutoDiscovery = true; |
| | 6 | 60 | | options.Assemblies = assemblies; |
| | 12 | 61 | | }); |
| | | 62 | | |
| | 6 | 63 | | return module; |
| | | 64 | | } |
| | | 65 | | } |