From fc3a53ec7b2af538d5e9748173fb11b5a09a4e84 Mon Sep 17 00:00:00 2001
From: kumavis <aaron@kumavis.me>
Date: Thu, 26 Jan 2017 20:18:28 -0800
Subject: background - stream disconnection fix

---
 app/scripts/background.js | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

(limited to 'app/scripts/background.js')

diff --git a/app/scripts/background.js b/app/scripts/background.js
index 18882e5d5..744a0caee 100644
--- a/app/scripts/background.js
+++ b/app/scripts/background.js
@@ -122,17 +122,16 @@ function setupController (initState) {
   // remote features
   //
 
-  function setupControllerConnection (stream) {
-    controller.stream = stream
+  function setupControllerConnection (outStream) {
     var api = controller.getApi()
     var dnode = Dnode(api)
-    stream.pipe(dnode).pipe(stream)
+    outStream.pipe(dnode).pipe(outStream)
     dnode.on('remote', (remote) => {
       // push updates to popup
       var sendUpdate = remote.sendUpdate.bind(remote)
       controller.on('update', sendUpdate)
       // teardown on disconnect
-      eos(stream, () => {
+      eos(outStream, () => {
         controller.removeListener('update', sendUpdate)
         popupIsOpen = false
       })
-- 
cgit 


From 2113979be7ee257c2650916c2c48f68c6390ef1f Mon Sep 17 00:00:00 2001
From: kumavis <aaron@kumavis.me>
Date: Thu, 26 Jan 2017 20:52:46 -0800
Subject: metamask controller - adopt setup*Communication from background

---
 app/scripts/background.js | 52 +++++++++--------------------------------------
 1 file changed, 10 insertions(+), 42 deletions(-)

(limited to 'app/scripts/background.js')

diff --git a/app/scripts/background.js b/app/scripts/background.js
index 744a0caee..da9c4f24b 100644
--- a/app/scripts/background.js
+++ b/app/scripts/background.js
@@ -1,6 +1,5 @@
 const urlUtil = require('url')
-const Dnode = require('dnode')
-const eos = require('end-of-stream')
+const endOfStream = require('end-of-stream')
 const asyncQ = require('async-q')
 const pipe = require('pump')
 const LocalStorageStore = require('obs-store/lib/localStorage')
@@ -10,7 +9,6 @@ const migrations = require('./migrations/')
 const PortStream = require('./lib/port-stream.js')
 const notification = require('./lib/notifications.js')
 const messageManager = require('./lib/message-manager')
-const setupMultiplex = require('./lib/stream-utils.js').setupMultiplex
 const MetamaskController = require('./metamask-controller')
 const extension = require('./lib/extension')
 const firstTimeState = require('./first-time-state')
@@ -93,51 +91,21 @@ function setupController (initState) {
     var portStream = new PortStream(remotePort)
     if (isMetaMaskInternalProcess) {
       // communication with popup
-      popupIsOpen = remotePort.name === 'popup'
-      setupTrustedCommunication(portStream, 'MetaMask', remotePort.name)
+      popupIsOpen = popupIsOpen || (remotePort.name === 'popup')
+      controller.setupTrustedCommunication(portStream, 'MetaMask', remotePort.name)
+      // record popup as closed
+      if (remotePort.name === 'popup') {
+        endOfStream(portStream, () => {
+          popupIsOpen = false
+        })
+      }
     } else {
       // communication with page
       var originDomain = urlUtil.parse(remotePort.sender.url).hostname
-      setupUntrustedCommunication(portStream, originDomain)
+      controller.setupUntrustedCommunication(portStream, originDomain)
     }
   }
 
-  function setupUntrustedCommunication (connectionStream, originDomain) {
-    // setup multiplexing
-    var mx = setupMultiplex(connectionStream)
-    // connect features
-    controller.setupProviderConnection(mx.createStream('provider'), originDomain)
-    controller.setupPublicConfig(mx.createStream('publicConfig'))
-  }
-
-  function setupTrustedCommunication (connectionStream, originDomain) {
-    // setup multiplexing
-    var mx = setupMultiplex(connectionStream)
-    // connect features
-    setupControllerConnection(mx.createStream('controller'))
-    controller.setupProviderConnection(mx.createStream('provider'), originDomain)
-  }
-
-  //
-  // remote features
-  //
-
-  function setupControllerConnection (outStream) {
-    var api = controller.getApi()
-    var dnode = Dnode(api)
-    outStream.pipe(dnode).pipe(outStream)
-    dnode.on('remote', (remote) => {
-      // push updates to popup
-      var sendUpdate = remote.sendUpdate.bind(remote)
-      controller.on('update', sendUpdate)
-      // teardown on disconnect
-      eos(outStream, () => {
-        controller.removeListener('update', sendUpdate)
-        popupIsOpen = false
-      })
-    })
-  }
-
   //
   // User Interface setup
   //
-- 
cgit