MyCrypto/common/reducers/wallet.js

// @flow
import type { WalletAction, SaveWalletAction, InitWalletAction } from 'actions/wallet';
import BaseWallet from 'libs/wallet/base';

export type State = {
    inst: ?BaseWallet,
    balance: number,
    tokens: {
        [string]: number
    }
};

const initialState: State = {
    inst: null,
    balance: 0,
    tokens: {}
};

function saveWallet(state: State, action: SaveWalletAction): State {
    return { ...state, inst: action.payload };
}

function initWallet(state: State): State {
    return { ...state, balance: 0, tokens: {} };
}

export function wallet(state: State = initialState, action: WalletAction): State {
    switch (action.type) {
        case 'WALLET_SAVE':
            return saveWallet(state, action);
        case 'WALLET_INIT':
            return initWallet(state);
        default:
            return state;
    }
}