import { safeJsonParse, safeJsonStringify } from "@connext/utils";
import localStorage from "localStorage";

import { storeDefaults } from "../constants";
import { KeyValueStorage } from "../types";

export class WrappedLocalStorage implements KeyValueStorage {
  private localStorage: Storage = localStorage;

  constructor(
    private readonly prefix: string = storeDefaults.PREFIX,
    private readonly separator: string = storeDefaults.SEPARATOR,
  ) {}

  init(): Promise<void> {
    return Promise.resolve();
  }

  close(): Promise<void> {
    return Promise.resolve();
  }

  async getItem<T>(key: string): Promise<T | undefined> {
    const item = this.localStorage.getItem(`${this.prefix}${this.separator}${key}`);
    return safeJsonParse(item);
  }

  async setItem<T>(key: string, value: T): Promise<void> {
    this.localStorage.setItem(`${this.prefix}${this.separator}${key}`, safeJsonStringify(value));
  }

  async removeItem(key: string): Promise<void> {
    this.localStorage.removeItem(`${this.prefix}${this.separator}${key}`);
  }

  async getKeys(): Promise<string[]> {
    const relevantKeys = Object.keys(this.localStorage).filter((key) =>
      key.startsWith(this.prefix),
    );
    return relevantKeys.map((key) => key.split(`${this.prefix}${this.separator}`)[1]);
  }

  async getEntries(): Promise<[string, any][]> {
    return Object.entries(this.localStorage)
      .filter(([name, _]) => name.startsWith(this.prefix))
      .map(([name, value]) => [
        name.replace(`${this.prefix}${this.separator}`, ""),
        safeJsonParse(value),
      ]);
  }

  getKey(...args: string[]) {
    return args.join(this.separator);
  }
}

export default WrappedLocalStorage;