From 789d7047c703c2dc2598f7d145736488bedbd633 Mon Sep 17 00:00:00 2001
From: Alex Harley <alex@alexbh.dev>
Date: Wed, 12 Jun 2024 18:29:10 +0200
Subject: [PATCH 1/2] feat: only show wallets useful to loaded warp routes

---
 .../wallet/WalletEnvSelectionModal.tsx        | 68 +++++++++++++------
 1 file changed, 46 insertions(+), 22 deletions(-)

diff --git a/src/features/wallet/WalletEnvSelectionModal.tsx b/src/features/wallet/WalletEnvSelectionModal.tsx
index c2872f55..2b6404e7 100644
--- a/src/features/wallet/WalletEnvSelectionModal.tsx
+++ b/src/features/wallet/WalletEnvSelectionModal.tsx
@@ -1,16 +1,34 @@
 import Image from 'next/image';
-import { PropsWithChildren } from 'react';
+import { PropsWithChildren, useMemo } from 'react';
 
 import { ethereum, solana } from '@hyperlane-xyz/registry';
 import { ProtocolType } from '@hyperlane-xyz/utils';
 
 import { ChainLogo } from '../../components/icons/ChainLogo';
 import { Modal } from '../../components/layout/Modal';
+import { getWarpCore } from '../../context/context';
 
 import { useConnectFns } from './hooks/multiProtocol';
 
 export function WalletEnvSelectionModal({ isOpen, close }: { isOpen: boolean; close: () => void }) {
   const connectFns = useConnectFns();
+  const protocolsEnabled = useMemo(() => {
+    const wc = getWarpCore();
+
+    return wc.getTokenChains().reduce(
+      (accum, chainName) => {
+        return {
+          ...accum,
+          [wc.multiProvider.getChainMetadata(chainName).protocol]: true,
+        };
+      },
+      {
+        [ProtocolType.Ethereum]: false,
+        [ProtocolType.Cosmos]: false,
+        [ProtocolType.Sealevel]: false,
+      },
+    );
+  }, []);
 
   const onClickEnv = (env: ProtocolType) => () => {
     close();
@@ -21,27 +39,33 @@ export function WalletEnvSelectionModal({ isOpen, close }: { isOpen: boolean; cl
   return (
     <Modal title="Select Wallet Environment" isOpen={isOpen} close={close} width="max-w-sm">
       <div className="pt-4 pb-2 flex flex-col space-y-2.5">
-        <EnvButton
-          onClick={onClickEnv(ProtocolType.Ethereum)}
-          subTitle="an EVM"
-          logoChainId={ethereum.chainId}
-        >
-          Ethereum
-        </EnvButton>
-        <EnvButton
-          onClick={onClickEnv(ProtocolType.Sealevel)}
-          subTitle="a Solana"
-          logoChainId={solana.chainId}
-        >
-          Solana
-        </EnvButton>
-        <EnvButton
-          onClick={onClickEnv(ProtocolType.Cosmos)}
-          subTitle="a Cosmos"
-          logo={<Image src={'/logos/cosmos.svg'} width={34} height={34} alt="" />}
-        >
-          Cosmos
-        </EnvButton>
+        {protocolsEnabled[ProtocolType.Ethereum] && (
+          <EnvButton
+            onClick={onClickEnv(ProtocolType.Ethereum)}
+            subTitle="an EVM"
+            logoChainId={ethereum.chainId}
+          >
+            Ethereum
+          </EnvButton>
+        )}
+        {protocolsEnabled[ProtocolType.Sealevel] && (
+          <EnvButton
+            onClick={onClickEnv(ProtocolType.Sealevel)}
+            subTitle="a Solana"
+            logoChainId={solana.chainId}
+          >
+            Solana
+          </EnvButton>
+        )}
+        {protocolsEnabled[ProtocolType.Cosmos] && (
+          <EnvButton
+            onClick={onClickEnv(ProtocolType.Cosmos)}
+            subTitle="a Cosmos"
+            logo={<Image src={'/logos/cosmos.svg'} width={34} height={34} alt="" />}
+          >
+            Cosmos
+          </EnvButton>
+        )}
       </div>
     </Modal>
   );

From 8390eb649b202e650ac7ace16112e87f53c82e53 Mon Sep 17 00:00:00 2001
From: Alex <alex@alexbh.dev>
Date: Thu, 13 Jun 2024 16:35:09 +0200
Subject: [PATCH 2/2] Update src/features/wallet/WalletEnvSelectionModal.tsx

Co-authored-by: J M Rossy <jm.rossy@gmail.com>
---
 src/features/wallet/WalletEnvSelectionModal.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/features/wallet/WalletEnvSelectionModal.tsx b/src/features/wallet/WalletEnvSelectionModal.tsx
index 2b6404e7..28895838 100644
--- a/src/features/wallet/WalletEnvSelectionModal.tsx
+++ b/src/features/wallet/WalletEnvSelectionModal.tsx
@@ -19,7 +19,7 @@ export function WalletEnvSelectionModal({ isOpen, close }: { isOpen: boolean; cl
       (accum, chainName) => {
         return {
           ...accum,
-          [wc.multiProvider.getChainMetadata(chainName).protocol]: true,
+          [wc.multiProvider.getProtocol(chainName)]: true,
         };
       },
       {