In Auto setup, when IsUninitialized, response http status code 503

src/OrchardCore.Modules/OrchardCore.AutoSetup/AutoSetupMiddleware.cs

@@ -1,14 +1,11 @@
-using System.Text;
 using Microsoft.AspNetCore.Http;
 using Microsoft.Extensions.DependencyInjection;
-using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
-using OrchardCore.Abstractions.Setup;
 using OrchardCore.AutoSetup.Extensions;
 using OrchardCore.AutoSetup.Options;
+using OrchardCore.AutoSetup.Services;
 using OrchardCore.Environment.Shell;
 using OrchardCore.Locking.Distributed;
-using OrchardCore.Setup.Services;
 
 namespace OrchardCore.AutoSetup;
 
@@ -47,11 +44,6 @@ public class AutoSetupMiddleware
     /// </summary>
     private readonly AutoSetupOptions _options;
 
-    /// <summary>
-    /// The logger.
-    /// </summary>
-    private readonly ILogger _logger;
-
     /// <summary>
     /// The auto setup lock options.
     /// </summary>
@@ -70,25 +62,21 @@ public class AutoSetupMiddleware
     /// <param name="shellSettings">The shell settings.</param>
     /// <param name="shellSettingsManager">The shell settings manager.</param>
     /// <param name="distributedLock">The distributed lock.</param>
-    /// <param name="options">The auto setup options.</param>
-    /// <param name="logger">The logger.</param>
+    /// <param name="options">The auto setup options.</param>    
     public AutoSetupMiddleware(
         RequestDelegate next,
         IShellHost shellHost,
         ShellSettings shellSettings,
         IShellSettingsManager shellSettingsManager,
         IDistributedLock distributedLock,
-        IOptions<AutoSetupOptions> options,
-        ILogger<AutoSetupMiddleware> logger)
+        IOptions<AutoSetupOptions> options)
     {
         _next = next;
         _shellHost = shellHost;
         _shellSettings = shellSettings;
         _shellSettingsManager = shellSettingsManager;
         _distributedLock = distributedLock;
         _options = options.Value;
-        _logger = logger;
-
         _lockOptions = _options.LockOptions;
         _setupOptions = _options.Tenants.FirstOrDefault(options => _shellSettings.Name == options.ShellName);
     }
@@ -124,20 +112,23 @@ public async Task InvokeAsync(HttpContext httpContext)
                 }
 
                 // Check if the tenant was installed by another instance.
-                using var settings = (await _shellSettingsManager
-                    .LoadSettingsAsync(_shellSettings.Name))
-                    .AsDisposable();
+                using var settings = await _shellSettingsManager.LoadSettingsAsync(_shellSettings.Name);
 
-                if (!settings.IsUninitialized())
+                if (settings != null)
                 {
-                    await _shellHost.ReloadShellContextAsync(_shellSettings, eventSource: false);
-                    httpContext.Response.Redirect(pathBase);
-
-                    return;
+                    settings.AsDisposable();
+                    if (!settings.IsUninitialized())
+                    {
+                        await _shellHost.ReloadShellContextAsync(_shellSettings, eventSource: false);
+                        httpContext.Response.StatusCode = StatusCodes.Status503ServiceUnavailable;
+                        await httpContext.Response.WriteAsync("The requested tenant is not initialized.");
+                        return;
+                    }
                 }
 
-                var setupService = httpContext.RequestServices.GetRequiredService<ISetupService>();
-                if (await SetupTenantAsync(setupService, _setupOptions, _shellSettings))
+                var autoSetupService = httpContext.RequestServices.GetRequiredService<IAutoSetupService>();
+                (var setupContext, var isSuccess) = await autoSetupService.SetupTenantAsync(_setupOptions, _shellSettings);
+                if (isSuccess)
                 {
                     if (_setupOptions.IsDefault)
                     {
@@ -146,123 +137,22 @@ public async Task InvokeAsync(HttpContext httpContext)
                         {
                             if (_setupOptions != setupOptions)
                             {
-                                await CreateTenantSettingsAsync(setupOptions);
+                                await autoSetupService.CreateTenantSettingsAsync(setupOptions);
                             }
                         }
                     }
 
                     httpContext.Response.Redirect(pathBase);
-
+                }
+                else
+                {
+                    httpContext.Response.StatusCode = StatusCodes.Status503ServiceUnavailable;
+                    await httpContext.Response.WriteAsync($"The AutoSetup failed installing the site.");
                     return;
                 }
             }
         }
 
         await _next.Invoke(httpContext);
     }
