vue-devtools/packages/shared-utils/src/storage.ts

import { target } from './env'

// If we can, we use the browser extension API to store data
// it's async though, so we synchronize changes from an intermediate
// storageData object
const useStorage = typeof target.chrome !== 'undefined' && typeof target.chrome.storage !== 'undefined'

let storageData = null

export function initStorage(): Promise<void> {
  return new Promise((resolve) => {
    if (useStorage) {
      target.chrome.storage.local.get(null, (result) => {
        storageData = result
        resolve()
      })
    }
    else {
      storageData = {}
      resolve()
    }
  })
}

export function getStorage(key: string, defaultValue: any = null) {
  checkStorage()
  if (useStorage) {
    return getDefaultValue(storageData[key], defaultValue)
  }
  else {
    try {
      return getDefaultValue(JSON.parse(localStorage.getItem(key)), defaultValue)
    }
    catch (e) {}
  }
}

export function setStorage(key: string, val: any) {
  checkStorage()
  if (useStorage) {
    storageData[key] = val
    target.chrome.storage.local.set({ [key]: val })
  }
  else {
    try {
      localStorage.setItem(key, JSON.stringify(val))
    }
    catch (e) {}
  }
}

export function removeStorage(key: string) {
  checkStorage()
  if (useStorage) {
    delete storageData[key]
    target.chrome.storage.local.remove([key])
  }
  else {
    try {
      localStorage.removeItem(key)
    }
    catch (e) {}
  }
}

export function clearStorage() {
  checkStorage()
  if (useStorage) {
    storageData = {}
    target.chrome.storage.local.clear()
  }
  else {
    try {
      localStorage.clear()
    }
    catch (e) {}
  }
}

function checkStorage() {
  if (!storageData) {
    throw new Error('Storage wasn\'t initialized with \'init()\'')
  }
}

function getDefaultValue(value, defaultValue) {
  if (value == null) {
    return defaultValue
  }
  return value
}