const Dnode = require('dnode') const Multiplex = require('multiplex') const eos = require('end-of-stream') const combineStreams = require('pumpify') const extend = require('xtend') const EthStore = require('eth-store') const PortStream = require('./lib/port-stream.js') const MetaMaskProvider = require('./lib/zero.js') const IdentityStore = require('./lib/idStore') const createTxNotification = require('./lib/tx-notification.js') const configManager = require('./lib/config-manager-singleton') const jsonParseStream = require('./lib/stream-utils.js').jsonParseStream const jsonStringifyStream = require('./lib/stream-utils.js').jsonStringifyStream // // connect to other contexts // chrome.runtime.onConnect.addListener(connectRemote) function connectRemote(remotePort){ var isMetaMaskInternalProcess = (remotePort.name === 'popup') var portStream = new PortStream(remotePort) if (isMetaMaskInternalProcess) { // communication with popup handleInternalCommunication(portStream) } else { // communication with page handleEthRpcRequestStream(portStream) } } function handleEthRpcRequestStream(stream){ stream.on('data', onRpcRequest.bind(null, stream)) } // // state and network // var providerConfig = configManager.getProvider() var idStore = new IdentityStore() var providerOpts = { rpcUrl: configManager.getCurrentRpcAddress(), getAccounts: function(cb){ var selectedAddress = idStore.getSelectedAddress() var result = selectedAddress ? [selectedAddress] : [] cb(null, result) }, approveTransaction: addUnconfirmedTx, signTransaction: idStore.signTransaction.bind(idStore), } var provider = MetaMaskProvider(providerOpts) // log new blocks provider.on('block', function(block){ console.log('BLOCK CHANGED:', '#'+block.number.toString('hex'), '0x'+block.hash.toString('hex')) }) var ethStore = new EthStore(provider) idStore.setStore(ethStore) function getState(){ var state = extend( ethStore.getState(), idStore.getState(), configManager.getConfig() ) return state } // handle rpc requests function onRpcRequest(remoteStream, payload){ // console.log('MetaMaskPlugin - incoming payload:', payload) provider.sendAsync(payload, function onPayloadHandled(err, response){ // provider engine errors are included in response objects if (!payload.isMetamaskInternal) console.log('MetaMaskPlugin - RPC complete:', payload, '->', response) try { remoteStream.write(response) } catch (err) { console.error(err) } }) } // // popup integration // function handleInternalCommunication(portStream){ // setup multiplexing var mx = Multiplex() portStream.pipe(mx).pipe(portStream) mx.on('error', function(err) { console.error(err) // portStream.destroy() }) portStream.on('error', function(err) { console.error(err) mx.destroy() }) var dnodeStream = mx.createSharedStream('dnode') var providerStream = combineStreams.obj( jsonStringifyStream(), mx.createSharedStream('provider'), jsonParseStream() ) linkDnode(dnodeStream) handleEthRpcRequestStream(providerStream) } function linkDnode(stream){ var connection = Dnode({ getState: function(cb){ cb(null, getState()) }, setRpcTarget: setRpcTarget, useEtherscanProvider: useEtherscanProvider, // forward directly to idStore createNewVault: idStore.createNewVault.bind(idStore), recoverFromSeed: idStore.recoverFromSeed.bind(idStore), submitPassword: idStore.submitPassword.bind(idStore), setSelectedAddress: idStore.setSelectedAddress.bind(idStore), approveTransaction: idStore.approveTransaction.bind(idStore), cancelTransaction: idStore.cancelTransaction.bind(idStore), setLocked: idStore.setLocked.bind(idStore), clearSeedWordCache: idStore.clearSeedWordCache.bind(idStore), exportAccount: idStore.exportAccount.bind(idStore), }) stream.pipe(connection).pipe(stream) connection.on('remote', function(remote){ // push updates to popup ethStore.on('update', sendUpdate) idStore.on('update', sendUpdate) // teardown on disconnect eos(stream, function unsubscribe(){ ethStore.removeListener('update', sendUpdate) }) function sendUpdate(){ var state = getState() remote.sendUpdate(state) } }) } // // plugin badge text // idStore.on('update', updateBadge) function updateBadge(state){ var label = '' var count = Object.keys(state.unconfTxs).length if (count) { label = String(count) } chrome.browserAction.setBadgeText({ text: label }) chrome.browserAction.setBadgeBackgroundColor({ color: '#506F8B' }) } // // Add unconfirmed Tx // function addUnconfirmedTx(txParams, cb){ var txId = idStore.addUnconfirmedTransaction(txParams, cb) createTxNotification({ title: 'New Unsigned Transaction', txParams: txParams, confirm: idStore.approveTransaction.bind(idStore, txId, noop), cancel: idStore.cancelTransaction.bind(idStore, txId), }) } // // config // // called from popup function setRpcTarget(rpcTarget){ configManager.setRpcTarget(rpcTarget) chrome.runtime.reload() } function useEtherscanProvider() { configManager.useEtherscanProvider() chrome.runtime.reload() } // util function noop(){}