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 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 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) { 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 callback({}) let items = {} keys.forEach(function (key) { var value = storage[key] if (value == null) value = defaults[key] items[key] = value }) callback(items) }) }, set (items, callback) { getStorage(storageType, extensionId, storage => { Object.keys(items).forEach(function (name) { storage[name] = items[name] }) setStorage(storageType, extensionId, storage, callback) }) }, remove (keys, callback) { getStorage(storageType, extensionId, storage => { if (!Array.isArray(keys)) { keys = [keys] } keys.forEach(function (key) { delete storage[key] }) setStorage(storageType, extensionId, storage, callback) }) }, clear (callback) { setStorage(storageType, extensionId, {}, callback) } } } module.exports = { setup: extensionId => ({ sync: getStorageManager('sync', extensionId), local: getStorageManager('local', extensionId) }) }