import localForage from 'localforage';
import { PrexStorage } from './PrexStorage';

export class LocalForageStorage implements PrexStorage {
  private store: LocalForage;

  constructor() {
    this.store = localForage.createInstance({
      name: 'prex',
      version: 1.0,
      storeName: 'prex_address', // Should be alphanumeric, with underscores.
      description: 'address store',
    });
  }

  async getItem<T>(key: string): Promise<T | null> {
    return this.store.getItem<T>(key);
  }

  async setItem<T>(key: string, value: T): Promise<void> {
    await this.store.setItem(key, value);
  }

  async removeItem(key: string): Promise<void> {
    await this.store.removeItem(key);
  }

  async getItemFromSessionStorage<T>(key: string): Promise<T | null> {
    const encoded = sessionStorage.getItem(key);

    if (!encoded) {
      return null;
    }

    return JSON.parse(encoded) as T;
  }

  async setItemToSessionStorage<T>(key: string, value: T): Promise<void> {
    sessionStorage.setItem(
      key,
      JSON.stringify(value, (_key, value) => {
        if (typeof value === 'bigint') {
          return value.toString();
        }
        return value;
      })
    );
  }

  async removeItemFromSessionStorage(key: string): Promise<void> {
    sessionStorage.removeItem(key);
  }
}