registration fixes.

ffMathy · Mar 7, 2020 · a4adfed · a4adfed
1 parent ce54ab1
commit a4adfed
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/src/FluffySpoon.AspNet.NGrok/NGrokHostedService.cs b/src/FluffySpoon.AspNet.NGrok/NGrokHostedService.cs
@@ -54,6 +54,9 @@ internal void InjectServerAddressesFeature(IServerAddressesFeature? feature)
 
         private async void RunAsync()
         {
+            if (_options.Disable)
+                return;
+
             await _nGrokDownloader.DownloadExecutableAsync();
 
             var url = await AdjustApplicationHttpUrlIfNeededAsync();

diff --git a/src/FluffySpoon.AspNet.NGrok/NgrokAspNetCoreExtensions.cs b/src/FluffySpoon.AspNet.NGrok/NgrokAspNetCoreExtensions.cs
@@ -48,9 +48,6 @@ public static IWebHostBuilder UseNGrok(this IWebHostBuilder builder)
             return builder
                 .ConfigureServices((context, services) =>
                 {
-                    if (services.All(x => x.ImplementationType != typeof(NGrokHostedService)))
-                        return;
-
                     services.AddSingleton<IHostedService>(p => p.GetRequiredService<NGrokHostedService>());
                 });
         }

diff --git a/src/FluffySpoon.AspNet.NGrok/NgrokOptions.cs b/src/FluffySpoon.AspNet.NGrok/NgrokOptions.cs
@@ -6,6 +6,7 @@ namespace FluffySpoon.AspNet.NGrok
 {
 	public class NGrokOptions
 	{
+		public bool Disable { get; set; }
 
 		public bool ShowNGrokWindow { get; set; }