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)) }