@tsed/schema/src/utils/getOperationsStores.ts

JsonSchema module for Ts.ED Framework

import {Type} from "@tsed/core";

import type {JsonClassStore} from "../domain/JsonClassStore.js";
import {JsonMethodStore} from "../domain/JsonMethodStore.js";
import {JsonOperationPathsMap} from "../domain/JsonOperationPathsMap.js";
import {getInheritedStores} from "./getInheritedStores.js";
import {getJsonEntityStore} from "./getJsonEntityStore.js";

/**
 * @ignore
 */
export function getOperationsStores<T extends JsonMethodStore = JsonMethodStore>(target: Type<any> | any): Map<string, T> {
  const store: JsonClassStore = target.isStore ? target : getJsonEntityStore(target);

  if (!store.$operations) {
    const operationPaths = new JsonOperationPathsMap();
    store.$operations = new Map<string, T>();

    getInheritedStores(store).forEach((currentStore) => {
      currentStore.children.forEach((propStore) => {
        if (propStore instanceof JsonMethodStore && !store.$operations.has(propStore.propertyKey)) {
          store.$operations.set(propStore.propertyKey, propStore);
        }
      });
    });

    store.$operations.forEach((store: JsonMethodStore) => {
      store.operation.operationPaths.forEach((operationPath) => {
        operationPaths.setOperationPath(operationPath);
      });
    });

    operationPaths.clear();
  }

  return store.$operations;
}