Merge pull request #9951 from alexstrat/fix-chrome-storage

Fix chrome storage access scope

lib/renderer/chrome-api.js

@@ -173,7 +173,7 @@ exports.injectTo = function (extensionId, isBackgroundPage, context) {
     onMessage: chrome.runtime.onMessage
   }
 
-  chrome.storage = require('./extensions/storage')
+  chrome.storage = require('./extensions/storage').setup(extensionId)
 
   chrome.pageAction = {
     show () {},

lib/renderer/extensions/storage.js

@@ -1,88 +1,131 @@
-const getStorage = (storageType) => {
-  const data = window.localStorage.getItem(`__chrome.storage.${storageType}__`)
-  if (data != null) {
-    return JSON.parse(data)
-  } else {
-    return {}
-  }
+const fs = require('fs')
+const path = require('path')
+const { remote } = require('electron')
+const { app } = remote
+
+const getChromeStoragePath = (storageType, extensionId) => {
+  return path.join(
+    app.getPath('userData'), `/Chrome Storage/${extensionId}-${storageType}.json`)
 }
 
-const setStorage = (storageType, storage) => {
-  const json = JSON.stringify(storage)
-  window.localStorage.setItem(`__chrome.storage.${storageType}__`, json)
-}
-
-const scheduleCallback = (items, callback) => {
-  setTimeout(function () {
-    callback(items)
+const mkdirp = (dir, callback) => {
+  fs.mkdir(dir, (error) => {
+    if (error && error.code === 'ENOENT') {
+      mkdirp(path.dirname(dir), (error) => {
+        if (!error) {
+          mkdirp(dir, callback)
+        }
+      })
+    } else if (error && error.code === 'EEXIST') {
+      callback(null)
+    } else {
+      callback(error)
+    }
   })
 }
 
-const getStorageManager = (storageType) => {
+const readChromeStorageFile = (storageType, extensionId, cb) => {
+  const filePath = getChromeStoragePath(storageType, extensionId)
+  fs.readFile(filePath, 'utf8', (err, data) => {
+    if (err && err.code === 'ENOENT') {
+      return cb(null, null)
+    }
+    cb(err, data)
+  })
+}
+
+const writeChromeStorageFile = (storageType, extensionId, data, cb) => {
+  const filePath = getChromeStoragePath(storageType, extensionId)
+
+  mkdirp(path.dirname(filePath), err => {
+    if (err) { /* we just ignore the errors of mkdir or mkdirp */ }
+    fs.writeFile(filePath, data, cb)
+  })
+}
+
+const getStorage = (storageType, extensionId, cb) => {
+  readChromeStorageFile(storageType, extensionId, (err, data) => {
+    if (err) throw err
+    if (!cb) throw new TypeError('No callback provided')
+
+    if (data !== null) {
+      cb(JSON.parse(data))
+    } else {
+      cb({})
+    }
+  })
+}
+
+const setStorage = (storageType, extensionId, storage, cb) => {
+  const json = JSON.stringify(storage)
+  writeChromeStorageFile(storageType, extensionId, json, err => {
+    if (err) throw err
+    if (cb) cb()
+  })
+}
+
+const getStorageManager = (storageType, extensionId) => {
   return {
     get (keys, callback) {
-      const storage = getStorage(storageType)
-      if (keys == null) return scheduleCallback(storage, callback)
+      getStorage(storageType, extensionId, storage => {
+        if (keys == null) return callback(storage)
 
-      let defaults = {}
-      switch (typeof keys) {
-        case 'string':
-          keys = [keys]
-          break
-        case 'object':
-          if (!Array.isArray(keys)) {
-            defaults = keys
-            keys = Object.keys(keys)
-          }
-          break
-      }
-      if (keys.length === 0) return scheduleCallback({}, callback)
+        let defaults = {}
+        switch (typeof keys) {
+          case 'string':
+            keys = [keys]
+            break
+          case 'object':
+            if (!Array.isArray(keys)) {
+              defaults = keys
+              keys = Object.keys(keys)
+            }
+            break
+        }
+        if (keys.length === 0) return callback({})
 
-      let items = {}
-      keys.forEach(function (key) {
-        var value = storage[key]
-        if (value == null) value = defaults[key]
-        items[key] = value
+        let items = {}
+        keys.forEach(function (key) {
+          var value = storage[key]
+          if (value == null) value = defaults[key]
+          items[key] = value
+        })
+        callback(items)
       })
-      scheduleCallback(items, callback)
     },
 
     set (items, callback) {
-      const storage = getStorage(storageType)
+      getStorage(storageType, extensionId, storage => {
+        Object.keys(items).forEach(function (name) {
+          storage[name] = items[name]
+        })
 
-      Object.keys(items).forEach(function (name) {
-        storage[name] = items[name]
+        setStorage(storageType, extensionId, storage, callback)
       })
-
-      setStorage(storageType, storage)
-
-      setTimeout(callback)
     },
 
     remove (keys, callback) {
-      const storage = getStorage(storageType)
+      getStorage(storageType, extensionId, storage => {
+        if (!Array.isArray(keys)) {
+          keys = [keys]
+        }
+        keys.forEach(function (key) {
+          delete storage[key]
+        })
 
-      if (!Array.isArray(keys)) {
-        keys = [keys]
-      }
-      keys.forEach(function (key) {
-        delete storage[key]
+        setStorage(storageType, extensionId, storage, callback)
       })
-
-      setStorage(storageType, storage)
-
-      setTimeout(callback)
     },
 
     clear (callback) {
-      setStorage(storageType, {})
-
-      setTimeout(callback)
+      setStorage(storageType, extensionId, {}, callback)
     }
   }
 }
 
 module.exports = {
-  sync: getStorageManager('sync'),
-  local: getStorageManager('local')
+  setup: extensionId => ({
+    sync: getStorageManager('sync', extensionId),
+    local: getStorageManager('local', extensionId)
+  })
 }