mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-12-23 09:52:26 +01:00
e6c4d63ccd
Calls to `eth.sign` are now transiently persisted in memory, and displayed in a chronological stack with pending transactions (which are still persisted to disk). This allows the user a method to sign/cancel transactions even if they miss the Chrome notification. Improved a lot of the view routing, to avoid cases where routes would show an empty account view, or transition to the accounts list when it shouldn't. Broke the transaction approval view into a couple components so messages and transactions could have their own templates.
266 lines
7.2 KiB
JavaScript
266 lines
7.2 KiB
JavaScript
const Dnode = require('dnode')
|
|
const ObjectMultiplex = require('./lib/obj-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('web3-provider-engine/zero.js')
|
|
const IdentityStore = require('./lib/idStore')
|
|
const createTxNotification = require('./lib/notifications.js').createTxNotification
|
|
const createMsgNotification = require('./lib/notifications.js').createMsgNotification
|
|
const configManager = require('./lib/config-manager-singleton')
|
|
const messageManager = require('./lib/message-manager')
|
|
const setupMultiplex = require('./lib/stream-utils.js').setupMultiplex
|
|
const HostStore = require('./lib/remote-store.js').HostStore
|
|
const Web3 = require('web3')
|
|
|
|
//
|
|
// 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
|
|
setupTrustedCommunication(portStream)
|
|
} else {
|
|
// communication with page
|
|
setupUntrustedCommunication(portStream)
|
|
}
|
|
}
|
|
|
|
function setupUntrustedCommunication(connectionStream){
|
|
// setup multiplexing
|
|
var mx = setupMultiplex(connectionStream)
|
|
// connect features
|
|
setupProviderConnection(mx.createStream('provider'))
|
|
setupPublicConfig(mx.createStream('publicConfig'))
|
|
}
|
|
|
|
function setupTrustedCommunication(connectionStream){
|
|
// setup multiplexing
|
|
var mx = setupMultiplex(connectionStream)
|
|
// connect features
|
|
setupControllerConnection(mx.createStream('controller'))
|
|
setupProviderConnection(mx.createStream('provider'))
|
|
}
|
|
|
|
//
|
|
// state and network
|
|
//
|
|
|
|
var providerConfig = configManager.getProvider()
|
|
var idStore = new IdentityStore()
|
|
|
|
var providerOpts = {
|
|
rpcUrl: configManager.getCurrentRpcAddress(),
|
|
// account mgmt
|
|
getAccounts: function(cb){
|
|
var selectedAddress = idStore.getSelectedAddress()
|
|
var result = selectedAddress ? [selectedAddress] : []
|
|
cb(null, result)
|
|
},
|
|
// tx signing
|
|
approveTransaction: addUnconfirmedTx,
|
|
signTransaction: idStore.signTransaction.bind(idStore),
|
|
// msg signing
|
|
approveMessage: addUnconfirmedMsg,
|
|
signMessage: idStore.signMessage.bind(idStore),
|
|
}
|
|
var provider = MetaMaskProvider(providerOpts)
|
|
var web3 = new Web3(provider)
|
|
idStore.web3 = web3
|
|
idStore.getNetwork(3)
|
|
|
|
// 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
|
|
}
|
|
|
|
//
|
|
// public store
|
|
//
|
|
|
|
// get init state
|
|
var initPublicState = extend(
|
|
idStoreToPublic(idStore.getState()),
|
|
configToPublic(configManager.getConfig())
|
|
)
|
|
|
|
var publicConfigStore = new HostStore(initPublicState)
|
|
|
|
// subscribe to changes
|
|
configManager.subscribe(function(state){
|
|
storeSetFromObj(publicConfigStore, configToPublic(state))
|
|
})
|
|
idStore.on('update', function(state){
|
|
storeSetFromObj(publicConfigStore, idStoreToPublic(state))
|
|
})
|
|
|
|
// idStore substate
|
|
function idStoreToPublic(state){
|
|
return {
|
|
selectedAddress: state.selectedAddress,
|
|
}
|
|
}
|
|
// config substate
|
|
function configToPublic(state){
|
|
return {
|
|
provider: state.provider,
|
|
}
|
|
}
|
|
// dump obj into store
|
|
function storeSetFromObj(store, obj){
|
|
Object.keys(obj).forEach(function(key){
|
|
store.set(key, obj[key])
|
|
})
|
|
}
|
|
|
|
|
|
|
|
// 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)
|
|
if (response.error) console.error('Error in RPC response:\n'+response.error.message)
|
|
}
|
|
try {
|
|
remoteStream.write(response)
|
|
} catch (err) {
|
|
console.error(err)
|
|
}
|
|
})
|
|
}
|
|
|
|
|
|
//
|
|
// remote features
|
|
//
|
|
|
|
function setupPublicConfig(stream){
|
|
var storeStream = publicConfigStore.createStream()
|
|
stream.pipe(storeStream).pipe(stream)
|
|
}
|
|
|
|
function setupProviderConnection(stream){
|
|
stream.on('data', onRpcRequest.bind(null, stream))
|
|
}
|
|
|
|
function setupControllerConnection(stream){
|
|
var dnode = 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),
|
|
signMessage: idStore.signMessage.bind(idStore),
|
|
cancelMessage: idStore.cancelMessage.bind(idStore),
|
|
setLocked: idStore.setLocked.bind(idStore),
|
|
clearSeedWordCache: idStore.clearSeedWordCache.bind(idStore),
|
|
exportAccount: idStore.exportAccount.bind(idStore),
|
|
})
|
|
stream.pipe(dnode).pipe(stream)
|
|
dnode.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 unconfTxs = configManager.unconfirmedTxs()
|
|
var unconfTxLen = Object.keys(unconfTxs).length
|
|
var unconfMsgs = messageManager.unconfirmedMsgs()
|
|
var unconfMsgLen = Object.keys(unconfMsgs).length
|
|
var count = unconfTxLen + unconfMsgLen
|
|
if (count) {
|
|
label = String(count)
|
|
}
|
|
chrome.browserAction.setBadgeText({ text: label })
|
|
chrome.browserAction.setBadgeBackgroundColor({ color: '#506F8B' })
|
|
}
|
|
|
|
//
|
|
// Add unconfirmed Tx + Msg
|
|
//
|
|
|
|
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),
|
|
})
|
|
}
|
|
|
|
function addUnconfirmedMsg(msgParams, cb){
|
|
var msgId = idStore.addUnconfirmedMessage(msgParams, cb)
|
|
createMsgNotification({
|
|
title: 'New Unsigned Message',
|
|
msgParams: msgParams,
|
|
confirm: idStore.approveMessage.bind(idStore, msgId, noop),
|
|
cancel: idStore.cancelMessage.bind(idStore, msgId),
|
|
})
|
|
}
|
|
|
|
//
|
|
// config
|
|
//
|
|
|
|
// called from popup
|
|
function setRpcTarget(rpcTarget){
|
|
configManager.setRpcTarget(rpcTarget)
|
|
chrome.runtime.reload()
|
|
idStore.getNetwork(3) // 3 retry attempts
|
|
}
|
|
|
|
function useEtherscanProvider() {
|
|
configManager.useEtherscanProvider()
|
|
chrome.runtime.reload()
|
|
}
|
|
|
|
// util
|
|
|
|
function noop(){}
|