LocalForageStorage

LocalForageStorage.ts

import localforage from 'localforage';
import { Auth } from 'aws-amplify';

export class LocalForageStorage {
  syncPromise = null;
  memoryStore = {};
  asyncStore = localforage.createInstance({
    name: 'amplifyStore'
  });

  setItem(key: string, value: any) {
    this.asyncStore.setItem(key, value);
    this.memoryStore[key] = value;
    return this.memoryStore[key];
  }

  getItem(key: string) {
    return Object.prototype.hasOwnProperty.call(this.memoryStore, key) ? this.memoryStore[key] : undefined;
  }

  removeItem(key: string) {
    this.asyncStore.removeItem(key);
    return delete this.memoryStore[key];
  }

  clear() {
    this.asyncStore.clear();
    return {};
  }

  sync(): Promise<void> {
    if (!this.syncPromise) {
      this.syncPromise = this.asyncStore.iterate((val: string, key) => {
        this.memoryStore[key] = val;
      });
    }
    return this.syncPromise;
  }
}

Auth.configure({
  storage: new LocalForageStorage()
});