chore: track walletType usage

src/db/wallets/getWalletDetails.ts

@@ -1,3 +1,4 @@
+import LRUMap from "mnemonist/lru-map";
 import { getAddress } from "thirdweb";
 import { z } from "zod";
 import type { PrismaTransaction } from "../../schema/prisma";
@@ -8,7 +9,7 @@ import { getPrismaWithPostgresTx } from "../client";
 
 interface GetWalletDetailsParams {
   pgtx?: PrismaTransaction;
-  address: string;
+  walletAddress: string;
 }
 
 export class WalletDetailsError extends Error {
@@ -130,6 +131,8 @@ export type SmartBackendWalletType = (typeof SmartBackendWalletTypes)[number];
 export type BackendWalletType = (typeof BackendWalletTypes)[number];
 export type ParsedWalletDetails = z.infer<typeof walletDetailsSchema>;
 
+export const walletDetailsCache = new LRUMap<string, ParsedWalletDetails>(2048);
+
 /**
  * Return the wallet details for the given address.
  *
@@ -143,20 +146,28 @@ export type ParsedWalletDetails = z.infer<typeof walletDetailsSchema>;
  */
 export const getWalletDetails = async ({
   pgtx,
-  address,
+  walletAddress: _walletAddress,
 }: GetWalletDetailsParams) => {
+  // Wallet details are stored in lowercase.
+  const walletAddress = _walletAddress.toLowerCase();
+
+  const cachedDetails = walletDetailsCache.get(walletAddress);
+  if (cachedDetails) {
+    return cachedDetails;
+  }
+
   const prisma = getPrismaWithPostgresTx(pgtx);
   const config = await getConfig();
 
   const walletDetails = await prisma.walletDetails.findUnique({
     where: {
-      address: address.toLowerCase(),
+      address: walletAddress,
     },
   });
 
   if (!walletDetails) {
     throw new WalletDetailsError(
-      `No wallet details found for address ${address}`,
+      `No wallet details found for address ${walletAddress}`,
     );
   }
 
@@ -167,7 +178,7 @@ export const getWalletDetails = async ({
   ) {
     if (!walletDetails.awsKmsArn) {
       throw new WalletDetailsError(
-        `AWS KMS ARN is missing for the wallet with address ${address}`,
+        `AWS KMS ARN is missing for the wallet with address ${walletAddress}`,
       );
     }
 
@@ -188,7 +199,7 @@ export const getWalletDetails = async ({
   ) {
     if (!walletDetails.gcpKmsResourcePath) {
       throw new WalletDetailsError(
-        `GCP KMS resource path is missing for the wallet with address ${address}`,
+        `GCP KMS resource path is missing for the wallet with address ${walletAddress}`,
       );
     }
 
@@ -209,14 +220,17 @@ export const getWalletDetails = async ({
 
   // zod schema can validate all necessary fields are populated after decryption
   try {
-    return walletDetailsSchema.parse(walletDetails, {
+    const result = walletDetailsSchema.parse(walletDetails, {
       errorMap: (issue) => {
         const fieldName = issue.path.join(".");
         return {
-          message: `${fieldName} is necessary for wallet ${address} of type ${walletDetails.type}, but not found in wallet details or configuration`,
+          message: `${fieldName} is necessary for wallet ${walletAddress} of type ${walletDetails.type}, but not found in wallet details or configuration`,
         };
       },
     });
+
+    walletDetailsCache.set(walletAddress, result);
+    return result;
   } catch (e) {
     if (e instanceof z.ZodError) {
       throw new WalletDetailsError(

src/server/routes/backend-wallet/signMessage.ts

@@ -56,7 +56,7 @@ export async function signMessageRoute(fastify: FastifyInstance) {
       }
 
       const walletDetails = await getWalletDetails({
-        address: walletAddress,
+        walletAddress,
       });
 
       if (isSmartBackendWallet(walletDetails) && !chainId) {

src/server/utils/wallets/getLocalWallet.ts

@@ -49,7 +49,9 @@ export const getLocalWallet = async ({
     });
 
     // If that works, save the wallet using the encryption password for the future
-    const walletDetails = await getWalletDetails({ address: walletAddress });
+    const walletDetails = await getWalletDetails({
+      walletAddress,
+    });
 
     logger({
       service: "worker",
@@ -73,7 +75,9 @@ export const getLocalWallet = async ({
 export const getLocalWalletAccount = async (
   walletAddress: Address,
 ): Promise<Account> => {
-  const walletDetails = await getWalletDetails({ address: walletAddress });
+  const walletDetails = await getWalletDetails({
+    walletAddress,
+  });
 
   if (walletDetails.type !== "local") {
     throw new Error(`Local Wallet not found for address ${walletAddress}`);

src/utils/account.ts

@@ -39,7 +39,7 @@ export const getAccount = async (args: {
   }
 
   const walletDetails = await getWalletDetails({
-    address: from,
+    walletAddress: from,
   });
 
   const { account } = await walletDetailsToAccount({ walletDetails, chain });
@@ -180,7 +180,7 @@ export const getSmartBackendWalletAdminAccount = async ({
   }
 
   const walletDetails = await getWalletDetails({
-    address: accountAddress,
+    walletAddress: accountAddress,
   });
 
   if (!isSmartBackendWallet(walletDetails)) {

src/utils/cache/clearCache.ts

@@ -1,7 +1,9 @@
+import { walletDetailsCache } from "../../db/wallets/getWalletDetails";
 import type { env } from "../env";
 import { accessTokenCache } from "./accessToken";
 import { invalidateConfig } from "./getConfig";
 import { sdkCache } from "./getSdk";
+import { smartWalletsCache } from "./getSmartWalletV5";
 import { walletsCache } from "./getWallet";
 import { webhookCache } from "./getWebhook";
 import { keypairCache } from "./keypair";
@@ -15,4 +17,6 @@ export const clearCache = async (
   walletsCache.clear();
   accessTokenCache.clear();
   keypairCache.clear();
+  smartWalletsCache.clear();
+  walletDetailsCache.clear();
 };

src/utils/cache/getWallet.ts

@@ -44,7 +44,7 @@ export const getWallet = async <TWallet extends EVMWallet>({
   try {
     walletDetails = await getWalletDetails({
       pgtx,
-      address: walletAddress,
+      walletAddress,
     });
   } catch (e) {
     if (e instanceof WalletDetailsError) {