port recovery flow, create logout view

josxha · Oct 22, 2023 · 169d78e · 169d78e
1 parent d64c3e2
commit 169d78e
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 19 deletions.
diff --git a/KratosSelfService/Controllers/RecoveryController.cs b/KratosSelfService/Controllers/RecoveryController.cs
@@ -1,3 +1,4 @@
+using KratosSelfService.Models;
 using KratosSelfService.Services;
 using Microsoft.AspNetCore.Mvc;
 using Ory.Kratos.Client.Client;
@@ -8,9 +9,18 @@ namespace KratosSelfService.Controllers;
 public class RecoveryController(ILogger<RecoveryController> logger, ApiService api) : Controller
 {
     [HttpGet("recovery")]
-    public async Task<IActionResult> Recovery([FromQuery(Name = "flow")] string? flowId)
+    public async Task<IActionResult> Recovery(
+        [FromQuery(Name = "flow")] string? flowId,
+        [FromQuery(Name = "return_to")] string? returnTo)
     {
-        if (flowId == null) return Redirect(api.GetUrlForBrowserFlow("recovery"));
+        if (flowId == null)
+        {
+            logger.LogDebug("No flow ID found in URL query initializing login flow");
+            return Redirect(api.GetUrlForBrowserFlow("recovery", new Dictionary<string, string?>()
+            {
+                ["return_to"] = returnTo
+            }));
+        }
 
         KratosRecoveryFlow flow;
         try
@@ -21,9 +31,17 @@ public async Task<IActionResult> Recovery([FromQuery(Name = "flow")] string? flo
         {
             logger.LogError(exception.Message);
             // restart flow
-            return Redirect("recovery");
+            return Redirect(api.GetUrlForBrowserFlow("recovery", new Dictionary<string, string?>()
+            {
+                ["return_to"] = returnTo
+            }));
         }
 
-        return View("Recovery", flow);
+        var loginUrl = api.GetUrlForBrowserFlow("login", new Dictionary<string, string?>()
+        {
+            ["return_to"] = returnTo
+        });
+        var model = new RecoveryModel(flow, loginUrl);
+        return View("Recovery", model);
     }
 }
diff --git a/KratosSelfService/KratosSelfService.csproj b/KratosSelfService/KratosSelfService.csproj
@@ -72,13 +72,8 @@
 
     <ItemGroup>
         <AdditionalFiles Include="Views\Login\Login.cshtml" />
-        <AdditionalFiles Include="Views\Login\Recovery.cshtml" />
-        <AdditionalFiles Include="Views\Login\Registration.cshtml" />
-        <AdditionalFiles Include="Views\Recovery\Login.cshtml" />
+        <AdditionalFiles Include="Views\Logout\Logout.cshtml" />
         <AdditionalFiles Include="Views\Recovery\Recovery.cshtml" />
-        <AdditionalFiles Include="Views\Recovery\Registration.cshtml" />
-        <AdditionalFiles Include="Views\Registration\Login.cshtml" />
-        <AdditionalFiles Include="Views\Registration\Recovery.cshtml" />
         <AdditionalFiles Include="Views\Registration\Registration.cshtml" />
         <AdditionalFiles Include="Views\Verification\Verification.cshtml" />
         <AdditionalFiles Include="Views\Settings\Settings.cshtml"/>

diff --git a/KratosSelfService/Models/models.cs b/KratosSelfService/Models/models.cs
@@ -17,9 +17,7 @@ public record LogoutModel(
     KratosLogoutFlow flow
 );
 
-public record RecoveryModel(
-    KratosRecoveryFlow flow
-);
+public record RecoveryModel(KratosRecoveryFlow flow, string loginUrl);
 
 public record RegistrationModel(
     KratosRegistrationFlow flow

diff --git a/KratosSelfService/Views/Logout/Logout.cshtml b/KratosSelfService/Views/Logout/Logout.cshtml
@@ -0,0 +1,15 @@
+@model LogoutModel
+@{
+    ViewData["Title"] = OryTranslator.Get("logout.title");
+    Layout = "_CardLayout";
+}
+
+<h1 class="title">@OryTranslator.Get("logout.title")</h1>
+<div class="buttons mt-5 is-fullwidth">
+    <a class="button is-dark" href="/">
+        @CustomTranslator.Get("logout.reject-button")
+    </a>
+    <a class="button is-warning" href="@Model.flow.LogoutUrl">
+        @CustomTranslator.Get("logout.accept-button")
+    </a>
+</div>
diff --git a/KratosSelfService/Views/Recovery/Recovery.cshtml b/KratosSelfService/Views/Recovery/Recovery.cshtml
@@ -1,4 +1,4 @@
-@model KratosRecoveryFlow
+@model RecoveryModel
 @{
     ViewData["Title"] = OryTranslator.Get("recovery.title");
     Layout = "_CardLayout";
@@ -7,20 +7,20 @@
 <h1 class="title has-text-centered">@OryTranslator.Get("recovery.title")</h1>
 <p class="has-text-centered">
     @OryTranslator.Get("recovery.login-label")
-    <a href="login">
+    <a href="@Model.loginUrl">
         @OryTranslator.Get("recovery.login-button")
     </a>
 </p>
 <hr/>
-<form method="@Model.Ui.Method.ToLower()" action="@Model.Ui.Action">
-    @if (Model.Ui.Messages != null)
+<form method="@Model.flow.Ui.Method.ToLower()" action="@Model.flow.Ui.Action">
+    @if (Model.flow.Ui.Messages != null)
     {
-        foreach (var message in Model.Ui.Messages)
+        foreach (var message in Model.flow.Ui.Messages)
         {
             @await Component.InvokeAsync("KratosUiTextMessage", message)
         }
     }
-    @foreach (var node in Model.Ui.Nodes)
+    @foreach (var node in Model.flow.Ui.Nodes)
     {
         var model = new KratosUiNodeModel(node);
         @await Component.InvokeAsync("KratosUiNodeInput", model)