nocobase/packages/core/utils/src/requireModule.ts

/**
 * This file is part of the NocoBase (R) project.
 * Copyright (c) 2020-2024 NocoBase Co., Ltd.
 * Authors: NocoBase Team.
 *
 * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
 * For more information, please refer to: https://www.nocobase.com/agreement.
 */

import fs from 'fs';
import path from 'path';
import { pathToFileURL } from 'url';

export async function requireResolve(m: any) {
  if (!process.env.VITEST) {
    return require.resolve(m);
  }
  // 以下逻辑仅用于测试环境，因为 vitest 不支持对 require 调用进行别名
  const json = JSON.parse(
    await fs.promises.readFile(path.resolve(process.cwd(), './tsconfig.paths.json'), { encoding: 'utf8' }),
  );
  const paths = json.compilerOptions.paths;
  if (paths[m]) {
    return require.resolve(path.resolve(process.cwd(), paths[m][0], 'index.ts'));
  }
  return require.resolve(m);
}

export function requireModule(m: any) {
  if (typeof m === 'string') {
    m = require(m);
  }

  if (typeof m !== 'object') {
    return m;
  }

  return m.__esModule ? m.default : m;
}

export default requireModule;

export async function importModule(m: string) {
  if (!process.env.VITEST) {
    return requireModule(m);
  }

  if (path.isAbsolute(m)) {
    m = pathToFileURL(m).href;
  }

  const r = (await import(m)).default;
  return r.__esModule ? r.default : r;
}