From 4443ec9c845bd3e4bfcefb14046fc8c4331e9a0c Mon Sep 17 00:00:00 2001
From: Gisle Aune <dev@gisle.me>
Date: Sun, 24 Jan 2021 18:56:10 +0100
Subject: [PATCH] add password reset.

---
 svelte-ui/src/clients/amplify.ts     |  8 ++++
 svelte-ui/src/forms/LoginForm.svelte | 55 +++++++++++++++++++++-------
 2 files changed, 50 insertions(+), 13 deletions(-)

diff --git a/svelte-ui/src/clients/amplify.ts b/svelte-ui/src/clients/amplify.ts
index 94774b4..cc60726 100644
--- a/svelte-ui/src/clients/amplify.ts
+++ b/svelte-ui/src/clients/amplify.ts
@@ -19,6 +19,14 @@ export async function signOut(): Promise<void> {
   await Auth.signOut();
 }
 
+export async function forgotPassword(username: string): Promise<void> {
+  const res = await Auth.forgotPassword(username);
+}
+
+export async function forgotPasswordSubmit(username: string, code: string, password: string): Promise<void> {
+  await Auth.forgotPasswordSubmit(username, code, password);
+}
+
 async function getAccessToken(): Promise<CognitoAccessToken | null> {
   try {
     const u = await Auth.currentSession();
diff --git a/svelte-ui/src/forms/LoginForm.svelte b/svelte-ui/src/forms/LoginForm.svelte
index 865c763..468386d 100644
--- a/svelte-ui/src/forms/LoginForm.svelte
+++ b/svelte-ui/src/forms/LoginForm.svelte
@@ -1,7 +1,7 @@
 <script lang="ts">
   import type {CognitoUser} from "amazon-cognito-identity-js";
 
-  import { signIn } from "../clients/amplify";  
+  import { forgotPassword, forgotPasswordSubmit, signIn } from "../clients/amplify";  
   import authStore from "../stores/auth";
   import Modal from "../components/Modal.svelte";
 
@@ -10,10 +10,21 @@
   let password = "";
   let newPassword = "";
   let newPasswordRepeat = "";
+  let resetCode = "";
   let settingNewPassword = false;
+  let resettingPassword = false;
   let error = null;
   let done = false;
 
+  function startPasswordReset() {
+    forgotPassword(username).then(() => {
+      settingNewPassword = true;
+      resettingPassword = true;
+    }).catch(err => {
+      error = "Reset request failed: " + err.message;
+    })
+  }
+
   function login() {
     error = null;
 
@@ -23,15 +34,25 @@
         return;
       }
 
-      user.completeNewPasswordChallenge(newPassword, null, {
-        onSuccess: () => {
-          done = true;
+      if (resettingPassword) {
+        forgotPasswordSubmit(username, resetCode, newPassword).then(() => {
           authStore.check();
-        },
-        onFailure: err => {
-          error = err
-        },
-      })
+          settingNewPassword = false;
+          resettingPassword = false;
+        }).catch(err => {
+          error = "Reset submit failed: " + err.message;
+        });
+      } else {  
+        user.completeNewPasswordChallenge(newPassword, null, {
+          onSuccess: () => {
+            done = true;
+            authStore.check();
+          },
+          onFailure: err => {
+            error = err
+          },
+        })
+      }
     } else {
       signIn(username, password).then(newUser => {
         if (!newUser) {
@@ -49,7 +70,7 @@
 
         user = newUser;
       }).catch(err => {
-        error = err
+        error = err.message
       });
     }
   }
@@ -59,16 +80,24 @@
   <form on:submit|preventDefault={login}>
     <label for="username">Username</label>
     <input name="username" type="text" bind:value={username} />
-    <label for="password">Password</label>
-    <input name="password" type="password" bind:value={password} />
+    {#if !resettingPassword}
+      <label for="password">Password</label>
+      <input name="password" type="password" bind:value={password} />
+    {:else}
+      <label for="resetCode">Code From Email</label>
+      <input name="resetCode" type="text" bind:value={resetCode} />
+    {/if}
     {#if settingNewPassword}
       <label for="newPassword">New Password</label>
       <input name="newPassword" type="password" bind:value={newPassword} />
-      <label for="newPasswordRepeat">New Password</label>
+      <label for="newPasswordRepeat">Repeat Password</label>
       <input name="newPasswordRepeat" type="password" bind:value={newPasswordRepeat} />
     {/if}
 
     <hr />
     <button type="submit">Login</button>
+    {#if !settingNewPassword}
+      <button disabled={username === ""} on:click|preventDefault={startPasswordReset}>Forgot Password</button>
+    {/if}
   </form>
 </Modal>
\ No newline at end of file