1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-10-22 19:26:13 +02:00
metamask-extension/app/scripts/controllers/detect-tokens.js

94 lines
2.7 KiB
JavaScript
Raw Normal View History

2018-06-27 22:29:24 +02:00
const contracts = require('eth-contract-metadata')
const {
MAINNET,
} = require('./network/enums')
// By default, poll every 3 minutes
2018-06-28 04:18:06 +02:00
const DEFAULT_INTERVAL = 180 * 1000
2018-06-27 22:29:24 +02:00
/**
* A controller that polls for token exchange
* rates based on a user's current token list
*/
class DetectTokensController {
/**
* Creates a DetectTokensController
*
* @param {Object} [config] - Options to configure controller
*/
constructor ({ interval = DEFAULT_INTERVAL, preferences, network } = {}) {
this.preferences = preferences
this.interval = interval
this.network = network
2018-06-28 01:54:43 +02:00
this.contracts = contracts
2018-06-27 22:29:24 +02:00
}
/**
2018-06-28 01:54:43 +02:00
* For each token in eth-contract-metada, find check selectedAddress balance.
2018-06-27 22:29:24 +02:00
*
*/
async exploreNewTokens () {
if (!this.isActive) { return }
2018-06-28 01:54:43 +02:00
if (this._network.getState().provider.type !== MAINNET) { return }
2018-06-27 22:29:24 +02:00
let detectedTokenAddress, token
2018-06-28 01:54:43 +02:00
for (const address in this.contracts) {
const contract = this.contracts[address]
if (contract.erc20 && !(address in this.tokens)) {
2018-06-27 22:29:24 +02:00
detectedTokenAddress = await this.fetchContractAccountBalance(address)
if (detectedTokenAddress) {
2018-06-28 01:54:43 +02:00
token = this.contracts[detectedTokenAddress]
2018-06-27 22:29:24 +02:00
this._preferences.addToken(detectedTokenAddress, token['symbol'], token['decimals'])
}
}
// etherscan restriction, 5 request/second, lazy scan
setTimeout(() => {}, 200)
}
}
/**
* Find if selectedAddress has tokens with contract in contractAddress.
*
* @param {string} contractAddress Hex address of the token contract to explore.
* @returns {string} Contract address to be added to tokens.
*
*/
async fetchContractAccountBalance (contractAddress) {
const address = this._preferences.store.getState().selectedAddress
const response = await fetch(`https://api.etherscan.io/api?module=account&action=tokenbalance&contractaddress=${contractAddress}&address=${address}&tag=latest&apikey=NCKS6GTY41KPHWRJB62ES1MDNRBIT174PV`)
const parsedResponse = await response.json()
if (parsedResponse.result !== '0') {
return contractAddress
}
return null
}
/**
* @type {Number}
*/
set interval (interval) {
this._handle && clearInterval(this._handle)
if (!interval) { return }
this._handle = setInterval(() => { this.exploreNewTokens() }, interval)
}
/**
* @type {Object}
*/
set preferences (preferences) {
if (!preferences) { return }
this._preferences = preferences
2018-06-28 01:54:43 +02:00
this.tokens = preferences.store.getState().tokens
2018-06-27 22:29:24 +02:00
}
/**
* @type {Object}
*/
set network (network) {
if (!network) { return }
this._network = network
}
}
module.exports = DetectTokensController