From 80a925ab4dab576f629e65b9344e0b915517ea39 Mon Sep 17 00:00:00 2001
From: David <info@thedaviddelta.com>
Date: Sun, 10 Oct 2021 21:45:14 +0200
Subject: [PATCH] Save languages to LocalStorage (#52)

* Saving & restoring implemented
---
 .dockerignore                    |  4 ++++
 pages/[[...slug]].tsx            | 36 +++++++++++++++++++++++++++++---
 tests/pages/[[...slug]].test.tsx |  2 ++
 3 files changed, 39 insertions(+), 3 deletions(-)

diff --git a/.dockerignore b/.dockerignore
index f93f642..7204a22 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -48,3 +48,7 @@ cypress/screenshots
 .git/
 .gitignore
 .gitattributes
+
+# docker
+Dockerfile
+.dockerignore
diff --git a/pages/[[...slug]].tsx b/pages/[[...slug]].tsx
index 0fefa05..2c3d459 100644
--- a/pages/[[...slug]].tsx
+++ b/pages/[[...slug]].tsx
@@ -11,6 +11,7 @@ import { useToastOnLoad } from "@hooks";
 import { googleScrape, extractSlug, textToSpeechScrape } from "@utils/translate";
 import { retrieveFromType, replaceBoth } from "@utils/language";
 import langReducer, { Actions, initialState } from "@utils/reducer";
+import { localGetItem, localSetItem } from "@utils/storage";
 
 const AutoTranslateButton = dynamic(() => import("@components/AutoTranslateButton"), { ssr: false });
 
@@ -37,19 +38,39 @@ const Page: FC<InferGetStaticPropsType<typeof getStaticProps>> = ({ home, transl
         if (!home && customQuery === initial.query && source === initial.source && target === initial.target)
             return;
 
+        localSetItem("source", source);
+        localSetItem("target", target);
+
         dispatch({ type: Actions.SET_FIELD, payload: { key: "isLoading", value: true }});
         Router.push(`/${source}/${target}/${encodeURIComponent(customQuery)}`);
     }, [isLoading, source, target, home, initial]);
 
     useEffect(() => {
         if (home)
-            return dispatch({ type: Actions.SET_ALL, payload: { state: { ...initialState, isLoading: false } } });
+            return dispatch({
+                type: Actions.SET_ALL,
+                payload: {
+                    state: {
+                        ...initialState,
+                        source: localGetItem("source") || initialState.source,
+                        target: localGetItem("target") || initialState.target,
+                        isLoading: false
+                    }
+                }
+            });
         if (!initial)
             return;
 
         dispatch({
             type: Actions.SET_ALL,
-            payload: { state: { ...initial, delayedQuery: initial.query, translation: translationRes, isLoading: false } }
+            payload: {
+                state: {
+                    ...initial,
+                    delayedQuery: initial.query,
+                    translation: translationRes,
+                    isLoading: false
+                }
+            }
         });
     }, [initial, translationRes, home]);
 
@@ -61,7 +82,16 @@ const Page: FC<InferGetStaticPropsType<typeof getStaticProps>> = ({ home, transl
     }, [query]);
 
     useEffect(() => {
-        const handler = (url: string) => url === Router.asPath || dispatch({ type: Actions.SET_FIELD, payload: { key: "isLoading", value: true }});
+        const handler = (url: string) => {
+            url === Router.asPath || dispatch({ type: Actions.SET_FIELD, payload: { key: "isLoading", value: true }});
+
+            if (url !== "/")
+                return;
+            dispatch({ type: Actions.SET_FIELD, payload: { key: "source", value: initialState.source }});
+            localSetItem("source", initialState.source);
+            dispatch({ type: Actions.SET_FIELD, payload: { key: "target", value: initialState.target }});
+            localSetItem("target", initialState.target);
+        };
         Router.events.on("beforeHistoryChange", handler);
         return () => Router.events.off("beforeHistoryChange", handler);
     }, []);
diff --git a/tests/pages/[[...slug]].test.tsx b/tests/pages/[[...slug]].test.tsx
index 5a6f114..fafacd5 100644
--- a/tests/pages/[[...slug]].test.tsx
+++ b/tests/pages/[[...slug]].test.tsx
@@ -156,6 +156,7 @@ describe("Page", () => {
         expect(source).toHaveValue(sourceVal);
 
         await waitFor(() => expect(routerPushMock).toHaveBeenCalledTimes(1));
+        expect(localStorageSetMock).toHaveBeenCalledWith("source", sourceVal);
     });
 
     it("doesn't switch the page on language change on the start page", async () => {
@@ -193,6 +194,7 @@ describe("Page", () => {
         expect(screen.getByRole("textbox", { name: /translation result/i })).toHaveValue(initial.query);
 
         await waitFor(() => expect(routerPushMock).toHaveBeenCalledTimes(1));
+        expect(localStorageSetMock).toHaveBeenLastCalledWith("target", initial.source);
     });
 
     it("translates & loads initials correctly", async () => {