try to load typescript fallback for specifier with extensions.

Author: mshima

src/loaders.ts

@@ -35,26 +35,52 @@ type resolve = (
 	recursiveCall?: boolean,
 ) => MaybePromise<Resolved>;
 
-const extensions = ['.js', '.json', '.ts', '.tsx', '.jsx'] as const;
+const mappedExtensions: Record<string, string[]> = {
+	'.js': ['.ts', '.tsx'],
+	'.jsx': ['.ts', '.tsx'],
+	'.cjs': ['.cts'],
+	'.mjs': ['.mts'],
+	'': ['.ts', '.js', '.json', '.tsx', '.jsx'],
+};
+
+function resolveTsAlternativeSpecifiers(specifier: string) {
+	const specifierExtension = path.extname(specifier);
+	let extensionsToTry = mappedExtensions[specifierExtension] ?? [];
+	if (specifierExtension) {
+		specifier = specifier.slice(0, specifier.length - specifierExtension.length);
+		extensionsToTry = [
+			// Try subextension first
+			...mappedExtensions[''].map(extension => `${specifierExtension}${extension}`),
+			...extensionsToTry,
+		];
+	}
+	return extensionsToTry.map(alternativeExtension => ({
+		specifier,
+		specifierExtension,
+		altenativeSpecifier: `${specifier}${alternativeExtension}`,
+		alternativeExtension,
+	}));
+}
 
 async function tryExtensions(
 	specifier: string,
 	context: Context,
 	defaultResolve: resolve,
 ) {
 	let error;
-	for (const extension of extensions) {
+	for (const alternativePath of resolveTsAlternativeSpecifiers(specifier)) {
+		const { specifierExtension, altenativeSpecifier, alternativeExtension } = alternativePath;
 		try {
 			return await resolve(
-				specifier + extension,
+				altenativeSpecifier,
 				context,
 				defaultResolve,
 				true,
 			);
 		} catch (_error: any) {
 			if (error === undefined) {
 				const { message } = _error;
-				_error.message = _error.message.replace(`${extension}'`, "'");
+				_error.message = _error.message.replace(`${alternativeExtension}'`, `${specifierExtension}'`);
 				_error.stack = _error.stack.replace(message, _error.message);
 				error = _error;
 			}

tests/specs/typescript/cts.ts

@@ -28,9 +28,9 @@ export default testSuite(async ({ describe }, node: NodeApis) => {
 		describe('full path via .cjs', ({ test }) => {
 			const importPath = './lib/ts-ext-cts/index.cjs';
 
-			test('Load - should not work', async () => {
+			test('Load', async () => {
 				const nodeProcess = await node.load(importPath);
-				expect(nodeProcess.stderr).toMatch('Cannot find module');
+				expect(nodeProcess.stderr).toMatch('SyntaxError: Unexpected token \':\'');
 			});
 
 			test('Import', async () => {

tests/specs/typescript/mts.ts

@@ -41,7 +41,7 @@ export default testSuite(async ({ describe }, node: NodeApis) => {
 
 			test('Load - should not work', async () => {
 				const nodeProcess = await node.load(importPath);
-				expect(nodeProcess.stderr).toMatch('Cannot find module');
+				assertResults(nodeProcess.stdout);
 			});
 
 			test('Import', async () => {

tests/specs/typescript/ts.ts

@@ -49,7 +49,7 @@ export default testSuite(async ({ describe }, node: NodeApis) => {
 
 			test('Load - should not work', async () => {
 				const nodeProcess = await node.load(importPath);
-				expect(nodeProcess.stderr).toMatch('Cannot find module');
+				assertResults(nodeProcess.stdout, 'ts-ext-ts/index.ts');
 			});
 
 			test('Import', async () => {