fix: allow callback on chain switch success

0xflair · Jan 8, 2023 · 0ee2048 · 0ee2048
1 parent 7c853ef
commit 0ee2048
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/packages/react/src/core/wallet/components/SwitchChainButton.tsx b/packages/react/src/core/wallet/components/SwitchChainButton.tsx
@@ -1,23 +1,32 @@
 import React from 'react';
-import { useNetwork, useSwitchNetwork } from 'wagmi';
+import { Chain, useNetwork, useSwitchNetwork } from 'wagmi';
 
 import { BareComponentProps, useChainInfo } from '../../../common';
 
 type Props = BareComponentProps & {
   label?: React.ReactNode;
   children?: React.ReactNode;
   requiredChainId: number;
+  onSuccess?: (chain: Chain) => void;
 };
 
 export const SwitchChainButton = ({
   as,
   label,
   children,
   requiredChainId,
+  onSuccess,
   ...attributes
 }: Props) => {
   const { chain } = useNetwork();
-  const { switchNetwork } = useSwitchNetwork();
+  const { switchNetwork } = useSwitchNetwork({
+    throwForSwitchChainNotSupported: false,
+    onSuccess(data, variables, context) {
+      if (data.id.toString() == requiredChainId.toString()) {
+        onSuccess?.(data);
+      }
+    },
+  });
   const requiredChain = useChainInfo(Number(requiredChainId));
 
   const Component = as || 'button';