nx-plugin.ts

import { sync } from 'fast-glob';
import { existsSync } from 'fs';
import * as path from 'path';
import { register } from '@swc-node/register/register';
import { readDefaultTsConfig } from '@swc-node/register/read-default-tsconfig';

import { appRootPath } from '../utils/app-root';
import { readJsonFile } from '../utils/fileutils';
import { PackageJson } from './package-json';
import { ProjectGraphProcessor } from './project-graph';
import { Workspaces } from './workspace';
import {
  ProjectConfiguration,
  TargetConfiguration,
  WorkspaceJsonConfiguration,
} from './workspace.model';

export type ProjectTargetConfigurator = (
  file: string
) => Record<string, TargetConfiguration>;

/**
 * A plugin for Nx
 */
export interface NxPlugin {
  name: string;
  processProjectGraph?: ProjectGraphProcessor;
  registerProjectTargets?: ProjectTargetConfigurator;

  /**
   * A glob pattern to search for non-standard project files.
   * @example: ["*.csproj", "pom.xml"]
   */
  projectFilePatterns?: string[];
}

// Short lived cache (cleared between cmd runs)
// holding resolved nx plugin objects.
// Allows loadNxPlugins to be called multiple times w/o
// executing resolution mulitple times.
let nxPluginCache: NxPlugin[] = null;
export function loadNxPlugins(
  plugins?: string[],
  paths = [appRootPath]
): NxPlugin[] {
  return plugins?.length
    ? nxPluginCache ||
        (nxPluginCache = plugins.map((moduleName) => {
          let pluginPath: string;
          try {
            pluginPath = require.resolve(moduleName, {
              paths,
            });
          } catch (e) {
            if (e.code === 'MODULE_NOT_FOUND') {
              const plugin = resolveLocalNxPlugin(moduleName);
              const { main } = Object.values(plugin.projectConfig.targets).find(
                (x) =>
                  [
                    '@nrwl/js:tsc',
                    '@nrwl/js:swc',
                    '@nrwl/node:package',
                  ].includes(x.executor)
              )?.options;
              pluginPath = main ? path.join(appRootPath, main) : plugin.path;
            } else {
              throw e;
            }
          }
          const packageJsonPath = path.join(pluginPath, 'package.json');
          const { name } =
            !['.ts', '.js'].some((x) => x === path.extname(pluginPath)) && // Not trying to point to a ts or js file
            existsSync(packageJsonPath) // plugin has a package.json
              ? readJsonFile(packageJsonPath) // read name from package.json
              : { name: path.basename(pluginPath) }; // use the name of the file we point to
          const plugin = require(pluginPath) as NxPlugin;
          plugin.name = name;

          return plugin;
        }))
    : [];
}

export function mergePluginTargetsWithNxTargets(
  projectRoot: string,
  targets: Record<string, TargetConfiguration>,
  plugins: NxPlugin[]
): Record<string, TargetConfiguration> {
  let newTargets: Record<string, TargetConfiguration> = {};
  for (const plugin of plugins) {
    if (!plugin.projectFilePatterns?.length || !plugin.registerProjectTargets) {
      continue;
    }

    const projectFiles = sync(`+(${plugin.projectFilePatterns.join('|')})`, {
      cwd: path.join(appRootPath, projectRoot),
    });
    for (const projectFile of projectFiles) {
      newTargets = {
        ...newTargets,
        ...plugin.registerProjectTargets(path.join(projectRoot, projectFile)),
      };
    }
  }
  return { ...newTargets, ...targets };
}

export function readPluginPackageJson(
  pluginName: string,
  paths = [appRootPath]
): {
  path: string;
  json: PackageJson;
} {
  let packageJsonPath: string;
  try {
    packageJsonPath = require.resolve(`${pluginName}/package.json`, {
      paths,
    });
  } catch (e) {
    if (e.code === 'MODULE_NOT_FOUND') {
      const localPluginPath = resolveLocalNxPlugin(pluginName);
      if (localPluginPath) {
        const localPluginPackageJson = path.join(
          localPluginPath.path,
          'package.json'
        );
        return {
          path: localPluginPackageJson,
          json: readJsonFile(localPluginPackageJson),
        };
      }
    }
    throw e;
  }
  return { json: readJsonFile(packageJsonPath), path: packageJsonPath };
}

/**
 * Builds a plugin package and returns the path to output
 * @param importPath What is the import path that refers to a potential plugin?
 * @returns The path to the built plugin, or null if it doesn't exist
 */
const localPluginCache: Record<
  string,
  { path: string; projectConfig: ProjectConfiguration }
> = {};
export function resolveLocalNxPlugin(
  importPath: string,
  root = appRootPath
): { path: string; projectConfig: ProjectConfiguration } | null {
  localPluginCache[importPath] ??= lookupLocalPlugin(importPath, root);
  return localPluginCache[importPath];
}

let tsNodeAndPathsRegistered = false;
function registerTSTranspiler() {
  try {
    const tsConfigOptions = readDefaultTsConfig(
      path.join(appRootPath, 'tsconfig.base.json')
    );
    register(tsConfigOptions);

    const tsconfigPaths: typeof import('tsconfig-paths') = require('tsconfig-paths');

    /**
     * Register the custom workspace path mappings with node so that workspace libraries
     * can be imported and used within custom workspace lint rules.
     */
    return tsconfigPaths.register({
      baseUrl: appRootPath,
      paths: tsConfigOptions.paths,
    });
  } catch (err) {}
}

function lookupLocalPlugin(importPath: string, root = appRootPath) {
  const workspace = new Workspaces(root).readWorkspaceConfiguration({
    ignorePluginInference: true,
  });
  const plugin = findNxProjectForImportPath(importPath, workspace, root);
  if (!plugin) {
    return null;
  }

  if (!tsNodeAndPathsRegistered) {
    registerTSTranspiler();
  }

  const projectConfig = workspace.projects[plugin];
  return { path: path.join(root, projectConfig.root), projectConfig };
}

function findNxProjectForImportPath(
  importPath: string,
  workspace: WorkspaceJsonConfiguration,
  root = appRootPath
): string | null {
  const tsConfigPaths: Record<string, string[]> = readJsonFile(
    path.join(root, 'tsconfig.base.json')
  )?.compilerOptions?.paths;
  const possiblePaths = tsConfigPaths[importPath]?.map((p) =>
    path.resolve(root, p)
  );
  if (tsConfigPaths[importPath]) {
    const projectRootMappings = Object.entries(workspace.projects).reduce(
      (m, [project, config]) => {
        m[path.resolve(root, config.root)] = project;
        return m;
      },
      {}
    );
    for (const root of Object.keys(projectRootMappings)) {
      if (possiblePaths.some((p) => p.startsWith(root))) {
        return projectRootMappings[root];
      }
    }
    if (process.env.NX_VERBOSE_LOGGING) {
      console.log(
        'Unable to find local plugin',
        possiblePaths,
        projectRootMappings
      );
    }
    throw new Error(
      'Unable to resolve local plugin with import path ' + importPath
    );
  }
}