From dfb56b62b19e751f3b6196a633f9454ab3ac827f Mon Sep 17 00:00:00 2001
From: static <kmc7468@naver.com>
Date: Sat, 28 Dec 2024 18:55:20 +0900
Subject: [PATCH] =?UTF-8?q?=EC=95=94=ED=98=B8=20=ED=82=A4=20=EC=9C=A0?=
 =?UTF-8?q?=EB=AC=B4=EC=97=90=20=EB=94=B0=EB=A5=B8=20=EC=9E=90=EB=8F=99=20?=
 =?UTF-8?q?=EB=A6=AC=EB=8B=A4=EC=9D=B4=EB=A0=89=EC=85=98=20=EA=B5=AC?=
 =?UTF-8?q?=ED=98=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/hooks.server.ts                             |  8 +++++---
 src/routes/(fullscreen)/auth/login/+page.svelte | 11 ++++++++---
 src/routes/(fullscreen)/key/generate/service.ts |  2 --
 src/routes/+layout.svelte                       | 16 ++++++++++++++++
 4 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/src/hooks.server.ts b/src/hooks.server.ts
index f9237c5..c915d9b 100644
--- a/src/hooks.server.ts
+++ b/src/hooks.server.ts
@@ -9,8 +9,7 @@ export const init: ServerInit = () => {
 };
 
 export const handle: Handle = async ({ event, resolve }) => {
-  const path = event.url.pathname;
-  if (path.startsWith("/api") || path.startsWith("/auth")) {
+  if (["/api", "/auth"].some((path) => event.url.pathname.startsWith(path))) {
     return await resolve(event);
   }
 
@@ -18,6 +17,9 @@ export const handle: Handle = async ({ event, resolve }) => {
   if (accessToken) {
     return await resolve(event);
   } else {
-    redirect(302, "/auth/login?redirect=" + encodeURIComponent(path));
+    redirect(
+      302,
+      "/auth/login?redirect=" + encodeURIComponent(event.url.pathname + event.url.search),
+    );
   }
 };
diff --git a/src/routes/(fullscreen)/auth/login/+page.svelte b/src/routes/(fullscreen)/auth/login/+page.svelte
index 5b13bab..ffac02d 100644
--- a/src/routes/(fullscreen)/auth/login/+page.svelte
+++ b/src/routes/(fullscreen)/auth/login/+page.svelte
@@ -1,8 +1,10 @@
 <script lang="ts">
+  import { get } from "svelte/store";
   import { goto } from "$app/navigation";
   import { Button, TextButton } from "$lib/components/buttons";
   import { TitleDiv, BottomDiv } from "$lib/components/divs";
   import { TextInput } from "$lib/components/inputs";
+  import { keyPairStore } from "$lib/stores";
   import { requestLogin } from "./service";
 
   let { data } = $props();
@@ -13,9 +15,12 @@
   const login = async () => {
     // TODO: Validation
 
-    const ok = await requestLogin(email, password);
-    if (ok) {
-      goto(data.redirectPath);
+    if (await requestLogin(email, password)) {
+      await goto(
+        get(keyPairStore)
+          ? data.redirectPath
+          : "/key/generate?redirect=" + encodeURIComponent(data.redirectPath),
+      );
     } else {
       // TODO: Alert
     }
diff --git a/src/routes/(fullscreen)/key/generate/service.ts b/src/routes/(fullscreen)/key/generate/service.ts
index 8c38abc..f43b66d 100644
--- a/src/routes/(fullscreen)/key/generate/service.ts
+++ b/src/routes/(fullscreen)/key/generate/service.ts
@@ -1,4 +1,3 @@
-import { storeKeyPairIntoIndexedDB } from "$lib/indexedDB";
 import { keyPairStore } from "$lib/stores";
 
 type KeyType = "public" | "private";
@@ -48,7 +47,6 @@ export const generateKeyPair = async () => {
     publicKey: keyPair.publicKey,
     privateKey: privKeySecured,
   });
-  await storeKeyPairIntoIndexedDB(keyPair.publicKey, privKeySecured);
 
   return {
     pubKeyBase64: await exportKeyToBase64(keyPair.publicKey, "public"),
diff --git a/src/routes/+layout.svelte b/src/routes/+layout.svelte
index 3fa208a..44112c1 100644
--- a/src/routes/+layout.svelte
+++ b/src/routes/+layout.svelte
@@ -1,6 +1,22 @@
 <script lang="ts">
+  import { onMount } from "svelte";
+  import { goto } from "$app/navigation";
+  import { getKeyPairFromIndexedDB } from "$lib/indexedDB";
+  import { keyPairStore } from "$lib/stores";
   import "../app.css";
+
   let { children } = $props();
+
+  onMount(async () => {
+    const { pubKey, privKey } = await getKeyPairFromIndexedDB();
+    if (pubKey && privKey) {
+      keyPairStore.set({ publicKey: pubKey, privateKey: privKey });
+    } else if (!["/auth", "/key/generate"].some((path) => location.pathname.startsWith(path))) {
+      await goto(
+        "/key/generate?redirect=" + encodeURIComponent(location.pathname + location.search),
+      );
+    }
+  });
 </script>
 
 {@render children()}