@@ -5,7 +5,7 @@ import { makeStyles, usePopupCustomSnackbar } from '@masknet/theme'
55import { useWallet } from '@masknet/web3-hooks-base'
66import { Box , Typography } from '@mui/material'
77import { useMutation , useQueryClient } from '@tanstack/react-query'
8- import { memo , useCallback } from 'react'
8+ import { memo } from 'react'
99
1010const useStyles = makeStyles ( ) ( ( theme ) => ( {
1111 container : {
@@ -93,13 +93,12 @@ const DisconnectModal = memo(function DisconnectModal({ origin, onClose }: Disco
9393 const queryClient = useQueryClient ( )
9494 const { classes, cx } = useStyles ( )
9595 const { showSnackbar } = usePopupCustomSnackbar ( )
96- const wallet = useWallet ( )
97- const address = wallet ?. address
96+ const address = useWallet ( ) ?. address
9897 const { mutate : onDisconnect } = useMutation ( {
99- mutationFn : useCallback ( async ( ) : Promise < void > => {
98+ mutationFn : async ( ) : Promise < void > => {
10099 if ( ! address ) return
101100 await Services . Wallet . disconnectWalletFromOrigin ( address , origin , 'any' )
102- } , [ address ] ) ,
101+ } ,
103102 onMutate : async ( ) => {
104103 await queryClient . invalidateQueries ( { queryKey : [ 'wallet-granted-origins' , address ] } )
105104 showSnackbar (
@@ -116,10 +115,10 @@ const DisconnectModal = memo(function DisconnectModal({ origin, onClose }: Disco
116115 } ,
117116 } )
118117 const { mutate : onDisconnectAll } = useMutation ( {
119- mutationFn : useCallback ( async ( ) : Promise < void > => {
120- if ( ! wallet ) return
118+ mutationFn : async ( ) => {
119+ if ( ! address ) return
121120 await Services . Wallet . disconnectAllOriginsConnectedFromWallet ( address ! , 'any' )
122- } , [ wallet ?. address ] ) ,
121+ } ,
123122 onMutate : async ( ) => {
124123 await queryClient . invalidateQueries ( { queryKey : [ 'wallet-granted-origins' , address ! ] } )
125124 showSnackbar (
@@ -132,7 +131,7 @@ const DisconnectModal = memo(function DisconnectModal({ origin, onClose }: Disco
132131 onClose ( )
133132 } ,
134133 onSettled : ( ) => {
135- queryClient . invalidateQueries ( { queryKey : [ 'wallet-granted-origins' , wallet ?. address ] } )
134+ queryClient . invalidateQueries ( { queryKey : [ 'wallet-granted-origins' , address ] } )
136135 } ,
137136 } )
138137 return (
@@ -150,7 +149,7 @@ const DisconnectModal = memo(function DisconnectModal({ origin, onClose }: Disco
150149 < button
151150 type = "button"
152151 className = { cx ( classes . button , classes . confirmButton ) }
153- disabled = { ! wallet }
152+ disabled = { ! address }
154153 onClick = { ( ) => onDisconnect ( ) } >
155154 < Trans > Confirm</ Trans >
156155 </ button >
@@ -161,7 +160,7 @@ const DisconnectModal = memo(function DisconnectModal({ origin, onClose }: Disco
161160 < button
162161 type = "button"
163162 className = { classes . disconnectAll }
164- disabled = { ! wallet }
163+ disabled = { ! address }
165164 onClick = { ( ) => onDisconnectAll ( ) } >
166165 < Trans > Disconnect all accounts</ Trans >
167166 </ button >
0 commit comments