Don't refresh all the page

Author: habarthierry-hue

src/IdServer/SimpleIdServer.IdServer.Website/Infrastructures/RealmRouter.cs

@@ -7,49 +7,119 @@
 using Microsoft.Extensions.Options;
 using SimpleIdServer.IdServer.Domains;
 using System.Reflection;
+using System.Runtime.ExceptionServices;
 using System.Text.RegularExpressions;
 
 namespace SimpleIdServer.IdServer.Website.Infrastructures;
 
 public class RealmRouter : IComponent, IHandleAfterRender, IDisposable
 {
+    string _baseUri;
+    string _locationAbsolute;
+    bool _navigationInterceptionEnabled;
     private RenderHandle _renderHandle;
     private Dictionary<Type, Dictionary<string, string>> _routeableComponents;
     internal static IServiceProvider _serviceProvider;
+    private CancellationTokenSource _onNavigateCts;
+    private Task _previousOnNavigateTask = Task.CompletedTask;
+    private IRoutingStateProvider? RoutingStateProvider { get; set; }
+
     [Inject] private NavigationManager NavigationManager { get; set; }
+    [Inject] IServiceProvider ServiceProvider { get; set; }
+    [Inject] private IScrollToLocationHash ScrollToLocationHash { get; set; }
+    [Inject] private INavigationInterception NavigationInterception { get; set; }
     [Parameter] public Assembly AppAssembly { get; set; }
     [Parameter] public RenderFragment<RouteData> Found { get; set; }
     [Parameter] public RenderFragment NotFound { get; set; }
+    [Parameter] public RenderFragment? Navigating { get; set; }
+    [Parameter] public EventCallback<CustomNavigationContext> OnNavigateAsync { get; set; }
+    private bool _onNavigateCalled;
 
     public void Attach(RenderHandle renderHandle)
     {
         _renderHandle = renderHandle;
-        NavigationManager.LocationChanged += OnLocationChanged;
         _routeableComponents = GetRouteableComponents();
+        _baseUri = NavigationManager.BaseUri;
+        _locationAbsolute = NavigationManager.Uri;
+        NavigationManager.LocationChanged += OnLocationChanged;
+        RoutingStateProvider = ServiceProvider.GetService<IRoutingStateProvider>();
     }
 
     public async Task SetParametersAsync(ParameterView parameters)
     {
         parameters.SetParameterProperties(this);
         var locationPath = NavigationManager.ToBaseRelativePath(NavigationManager.Uri);
-        await Navigate(locationPath);
+        if (!_onNavigateCalled)
+        {
+            _onNavigateCalled = true;
+            await RunOnNavigateAsync(locationPath, false);
+        }
+        else
+        {
+            await Refresh(locationPath, false);
+        }
     }
 
     public void Dispose()
     {
         NavigationManager.LocationChanged -= OnLocationChanged;
     }
 
-    public Task OnAfterRenderAsync()
+    public async Task OnAfterRenderAsync()
+    {
+        if (!_navigationInterceptionEnabled)
+        {
+            _navigationInterceptionEnabled = true;
+            await NavigationInterception.EnableNavigationInterceptionAsync();
+        }
+    }
+
+    private async Task RunOnNavigateAsync(string locationPath, bool isNavigationIntercepted)
     {
-        return Task.CompletedTask;
+        _onNavigateCts?.Cancel();
+        await _previousOnNavigateTask;
+        var tcs = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
+        _previousOnNavigateTask = tcs.Task;
+        if (!OnNavigateAsync.HasDelegate)
+        {
+            await Refresh(locationPath, isNavigationIntercepted);
+        }
+
+        _onNavigateCts = new CancellationTokenSource();
+        var navigateContext = new CustomNavigationContext(locationPath,  _onNavigateCts.Token);
+        var cancellationTcs = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
+        navigateContext.CancellationToken.Register(state =>
+            ((TaskCompletionSource)state).SetResult(), cancellationTcs);
+
+        try
+        {
+            var task = await Task.WhenAny(OnNavigateAsync.InvokeAsync(navigateContext), cancellationTcs.Task);
+            await task;
+            tcs.SetResult();
+            await Refresh(locationPath, isNavigationIntercepted);
+        }
+        catch (Exception e)
+        {
+            _renderHandle.Render(builder => ExceptionDispatchInfo.Throw(e));
+        }
     }
 
-    private async Task Navigate(string locationPath)
+    private async Task Refresh(string locationPath, bool isNavigationIntercepted)
     {
+        if (_previousOnNavigateTask.Status != TaskStatus.RanToCompletion)
+        {
+            if (Navigating != null)
+            {
+                _renderHandle.Render(Navigating);
+            }
+            return;
+        }
+
+
         var options = _serviceProvider.GetRequiredService<IOptions<IdServerWebsiteOptions>>();
         var routeParameters = new Dictionary<string, object>();
         Type handlerContext = null;
+        var relativePath = NavigationManager.ToBaseRelativePath(_locationAbsolute);
         if (!options.Value.IsReamEnabled)
         {
             if (!locationPath.StartsWith("/"))
@@ -61,7 +131,7 @@ private async Task Navigate(string locationPath)
             }
 
             var routeData = new RouteData(handlerContext, routeParameters);
-            _renderHandle.Render(Found(routeData));
+            _renderHandle.Render(Found(routeData)); 
             return;
         }
         else
@@ -85,7 +155,7 @@ private async void OnLocationChanged(object sender, LocationChangedEventArgs arg
         if(_renderHandle.IsInitialized && _routeableComponents != null)
         {
             var locationPath = NavigationManager.ToBaseRelativePath(args.Location);
-            _ = Navigate(locationPath);
+            _ = RunOnNavigateAsync((locationPath), args.IsNavigationIntercepted);
         }
     }
 
@@ -144,3 +214,23 @@ private Dictionary<Type, Dictionary<string, string>> GetRouteableComponents()
         return dic;
     }
 }
