From 82dd3be25e053004143a931e2dbc0fb74f8206ab Mon Sep 17 00:00:00 2001
From: Aaron McCloud <mcclouda@gmail.com>
Date: Fri, 10 Dec 2021 13:43:55 -0800
Subject: [PATCH] Added fix for Signup.js refresh losing state when user
 navigates back to the page on mobile devices

---
 _chapters/signup-with-aws-cognito.md | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/_chapters/signup-with-aws-cognito.md b/_chapters/signup-with-aws-cognito.md
index 3b37a3586..bdc91a84a 100644
--- a/_chapters/signup-with-aws-cognito.md
+++ b/_chapters/signup-with-aws-cognito.md
@@ -8,7 +8,27 @@ description: To implement a signup form in our React.js app using Amazon Cognito
 comments_id: signup-with-aws-cognito/130
 ---
 
-Now let's go ahead and implement the `handleSubmit` and `handleConfirmationSubmit` functions and connect it up with our AWS Cognito setup.
+Now let's go ahead and implement the `handleSubmit` and `handleConfirmationSubmit` functions and connect it up with our AWS Cognito setup. We will also add some code to control for the user navigating away from the page to get their AWS confirmation code, which otherwise may cause your page to refresh and lose state (this issue usually appears on mobile devices).
+
+{%change%} First, add `useEffect` to the `import`s in the top line of `src/containers/Signup.js`, so it looks like this:
+
+``` javascript
+import React, { useState, useEffect } from "react";
+```
+
+{%change%} Then we will add a variable that will serve as our key to store and retrieve user data from `localStorage`, and a check for that data when someone navigates to, or refreshes, the page. This should be added right below the `useState()` variables:
+
+``` javascript
+  const tempAddressForUserInfo = 'tempNewUserData';
+ 
+  useEffect(() => {
+    const tempNewUser = window.localStorage.getItem(tempAddressForUserInfo);
+ 
+    if (tempNewUser) {
+      setNewUser(JSON.parse(tempNewUser));
+    }
+  }, []);
+```
 
 {%change%} Replace our `handleSubmit` and `handleConfirmationSubmit` functions in `src/containers/Signup.js` with the following.
 
@@ -23,6 +43,8 @@ async function handleSubmit(event) {
       username: fields.email,
       password: fields.password,
     });
+
+    window.localStorage.setItem(tempAddressForUserInfo, JSON.stringify(newUser));
     setIsLoading(false);
     setNewUser(newUser);
   } catch (e) {
@@ -42,6 +64,7 @@ async function handleConfirmationSubmit(event) {
 
     userHasAuthenticated(true);
     history.push("/");
+    window.localStorage.removeItem(tempAddressForUserInfo);
   } catch (e) {
     onError(e);
     setIsLoading(false);