2016-05-25 17:27:49 -07:00
|
|
|
const urlUtil = require('url')
|
2017-01-26 20:52:46 -08:00
|
|
|
const endOfStream = require('end-of-stream')
|
2017-01-24 19:47:00 -08:00
|
|
|
const pipe = require('pump')
|
2017-09-13 10:28:29 -07:00
|
|
|
const log = require('loglevel')
|
|
|
|
const extension = require('extensionizer')
|
2017-01-24 19:47:00 -08:00
|
|
|
const LocalStorageStore = require('obs-store/lib/localStorage')
|
|
|
|
const storeTransform = require('obs-store/lib/transform')
|
2017-03-30 18:33:19 -07:00
|
|
|
const ExtensionPlatform = require('./platforms/extension')
|
2017-01-12 01:17:05 -08:00
|
|
|
const Migrator = require('./lib/migrator/')
|
2017-01-24 19:47:00 -08:00
|
|
|
const migrations = require('./migrations/')
|
2016-05-22 18:02:27 -07:00
|
|
|
const PortStream = require('./lib/port-stream.js')
|
2017-03-30 18:33:19 -07:00
|
|
|
const NotificationManager = require('./lib/notification-manager.js')
|
2016-06-24 15:55:11 -07:00
|
|
|
const MetamaskController = require('./metamask-controller')
|
2017-01-11 22:47:56 -08:00
|
|
|
const firstTimeState = require('./first-time-state')
|
2016-06-24 13:05:21 -07:00
|
|
|
|
2016-06-24 15:52:56 -07:00
|
|
|
const STORAGE_KEY = 'metamask-config'
|
2016-11-01 11:21:46 -07:00
|
|
|
const METAMASK_DEBUG = 'GULP_METAMASK_DEBUG'
|
2017-01-14 21:29:46 -08:00
|
|
|
|
2017-02-23 14:23:45 -08:00
|
|
|
window.log = log
|
|
|
|
log.setDefaultLevel(METAMASK_DEBUG ? 'debug' : 'warn')
|
|
|
|
|
2017-03-30 18:33:19 -07:00
|
|
|
const platform = new ExtensionPlatform()
|
|
|
|
const notificationManager = new NotificationManager()
|
|
|
|
global.METAMASK_NOTIFIER = notificationManager
|
|
|
|
|
2017-01-11 22:47:56 -08:00
|
|
|
let popupIsOpen = false
|
2016-08-11 19:44:59 -07:00
|
|
|
|
2017-01-12 01:17:05 -08:00
|
|
|
// state persistence
|
|
|
|
const diskStore = new LocalStorageStore({ storageKey: STORAGE_KEY })
|
|
|
|
|
|
|
|
// initialization flow
|
2017-09-13 10:28:29 -07:00
|
|
|
initialize().catch(log.error)
|
2017-05-12 13:09:23 -07:00
|
|
|
|
2017-05-28 11:18:07 -07:00
|
|
|
async function initialize () {
|
2017-05-12 13:09:23 -07:00
|
|
|
const initState = await loadStateFromPersistence()
|
|
|
|
await setupController(initState)
|
2017-09-13 10:28:29 -07:00
|
|
|
log.debug('MetaMask initialization complete.')
|
2017-05-12 13:09:23 -07:00
|
|
|
}
|
2017-01-11 19:04:19 -08:00
|
|
|
|
2017-01-11 22:47:56 -08:00
|
|
|
//
|
|
|
|
// State and Persistence
|
|
|
|
//
|
|
|
|
|
2017-05-12 13:09:23 -07:00
|
|
|
async function loadStateFromPersistence () {
|
2017-01-12 01:17:05 -08:00
|
|
|
// migrations
|
2017-04-26 21:05:45 -07:00
|
|
|
const migrator = new Migrator({ migrations })
|
2017-05-12 13:09:23 -07:00
|
|
|
// read from disk
|
|
|
|
let versionedData = diskStore.getState() || migrator.generateInitialState(firstTimeState)
|
|
|
|
// migrate data
|
|
|
|
versionedData = await migrator.migrateData(versionedData)
|
|
|
|
// write to disk
|
|
|
|
diskStore.putState(versionedData)
|
|
|
|
// return just the data
|
|
|
|
return versionedData.data
|
2017-01-11 22:47:56 -08:00
|
|
|
}
|
|
|
|
|
2017-01-12 01:17:05 -08:00
|
|
|
function setupController (initState) {
|
|
|
|
//
|
|
|
|
// MetaMask Controller
|
|
|
|
//
|
2017-01-11 22:47:56 -08:00
|
|
|
|
2017-01-12 01:17:05 -08:00
|
|
|
const controller = new MetamaskController({
|
|
|
|
// User confirmation callbacks:
|
|
|
|
showUnconfirmedMessage: triggerUi,
|
|
|
|
unlockAccountMessage: triggerUi,
|
|
|
|
showUnapprovedTx: triggerUi,
|
|
|
|
// initial state
|
|
|
|
initState,
|
2017-03-30 18:33:19 -07:00
|
|
|
// platform specific api
|
|
|
|
platform,
|
2017-01-12 01:17:05 -08:00
|
|
|
})
|
2017-01-12 02:24:33 -08:00
|
|
|
global.metamaskController = controller
|
2017-01-11 22:47:56 -08:00
|
|
|
|
2017-01-12 01:17:05 -08:00
|
|
|
// setup state persistence
|
2017-01-24 19:47:00 -08:00
|
|
|
pipe(
|
|
|
|
controller.store,
|
|
|
|
storeTransform(versionifyData),
|
|
|
|
diskStore
|
|
|
|
)
|
|
|
|
|
2017-04-26 21:05:45 -07:00
|
|
|
function versionifyData (state) {
|
|
|
|
const versionedData = diskStore.getState()
|
2017-01-12 14:40:04 -08:00
|
|
|
versionedData.data = state
|
|
|
|
return versionedData
|
2017-01-24 19:47:00 -08:00
|
|
|
}
|
2017-01-12 01:17:05 -08:00
|
|
|
|
|
|
|
//
|
|
|
|
// connect to other contexts
|
|
|
|
//
|
|
|
|
|
|
|
|
extension.runtime.onConnect.addListener(connectRemote)
|
|
|
|
function connectRemote (remotePort) {
|
2017-08-02 14:26:10 -07:00
|
|
|
const isMetaMaskInternalProcess = remotePort.name === 'popup' || remotePort.name === 'notification'
|
|
|
|
const portStream = new PortStream(remotePort)
|
2017-01-12 01:17:05 -08:00
|
|
|
if (isMetaMaskInternalProcess) {
|
|
|
|
// communication with popup
|
2017-01-26 20:52:46 -08:00
|
|
|
popupIsOpen = popupIsOpen || (remotePort.name === 'popup')
|
2017-08-02 14:26:10 -07:00
|
|
|
controller.setupTrustedCommunication(portStream, 'MetaMask')
|
2017-01-26 20:52:46 -08:00
|
|
|
// record popup as closed
|
|
|
|
if (remotePort.name === 'popup') {
|
|
|
|
endOfStream(portStream, () => {
|
|
|
|
popupIsOpen = false
|
|
|
|
})
|
|
|
|
}
|
2017-01-12 01:17:05 -08:00
|
|
|
} else {
|
|
|
|
// communication with page
|
2017-08-02 14:26:10 -07:00
|
|
|
const originDomain = urlUtil.parse(remotePort.sender.url).hostname
|
2017-01-26 20:52:46 -08:00
|
|
|
controller.setupUntrustedCommunication(portStream, originDomain)
|
2017-01-12 01:17:05 -08:00
|
|
|
}
|
|
|
|
}
|
2016-09-06 10:24:31 -07:00
|
|
|
|
2017-01-12 01:17:05 -08:00
|
|
|
//
|
|
|
|
// User Interface setup
|
|
|
|
//
|
|
|
|
|
2017-01-21 10:06:50 -08:00
|
|
|
updateBadge()
|
2017-09-26 16:52:08 -07:00
|
|
|
controller.txController.on('update:badge', updateBadge)
|
2017-01-27 16:11:59 -08:00
|
|
|
controller.messageManager.on('updateBadge', updateBadge)
|
2017-06-20 15:38:23 -07:00
|
|
|
controller.personalMessageManager.on('updateBadge', updateBadge)
|
2017-01-12 01:17:05 -08:00
|
|
|
|
|
|
|
// plugin badge text
|
|
|
|
function updateBadge () {
|
|
|
|
var label = ''
|
2017-08-02 15:58:05 -07:00
|
|
|
var unapprovedTxCount = controller.txController.getUnapprovedTxCount()
|
2017-01-27 16:11:59 -08:00
|
|
|
var unapprovedMsgCount = controller.messageManager.unapprovedMsgCount
|
2017-06-20 15:38:23 -07:00
|
|
|
var unapprovedPersonalMsgs = controller.personalMessageManager.unapprovedPersonalMsgCount
|
2017-09-29 09:24:08 -07:00
|
|
|
var unapprovedTypedMsgs = controller.typedMessageManager.unapprovedTypedMessagesCount
|
|
|
|
var count = unapprovedTxCount + unapprovedMsgCount + unapprovedPersonalMsgs + unapprovedTypedMsgs
|
2017-01-12 01:17:05 -08:00
|
|
|
if (count) {
|
|
|
|
label = String(count)
|
|
|
|
}
|
|
|
|
extension.browserAction.setBadgeText({ text: label })
|
|
|
|
extension.browserAction.setBadgeBackgroundColor({ color: '#506F8B' })
|
|
|
|
}
|
2016-04-15 13:04:17 -07:00
|
|
|
|
2017-01-12 01:17:05 -08:00
|
|
|
return Promise.resolve()
|
2016-02-10 17:44:46 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
//
|
2017-01-12 01:17:05 -08:00
|
|
|
// Etc...
|
2016-02-10 17:44:46 -08:00
|
|
|
//
|
|
|
|
|
2017-01-11 22:47:56 -08:00
|
|
|
// popup trigger
|
|
|
|
function triggerUi () {
|
2017-03-31 18:41:51 -07:00
|
|
|
if (!popupIsOpen) notificationManager.showPopup()
|
2017-01-11 22:47:56 -08:00
|
|
|
}
|
2015-12-18 22:05:16 -08:00
|
|
|
|
2017-01-11 22:47:56 -08:00
|
|
|
// On first install, open a window to MetaMask website to how-it-works.
|
|
|
|
extension.runtime.onInstalled.addListener(function (details) {
|
|
|
|
if ((details.reason === 'install') && (!METAMASK_DEBUG)) {
|
|
|
|
extension.tabs.create({url: 'https://metamask.io/#how-it-works'})
|
|
|
|
}
|
2017-01-27 16:11:59 -08:00
|
|
|
})
|