@bajetech/digitalbits-wallet-sdk/src/plugins/MemoryKeyStore.ts

A library to make it easier to write wallets that interact with the DigitalBits blockchain

import { getKeyMetadata } from "../helpers/getKeyMetadata";
import { EncryptedKey, KeyMetadata, KeyStore } from "../types";

interface MemoryStorer {
  [id: string]: EncryptedKey;
}

export class MemoryKeyStore implements KeyStore {
  public name: string;
  private keyStore: MemoryStorer;

  constructor() {
    this.name = "MemoryKeyStore";
    this.keyStore = {};
  }

  public configure() {
    return Promise.resolve();
  }

  public storeKeys(keys: EncryptedKey[]) {
    // We can't store keys if they're already there
    const invalidKeys: EncryptedKey[] = keys.filter(
      (encryptedKey: EncryptedKey) => !!this.keyStore[encryptedKey.id]
    );

    if (invalidKeys.length) {
      return Promise.reject(
        `Some keys were already stored in the keystore: ${invalidKeys
          .map(k => k.id)
          .join(", ")}`
      );
    }

    const keysMetadata = keys.map((encryptedKey: EncryptedKey) => {
      this.keyStore[encryptedKey.id] = {
        ...encryptedKey,
      };

      return getKeyMetadata(this.keyStore[encryptedKey.id] as EncryptedKey);
    });

    return Promise.resolve(keysMetadata);
  }

  public updateKeys(keys: EncryptedKey[]) {
    // we can't update keys if they're already stored
    const invalidKeys: EncryptedKey[] = keys.filter(
      (encryptedKey: EncryptedKey) => !this.keyStore[encryptedKey.id]
    );

    if (invalidKeys.length) {
      return Promise.reject(
        `Some keys couldn't be found in the keystore: ${invalidKeys
          .map(k => k.id)
          .join(", ")}`
      );
    }

    const keysMetadata = keys.map((encryptedKey: EncryptedKey) => {
      const id = encryptedKey.id;

      this.keyStore[id] = {
        ...encryptedKey,
      };

      return getKeyMetadata(this.keyStore[id] as EncryptedKey);
    });

    return Promise.resolve(keysMetadata);
  }

  public loadKey(id: string) {
    return Promise.resolve(this.keyStore[id]);
  }

  public removeKey(id: string) {
    if (!this.keyStore[id]) {
      return Promise.reject(id);
    }

    const metadata: KeyMetadata = getKeyMetadata(
      this.keyStore[id] as EncryptedKey
    );
    delete this.keyStore[id];

    return Promise.resolve(metadata);
  }

  public loadAllKeys() {
    return Promise.resolve(
      Object.values(this.keyStore).map((item: EncryptedKey) => item)
    );
  }
}