@nestia/core/src/decorators/internal/load_controller.ts

Super-fast validation decorators of NestJS

import is_ts_node from "detect-ts-node";

import { Creator } from "../../typings/Creator";
import { SourceFinder } from "../../utils/SourceFinder";

export const load_controllers = async (
  path: string | string[] | { include: string[]; exclude?: string[] },
  isTsNode?: boolean,
): Promise<Creator<object>[]> => {
  const sources: string[] = await SourceFinder.find({
    include: Array.isArray(path)
      ? path
      : typeof path === "object"
        ? path.include
        : [path],
    exclude:
      typeof path === "object" && !Array.isArray(path)
        ? (path.exclude ?? [])
        : [],
    filter:
      isTsNode === true || EXTENSION === "ts"
        ? (file) =>
            file.substring(file.length - 3) === ".ts" &&
            file.substring(file.length - 5) !== ".d.ts"
        : (flle) => flle.substring(flle.length - 3) === ".js",
  });
  return mount(sources);
};

/** @internal */
async function mount(sources: string[]): Promise<any[]> {
  const controllers: any[] = [];
  for (const file of sources) {
    const external: any = await import(file);
    for (const key in external) {
      const instance: Creator<object> = external[key];
      if (Reflect.getMetadata("path", instance) !== undefined)
        controllers.push(instance);
    }
  }
  return controllers;
}

/** @internal */
const EXTENSION = is_ts_node ? "ts" : "js";