From 9e2ed2f9a43b24d7a94dfe1548c3e081ef59d5d9 Mon Sep 17 00:00:00 2001
From: PiotrMatyjasik <matyjasik.piotr@wp.pl>
Date: Sat, 1 Apr 2023 00:28:00 +0200
Subject: [PATCH] On send survey add surveyId to LocalStorage and block
 multiple replies

---
 src/features/surveys/constants/types.ts       |  3 +++
 src/features/surveys/hooks/useLocalStorage.ts | 21 +++++++++++++++++++
 .../surveys/managers/surveyAnswerManager.ts   | 16 ++++++++++++--
 3 files changed, 38 insertions(+), 2 deletions(-)
 create mode 100644 src/features/surveys/constants/types.ts
 create mode 100644 src/features/surveys/hooks/useLocalStorage.ts

diff --git a/src/features/surveys/constants/types.ts b/src/features/surveys/constants/types.ts
new file mode 100644
index 00000000..10cf468a
--- /dev/null
+++ b/src/features/surveys/constants/types.ts
@@ -0,0 +1,3 @@
+export enum LocalStorageKeys {
+  LocalStorageKey = 'surveyId',
+}
diff --git a/src/features/surveys/hooks/useLocalStorage.ts b/src/features/surveys/hooks/useLocalStorage.ts
new file mode 100644
index 00000000..269b8bfb
--- /dev/null
+++ b/src/features/surveys/hooks/useLocalStorage.ts
@@ -0,0 +1,21 @@
+import { useEffect, useState } from 'react';
+
+function useLocalStorage<T>(defaultValue: T, key: string) {
+  const [value, setValue] = useState(defaultValue);
+
+  useEffect(() => {
+    if (defaultValue === value) {
+      return;
+    }
+    window.localStorage.setItem(key, JSON.stringify(value));
+  }, [defaultValue, key, value]);
+
+  useEffect(() => {
+    const storageValue = window.localStorage.getItem(key);
+    setValue(storageValue != null ? JSON.parse(storageValue) : defaultValue);
+  }, [defaultValue, key]);
+
+  return [value, setValue] as const;
+}
+
+export default useLocalStorage;
diff --git a/src/features/surveys/managers/surveyAnswerManager.ts b/src/features/surveys/managers/surveyAnswerManager.ts
index 02e41de3..f5f42d3b 100644
--- a/src/features/surveys/managers/surveyAnswerManager.ts
+++ b/src/features/surveys/managers/surveyAnswerManager.ts
@@ -3,6 +3,10 @@ import { useRouter } from 'next/router';
 import { useState, useEffect, useCallback } from 'react';
 import toast from 'react-hot-toast';
 import { db } from 'firebaseConfiguration';
+import { LocalStorageKeys } from 'features/surveys/constants/types';
+import useLocalStorage from 'features/surveys/hooks/useLocalStorage';
+
+const DEFAULT_VALUE: string[] = [];
 
 export const useSurveyAnswerManager = () => {
   const router = useRouter();
@@ -16,6 +20,10 @@ export const useSurveyAnswerManager = () => {
   const [buttonDisable, setButtonDisable] = useState(false);
   const [isLoading, setIsLoading] = useState(true);
   const [isAnswering, setIsAnswering] = useState(false);
+  const [localStorageValue, setLocalStorageValue] = useLocalStorage<string[]>(
+    DEFAULT_VALUE,
+    LocalStorageKeys.LocalStorageKey
+  );
 
   const getSurveyData = useCallback(async () => {
     const surveyData = await getDoc(doc(db, 'surveys', surveyId));
@@ -33,9 +41,12 @@ export const useSurveyAnswerManager = () => {
     if (surveyId) {
       getSurveyData();
     }
-
+    if (localStorageValue.includes(surveyId) && !isAnswering) {
+      router.replace('/');
+      toast.success('You have answered this survey');
+    }
     // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, []);
+  }, [localStorageValue]);
 
   const handleIconClick = (icon: string) => {
     setSelectedIcon(icon);
@@ -55,6 +66,7 @@ export const useSurveyAnswerManager = () => {
         answer,
         answerDate: new Date(),
       });
+      setLocalStorageValue([...localStorageValue, surveyId]);
       await router.replace('/');
       toast.success('The reply has been sent');
     } catch (error) {