feat: allow filtering by ids or names

starknet-io · Dec 6, 2024 · 2cccbbb · 2cccbbb
1 parent 7144af2
commit 2cccbbb
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/packages/core/src/main.ts b/packages/core/src/main.ts
@@ -117,9 +117,22 @@ export function getStarknet(
 
       return firstAuthorizedWallet
     },
-    discoverVirtualWallets: async (): Promise<void> => {
+    discoverVirtualWallets: async (
+      walletNamesOrIds: string[] = [],
+    ): Promise<void> => {
+      const walletNamesOrIdsSet = new Set(walletNamesOrIds)
+
+      const virtualWalletToDiscover =
+        walletNamesOrIdsSet.size > 0
+          ? virtualWallets.filter(
+              (virtualWallet) =>
+                walletNamesOrIdsSet.has(virtualWallet.name) ||
+                walletNamesOrIdsSet.has(virtualWallet.id),
+            )
+          : virtualWallets
+
       await Promise.all(
-        virtualWallets.map(async (virtualWallet) => {
+        virtualWalletToDiscover.map(async (virtualWallet) => {
           const hasSupport = await virtualWallet.hasSupport(windowObject)
           if (hasSupport) {
             windowObject[virtualWallet.windowKey] = virtualWallet