-
-    /// <summary>
-    /// Sets up a tenant.
-    /// </summary>
-    /// <param name="setupService">The setup service.</param>
-    /// <param name="setupOptions">The tenant setup options.</param>
-    /// <param name="shellSettings">The tenant shell settings.</param>
-    /// <returns>
-    /// Returns <c>true</c> if successfully setup.
-    /// </returns>
-    public async Task<bool> SetupTenantAsync(ISetupService setupService, TenantSetupOptions setupOptions, ShellSettings shellSettings)
-    {
-        var setupContext = await GetSetupContextAsync(setupOptions, setupService, shellSettings);
-
-        _logger.LogInformation("AutoSetup is initializing the site");
-
-        await setupService.SetupAsync(setupContext);
-
-        if (setupContext.Errors.Count == 0)
-        {
-            _logger.LogInformation("AutoSetup successfully provisioned the site '{SiteName}'.", setupOptions.SiteName);
-
-            return true;
-        }
-
-        var stringBuilder = new StringBuilder();
-        foreach (var error in setupContext.Errors)
-        {
-            stringBuilder.AppendLine($"{error.Key} : '{error.Value}'");
-        }
-
-        _logger.LogError("AutoSetup failed installing the site '{SiteName}' with errors: {Errors}", setupOptions.SiteName, stringBuilder);
-
-        return false;
-    }
-
-    /// <summary>
-    /// Creates a tenant shell settings.
-    /// </summary>
-    /// <param name="setupOptions">The setup options.</param>
-    /// <returns>The <see cref="ShellSettings"/>.</returns>
-    public async Task<ShellSettings> CreateTenantSettingsAsync(TenantSetupOptions setupOptions)
-    {
-        using var shellSettings = _shellSettingsManager
-            .CreateDefaultSettings()
-            .AsUninitialized()
-            .AsDisposable();
-
-        shellSettings.Name = setupOptions.ShellName;
-        shellSettings.RequestUrlHost = setupOptions.RequestUrlHost;
-        shellSettings.RequestUrlPrefix = setupOptions.RequestUrlPrefix;
-
-        shellSettings["ConnectionString"] = setupOptions.DatabaseConnectionString;
-        shellSettings["TablePrefix"] = setupOptions.DatabaseTablePrefix;
-        shellSettings["Schema"] = setupOptions.DatabaseSchema;
-        shellSettings["DatabaseProvider"] = setupOptions.DatabaseProvider;
-        shellSettings["Secret"] = Guid.NewGuid().ToString();
-        shellSettings["RecipeName"] = setupOptions.RecipeName;
-        shellSettings["FeatureProfile"] = setupOptions.FeatureProfile;
-
-        await _shellHost.UpdateShellSettingsAsync(shellSettings);
-
-        return shellSettings;
-    }
-
-    /// <summary>
-    /// Gets a setup context from the configuration.
-    /// </summary>
-    /// <param name="options">The tenant setup options.</param>
-    /// <param name="setupService">The setup service.</param>
-    /// <param name="shellSettings">The tenant shell settings.</param>
-    /// <returns> The <see cref="SetupContext"/> used to setup the site.</returns>
-    private static async Task<SetupContext> GetSetupContextAsync(TenantSetupOptions options, ISetupService setupService, ShellSettings shellSettings)
-    {
-        var recipes = await setupService.GetSetupRecipesAsync();
-
-        var recipe = recipes.SingleOrDefault(r => r.Name == options.RecipeName);
-
-        var setupContext = new SetupContext
-        {
-            Recipe = recipe,
-            ShellSettings = shellSettings,
-            Errors = new Dictionary<string, string>()
-        };
-
-        if (shellSettings.IsDefaultShell())
-        {
-            // The 'Default' shell is first created by the infrastructure,
-            // so the following 'Autosetup' options need to be passed.
-            shellSettings.RequestUrlHost = options.RequestUrlHost;
-            shellSettings.RequestUrlPrefix = options.RequestUrlPrefix;
-        }
-
-        setupContext.Properties[SetupConstants.AdminEmail] = options.AdminEmail;
-        setupContext.Properties[SetupConstants.AdminPassword] = options.AdminPassword;
-        setupContext.Properties[SetupConstants.AdminUsername] = options.AdminUsername;
-        setupContext.Properties[SetupConstants.DatabaseConnectionString] = options.DatabaseConnectionString;
-        setupContext.Properties[SetupConstants.DatabaseProvider] = options.DatabaseProvider;
-        setupContext.Properties[SetupConstants.DatabaseTablePrefix] = options.DatabaseTablePrefix;
-        setupContext.Properties[SetupConstants.DatabaseSchema] = options.DatabaseSchema;
-        setupContext.Properties[SetupConstants.SiteName] = options.SiteName;
-        setupContext.Properties[SetupConstants.SiteTimeZone] = options.SiteTimeZone;
-
-        return setupContext;
-    }
 }

