mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-11-27 04:46:10 +01:00
44effa0d91
We don't need to store the current UI type as a global. We're already using the `getEnvironmentType` helper function throughout the UI, so we'd might as well use that instead of this global state.
46 lines
1.5 KiB
JavaScript
46 lines
1.5 KiB
JavaScript
import querystring from 'querystring'
|
|
import dnode from 'dnode'
|
|
import { EventEmitter } from 'events'
|
|
import PortStream from 'extension-port-stream'
|
|
import extension from 'extensionizer'
|
|
import { setupMultiplex } from './lib/stream-utils.js'
|
|
import { getEnvironmentType } from './lib/util'
|
|
import ExtensionPlatform from './platforms/extension'
|
|
|
|
document.addEventListener('DOMContentLoaded', start)
|
|
|
|
function start () {
|
|
const hash = window.location.hash.substring(1)
|
|
const suspect = querystring.parse(hash)
|
|
|
|
document.getElementById('csdbLink').href = `https://cryptoscamdb.org/search`
|
|
|
|
global.platform = new ExtensionPlatform()
|
|
|
|
const extensionPort = extension.runtime.connect({ name: getEnvironmentType() })
|
|
const connectionStream = new PortStream(extensionPort)
|
|
const mx = setupMultiplex(connectionStream)
|
|
setupControllerConnection(mx.createStream('controller'), (err, metaMaskController) => {
|
|
if (err) {
|
|
return
|
|
}
|
|
|
|
const continueLink = document.getElementById('unsafe-continue')
|
|
continueLink.addEventListener('click', () => {
|
|
metaMaskController.whitelistPhishingDomain(suspect.hostname)
|
|
window.location.href = suspect.href
|
|
})
|
|
})
|
|
}
|
|
|
|
function setupControllerConnection (connectionStream, cb) {
|
|
const eventEmitter = new EventEmitter()
|
|
const metaMaskControllerDnode = dnode({
|
|
sendUpdate (state) {
|
|
eventEmitter.emit('update', state)
|
|
},
|
|
})
|
|
connectionStream.pipe(metaMaskControllerDnode).pipe(connectionStream)
|
|
metaMaskControllerDnode.once('remote', (backgroundConnection) => cb(null, backgroundConnection))
|
|
}
|