@hashgraph/hedera-identify-snap/src/plugins/veramo/verifiable-creds-manager/data-store/memoryDataStore.ts

A snap for managing Decentralized Identifiers(DIDs)

/*-
 *
 * Hedera Identify Snap
 *
 * Copyright (C) 2024 Hedera Hashgraph, LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

import { sha256 } from 'js-sha256';
import jsonpath from 'jsonpath';
import {
  AbstractDataStore,
  IDeleteArgs,
  IFilterArgs,
  IQueryResult,
  ISaveArgs,
} from './abstractDataStore';

/**
 * An implementation of {@link AbstractDataStore} that stores everything in memory.
 */
export class MemoryDataStore extends AbstractDataStore {
  private data: Record<string, unknown> = {};

  configure: undefined;

  // eslint-disable-next-line @typescript-eslint/require-await
  public async saveVC(args: ISaveArgs): Promise<string[]> {
    const id = sha256(JSON.stringify(args));
    this.data[id] = args.data;
    return [id];
  }

  // eslint-disable-next-line @typescript-eslint/require-await
  public async deleteVC(args: IDeleteArgs): Promise<boolean> {
    const { id } = args;
    if (id in this.data) {
      delete this.data[id];
      return true;
    }
    return false;
  }

  // eslint-disable-next-line @typescript-eslint/require-await
  public async queryVC(args: IFilterArgs): Promise<IQueryResult[]> {
    const { filter } = args;
    if (filter && filter.type === 'id') {
      try {
        if (this.data[filter.filter as string]) {
          const obj = [
            {
              metadata: { id: filter.filter as string },
              data: this.data[filter.filter as string],
            },
          ];
          return obj;
        }
        return [];
      } catch (e) {
        throw new Error('Invalid id');
      }
    }

    if (filter === undefined || (filter && filter.type === 'none')) {
      return Object.keys(this.data).map((k) => {
        return {
          metadata: { id: k },
          data: this.data[k],
        };
      });
    }

    if (filter && filter.type === 'jsonpath') {
      const objects = Object.keys(this.data).map((k) => {
        return {
          metadata: { id: k },
          data: this.data[k],
        };
      });
      const filteredObjects = jsonpath.query(objects, filter.filter as string);
      return filteredObjects as IQueryResult[];
    }
    return [];
  }

  // eslint-disable-next-line @typescript-eslint/require-await
  public async clearVCs(_args: IFilterArgs): Promise<boolean> {
    this.data = {};
    return true;
  }
}