src/OrchardCore.Modules/OrchardCore.AutoSetup/Services/AutoSetupService.cs

@@ -0,0 +1,112 @@
+using System.Text;
+using Microsoft.Extensions.Logging;
+using OrchardCore.Abstractions.Setup;
+using OrchardCore.AutoSetup.Options;
+using OrchardCore.Environment.Shell;
+using OrchardCore.Setup.Services;
+
+namespace OrchardCore.AutoSetup.Services;
+
+public class AutoSetupService : IAutoSetupService
+{
+    private readonly IShellHost _shellHost;
+    private readonly IShellSettingsManager _shellSettingsManager;
+    private readonly ISetupService _setupService;
+    private readonly ILogger _logger;
+
+    public AutoSetupService(
+        IShellHost shellHost,
+        IShellSettingsManager shellSettingsManager,
+        ISetupService setupService,
+        ILogger<AutoSetupService> logger
+        )
+    {
+        _shellHost = shellHost;
+        _shellSettingsManager = shellSettingsManager;
+        _setupService = setupService;
+        _logger = logger;
+    }
+
+    public async Task<(SetupContext, bool)> SetupTenantAsync(TenantSetupOptions setupOptions, ShellSettings shellSettings)
+    {
+        var setupContext = await GetSetupContextAsync(setupOptions, shellSettings);
+
+        _logger.LogInformation("The AutoSetup is initializing the site.");
+
+        await _setupService.SetupAsync(setupContext);
+
+        if (setupContext.Errors.Count == 0)
+        {
+            _logger.LogInformation("The AutoSetup successfully provisioned the site '{SiteName}'.", setupOptions.SiteName);
+
+            return (setupContext, true);
+        }
+
+        var stringBuilder = new StringBuilder();
+        foreach (var error in setupContext.Errors)
+        {
+            stringBuilder.AppendLine($"{error.Key} : '{error.Value}'");
+        }
+
+        _logger.LogError("The AutoSetup failed installing the site '{SiteName}' with errors: {Errors}.", setupOptions.SiteName, stringBuilder);
+
+        return (setupContext, false);
+    }
+
+    public async Task<ShellSettings> CreateTenantSettingsAsync(TenantSetupOptions setupOptions)
+    {
+        using var shellSettings = _shellSettingsManager
+            .CreateDefaultSettings()
+            .AsUninitialized()
+            .AsDisposable();
+
+        shellSettings.Name = setupOptions.ShellName;
+        shellSettings.RequestUrlHost = setupOptions.RequestUrlHost;
+        shellSettings.RequestUrlPrefix = setupOptions.RequestUrlPrefix;
+        shellSettings["ConnectionString"] = setupOptions.DatabaseConnectionString;
+        shellSettings["TablePrefix"] = setupOptions.DatabaseTablePrefix;
+        shellSettings["Schema"] = setupOptions.DatabaseSchema;
+        shellSettings["DatabaseProvider"] = setupOptions.DatabaseProvider;
+        shellSettings["Secret"] = Guid.NewGuid().ToString();
+        shellSettings["RecipeName"] = setupOptions.RecipeName;
+        shellSettings["FeatureProfile"] = setupOptions.FeatureProfile;
+
+        await _shellHost.UpdateShellSettingsAsync(shellSettings);
+
+        return shellSettings;
+    }
+
+    public async Task<SetupContext> GetSetupContextAsync(TenantSetupOptions options, ShellSettings shellSettings)
+    {
+        var recipes = await _setupService.GetSetupRecipesAsync();
+
+        var recipe = recipes.SingleOrDefault(r => r.Name == options.RecipeName);
+
+        var setupContext = new SetupContext
+        {
+            Recipe = recipe,
+            ShellSettings = shellSettings,
+            Errors = new Dictionary<string, string>()
+        };
+
+        if (shellSettings.IsDefaultShell())
+        {
+            // The 'Default' shell is first created by the infrastructure,
+            // so the following 'Autosetup' options need to be passed.
+            shellSettings.RequestUrlHost = options.RequestUrlHost;
+            shellSettings.RequestUrlPrefix = options.RequestUrlPrefix;
+        }
+
+        setupContext.Properties[SetupConstants.AdminEmail] = options.AdminEmail;
+        setupContext.Properties[SetupConstants.AdminPassword] = options.AdminPassword;
+        setupContext.Properties[SetupConstants.AdminUsername] = options.AdminUsername;
+        setupContext.Properties[SetupConstants.DatabaseConnectionString] = options.DatabaseConnectionString;
+        setupContext.Properties[SetupConstants.DatabaseProvider] = options.DatabaseProvider;
+        setupContext.Properties[SetupConstants.DatabaseTablePrefix] = options.DatabaseTablePrefix;
+        setupContext.Properties[SetupConstants.DatabaseSchema] = options.DatabaseSchema;
+        setupContext.Properties[SetupConstants.SiteName] = options.SiteName;
+        setupContext.Properties[SetupConstants.SiteTimeZone] = options.SiteTimeZone;
+
+        return setupContext;
+    }
+}

