packages/core/database/src/lifecycles/index.ts - develop - githubmirror/strapi

packages/core/database/src/lifecycles/index.ts
107 строк3 KB
Bassel Kanso
fix: disable lifecycle hooks on transfers [v5] (#22651)
21 янв 2025, 15:44
21 янв 2025, 15:44a28002c
import { strict as assert } from 'assert';

import * as subscriberUtils from './subscribers';

import type { Action, Event, Params, Subscriber } from './types';
import type { Database } from '..';

export type * from './types';

export type State = Record<string, unknown>;
export type States = Map<Subscriber, State>;

export interface Properties {
  params: Params;
  result?: unknown;
}

export interface LifecycleProvider {
  subscribe(subscriber: Subscriber): () => void;
  clear(): void;
  run(action: Action, uid: string, properties: Properties, states?: States): Promise<States>;
  createEvent(action: Action, uid: string, properties: Properties, state: State): Event;
  disable(): void;
  enable(): void;
}

export const createLifecyclesProvider = (db: Database): LifecycleProvider => {
  let subscribers = [
    subscriberUtils.timestampsLifecyclesSubscriber,
    subscriberUtils.modelsLifecyclesSubscriber,
  ];

  let isLifecycleHooksDisabled = false;

  return {
    subscribe(subscriber) {
      assert(
        subscriberUtils.isValidSubscriber(subscriber),
        'Invalid subscriber. Expected function or object'
      );

      subscribers.push(subscriber);

      return () => subscribers.splice(subscribers.indexOf(subscriber), 1);
    },

    clear() {
      subscribers = [];
    },

    disable() {
      isLifecycleHooksDisabled = true;
    },

    enable() {
      isLifecycleHooksDisabled = false;
    },

    createEvent(action, uid, properties, state): Event {
      const model = db.metadata.get(uid);

      return {
        action,
        model,
        state,
        ...properties,
      };
    },

    /**
     * @param {string} action
     * @param {string} uid
     * @param {{ params?: any, result?: any }} properties
     * @param {Map<any, any>} states
     */
    async run(action, uid, properties, states = new Map()) {
      if (isLifecycleHooksDisabled) return states;
      for (let i = 0; i < subscribers.length; i += 1) {
        const subscriber = subscribers[i];
        if (typeof subscriber === 'function') {
          const state = states.get(subscriber) || {};
          const event = this.createEvent(action, uid, properties, state);
          await subscriber(event);
          if (event.state) {
            states.set(subscriber, event.state || state);
          }
          continue;
        }

        const hasAction = action in subscriber;
        const hasModel = !subscriber.models || subscriber.models.includes(uid);

        if (hasAction && hasModel) {
          const state = states.get(subscriber) || {};
          const event = this.createEvent(action, uid, properties, state);

          await subscriber[action]?.(event);
          if (event.state) {
            states.set(subscriber, event.state);
          }
        }
      }

      return states;
    },
  };
};