pinafore/routes/_database/databaseLifecycle.js

import {
  META_STORE,
  STATUS_TIMELINES_STORE,
  STATUSES_STORE,
  ACCOUNTS_STORE,
  RELATIONSHIPS_STORE,
  NOTIFICATIONS_STORE,
  NOTIFICATION_TIMELINES_STORE,
  PINNED_STATUSES_STORE,
  TIMESTAMP,
  REBLOG_ID,
  THREADS_STORE,
  STATUS_ID,
  USERNAME_LOWERCASE
} from './constants'
import { addKnownInstance, deleteKnownInstance } from './knownInstances'

if (process.browser) {
  require('indexeddb-getall-shim') // needed for Edge
}

const openReqs = {}
const databaseCache = {}

const DB_VERSION_INITIAL = 9
const DB_VERSION_SEARCH_ACCOUNTS = 10
const DB_VERSION_CURRENT = 10

function createDatabase (instanceName) {
  return new Promise((resolve, reject) => {
    let req = indexedDB.open(instanceName, DB_VERSION_CURRENT)
    openReqs[instanceName] = req
    req.onerror = reject
    req.onblocked = () => {
      console.log('idb blocked')
    }
    req.onupgradeneeded = (e) => {
      let db = req.result
      let tx = e.currentTarget.transaction

      function createObjectStore (name, init, indexes) {
        let store = init
          ? db.createObjectStore(name, init)
          : db.createObjectStore(name)
        if (indexes) {
          Object.keys(indexes).forEach(indexKey => {
            store.createIndex(indexKey, indexes[indexKey])
          })
        }
      }

      if (e.oldVersion < DB_VERSION_INITIAL) {
        createObjectStore(STATUSES_STORE, { keyPath: 'id' }, {
          [TIMESTAMP]: TIMESTAMP,
          [REBLOG_ID]: REBLOG_ID
        })
        createObjectStore(STATUS_TIMELINES_STORE, null, {
          'statusId': ''
        })
        createObjectStore(NOTIFICATIONS_STORE, { keyPath: 'id' }, {
          [TIMESTAMP]: TIMESTAMP,
          [STATUS_ID]: STATUS_ID
        })
        createObjectStore(NOTIFICATION_TIMELINES_STORE, null, {
          'notificationId': ''
        })
        createObjectStore(ACCOUNTS_STORE, { keyPath: 'id' }, {
          [TIMESTAMP]: TIMESTAMP
        })
        createObjectStore(RELATIONSHIPS_STORE, { keyPath: 'id' }, {
          [TIMESTAMP]: TIMESTAMP
        })
        createObjectStore(THREADS_STORE, null, {
          'statusId': ''
        })
        createObjectStore(PINNED_STATUSES_STORE, null, {
          'statusId': ''
        })
        createObjectStore(META_STORE)
      }
      if (e.oldVersion < DB_VERSION_SEARCH_ACCOUNTS) {
        tx.objectStore(ACCOUNTS_STORE)
          .createIndex(USERNAME_LOWERCASE, USERNAME_LOWERCASE)
      }
    }
    req.onsuccess = () => resolve(req.result)
  })
}

export async function getDatabase (instanceName) {
  if (!instanceName) {
    throw new Error('instanceName is undefined in getDatabase()')
  }
  if (!databaseCache[instanceName]) {
    databaseCache[instanceName] = await createDatabase(instanceName)
    await addKnownInstance(instanceName)
  }
  return databaseCache[instanceName]
}

export async function dbPromise (db, storeName, readOnlyOrReadWrite, cb) {
  return new Promise((resolve, reject) => {
    const tx = db.transaction(storeName, readOnlyOrReadWrite)
    let store = typeof storeName === 'string'
      ? tx.objectStore(storeName)
      : storeName.map(name => tx.objectStore(name))
    let res
    cb(store, (result) => {
      res = result
    })

    tx.oncomplete = () => resolve(res)
    tx.onerror = () => reject(tx.error)
  })
}

export function deleteDatabase (instanceName) {
  return new Promise((resolve, reject) => {
    // close any open requests
    let openReq = openReqs[instanceName]
    if (openReq && openReq.result) {
      openReq.result.close()
    }
    delete openReqs[instanceName]
    delete databaseCache[instanceName]
    let req = indexedDB.deleteDatabase(instanceName)
    req.onsuccess = () => resolve()
    req.onerror = () => reject(req.error)
    req.onblocked = () => console.error(`database ${instanceName} blocked`)
  }).then(() => deleteKnownInstance(instanceName))
}