@e22m4u/js-repository/src/adapter/decorator/data-sanitizing-decorator.js

Реализация репозитория для работы с базами данных в Node.js

import {Adapter} from '../adapter.js';
import {Service} from '@e22m4u/js-service';
import {InvalidArgumentError} from '../../errors/index.js';
import {ModelDataSanitizer} from '../../definition/index.js';

/**
 * Data sanitizing decorator.
 */
export class DataSanitizingDecorator extends Service {
  /**
   * Decorate.
   *
   * @param {Adapter} adapter
   */
  decorate(adapter) {
    if (!adapter || !(adapter instanceof Adapter))
      throw new InvalidArgumentError(
        'The first argument of DataSanitizingDecorator.decorate should be ' +
          'an Adapter instance, but %v given.',
        adapter,
      );

    const sanitizer = adapter.getService(ModelDataSanitizer);
    const sanitize = (...args) => sanitizer.sanitize(...args);

    const create = adapter.create;
    adapter.create = async function (modelName, modelData, filter) {
      modelData = sanitize(modelName, modelData);
      return create.call(this, modelName, modelData, filter);
    };

    const replaceById = adapter.replaceById;
    adapter.replaceById = async function (modelName, id, modelData, filter) {
      modelData = sanitize(modelName, modelData);
      return replaceById.call(this, modelName, id, modelData, filter);
    };

    const replaceOrCreate = adapter.replaceOrCreate;
    adapter.replaceOrCreate = async function (modelName, modelData, filter) {
      modelData = sanitize(modelName, modelData);
      return replaceOrCreate.call(this, modelName, modelData, filter);
    };

    const patch = adapter.patch;
    adapter.patch = async function (modelName, modelData, where) {
      modelData = sanitize(modelName, modelData);
      return patch.call(this, modelName, modelData, where);
    };

    const patchById = adapter.patchById;
    adapter.patchById = async function (modelName, id, modelData, filter) {
      modelData = sanitize(modelName, modelData);
      return patchById.call(this, modelName, id, modelData, filter);
    };
  }
}