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

SDK to work with Polygon ID

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

/**
 * Key Store to use in memory
 *
 * @public
 * @class InMemoryPrivateKeyStore
 * @implements implements AbstractPrivateKeyStore interface
 */
export class InMemoryPrivateKeyStore implements AbstractPrivateKeyStore {
  private _data: Map<string, string>;
  constructor() {
    this._data = new Map<string, string>();
  }

  list(): Promise<{ alias: string; key: string }[]> {
    return Promise.resolve(Array.from(this._data).map(([alias, key]) => ({ alias, key })));
  }

  async get(args: { alias: string }): Promise<string> {
    const privateKey = this._data.get(args.alias);
    if (!privateKey) {
      throw new Error('no key under given alias');
    }
    return privateKey;
  }

  async importKey(args: { alias: string; key: string }): Promise<void> {
    this._data.set(args.alias, args.key);
  }
}