fix for adding wallets dynamically for a new chain (#128)

thirdweb-dev · Sep 14, 2023 · 868c26d · 868c26d
1 parent f38658f
commit 868c26d
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 12 deletions.
diff --git a/core/database/dbOperation.ts b/core/database/dbOperation.ts
@@ -59,9 +59,10 @@ export const getWalletDetails = async (
 
 export const addWalletToDB = async (
   chainId: string,
+  dbInstance: Knex,
   walletAddress: string,
   slug: string,
-  dbInstance: Knex,
+  walletType: string,
 ): Promise<void> => {
   try {
     const sdk = await getSDK(chainId);
@@ -76,7 +77,8 @@ export const addWalletToDB = async (
       blockchainNonce: BigNumber.from(walletNonce ?? 0).toNumber(),
       lastSyncedTimestamp: new Date(),
       lastUsedNonce: -1,
-      walletType: slug,
+      walletType,
+      slug,
     };
 
     await insertIntoWallets(walletData, dbInstance);
@@ -176,3 +178,19 @@ export const addWalletDataWithSupportChainsNonceToDB = async (
     throw error;
   }
 };
+
+export const getWalletDetailsWithoutChain = async (
+  walletAddress: string,
+  database: Knex,
+): Promise<any> => {
+  try {
+    const walletDetails = await database("wallets")
+      .select("*")
+      .where({ walletAddress: walletAddress.toLowerCase() })
+      .first();
+
+    return walletDetails;
+  } catch (error) {
+    throw error;
+  }
+};
diff --git a/server/helpers/dbOperations.ts b/server/helpers/dbOperations.ts
@@ -10,7 +10,12 @@ import { FastifyInstance, FastifyRequest } from "fastify";
 import { StatusCodes } from "http-status-codes";
 import { Knex } from "knex";
 import { v4 as uuid } from "uuid";
-import { connectToDatabase, getWalletDetails } from "../../core";
+import {
+  addWalletToDB,
+  connectToDatabase,
+  getWalletDetails,
+  getWalletDetailsWithoutChain,
+} from "../../core";
 import { createCustomError } from "../../core/error/customError";
 import {
   TransactionSchema,
@@ -63,15 +68,22 @@ export const queueTransaction = async (
   );
 
   if (!walletDetails) {
-    // await addWalletToDB(
-    //   chainId,
-    //   dbInstance,
-    //   walletAddress,
-    //   chainData.slug,
-    //   getWalletType(),
-    // );
-    throw new Error(
-      `Import Wallet Address ${walletAddress} to DB using /wallet/import end-point`,
+    const walletData = await getWalletDetailsWithoutChain(
+      walletAddress.toLowerCase(),
+      dbInstance,
+    );
+
+    if (!walletData) {
+      throw new Error(
+        `Import Wallet Address ${walletAddress} to DB using /wallet/import end-point.`,
+      );
+    }
+    await addWalletToDB(
+      chainId,
+      dbInstance,
+      walletAddress,
+      chainData.slug,
+      walletData.walletType,
     );
   }
   // encode tx