fix: support wasm extension (#46)

src/loaders-deprecated.ts

@@ -14,11 +14,11 @@ import {
 	applySourceMap,
 	tsconfigRaw,
 	tsExtensionsPattern,
-	getFormatFromExtension,
+	getFormatFromFileUrl,
+	fileProtocol,
 	type ModuleFormat,
 	type MaybePromise,
 } from './utils';
-import { getPackageType } from './package-json';
 
 type getFormat = (
 	url: string,
@@ -35,12 +35,21 @@ const _getFormat: getFormat = async function (
 		return { format: 'module' };
 	}
 
-	if (url.startsWith('file:')) {
-		const format = getFormatFromExtension(url) ?? await getPackageType(url);
-		return { format };
-	}
+	try {
+		return await defaultGetFormat(url, context, defaultGetFormat);
+	} catch (error) {
+		if (
+			(error as any).code === 'ERR_UNKNOWN_FILE_EXTENSION'
+			&& url.startsWith(fileProtocol)
+		) {
+			const format = await getFormatFromFileUrl(url);
+			if (format) {
+				return { format };
+			}
+		}
 
-	return await defaultGetFormat(url, context, defaultGetFormat);
+		throw error;
+	}
 };
 
 type Source = string | SharedArrayBuffer | Uint8Array;
@@ -87,12 +96,15 @@ const _transformSource: transformSource = async function (
 	}
 
 	const result = await defaultTransformSource(source, context, defaultTransformSource);
-	const dynamicImportTransformed = transformDynamicImport(filePath, result.source.toString());
-	if (dynamicImportTransformed) {
-		result.source = applySourceMap(
-			dynamicImportTransformed,
-			url,
-		);
+
+	if (context.format === 'module') {
+		const dynamicImportTransformed = transformDynamicImport(filePath, result.source.toString());
+		if (dynamicImportTransformed) {
+			result.source = applySourceMap(
+				dynamicImportTransformed,
+				url,
+			);
+		}
 	}
 
 	return result;

src/loaders.ts

@@ -12,15 +12,15 @@ import {
 	tsconfigRaw,
 	tsconfigPathsMatcher,
 	tsExtensionsPattern,
-	getFormatFromExtension,
+	getFormatFromFileUrl,
+	fileProtocol,
 	type ModuleFormat,
 	type MaybePromise,
 } from './utils';
-import { getPackageType } from './package-json';
 
 type Resolved = {
 	url: string;
-	format: ModuleFormat;
+	format: ModuleFormat | undefined;
 };
 
 type Context = {
@@ -81,7 +81,6 @@ async function tryDirectory(
 	}
 }
 
-const fileProtocol = 'file://';
 const isPathPattern = /^\.{0,2}\//;
 
 const supportsNodePrefix = (
@@ -169,27 +168,14 @@ export const resolve: resolve = async function (
 		throw error;
 	}
 
-	if (resolved.url.endsWith('.json')) {
-		return {
-			...resolved,
-			format: 'json',
-		};
-	}
-
-	let { format } = resolved;
-
-	if (resolved.url.startsWith(fileProtocol)) {
-		format = getFormatFromExtension(resolved.url) ?? format;
-
-		if (!format) {
-			format = await getPackageType(resolved.url);
-		}
+	if (
+		!resolved.format
+		&& resolved.url.startsWith(fileProtocol)
+	) {
+		resolved.format = await getFormatFromFileUrl(resolved.url);
 	}
 
-	return {
-		...resolved,
-		format,
-	};
+	return resolved;
 };
 
 type load = (
@@ -250,12 +236,14 @@ export const load: load = async function (
 		};
 	}
 
-	const dynamicImportTransformed = transformDynamicImport(filePath, code);
-	if (dynamicImportTransformed) {
-		loaded.source = applySourceMap(
-			dynamicImportTransformed,
-			url,
-		);
+	if (loaded.format === 'module') {
+		const dynamicImportTransformed = transformDynamicImport(filePath, code);
+		if (dynamicImportTransformed) {
+			loaded.source = applySourceMap(
+				dynamicImportTransformed,
+				url,
+			);
+		}
 	}
 
 	return loaded;

src/utils.ts

@@ -5,6 +5,7 @@ import {
 	parseTsconfig,
 	createPathsMatcher,
 } from 'get-tsconfig';
+import { getPackageType } from './package-json';
 
 export const applySourceMap = installSourceMapSupport();
 
@@ -20,10 +21,16 @@ const tsconfig = (
 export const tsconfigRaw = tsconfig?.config;
 export const tsconfigPathsMatcher = tsconfig && createPathsMatcher(tsconfig);
 
+export const fileProtocol = 'file://';
+
 export const tsExtensionsPattern = /\.([cm]?ts|[tj]sx)$/;
 
-export const getFormatFromExtension = (filePath: string): ModuleFormat | undefined => {
-	const extension = path.extname(filePath);
+const getFormatFromExtension = (fileUrl: string): ModuleFormat | undefined => {
+	const extension = path.extname(fileUrl);
+
+	if (extension === '.json') {
+		return 'json';
+	}
 
 	if (extension === '.mjs' || extension === '.mts') {
 		return 'module';
@@ -34,6 +41,19 @@ export const getFormatFromExtension = (filePath: string): ModuleFormat | undefin
 	}
 };
 
+export const getFormatFromFileUrl = (fileUrl: string) => {
+	const format = getFormatFromExtension(fileUrl);
+
+	if (format) {
+		return format;
+	}
+
+	// ts, tsx, jsx
+	if (tsExtensionsPattern.test(fileUrl)) {
+		return getPackageType(fileUrl);
+	}
+};
+
 export type ModuleFormat =
 	| 'builtin'
 	| 'dynamic'

tests/fixtures/package-module/lib/wasm/index.js

@@ -0,0 +1,3 @@
+import { myValue } from './test.wasm';
+
+console.log(myValue.valueOf());

tests/index.ts

@@ -34,6 +34,10 @@ const nodeVersions = [
 					import('./specs/json'),
 					node,
 				);
+				runTestSuite(
+					import('./specs/wasm'),
+					node,
+				);
 			});
 
 			runTestSuite(

tests/specs/wasm.ts

@@ -0,0 +1,24 @@
+import { testSuite, expect } from 'manten';
+import type { NodeApis } from '../utils/node-with-loader';
+
+export default testSuite(async ({ describe }, node: NodeApis) => {
+	describe('WASM', async ({ test }) => {
+		const importPath = './lib/wasm/index.js';
+
+		test('Unsupported extension error', async () => {
+			const nodeProcess = await node.load(importPath);
+
+			expect(nodeProcess.exitCode).toBe(1);
+			expect(nodeProcess.stderr).toMatch('ERR_UNKNOWN_FILE_EXTENSION');
+		});
+
+		test('Loads with experimental flag', async () => {
+			const nodeProcess = await node.load(importPath, {
+				nodeOptions: ['--experimental-wasm-modules'],
+			});
+
+			expect(nodeProcess.exitCode).toBe(0);
+			expect(nodeProcess.stdout).toBe('1234');
+		});
+	});
+});