import { AssetBuyerError } from '@0x/asset-buyer'; import { AssetProxyId, ObjectMap } from '@0x/types'; import * as _ from 'lodash'; import { assetDataNetworkMapping } from '../data/asset_data_network_mapping'; import { Asset, AssetMetaData, ERC20Asset, Network, ZeroExInstantError } from '../types'; export const assetUtils = { createAssetsFromAssetDatas: ( assetDatas: string[], assetMetaDataMap: ObjectMap, network: Network, ): Asset[] => { const arrayOfAssetOrUndefined = _.map(assetDatas, assetData => assetUtils.createAssetFromAssetDataIfExists(assetData, assetMetaDataMap, network), ); return _.compact(arrayOfAssetOrUndefined); }, createAssetFromAssetDataIfExists: ( assetData: string, assetMetaDataMap: ObjectMap, network: Network, ): Asset | undefined => { const metaData = assetUtils.getMetaDataIfExists(assetData, assetMetaDataMap, network); if (_.isUndefined(metaData)) { return; } return { assetData, metaData, }; }, createAssetFromAssetDataOrThrow: ( assetData: string, assetMetaDataMap: ObjectMap, network: Network, ): Asset => { return { assetData, metaData: assetUtils.getMetaDataOrThrow(assetData, assetMetaDataMap, network), }; }, getMetaDataOrThrow: (assetData: string, metaDataMap: ObjectMap, network: Network): AssetMetaData => { const metaDataIfExists = assetUtils.getMetaDataIfExists(assetData, metaDataMap, network); if (_.isUndefined(metaDataIfExists)) { throw new Error(ZeroExInstantError.AssetMetaDataNotAvailable); } return metaDataIfExists; }, getMetaDataIfExists: ( assetData: string, metaDataMap: ObjectMap, network: Network, ): AssetMetaData | undefined => { let mainnetAssetData: string | undefined = assetData; if (network !== Network.Mainnet) { const mainnetAssetDataIfExists = assetUtils.getAssociatedAssetDataIfExists( assetData.toLowerCase(), network, ); // Just so we don't fail in the case where we are on a non-mainnet network, // but pass in a valid mainnet assetData. mainnetAssetData = mainnetAssetDataIfExists || assetData; } if (_.isUndefined(mainnetAssetData)) { return; } const metaData = metaDataMap[mainnetAssetData.toLowerCase()]; if (_.isUndefined(metaData)) { return; } return metaData; }, bestNameForAsset: (asset?: Asset, defaultName: string = '???'): string => { if (_.isUndefined(asset)) { return defaultName; } const metaData = asset.metaData; switch (metaData.assetProxyId) { case AssetProxyId.ERC20: return metaData.symbol.toUpperCase(); case AssetProxyId.ERC721: return metaData.name; } }, formattedSymbolForAsset: (asset?: ERC20Asset, defaultName: string = '???'): string => { if (_.isUndefined(asset)) { return defaultName; } const symbol = asset.metaData.symbol; if (symbol.length <= 5) { return symbol; } return `${symbol.slice(0, 3)}…`; }, getAssociatedAssetDataIfExists: (assetData: string, network: Network): string | undefined => { const assetDataGroupIfExists = _.find(assetDataNetworkMapping, value => value[network] === assetData); if (_.isUndefined(assetDataGroupIfExists)) { return; } return assetDataGroupIfExists[Network.Mainnet]; }, getERC20AssetsFromAssets: (assets: Asset[]): ERC20Asset[] => { const erc20sOrUndefined = _.map( assets, asset => (asset.metaData.assetProxyId === AssetProxyId.ERC20 ? (asset as ERC20Asset) : undefined), ); return _.compact(erc20sOrUndefined); }, assetBuyerErrorMessage: (asset: ERC20Asset, error: Error): string | undefined => { if (error.message === AssetBuyerError.InsufficientAssetLiquidity) { const assetName = assetUtils.bestNameForAsset(asset, 'of this asset'); return `Not enough ${assetName} available`; } else if (error.message === AssetBuyerError.InsufficientZrxLiquidity) { return 'Not enough ZRX available'; } else if ( error.message === AssetBuyerError.StandardRelayerApiError || error.message.startsWith(AssetBuyerError.AssetUnavailable) ) { const assetName = assetUtils.bestNameForAsset(asset, 'This asset'); return `${assetName} is currently unavailable`; } return undefined; }, };