path: root/packages/website/ts/local_storage/tracked_token_storage.ts

import * as _ from 'lodash';
import { localStorage } from 'ts/local_storage/local_storage';
import { Token, TokenByAddress, TrackedTokensByUserAddress } from 'ts/types';
import { configs } from 'ts/utils/configs';

const TRACKED_TOKENS_KEY = 'trackedTokens';
const TRACKED_TOKENS_CLEAR_KEY = 'lastClearTrackedTokensDate';

export const trackedTokenStorage = {
    // Clear trackedTokens localStorage if we've updated the config variable in an update
    // that introduced a backward incompatible change requiring the tracked tokens to be re-set
    clearIfRequired(): void {
        const lastClearFillDate = localStorage.getItemIfExists(TRACKED_TOKENS_CLEAR_KEY);
        if (lastClearFillDate !== configs.LAST_LOCAL_STORAGE_TRACKED_TOKEN_CLEARANCE_DATE) {
            localStorage.removeItem(TRACKED_TOKENS_KEY);
        }
        localStorage.setItem(TRACKED_TOKENS_CLEAR_KEY, configs.LAST_LOCAL_STORAGE_TRACKED_TOKEN_CLEARANCE_DATE);
    },
    addTrackedTokenToUser(userAddress: string, networkId: number, token: Token): void {
        const trackedTokensByUserAddress = trackedTokenStorage.getTrackedTokensByUserAddress();
        let trackedTokensByNetworkId = trackedTokensByUserAddress[userAddress];
        if (_.isUndefined(trackedTokensByNetworkId)) {
            trackedTokensByNetworkId = {};
        }
        const trackedTokens = !_.isUndefined(trackedTokensByNetworkId[networkId])
            ? trackedTokensByNetworkId[networkId]
            : [];
        trackedTokens.push(token);
        trackedTokensByNetworkId[networkId] = trackedTokens;
        trackedTokensByUserAddress[userAddress] = trackedTokensByNetworkId;
        const trackedTokensByUserAddressJSONString = JSON.stringify(trackedTokensByUserAddress);
        localStorage.setItem(TRACKED_TOKENS_KEY, trackedTokensByUserAddressJSONString);
    },
    getTrackedTokensByUserAddress(): TrackedTokensByUserAddress {
        const trackedTokensJSONString = localStorage.getItemIfExists(TRACKED_TOKENS_KEY);
        if (_.isEmpty(trackedTokensJSONString)) {
            return {};
        }
        const trackedTokensByUserAddress = JSON.parse(trackedTokensJSONString);
        return trackedTokensByUserAddress;
    },
    getTrackedTokensByAddress(userAddress: string, networkId: number): TokenByAddress {
        const trackedTokensByAddress: TokenByAddress = {};
        const trackedTokensJSONString = localStorage.getItemIfExists(TRACKED_TOKENS_KEY);
        if (_.isEmpty(trackedTokensJSONString)) {
            return trackedTokensByAddress;
        }
        const trackedTokensByUserAddress = JSON.parse(trackedTokensJSONString);
        const trackedTokensByNetworkId = trackedTokensByUserAddress[userAddress];
        if (_.isUndefined(trackedTokensByNetworkId)) {
            return trackedTokensByAddress;
        }
        const trackedTokens = trackedTokensByNetworkId[networkId];
        _.each(trackedTokens, (trackedToken: Token) => {
            trackedTokensByAddress[trackedToken.address] = trackedToken;
        });
        return trackedTokensByAddress;
    },
    removeTrackedToken(userAddress: string, networkId: number, tokenAddress: string): void {
        const trackedTokensByUserAddress = trackedTokenStorage.getTrackedTokensByUserAddress();
        const trackedTokensByNetworkId = trackedTokensByUserAddress[userAddress];
        const trackedTokens = trackedTokensByNetworkId[networkId];
        const remainingTrackedTokens = _.filter(trackedTokens, (token: Token) => {
            return token.address !== tokenAddress;
        });
        trackedTokensByNetworkId[networkId] = remainingTrackedTokens;
        trackedTokensByUserAddress[userAddress] = trackedTokensByNetworkId;
        const trackedTokensByUserAddressJSONString = JSON.stringify(trackedTokensByUserAddress);
        localStorage.setItem(TRACKED_TOKENS_KEY, trackedTokensByUserAddressJSONString);
    },
};