86 lines
2.7 KiB
TypeScript
86 lines
2.7 KiB
TypeScript
import { ipcRendererInternal } from '@electron/internal/renderer/ipc-renderer-internal'
|
|
|
|
const getStorage = (storageType: string, extensionId: number, callback: Function) => {
|
|
if (typeof callback !== 'function') throw new TypeError('No callback provided')
|
|
|
|
ipcRendererInternal.invoke<string>('CHROME_STORAGE_READ', storageType, extensionId)
|
|
.then(data => {
|
|
if (data !== null) {
|
|
callback(JSON.parse(data))
|
|
} else {
|
|
// Disabled due to false positive in StandardJS
|
|
// eslint-disable-next-line standard/no-callback-literal
|
|
callback({})
|
|
}
|
|
})
|
|
}
|
|
|
|
const setStorage = (storageType: string, extensionId: number, storage: Record<string, any>, callback: Function) => {
|
|
const json = JSON.stringify(storage)
|
|
ipcRendererInternal.invoke('CHROME_STORAGE_WRITE', storageType, extensionId, json)
|
|
.then(() => {
|
|
if (callback) callback()
|
|
})
|
|
}
|
|
|
|
const getStorageManager = (storageType: string, extensionId: number) => {
|
|
return {
|
|
get (keys: string[], callback: Function) {
|
|
getStorage(storageType, extensionId, (storage: Record<string, any>) => {
|
|
if (keys == null) return callback(storage)
|
|
|
|
let defaults: Record<string, any> = {}
|
|
switch (typeof keys) {
|
|
case 'string':
|
|
keys = [keys]
|
|
break
|
|
case 'object':
|
|
if (!Array.isArray(keys)) {
|
|
defaults = keys
|
|
keys = Object.keys(keys)
|
|
}
|
|
break
|
|
}
|
|
|
|
// Disabled due to false positive in StandardJS
|
|
// eslint-disable-next-line standard/no-callback-literal
|
|
if (keys.length === 0) return callback({})
|
|
|
|
const items: Record<string, any> = {}
|
|
keys.forEach((key: string) => {
|
|
let value = storage[key]
|
|
if (value == null) value = defaults[key]
|
|
items[key] = value
|
|
})
|
|
callback(items)
|
|
})
|
|
},
|
|
|
|
set (items: Record<string, any>, callback: Function) {
|
|
getStorage(storageType, extensionId, (storage: Record<string, any>) => {
|
|
Object.keys(items).forEach(name => { storage[name] = items[name] })
|
|
setStorage(storageType, extensionId, storage, callback)
|
|
})
|
|
},
|
|
|
|
remove (keys: string[], callback: Function) {
|
|
getStorage(storageType, extensionId, (storage: Record<string, any>) => {
|
|
if (!Array.isArray(keys)) keys = [keys]
|
|
keys.forEach((key: string) => {
|
|
delete storage[key]
|
|
})
|
|
|
|
setStorage(storageType, extensionId, storage, callback)
|
|
})
|
|
},
|
|
|
|
clear (callback: Function) {
|
|
setStorage(storageType, extensionId, {}, callback)
|
|
}
|
|
}
|
|
}
|
|
|
|
export const setup = (extensionId: number) => ({
|
|
sync: getStorageManager('sync', extensionId),
|
|
local: getStorageManager('local', extensionId)
|
|
})
|