Refetch wallets on change

Author: ekzyis

wallets/client/hooks/query.js

@@ -24,6 +24,7 @@ import { protocolTestSendPayment } from '@/wallets/client/protocols'
 import { timeoutSignal } from '@/lib/time'
 import { WALLET_SEND_PAYMENT_TIMEOUT_MS } from '@/lib/constants'
 import { useToast } from '@/components/toast'
+import { useMe } from '@/components/me'
 
 export function useWalletsQuery () {
   const query = useQuery(WALLETS)
@@ -38,6 +39,8 @@ export function useWalletsQuery () {
     ).then(wallets => setWallets(wallets))
   }, [query.data, decryptWallet])
 
+  useRefetchOnChange(query.refetch)
+
   return useMemo(() => ({
     ...query,
     // pretend query is still loading until we've decrypted the wallet
@@ -46,6 +49,13 @@ export function useWalletsQuery () {
   }), [query, wallets])
 }
 
+function useRefetchOnChange (refetch) {
+  const { me } = useMe()
+  useEffect(() => {
+    refetch()
+  }, [refetch, me?.privates?.walletsUpdatedAt])
+}
+
 export function useWalletQuery ({ id, name }) {
   const query = useQuery(WALLET, { variables: { id, name } })
   const [wallet, setWallet] = useState(null)