Iuri Matias
5 years ago
parent
b7acc307be
commit
42b8a5be63
@ -1,361 +0,0 @@ |
||||
const async = require('async') |
||||
const { BN, privateToAddress, isValidPrivate, stripHexPrefix, toChecksumAddress } = require('ethereumjs-util') |
||||
const crypto = require('crypto') |
||||
const { EventEmitter } = require('events') |
||||
|
||||
const remixLib = require('remix-lib') |
||||
|
||||
const TxRunner = remixLib.execution.txRunner |
||||
const EventManager = remixLib.EventManager |
||||
const Web3 = require('web3') |
||||
|
||||
const { resultToRemixTx } = require('./txResultHelper') |
||||
|
||||
class UniversalDApp { |
||||
|
||||
constructor (config, executionContext) { |
||||
this.events = new EventEmitter() |
||||
this.event = new EventManager() |
||||
// has a default for now for backwards compatability
|
||||
this.executionContext = executionContext |
||||
this.config = config |
||||
|
||||
this.txRunner = new TxRunner({}, { |
||||
config: config, |
||||
detectNetwork: (cb) => { |
||||
this.executionContext.detectNetwork(cb) |
||||
}, |
||||
personalMode: () => { |
||||
return this.executionContext.getProvider() === 'web3' ? this.config.get('settings/personal-mode') : false |
||||
} |
||||
}, this.executionContext) |
||||
this.accounts = {} |
||||
this.executionContext.event.register('contextChanged', this.resetEnvironment.bind(this)) |
||||
} |
||||
|
||||
// TODO : event should be triggered by Udapp instead of TxListener
|
||||
/** Listen on New Transaction. (Cannot be done inside constructor because txlistener doesn't exist yet) */ |
||||
startListening (txlistener) { |
||||
txlistener.event.register('newTransaction', (tx) => { |
||||
this.events.emit('newTransaction', tx) |
||||
}) |
||||
} |
||||
|
||||
resetEnvironment () { |
||||
this.accounts = {} |
||||
if (this.executionContext.isVM()) { |
||||
this._addAccount('3cd7232cd6f3fc66a57a6bedc1a8ed6c228fff0a327e169c2bcc5e869ed49511', '0x56BC75E2D63100000') |
||||
this._addAccount('2ac6c190b09897cd8987869cc7b918cfea07ee82038d492abce033c75c1b1d0c', '0x56BC75E2D63100000') |
||||
this._addAccount('dae9801649ba2d95a21e688b56f77905e5667c44ce868ec83f82e838712a2c7a', '0x56BC75E2D63100000') |
||||
this._addAccount('d74aa6d18aa79a05f3473dd030a97d3305737cbc8337d940344345c1f6b72eea', '0x56BC75E2D63100000') |
||||
this._addAccount('71975fbf7fe448e004ac7ae54cad0a383c3906055a65468714156a07385e96ce', '0x56BC75E2D63100000') |
||||
} |
||||
// TODO: most params here can be refactored away in txRunner
|
||||
this.txRunner = new TxRunner(this.accounts, { |
||||
// TODO: only used to check value of doNotShowTransactionConfirmationAgain property
|
||||
config: this.config, |
||||
// TODO: to refactor, TxRunner already has access to executionContext
|
||||
detectNetwork: (cb) => { |
||||
this.executionContext.detectNetwork(cb) |
||||
}, |
||||
personalMode: () => { |
||||
return this.executionContext.getProvider() === 'web3' ? this.config.get('settings/personal-mode') : false |
||||
} |
||||
}, this.executionContext) |
||||
this.txRunner.event.register('transactionBroadcasted', (txhash) => { |
||||
this.executionContext.detectNetwork((error, network) => { |
||||
if (error || !network) return |
||||
this.event.trigger('transactionBroadcasted', [txhash, network.name]) |
||||
}) |
||||
}) |
||||
} |
||||
|
||||
resetAPI (transactionContextAPI) { |
||||
this.transactionContextAPI = transactionContextAPI |
||||
} |
||||
|
||||
/** |
||||
* Create a VM Account |
||||
* @param {{privateKey: string, balance: string}} newAccount The new account to create |
||||
*/ |
||||
createVMAccount (newAccount) { |
||||
const { privateKey, balance } = newAccount |
||||
if (this.executionContext.getProvider() !== 'vm') { |
||||
throw new Error('plugin API does not allow creating a new account through web3 connection. Only vm mode is allowed') |
||||
} |
||||
this._addAccount(privateKey, balance) |
||||
const privKey = Buffer.from(privateKey, 'hex') |
||||
return '0x' + privateToAddress(privKey).toString('hex') |
||||
} |
||||
|
||||
newAccount (password, passwordPromptCb, cb) { |
||||
if (!this.executionContext.isVM()) { |
||||
if (!this.config.get('settings/personal-mode')) { |
||||
return cb('Not running in personal mode') |
||||
} |
||||
passwordPromptCb((passphrase) => { |
||||
this.executionContext.web3().personal.newAccount(passphrase, cb) |
||||
}) |
||||
} else { |
||||
let privateKey |
||||
do { |
||||
privateKey = crypto.randomBytes(32) |
||||
} while (!isValidPrivate(privateKey)) |
||||
this._addAccount(privateKey, '0x56BC75E2D63100000') |
||||
cb(null, '0x' + privateToAddress(privateKey).toString('hex')) |
||||
} |
||||
} |
||||
|
||||
/** Add an account to the list of account (only for Javascript VM) */ |
||||
_addAccount (privateKey, balance) { |
||||
if (!this.executionContext.isVM()) { |
||||
throw new Error('_addAccount() cannot be called in non-VM mode') |
||||
} |
||||
|
||||
if (this.accounts) { |
||||
privateKey = Buffer.from(privateKey, 'hex') |
||||
const address = privateToAddress(privateKey) |
||||
|
||||
// FIXME: we don't care about the callback, but we should still make this proper
|
||||
let stateManager = this.executionContext.vm().stateManager |
||||
stateManager.getAccount(address, (error, account) => { |
||||
if (error) return console.log(error) |
||||
account.balance = balance || '0xf00000000000000001' |
||||
stateManager.putAccount(address, account, function cb (error) { |
||||
if (error) console.log(error) |
||||
}) |
||||
}) |
||||
|
||||
this.accounts[toChecksumAddress('0x' + address.toString('hex'))] = { privateKey, nonce: 0 } |
||||
} |
||||
} |
||||
|
||||
/** Return the list of accounts */ |
||||
getAccounts (cb) { |
||||
return new Promise((resolve, reject) => { |
||||
const provider = this.executionContext.getProvider() |
||||
switch (provider) { |
||||
case 'vm': { |
||||
if (!this.accounts) { |
||||
if (cb) cb('No accounts?') |
||||
reject('No accounts?') |
||||
return |
||||
} |
||||
if (cb) cb(null, Object.keys(this.accounts)) |
||||
resolve(Object.keys(this.accounts)) |
||||
} |
||||
break |
||||
case 'web3': { |
||||
if (this.config.get('settings/personal-mode')) { |
||||
return this.executionContext.web3().personal.getListAccounts((error, accounts) => { |
||||
if (cb) cb(error, accounts) |
||||
if (error) return reject(error) |
||||
resolve(accounts) |
||||
}) |
||||
} else { |
||||
this.executionContext.web3().eth.getAccounts((error, accounts) => { |
||||
if (cb) cb(error, accounts) |
||||
if (error) return reject(error) |
||||
resolve(accounts) |
||||
}) |
||||
} |
||||
} |
||||
break |
||||
case 'injected': { |
||||
this.executionContext.web3().eth.getAccounts((error, accounts) => { |
||||
if (cb) cb(error, accounts) |
||||
if (error) return reject(error) |
||||
resolve(accounts) |
||||
}) |
||||
} |
||||
} |
||||
}) |
||||
} |
||||
|
||||
/** Get the balance of an address */ |
||||
getBalance (address, cb) { |
||||
address = stripHexPrefix(address) |
||||
|
||||
if (!this.executionContext.isVM()) { |
||||
this.executionContext.web3().eth.getBalance(address, (err, res) => { |
||||
if (err) { |
||||
cb(err) |
||||
} else { |
||||
cb(null, res.toString(10)) |
||||
} |
||||
}) |
||||
} else { |
||||
if (!this.accounts) { |
||||
return cb('No accounts?') |
||||
} |
||||
|
||||
this.executionContext.vm().stateManager.getAccount(Buffer.from(address, 'hex'), (err, res) => { |
||||
if (err) { |
||||
cb('Account not found') |
||||
} else { |
||||
cb(null, new BN(res.balance).toString(10)) |
||||
} |
||||
}) |
||||
} |
||||
} |
||||
|
||||
/** Get the balance of an address, and convert wei to ether */ |
||||
getBalanceInEther (address, callback) { |
||||
this.getBalance(address, (error, balance) => { |
||||
if (error) { |
||||
callback(error) |
||||
} else { |
||||
// callback(null, this.executionContext.web3().fromWei(balance, 'ether'))
|
||||
callback(null, Web3.utils.fromWei(balance.toString(10), 'ether')) |
||||
} |
||||
}) |
||||
} |
||||
|
||||
pendingTransactionsCount () { |
||||
return Object.keys(this.txRunner.pendingTxs).length |
||||
} |
||||
|
||||
/** |
||||
* deploy the given contract |
||||
* |
||||
* @param {String} data - data to send with the transaction ( return of txFormat.buildData(...) ). |
||||
* @param {Function} callback - callback. |
||||
*/ |
||||
createContract (data, confirmationCb, continueCb, promptCb, callback) { |
||||
this.runTx({data: data, useCall: false}, confirmationCb, continueCb, promptCb, (error, txResult) => { |
||||
// see universaldapp.js line 660 => 700 to check possible values of txResult (error case)
|
||||
callback(error, txResult) |
||||
}) |
||||
} |
||||
|
||||
/** |
||||
* call the current given contract |
||||
* |
||||
* @param {String} to - address of the contract to call. |
||||
* @param {String} data - data to send with the transaction ( return of txFormat.buildData(...) ). |
||||
* @param {Object} funAbi - abi definition of the function to call. |
||||
* @param {Function} callback - callback. |
||||
*/ |
||||
callFunction (to, data, funAbi, confirmationCb, continueCb, promptCb, callback) { |
||||
const useCall = funAbi.stateMutability === 'view' || funAbi.stateMutability === 'pure' |
||||
this.runTx({to, data, useCall}, confirmationCb, continueCb, promptCb, (error, txResult) => { |
||||
// see universaldapp.js line 660 => 700 to check possible values of txResult (error case)
|
||||
callback(error, txResult) |
||||
}) |
||||
} |
||||
|
||||
context () { |
||||
return (this.executionContext.isVM() ? 'memory' : 'blockchain') |
||||
} |
||||
|
||||
/** |
||||
* This function send a tx only to javascript VM or testnet, will return an error for the mainnet |
||||
* SHOULD BE TAKEN CAREFULLY! |
||||
* |
||||
* @param {Object} tx - transaction. |
||||
*/ |
||||
sendTransaction (tx) { |
||||
return new Promise((resolve, reject) => { |
||||
this.executionContext.detectNetwork((error, network) => { |
||||
if (error) return reject(error) |
||||
if (network.name === 'Main' && network.id === '1') { |
||||
return reject(new Error('It is not allowed to make this action against mainnet')) |
||||
} |
||||
this.silentRunTx(tx, (error, result) => { |
||||
if (error) return reject(error) |
||||
try { |
||||
resolve(resultToRemixTx(result)) |
||||
} catch (e) { |
||||
reject(e) |
||||
} |
||||
}) |
||||
}) |
||||
}) |
||||
} |
||||
|
||||
/** |
||||
* This function send a tx without alerting the user (if mainnet or if gas estimation too high). |
||||
* SHOULD BE TAKEN CAREFULLY! |
||||
* |
||||
* @param {Object} tx - transaction. |
||||
* @param {Function} callback - callback. |
||||
*/ |
||||
silentRunTx (tx, cb) { |
||||
this.txRunner.rawRun( |
||||
tx, |
||||
(network, tx, gasEstimation, continueTxExecution, cancelCb) => { continueTxExecution() }, |
||||
(error, continueTxExecution, cancelCb) => { if (error) { cb(error) } else { continueTxExecution() } }, |
||||
(okCb, cancelCb) => { okCb() }, |
||||
cb |
||||
) |
||||
} |
||||
|
||||
runTx (args, confirmationCb, continueCb, promptCb, cb) { |
||||
const self = this |
||||
async.waterfall([ |
||||
function getGasLimit (next) { |
||||
if (self.transactionContextAPI.getGasLimit) { |
||||
return self.transactionContextAPI.getGasLimit(next) |
||||
} |
||||
next(null, 3000000) |
||||
}, |
||||
function queryValue (gasLimit, next) { |
||||
if (args.value) { |
||||
return next(null, args.value, gasLimit) |
||||
} |
||||
if (args.useCall || !self.transactionContextAPI.getValue) { |
||||
return next(null, 0, gasLimit) |
||||
} |
||||
self.transactionContextAPI.getValue(function (err, value) { |
||||
next(err, value, gasLimit) |
||||
}) |
||||
}, |
||||
function getAccount (value, gasLimit, next) { |
||||
if (args.from) { |
||||
return next(null, args.from, value, gasLimit) |
||||
} |
||||
if (self.transactionContextAPI.getAddress) { |
||||
return self.transactionContextAPI.getAddress(function (err, address) { |
||||
next(err, address, value, gasLimit) |
||||
}) |
||||
} |
||||
self.getAccounts(function (err, accounts) { |
||||
let address = accounts[0] |
||||
|
||||
if (err) return next(err) |
||||
if (!address) return next('No accounts available') |
||||
if (self.executionContext.isVM() && !self.accounts[address]) { |
||||
return next('Invalid account selected') |
||||
} |
||||
next(null, address, value, gasLimit) |
||||
}) |
||||
}, |
||||
function runTransaction (fromAddress, value, gasLimit, next) { |
||||
const tx = { to: args.to, data: args.data.dataHex, useCall: args.useCall, from: fromAddress, value: value, gasLimit: gasLimit, timestamp: args.data.timestamp } |
||||
const payLoad = { funAbi: args.data.funAbi, funArgs: args.data.funArgs, contractBytecode: args.data.contractBytecode, contractName: args.data.contractName, contractABI: args.data.contractABI, linkReferences: args.data.linkReferences } |
||||
let timestamp = Date.now() |
||||
if (tx.timestamp) { |
||||
timestamp = tx.timestamp |
||||
} |
||||
|
||||
self.event.trigger('initiatingTransaction', [timestamp, tx, payLoad]) |
||||
self.txRunner.rawRun(tx, confirmationCb, continueCb, promptCb, |
||||
function (error, result) { |
||||
let eventName = (tx.useCall ? 'callExecuted' : 'transactionExecuted') |
||||
self.event.trigger(eventName, [error, tx.from, tx.to, tx.data, tx.useCall, result, timestamp, payLoad]) |
||||
|
||||
if (error && (typeof (error) !== 'string')) { |
||||
if (error.message) error = error.message |
||||
else { |
||||
try { error = 'error: ' + JSON.stringify(error) } catch (e) {} |
||||
} |
||||
} |
||||
next(error, result) |
||||
} |
||||
) |
||||
} |
||||
], cb) |
||||
} |
||||
} |
||||
|
||||
module.exports = UniversalDApp |
||||
Loading…
Reference in new issue