+
+public class CustomNavigationContext
+{
+    public CustomNavigationContext(string path, CancellationToken cancellationToken)
+    {
+        Path = path;
+        CancellationToken = cancellationToken;
+    }
+
+    /// <summary>
+    /// The target path for the navigation.
+    /// </summary>
+    public string Path { get; }
+
+    /// <summary>
+    /// The <see cref="CancellationToken"/> to use to cancel navigation.
+    /// </summary>
+    public CancellationToken CancellationToken { get; }
+
+}

src/IdServer/SimpleIdServer.IdServer.Website/Pages/Acrs.razor

@@ -127,7 +127,7 @@
                 StateHasChanged();
             });
             if (!acrsState.Value.IsLoading && acrsGrid != null)
-                acrsGrid.Reload();
+                acrsGrid?.Reload();
         }
     }
 

src/IdServer/SimpleIdServer.IdServer.Website/Pages/Auditing.razor

@@ -65,7 +65,7 @@
         if(firstRender)
         {
             if (!searchAuditingRecordsState.Value.IsLoading)
-                auditEvtsGrid.Reload();
+                auditEvtsGrid?.Reload();
         }
     }
 
@@ -92,6 +92,6 @@
     void DisplayErrors(bool a)
     {
         displayErrors = a;
-        auditEvtsGrid.Reload();
+        auditEvtsGrid?.Reload();
     }
 }

src/IdServer/SimpleIdServer.IdServer.Website/Pages/CertificateAuthorities.razor

@@ -108,7 +108,7 @@
                 await JS.InvokeVoidAsync("downloadFileFromStream", fileName, streamRef);
             });
             if (!searchCertificateAuthoritiesState.Value.IsLoading)
-                caGrid.Reload();
+                caGrid?.Reload();
         }
     }
 

src/IdServer/SimpleIdServer.IdServer.Website/Pages/Clients.razor

@@ -98,7 +98,7 @@
                 StateHasChanged();
             });
             if (!clientState.Value.IsLoading)
-                clientsGrid.Reload();
+                clientsGrid?.Reload();
         }
     }
 

src/IdServer/SimpleIdServer.IdServer.Website/Pages/Groups.razor

@@ -98,7 +98,7 @@
                 dialogService.Close();
             });
             if (!searchGroupsState.Value.IsLoading)
-                groupsGrid.Reload();
+                groupsGrid?.Reload();
         }
     }
 

src/IdServer/SimpleIdServer.IdServer.Website/Pages/IdProviders.razor

@@ -76,7 +76,7 @@
                 StateHasChanged();
             });
             if (!searchIdProvidersState.Value.IsLoading)
-                idProvidersGrid.Reload();
+                idProvidersGrid?.Reload();
         }
     }
 

src/IdServer/SimpleIdServer.IdServer.Website/Pages/IdentityProvisioningLst.razor

@@ -67,7 +67,7 @@
         if(firstRender)
         {
             if (!searchIdentityProvisioningState.Value.IsLoading)
-                identityProvisioningGrid.Reload();
+                identityProvisioningGrid?.Reload();
         }
     }
 

src/IdServer/SimpleIdServer.IdServer.Website/Pages/RealmRoles.razor

@@ -52,7 +52,7 @@
         if(firstRender)
         {
             if (realmScopesState.Value.IsLoading && scopesGrid != null)
-                scopesGrid.Reload();
+                scopesGrid?.Reload();
         }
     }
 

src/IdServer/SimpleIdServer.IdServer.Website/Pages/RegistrationWorkflows.razor

@@ -114,7 +114,7 @@
                 StateHasChanged();
             });
             if (!registrationWorkflowsState.Value.IsLoading && registrationWorkflowsGrid != null)
-                registrationWorkflowsGrid.Reload();
+                registrationWorkflowsGrid?.Reload();
         }
     }