feat(ironfish): Write encrypted accounts when toggling scanning

iron-fish · Sep 19, 2024 · eee04eb · eee04eb
1 parent 6fbfd31
commit eee04eb
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/ironfish/src/rpc/routes/wallet/setScanning.ts b/ironfish/src/rpc/routes/wallet/setScanning.ts
@@ -28,7 +28,7 @@ routes.register<typeof SetScanningRequestSchema, SetScanningResponse>(
     AssertHasRpcContext(request, context, 'wallet')
 
     const account = getAccount(context.wallet, request.data.account)
-    await account.updateScanningEnabled(request.data.enabled)
+    await context.wallet.setScanningEnabled(account, request.data.enabled)
     request.end()
   },
 )
diff --git a/ironfish/src/wallet/account/account.ts b/ironfish/src/wallet/account/account.ts
@@ -1299,10 +1299,20 @@ export class Account {
 
   async updateScanningEnabled(
     scanningEnabled: boolean,
+    options?: { masterKey: MasterKey | null },
     tx?: IDatabaseTransaction,
   ): Promise<void> {
+    const walletEncrypted = await this.walletDb.accountsEncrypted(tx)
+
     this.scanningEnabled = scanningEnabled
-    await this.walletDb.setAccount(this, tx)
+
+    if (walletEncrypted) {
+      Assert.isNotUndefined(options)
+      Assert.isNotNull(options?.masterKey)
+      await this.walletDb.setEncryptedAccount(this, options.masterKey, tx)
+    } else {
+      await this.walletDb.setAccount(this, tx)
+    }
   }
 
   async getTransactionNotes(

diff --git a/ironfish/src/wallet/wallet.ts b/ironfish/src/wallet/wallet.ts
@@ -1569,6 +1569,14 @@ export class Wallet {
     await account.setName(name, { masterKey: this.masterKey }, tx)
   }
 
+  async setScanningEnabled(
+    account: Account,
+    enabled: boolean,
+    tx?: IDatabaseTransaction,
+  ): Promise<void> {
+    await account.updateScanningEnabled(enabled, { masterKey: this.masterKey }, tx)
+  }
+
   get accounts(): Account[] {
     return Array.from(this.accountById.values())
   }