src/OrchardCore.Modules/OrchardCore.AutoSetup/Services/IAutoSetupService.cs

@@ -0,0 +1,33 @@
+using OrchardCore.AutoSetup.Options;
+using OrchardCore.Environment.Shell;
+using OrchardCore.Setup.Services;
+
+namespace OrchardCore.AutoSetup.Services;
+
+public interface IAutoSetupService
+{
+    /// <summary>
+    /// Creates a tenant shell settings.
+    /// </summary>
+    /// <param name="setupOptions">The setup options.</param>
+    /// <returns>The <see cref="ShellSettings"/>.</returns>
+    Task<ShellSettings> CreateTenantSettingsAsync(TenantSetupOptions setupOptions);
+
+    /// <summary>
+    /// Gets a setup context from the configuration.
+    /// </summary>
+    /// <param name="options">The tenant setup options.</param>    
+    /// <param name="shellSettings">The tenant shell settings.</param>
+    /// <returns> The <see cref="SetupContext"/> used to setup the site.</returns>
+    Task<SetupContext> GetSetupContextAsync(TenantSetupOptions options, ShellSettings shellSettings);
+
+    /// <summary>
+    /// Sets up a tenant.
+    /// </summary>    
+    /// <param name="setupOptions">The tenant setup options.</param>
+    /// <param name="shellSettings">The tenant shell settings.</param>
+    /// <returns>
+    /// Returns <see langword="true" /> if successfully setup.
+    /// </returns>
+    Task<(SetupContext, bool)> SetupTenantAsync(TenantSetupOptions setupOptions, ShellSettings shellSettings);
+}

src/OrchardCore.Modules/OrchardCore.AutoSetup/Startup.cs

@@ -7,6 +7,7 @@
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
 using OrchardCore.AutoSetup.Options;
+using OrchardCore.AutoSetup.Services;
 using OrchardCore.Environment.Shell;
 using OrchardCore.Environment.Shell.Configuration;
 using OrchardCore.Modules;
@@ -67,6 +68,8 @@ public override void ConfigureServices(IServiceCollection services)
         {
             services.Configure<AutoSetupOptions>(o => o.ConfigurationExists = true);
         }
+
+        services.AddScoped<IAutoSetupService, AutoSetupService>();
     }
 
     /// <summary>