packages/wallet/store/hooks/useBitcoinAddressPool.ts

   1 import { useCallback, useEffect } from 'react';
   2
   3 import { createSelector } from '@reduxjs/toolkit';
   4
   5 import { type WasmApiWalletBitcoinAddress } from '@proton/andromeda';
   6 import { baseUseSelector } from '@proton/react-redux-store';
   7 import { CacheType, createHooks } from '@proton/redux-utilities';
   8
   9 import { bitcoinAddressPoolThunk, selectBitcoinAddressPool } from '../slices';
  10
  11 const hooks = createHooks(bitcoinAddressPoolThunk, selectBitcoinAddressPool);
  12
  13 export const useGetBitcoinAddressPool = () => {
  14     const get = hooks.useGet();
  15
  16     return useCallback(
  17         async (walletId: string, walletAccountId: string) => {
  18             const results = await get({ thunkArg: [walletId, walletAccountId], cache: CacheType.None });
  19             return results[walletAccountId] ?? [];
  20         },
  21         [get]
  22     );
  23 };
  24
  25 export const useBitcoinAddressPool = (walletId: string, walletAccountId: string) => {
  26     const getBitcoinAddressPool = useGetBitcoinAddressPool();
  27     const [, loading] = hooks.useValue();
  28
  29     const bitcoinAddressPoolSelector = createSelector(
  30         selectBitcoinAddressPool,
  31         (result): [WasmApiWalletBitcoinAddress[] | undefined, boolean] => {
  32             const { value } = result;
  33
  34             const addresses = value?.[walletAccountId];
  35
  36             return [addresses, loading];
  37         }
  38     );
  39
  40     useEffect(() => {
  41         void getBitcoinAddressPool(walletId, walletAccountId);
  42     }, [walletId, walletAccountId]);
  43
  44     return baseUseSelector(bitcoinAddressPoolSelector);
  45 };