export async function initialize({ number, port }) {
  // 在注册的时候调用，用于接收参数和初始化。
}

export async function resolve(specifier, context, nextResolve) {
  // 解析导入的模块路径。
}

export async function load(url, context, nextLoad) {
  // 加载代码，在这里可以把 TS 编译成 JS 后返回。
} 

import { register } from "module";

register("./上面的模块.js", import.meta.url);

// 注册完即生效，接下来就可以导入 TS 文件了。
await import("./xxx.ts");

export const resolve: ResolveHook = async (specifier, context, nextResolve) => {
	try {
        // 先调用 Node 内置的解析，成功就直接返回。
		return await nextResolve(specifier, context);
	} catch (e) {
		// 如果失败且导入的是 JS 文件。
		const isFile = /^(?:file:|\.{1,2}\/)/i.test(specifier);
		const isJSFile = isFile && /\.[cm]?jsx?$/i.test(specifier);

		if (!isJSFile || e.code !== "ERR_MODULE_NOT_FOUND") {
			throw e;
		}
        // 则将文件扩展名中的 j 字母替换为 t 再次解析。
		if (specifier.at(-1) !== "x") {
			return nextResolve(specifier.slice(0, -2) + "ts", context);
		} else {
			return nextResolve(specifier.slice(0, -3) + "tsx", context);
		}
	}
};

import { LoadHook, ModuleFormat } from "module";

export type CompileFn = (code: string, filename: string, isESM: boolean) => Promise<string>;

let compile: CompileFn;

export const load: LoadHook = async (url, context, nextLoad) => {
	// 似乎是 Node 的 BUG，会丢失 importAttributes。
	if (context.format === "json") {
		context.importAttributes.type = "json";
		return nextLoad(url, context);
	}

    // 非 TS 文件咱不处理。
	const match = /\.[cm]?tsx?$/i.exec(url);
	if (!match || !url.startsWith("file:")) {
		return nextLoad(url, context);
	}

    // 调用 Node 自己的加载函数，设置自定义的 format 可避免文件内容被处理。
	context.format = "ts" as any;
	const ts = await nextLoad(url, context);
	const code = ts.source!.toString();
	const filename = fileURLToPath(url);

    // 根据文件名和 package.json 的 type 属性判断文件是 CJS 还是 ESM。
	let format: ModuleFormat;
	switch (match[0].charCodeAt(1)) {
		case 99: /* c */
			format = "commonjs";
			break;
		case 109: /* m */
			format = "module";
			break;
		default: /* t */
            // 这个函数就不写了，去源码里看吧。
			format = getPackageType(filename);
	}

    // 加载编译器，最后返回编译后的代码。
	compile ??= await detectTypeScriptCompiler();
	return {
		shortCircuit: true,
		format,
		source: await compile(code, filename, format === "module"),
	};
};

// 三个编译函数就省略了，里头就是转换下选项然后编译而已。
async function swcCompiler() {
	const swc = await import("@swc/core");
    return code => { /* 转换 code 为 JS 代码 */};
}

async function esbuildCompiler() {
	const esbuild = await import("esbuild");
    return code => { /* 转换 code 为 JS 代码 */};
}

async function tsCompiler() {
	const { default: ts } = await import("typescript");
    return code => { /* 转换 code 为 JS 代码 */};
}

// 速度快的在前面
export const compilers = [swcCompiler, esbuildCompiler, tsCompiler];

async function detectTypeScriptCompiler() {
	// 如果设置了环境变量，就用指定的编译器。
	const name = process.env.TS_COMPILER;
	if (name) {
		const i = ["swc", "esbuild", "tsc"].indexOf(name);
		return compilers[i]();
	}
	// 否则挨个尝试支持的。
	for (const create of compilers) {
		try {
			return await create();
		} catch (e) {
			if (e.code !== "ERR_MODULE_NOT_FOUND") throw e;
		}
	}
	throw new Error("No TypeScript transformer found");
}