platforms: put context for extension in platform extension class

3 files changed, 4 insertions, 2 deletions

diff --git a/app/scripts/background.js b/app/scripts/background.js
index a271db263..1b96d68b5 100644
--- a/app/scripts/background.js
+++ b/app/scripts/background.js
@@ -20,7 +20,6 @@ window.log = log
 log.setDefaultLevel(METAMASK_DEBUG ? 'debug' : 'warn')
 
 const platform = new ExtensionPlatform()
-platform.context = 'extension'
 const notificationManager = new NotificationManager()
 global.METAMASK_NOTIFIER = notificationManager
 
diff --git a/app/scripts/metamask-controller.js b/app/scripts/metamask-controller.js
index 3a2f3878c..4d149545a 100644
--- a/app/scripts/metamask-controller.js
+++ b/app/scripts/metamask-controller.js
@@ -196,7 +196,7 @@ module.exports = class MetamaskController extends EventEmitter {
       },
       // rpc data source
       rpcUrl: this.networkController.getCurrentRpcAddress(),
-      originHttpHeaderKey: this.platform.context === 'extension' ? 'X-Metamask-Origin' : undefined,
+      originHttpHeaderKey: this.platform.isExtension ? 'X-Metamask-Origin' : undefined,
       // account mgmt
       getAccounts: (cb) => {
         const isUnlocked = this.keyringController.memStore.getState().isUnlocked
diff --git a/app/scripts/platforms/extension.js b/app/scripts/platforms/extension.js
index 00c2aa275..d97138207 100644
--- a/app/scripts/platforms/extension.js
+++ b/app/scripts/platforms/extension.js
@@ -5,6 +5,9 @@ class ExtensionPlatform {
   //
   // Public
   //
+  get isExtension () {
+    return true
+  }
 
   reload () {
     extension.runtime.reload()