@0xpolygonid/js-sdk/src/kms/store/local-storage-key-store.ts

SDK to work with Polygon ID

import { AbstractPrivateKeyStore } from './abstract-key-store';
import { KmsKeyId } from './types';

/**
 * Allows storing keys in the local storage of the browser
 * (NOT ENCRYPTED: DO NOT USE IN THE PRODUCTION)
 *
 * @public
 * @class LocalStoragePrivateKeyStore
 * @implements implements AbstractPrivateKeyStore interface
 */
export class LocalStoragePrivateKeyStore implements AbstractPrivateKeyStore {
  static readonly storageKey = 'keystore';

  /**
   * get all keys
   *
   * @abstract
   * @returns `Promise<{ alias: string; key: string }[]>`
   */
  list(): Promise<{ alias: string; key: string }[]> {
    const dataStr = localStorage.getItem(LocalStoragePrivateKeyStore.storageKey);
    if (!dataStr) {
      throw new Error('no key under given alias');
    }
    const data = JSON.parse(dataStr);
    return data.map((i: { id: string; value: string }) => ({ alias: i.id, key: i.value }));
  }
  /**
   * Gets key from the local storage
   *
   * @param {{ alias: string }} args
   * @returns hex string
   */
  async get(args: { alias: string }): Promise<string> {
    const dataStr = localStorage.getItem(LocalStoragePrivateKeyStore.storageKey);
    if (!dataStr) {
      throw new Error('no key under given alias');
    }
    const data = JSON.parse(dataStr);
    const privateKey = data.find((d: KmsKeyId) => d.id === args.alias);
    if (!privateKey) {
      throw new Error('no key under given alias');
    }
    return privateKey.value;
  }

  /**
   * Import key to the local storage
   *
   * @param {{ alias: string; key: string }} args - alias and private key in the hex
   * @returns void
   */
  async importKey(args: { alias: string; key: string }): Promise<void> {
    const dataStr = localStorage.getItem(LocalStoragePrivateKeyStore.storageKey);
    let data = [];
    if (dataStr) {
      data = JSON.parse(dataStr);
    }

    const index = data.findIndex((d: KmsKeyId) => d.id === args.alias);
    if (index > -1) {
      data[index].value = args.key;
    } else {
      data.push({ id: args.alias, value: args.key });
    }
    localStorage.setItem(LocalStoragePrivateKeyStore.storageKey, JSON.stringify(data));
  }
}