password lock button

0xsequence · Nov 10, 2024 · 85b964b · 85b964b
1 parent ab20d40
commit 85b964b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git 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
@@ -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(