standardnotes-app-web/packages/snjs/mocha/lib/web_device_interface.js

/* eslint-disable no-undef */

const KEYCHAIN_STORAGE_KEY = 'keychain'

export default class WebDeviceInterface {
  environment = Environment.Web

  async getRawStorageValue(key) {
    return localStorage.getItem(key)
  }

  async getJsonParsedRawStorageValue(key) {
    const value = await this.getRawStorageValue(key)
    if (isNullOrUndefined(value)) {
      return undefined
    }
    try {
      return JSON.parse(value)
    } catch (e) {
      return value
    }
  }

  async setRawStorageValue(key, value) {
    localStorage.setItem(key, value)
  }

  async removeRawStorageValue(key) {
    localStorage.removeItem(key)
  }

  async removeAllRawStorageValues() {
    localStorage.clear()
  }

  async openDatabase(_identifier) {
    return {}
  }

  clearAllDataFromDevice() {
    localStorage.clear()
  }

  _getDatabaseKeyPrefix(identifier) {
    if (identifier) {
      return `${identifier}-item-`
    } else {
      return 'item-'
    }
  }

  _keyForPayloadId(id, identifier) {
    return `${this._getDatabaseKeyPrefix(identifier)}${id}`
  }

  async getAllDatabaseEntries(identifier) {
    const models = []
    for (const key in localStorage) {
      if (key.startsWith(this._getDatabaseKeyPrefix(identifier))) {
        models.push(JSON.parse(localStorage[key]))
      }
    }
    return models
  }

  async getDatabaseEntries(identifier, ids) {
    const models = []
    for (const id of ids) {
      const key = this._keyForPayloadId(id, identifier)
      const model = JSON.parse(localStorage[key])
      models.push(model)
    }
    return models
  }

  async getDatabaseLoadChunks(options, identifier) {
    const entries = await this.getAllDatabaseEntries(identifier)
    const {
      itemsKeyPayloads,
      keySystemRootKeyPayloads,
      keySystemItemsKeyPayloads,
      contentTypePriorityPayloads,
      remainingPayloads,
    } = GetSortedPayloadsByPriority(entries, options)

    const itemsKeysChunk = {
      entries: itemsKeyPayloads,
    }

    const keySystemRootKeysChunk = {
      entries: keySystemRootKeyPayloads,
    }

    const keySystemItemsKeysChunk = {
      entries: keySystemItemsKeyPayloads,
    }

    const contentTypePriorityChunk = {
      entries: contentTypePriorityPayloads,
    }

    const remainingPayloadsChunks = []
    for (let i = 0; i < remainingPayloads.length; i += options.batchSize) {
      remainingPayloadsChunks.push({
        entries: remainingPayloads.slice(i, i + options.batchSize),
      })
    }

    const result = {
      fullEntries: {
        itemsKeys: itemsKeysChunk,
        keySystemRootKeys: keySystemRootKeysChunk,
        keySystemItemsKeys: keySystemItemsKeysChunk,
        remainingChunks: [contentTypePriorityChunk, ...remainingPayloadsChunks],
      },
      remainingChunksItemCount: contentTypePriorityPayloads.length + remainingPayloads.length,
    }

    return result
  }

  async saveDatabaseEntry(payload, identifier) {
    localStorage.setItem(this._keyForPayloadId(payload.uuid, identifier), JSON.stringify(payload))
  }

  async saveDatabaseEntries(payloads, identifier) {
    for (const payload of payloads) {
      await this.saveDatabaseEntry(payload, identifier)
    }
  }

  async removeDatabaseEntry(id, identifier) {
    localStorage.removeItem(this._keyForPayloadId(id, identifier))
  }

  async removeAllDatabaseEntries(identifier) {
    for (const key in localStorage) {
      if (key.startsWith(this._getDatabaseKeyPrefix(identifier))) {
        delete localStorage[key]
      }
    }
  }

  /** @keychain */
  async getNamespacedKeychainValue(identifier) {
    const keychain = await this.getRawKeychainValue(identifier)
    if (!keychain) {
      return
    }
    return keychain[identifier]
  }

  async setNamespacedKeychainValue(value, identifier) {
    let keychain = await this.getRawKeychainValue()
    if (!keychain) {
      keychain = {}
    }
    localStorage.setItem(
      KEYCHAIN_STORAGE_KEY,
      JSON.stringify({
        ...keychain,
        [identifier]: value,
      }),
    )
  }

  async clearNamespacedKeychainValue(identifier) {
    const keychain = await this.getRawKeychainValue()
    if (!keychain) {
      return
    }
    delete keychain[identifier]
    localStorage.setItem(KEYCHAIN_STORAGE_KEY, JSON.stringify(keychain))
  }

  async getRawKeychainValue() {
    const keychain = localStorage.getItem(KEYCHAIN_STORAGE_KEY)
    return JSON.parse(keychain)
  }

  async clearRawKeychainValue() {
    localStorage.removeItem(KEYCHAIN_STORAGE_KEY)
  }

  async removeRawStorageValuesForIdentifier(identifier) {
    await this.removeRawStorageValue(namespacedKey(identifier, RawStorageKey.SnjsVersion))
    await this.removeRawStorageValue(namespacedKey(identifier, RawStorageKey.StorageObject))
  }

  performSoftReset() {}

  performHardReset() {}

  isDeviceDestroyed() {
    return false
  }

  deinit() {}
}