From 85b964bb06f4a602b778cba2b4ad2c29e051e898 Mon Sep 17 00:00:00 2001
From: William Hua <william@attente.ca>
Date: Sun, 10 Nov 2024 15:28:26 -0500
Subject: [PATCH] password lock button

---
 src/routes/Wallet.tsx   | 10 ++++++++++
 src/stores/AuthStore.ts |  9 +++++++++
 2 files changed, 19 insertions(+)

diff --git a/src/routes/Wallet.tsx b/src/routes/Wallet.tsx
index be86dae..e2e93bf 100644
--- a/src/routes/Wallet.tsx
+++ b/src/routes/Wallet.tsx
@@ -69,6 +69,7 @@ function Wallet() {
   const navigate = useNavigate()
 
   const accountAddress = useObservable(authStore.accountAddress)
+  const isPasswordSet = useObservable(authStore.isPasswordSet)
   const isSigningTxn = useObservable(walletStore.isSigningTxn)
   const isSigningMsg = useObservable(walletStore.isSigningMsg)
 
@@ -387,6 +388,15 @@ function Wallet() {
               onClick={() => setIsNetworkModalOpen(true)}
             />
 
+            {isPasswordSet && (
+              <Button
+                label="Password Lock"
+                variant="text"
+                marginRight="8"
+                onClick={() => window.location.reload()}
+              />
+            )}
+
             <Button
               label="Sign Out"
               variant="text"
diff --git a/src/stores/AuthStore.ts b/src/stores/AuthStore.ts
index 3f9809d..59850a9 100644
--- a/src/stores/AuthStore.ts
+++ b/src/stores/AuthStore.ts
@@ -35,9 +35,14 @@ export class AuthStore {
         networkStore.accountLoaded.set(true)
       }
     })
+
+    getIndexedDB(IndexedDBKey.SECURITY).then(async db => {
+      this.isPasswordSet.set(!db.get(IndexedDBKey.SECURITY, 'key'))
+    })
   }
 
   isLoadingAccount = observable(true)
+  isPasswordSet = observable(true)
 
   account: Account | undefined
 
@@ -133,6 +138,8 @@ export class AuthStore {
     const db = await getIndexedDB(IndexedDBKey.SECURITY)
     await db.put(IndexedDBKey.SECURITY, key, 'key')
     await db.put(IndexedDBKey.SECURITY, encrypted, 'mnemonic')
+
+    this.isPasswordSet.set(false)
   }
 
   async encryptRecoveryMnemonicWithPassword(mnemonic: string, address: string, password: string) {
@@ -148,6 +155,8 @@ export class AuthStore {
     // Store encrypted data (minus key) in indexed db
     const db = await getIndexedDB(IndexedDBKey.SECURITY)
     await db.put(IndexedDBKey.SECURITY, encrypted, 'mnemonic')
+
+    this.isPasswordSet.set(true)
   }
 
   async decryptRecoveryMnemonic(