diff --git a/src/components/PasswordForm.tsx b/src/components/PasswordForm.tsx
index 6be5aec08..f18d8436c 100644
--- a/src/components/PasswordForm.tsx
+++ b/src/components/PasswordForm.tsx
@@ -114,11 +114,16 @@ function SetPassword(props: Props) {
)}
-
-
-
+ {props.back && (
+
+
+
+ )}
diff --git a/src/components/Sidebar.tsx b/src/components/Sidebar.tsx
index 7ee2f4d0f..d075925a6 100644
--- a/src/components/Sidebar.tsx
+++ b/src/components/Sidebar.tsx
@@ -3,7 +3,7 @@ import React, { useEffect, useState } from 'react';
import { slide as Menu } from 'react-burger-menu';
import billingService, { Subscription } from 'services/billingService';
import constants from 'utils/strings/constants';
-import { getData, LS_KEYS } from 'utils/storage/localStorage';
+import { getData, LS_KEYS, setData } from 'utils/storage/localStorage';
import { getToken } from 'utils/common/key';
import { getEndpoint } from 'utils/common/apiUtil';
import { Button } from 'react-bootstrap';
@@ -214,7 +214,10 @@ export default function Sidebar(props: Props) {
>
router.push('changePassword')}
+ onClick={() => {
+ setData(LS_KEYS.SHOW_BACK_BUTTON, { value: true });
+ router.push('changePassword');
+ }}
>
{constants.CHANGE_PASSWORD}
diff --git a/src/pages/changePassword/index.tsx b/src/pages/changePassword/index.tsx
index d1474f84c..925adc306 100644
--- a/src/pages/changePassword/index.tsx
+++ b/src/pages/changePassword/index.tsx
@@ -1,6 +1,6 @@
import React, { useState, useEffect, useContext } from 'react';
import constants from 'utils/strings/constants';
-import { getData, LS_KEYS } from 'utils/storage/localStorage';
+import { getData, LS_KEYS, setData } from 'utils/storage/localStorage';
import { useRouter } from 'next/router';
import { getKey, SESSION_KEYS, setKey } from 'utils/storage/sessionStorage';
import { B64EncryptionResult } from 'services/uploadService';
@@ -66,14 +66,21 @@ export default function Generate() {
);
setSessionKeys(key);
+ redirectToGallery();
+ };
+ const redirectToGallery = () => {
+ setData(LS_KEYS.SHOW_BACK_BUTTON, { value: false });
router.push('/gallery');
};
-
return (
router.push('/gallery')}
+ back={
+ getData(LS_KEYS.SHOW_BACK_BUTTON)?.value
+ ? redirectToGallery
+ : null
+ }
/>
);
}
diff --git a/src/utils/storage/localStorage.ts b/src/utils/storage/localStorage.ts
index 44d5d1d6c..7ded2e198 100644
--- a/src/utils/storage/localStorage.ts
+++ b/src/utils/storage/localStorage.ts
@@ -6,6 +6,7 @@ export enum LS_KEYS {
PLANS = 'plans',
IS_FIRST_LOGIN = 'isFirstLogin',
JUST_SIGNED_UP = 'justSignedUp',
+ SHOW_BACK_BUTTON = 'showBackButton',
}
export const setData = (key: LS_KEYS, value: object) => {