feat: add custom wallet handler

Author: gin-lsl

packages/common/src/types.ts

@@ -102,7 +102,7 @@ export interface NFTMetadata {
 }
 
 export interface ConnectOptions {
-  connectType?: 'extension' | 'qrCode';
+  connectType?: 'extension' | 'qrCode' | 'custom';
 }
 
 export interface UniversalWeb3ProviderInterface {
@@ -137,6 +137,7 @@ export interface Wallet extends WalletMetadata {
   _standardWallet?: any;
   hasWalletReady?: () => Promise<boolean>;
   hasExtensionInstalled?: () => Promise<boolean>;
+  hasCustomHandler?: () => boolean;
   getQrCode?: () => Promise<{ uri: string }>;
   customQrCodePanel?: boolean;
 }

packages/solana/src/solana-provider/config-provider.tsx

@@ -49,6 +49,8 @@ export const AntDesignWeb3ConfigProvider: React.FC<
   const { connection } = useConnection();
   const connectAsyncRef = useRef<ConnectAsync>();
 
+  console.log('wallets:', wallets);
+
   const [balanceData, setBalanceData] = useState<bigint>();
   const [account, setAccount] = useState<Account>();
   const [currentWalletName, setCurrentWalletName] = useState(() => wallet?.adapter?.name ?? null);
@@ -188,11 +190,14 @@ export const AntDesignWeb3ConfigProvider: React.FC<
             remark: adapter.name,
             _standardWallet: adapter,
 
-            hasExtensionInstalled: async () => {
-              return true;
-            },
+            hasCustomHandler: () => {},
+
+            // hasExtensionInstalled: async () => {
+            //   return true;
+            // },
             hasWalletReady: async () => {
-              return hasWalletReady(adapter.readyState);
+              // return hasWalletReady(adapter.readyState);
+              return true;
             },
           };
         }
@@ -245,6 +250,7 @@ export const AntDesignWeb3ConfigProvider: React.FC<
         selectWallet(currentWalletName);
       }}
       connect={async (_wallet, options) => {
+        console.log('connect_wallet:', _wallet, options);
         let resolve: any;
         let reject: any;
 

packages/web3/src/connect-modal/components/ModalPanel.tsx

@@ -65,10 +65,13 @@ const ModalPanel: React.FC<ModalPanelProps> = (props) => {
     async (wallet?: Wallet, connectOptions?: ConnectOptions) => {
       setSelectedWallet(wallet);
       if (wallet && connectOptions) {
+        console.log('wallet:', wallet, connectOptions);
         if (connectOptions.connectType === 'qrCode' && !wallet.customQrCodePanel) {
           updatePanelRoute('qrCode', true);
         } else if (connectOptions.connectType === 'extension') {
           updatePanelRoute('link', true);
+        } else if (connectOptions.connectType === 'custom') {
+          updatePanelRoute('link', true);
         } else {
           setPanelRoute('init');
         }
@@ -77,6 +80,7 @@ const ModalPanel: React.FC<ModalPanelProps> = (props) => {
     },
     [onWalletSelected],
   );
+
   const panelRouteBack = React.useCallback(() => {
     routeStack.current.pop();
     const route = routeStack.current[routeStack.current.length - 1];

packages/web3/src/connect-modal/components/WalletList.tsx

@@ -61,24 +61,41 @@ const WalletList: ForwardRefRenderFunction<ConnectModalActionType, WalletListPro
   };
 
   const selectWallet = async (wallet: Wallet) => {
+    console.log('selectWallet:', wallet);
     const hasWalletReady = await wallet.hasWalletReady?.();
     if (hasWalletReady) {
       // wallet is ready, call ConnectModal's onWalletSelected
       const hasExtensionInstalled = await wallet?.hasExtensionInstalled?.();
-      if (hasExtensionInstalled) {
+
+      // first check if the wallet has custom handler
+      if (wallet.hasCustomHandler) {
+        wallet.hasCustomHandler();
+        updateSelectedWallet(wallet, {
+          connectType: 'custom',
+        });
+      }
+
+      // use extension to connect
+      else if (hasExtensionInstalled) {
         updateSelectedWallet(wallet, {
           connectType: 'extension',
         });
-      } else if (mobile()) {
-        // open in universal link
+      }
+
+      // open in universal link
+      else if (mobile()) {
         openInUniversalLink(wallet);
-      } else if (wallet.getQrCode) {
-        // Extension not installed and can use qr code to connect
+      }
+
+      // Extension not installed and can use qr code to connect
+      else if (wallet.getQrCode) {
         updateSelectedWallet(wallet, {
           connectType: 'qrCode',
         });
-      } else {
-        // use the default connect
+      }
+
+      // use the default connect
+      else {
         updateSelectedWallet(wallet, {});
       }
       return;