import { DeterministicWalletAction, DeterministicWalletData } from 'actions/deterministicWallets'; import { TypeKeys } from 'actions/deterministicWallets/constants'; export interface State { wallets: DeterministicWalletData[]; desiredToken: string | undefined; } export const INITIAL_STATE: State = { wallets: [], desiredToken: '' }; export function deterministicWallets( state: State = INITIAL_STATE, action: DeterministicWalletAction ): State { switch (action.type) { case TypeKeys.DW_SET_WALLETS: return { ...state, wallets: action.payload }; case TypeKeys.DW_SET_DESIRED_TOKEN: return { ...state, desiredToken: action.payload }; case TypeKeys.DW_UPDATE_WALLET: return { ...state, wallets: updateWalletValues(state.wallets, action.payload) }; default: return state; } } function updateWalletValues(wallets, newWallet) { return wallets.map(w => { if (w.address === newWallet.address) { return { ...w, ...newWallet }; } return w; }); }