diff options
author | Leonid Logvinov <logvinov.leon@gmail.com> | 2018-07-18 22:38:16 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-07-18 22:38:16 +0800 |
commit | 00d1622b3f087943e750f1b3f6ee6ceab7e80285 (patch) | |
tree | 44789fff181ec24846728d885ba541c23f682321 /packages/order-watcher/src/fetchers/asset_balance_and_proxy_allowance_fetcher.ts | |
parent | e2fb49a8f871fd4b631bb01bb641c632b7c19435 (diff) | |
parent | a1acf19ff3c1948214dde05a91eb3fbeac728d2c (diff) | |
download | dexon-sol-tools-00d1622b3f087943e750f1b3f6ee6ceab7e80285.tar dexon-sol-tools-00d1622b3f087943e750f1b3f6ee6ceab7e80285.tar.gz dexon-sol-tools-00d1622b3f087943e750f1b3f6ee6ceab7e80285.tar.bz2 dexon-sol-tools-00d1622b3f087943e750f1b3f6ee6ceab7e80285.tar.lz dexon-sol-tools-00d1622b3f087943e750f1b3f6ee6ceab7e80285.tar.xz dexon-sol-tools-00d1622b3f087943e750f1b3f6ee6ceab7e80285.tar.zst dexon-sol-tools-00d1622b3f087943e750f1b3f6ee6ceab7e80285.zip |
Merge pull request #887 from 0xProject/feature/order-watcher-v2
Order watcher v2
Diffstat (limited to 'packages/order-watcher/src/fetchers/asset_balance_and_proxy_allowance_fetcher.ts')
-rw-r--r-- | packages/order-watcher/src/fetchers/asset_balance_and_proxy_allowance_fetcher.ts | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/packages/order-watcher/src/fetchers/asset_balance_and_proxy_allowance_fetcher.ts b/packages/order-watcher/src/fetchers/asset_balance_and_proxy_allowance_fetcher.ts new file mode 100644 index 000000000..a1de22a5e --- /dev/null +++ b/packages/order-watcher/src/fetchers/asset_balance_and_proxy_allowance_fetcher.ts @@ -0,0 +1,74 @@ +// tslint:disable:no-unnecessary-type-assertion +import { BlockParamLiteral, ERC20TokenWrapper, ERC721TokenWrapper } from '@0xproject/contract-wrappers'; +import { AbstractBalanceAndProxyAllowanceFetcher, assetDataUtils } from '@0xproject/order-utils'; +import { AssetProxyId, ERC20AssetData, ERC721AssetData } from '@0xproject/types'; +import { BigNumber } from '@0xproject/utils'; + +export class AssetBalanceAndProxyAllowanceFetcher implements AbstractBalanceAndProxyAllowanceFetcher { + private readonly _erc20Token: ERC20TokenWrapper; + private readonly _erc721Token: ERC721TokenWrapper; + private readonly _stateLayer: BlockParamLiteral; + constructor(erc20Token: ERC20TokenWrapper, erc721Token: ERC721TokenWrapper, stateLayer: BlockParamLiteral) { + this._erc20Token = erc20Token; + this._erc721Token = erc721Token; + this._stateLayer = stateLayer; + } + public async getBalanceAsync(assetData: string, userAddress: string): Promise<BigNumber> { + const decodedAssetData = assetDataUtils.decodeAssetDataOrThrow(assetData); + if (decodedAssetData.assetProxyId === AssetProxyId.ERC20) { + const decodedERC20AssetData = decodedAssetData as ERC20AssetData; + const balance = await this._erc20Token.getBalanceAsync(decodedERC20AssetData.tokenAddress, userAddress, { + defaultBlock: this._stateLayer, + }); + return balance; + } else { + const decodedERC721AssetData = decodedAssetData as ERC721AssetData; + const tokenOwner = await this._erc721Token.getOwnerOfAsync( + decodedERC721AssetData.tokenAddress, + decodedERC721AssetData.tokenId, + { + defaultBlock: this._stateLayer, + }, + ); + const balance = tokenOwner === userAddress ? new BigNumber(1) : new BigNumber(0); + return balance; + } + } + public async getProxyAllowanceAsync(assetData: string, userAddress: string): Promise<BigNumber> { + const decodedAssetData = assetDataUtils.decodeAssetDataOrThrow(assetData); + if (decodedAssetData.assetProxyId === AssetProxyId.ERC20) { + const decodedERC20AssetData = decodedAssetData as ERC20AssetData; + const proxyAllowance = await this._erc20Token.getProxyAllowanceAsync( + decodedERC20AssetData.tokenAddress, + userAddress, + { + defaultBlock: this._stateLayer, + }, + ); + return proxyAllowance; + } else { + const decodedERC721AssetData = decodedAssetData as ERC721AssetData; + + const isApprovedForAll = await this._erc721Token.isProxyApprovedForAllAsync( + decodedERC721AssetData.tokenAddress, + userAddress, + { + defaultBlock: this._stateLayer, + }, + ); + if (isApprovedForAll) { + return new BigNumber(this._erc20Token.UNLIMITED_ALLOWANCE_IN_BASE_UNITS); + } else { + const isApproved = await this._erc721Token.isProxyApprovedAsync( + decodedERC721AssetData.tokenAddress, + decodedERC721AssetData.tokenId, + { + defaultBlock: this._stateLayer, + }, + ); + const proxyAllowance = isApproved ? new BigNumber(1) : new BigNumber(0); + return proxyAllowance; + } + } + } +} |