diff --git a/src/activities/Elsa.Activities.File/Extensions/ServiceCollectionExtensions.cs b/src/activities/Elsa.Activities.File/Extensions/ServiceCollectionExtensions.cs index d3021846d..b2631b2a8 100644 --- a/src/activities/Elsa.Activities.File/Extensions/ServiceCollectionExtensions.cs +++ b/src/activities/Elsa.Activities.File/Extensions/ServiceCollectionExtensions.cs @@ -1,5 +1,6 @@ using Elsa; using Elsa.Activities.File; +using Elsa.Activities.File.Bookmarks; using Elsa.Activities.File.Options; using Elsa.Activities.File.Services; using Microsoft.Extensions.DependencyInjection; @@ -16,20 +17,30 @@ namespace Microsoft.Extensions.DependencyInjection return services; } - public static ElsaOptionsBuilder AddFileActivities(this ElsaOptionsBuilder builder, Action configureOptions = null) => builder - .AddActivity() - .AddActivity() - .AddActivity() - .AddActivity() - .AddActivity(); + public static ElsaOptionsBuilder AddFileActivities(this ElsaOptionsBuilder builder, Action configureOptions = null) + { + if (builder == null) + throw new ArgumentNullException(nameof(builder)); + + builder.Services.AddFileServices(configureOptions); + + builder.AddActivity() + .AddActivity() + .AddActivity() + .AddActivity() + .AddActivity() + .AddActivity(); + + return builder; + } public static IServiceCollection AddFileServices(this IServiceCollection services, Action configureOptions = null) { if (configureOptions != null) - { - services.Configure(configureOptions); - } + services.Configure(configureOptions); + services.AddHostedService(); + services.AddBookmarkProvider(); return services; }