Merge pull request #20293 from abpframework/WebAssemblyMultiTenantUrl…

…Provider Add `WebAssemblyMultiTenantUrlProvider`.
abpframework · Jul 30, 2024 · fa8973b · fa8973b
2 parents eed75c0 + 91c6ee0
commit fa8973b
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 0 deletions.
diff --git a/...olo/Abp/AspNetCore/Components/WebAssembly/MultiTenant/WebAssemblyMultiTenantUrlOptions.cs b/...olo/Abp/AspNetCore/Components/WebAssembly/MultiTenant/WebAssemblyMultiTenantUrlOptions.cs
@@ -0,0 +1,6 @@
+namespace Volo.Abp.AspNetCore.Components.WebAssembly.MultiTenant;
+
+public class WebAssemblyMultiTenantUrlOptions
+{
+    public string DomainFormat { get; set; } = default!;
+}
diff --git a/...lo/Abp/AspNetCore/Components/WebAssembly/MultiTenant/WebAssemblyMultiTenantUrlProvider.cs b/...lo/Abp/AspNetCore/Components/WebAssembly/MultiTenant/WebAssemblyMultiTenantUrlProvider.cs
@@ -0,0 +1,49 @@
+using System;
+using System.Threading.Tasks;
+using Microsoft.AspNetCore.Components;
+using Microsoft.Extensions.Options;
+using Volo.Abp.DependencyInjection;
+using Volo.Abp.MultiTenancy;
+using Volo.Abp.Text.Formatting;
+
+namespace Volo.Abp.AspNetCore.Components.WebAssembly.MultiTenant;
+
+[Dependency(ReplaceServices = true)]
+[ExposeServices(typeof(IMultiTenantUrlProvider), typeof(MultiTenantUrlProvider))]
+public class WebAssemblyMultiTenantUrlProvider : MultiTenantUrlProvider
+{
+    private readonly static string[] ProtocolPrefixes = ["http://", "https://"];
+
+    protected NavigationManager NavigationManager { get; }
+    protected IOptions<WebAssemblyMultiTenantUrlOptions> Options { get; }
+
+    public WebAssemblyMultiTenantUrlProvider(
+        ICurrentTenant currentTenant,
+        ITenantStore tenantStore,
+        NavigationManager navigationManager,
+        IOptions<WebAssemblyMultiTenantUrlOptions> options)
+        : base(currentTenant, tenantStore)
+    {
+        NavigationManager = navigationManager;
+        Options = options;
+    }
+
+    public async override Task<string> GetUrlAsync(string templateUrl)
+    {
+        if (!Options.Value.DomainFormat.IsNullOrEmpty() && !CurrentTenant.IsAvailable)
+        {
+            // If the domain format is configured and the tenant is not available
+            // try to extract the tenant name from the current blazor URL.
+            var url = NavigationManager.ToAbsoluteUri(NavigationManager.Uri).Authority;
+            var domainFormat = Options.Value.DomainFormat.RemovePreFix(ProtocolPrefixes).RemovePostFix("/");
+            var extractResult = FormattedStringValueExtracter.Extract(url, domainFormat, ignoreCase: true);
+            if (extractResult.IsMatch)
+            {
+                var tenant = extractResult.Matches[0].Value;
+                return templateUrl.Replace(TenantPlaceHolder, tenant).Replace(TenantIdPlaceHolder, tenant).Replace(TenantNamePlaceHolder, tenant);
+            }
+        }
+
+        return await base.GetUrlAsync(templateUrl);
+    }
+}