From 929f622b3c111c0a80e6ba255a2473ab4cd1a1f8 Mon Sep 17 00:00:00 2001
From: 0xExp-po <lucaschang1127@gmail.com>
Date: Sun, 15 Sep 2024 23:26:34 +0900
Subject: [PATCH] fix: keep wallet connection

---
 dapp/src/components/ConnectWallet.astro    |  3 ++-
 dapp/src/components/stellar-wallets-kit.ts | 10 +++++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/dapp/src/components/ConnectWallet.astro b/dapp/src/components/ConnectWallet.astro
index 543f75d..c1d939d 100644
--- a/dapp/src/components/ConnectWallet.astro
+++ b/dapp/src/components/ConnectWallet.astro
@@ -4,9 +4,10 @@
 
 <script>
   import { truncateMiddle } from "../utils/utils";
-  import { kit, loadedPublicKey, setPublicKey } from "./stellar-wallets-kit";
+  import { initializeConnection, kit, loadedPublicKey, setPublicKey } from "./stellar-wallets-kit";
 
   document.addEventListener("astro:page-load", () => {
+    initializeConnection();
     const button = document.querySelector(
       "[data-connect]",
     ) as HTMLButtonElement;
diff --git a/dapp/src/components/stellar-wallets-kit.ts b/dapp/src/components/stellar-wallets-kit.ts
index ca984b9..2c9260c 100644
--- a/dapp/src/components/stellar-wallets-kit.ts
+++ b/dapp/src/components/stellar-wallets-kit.ts
@@ -21,6 +21,14 @@ function loadedPublicKey(): string | undefined {
 
 function setPublicKey(data: string): void {
   connectionState.publicKey = data;
+  localStorage.setItem('publicKey', data);
 }
 
-export { kit, loadedPublicKey, setPublicKey };
+function initializeConnection(): void {
+  const storedPublicKey = localStorage.getItem('publicKey');
+  if (storedPublicKey) {
+    setPublicKey(storedPublicKey);
+  }
+}
+
+export { kit, loadedPublicKey, setPublicKey, initializeConnection };