@athree/module-loader 2.0.0

package/package.json

@@ -0,0 +1,12 @@
+{
+    "name": "@athree/module-loader",
+    "version": "2.0.0",
+    "type": "module",
+    "exports": {
+        ".": "./out/main/index.js",
+        "./register": "./out/main/register.js"
+    },
+    "scripts": {
+        "clean": "rm -rf out/ *.tsbuildinfo"
+    }
+}

package/src/main/index.ts

@@ -0,0 +1 @@
+export * from './module-loader.js';

package/src/main/module-loader.ts

@@ -0,0 +1,50 @@
+import { readFile, stat } from 'node:fs/promises';
+import { LoadFnOutput, LoadHookContext, ResolveFnOutput, ResolveHookContext } from 'node:module';
+import { fileURLToPath } from 'node:url';
+
+type ResolveNext = (specifier: string, context: ResolveHookContext) => Promise<ResolveFnOutput>;
+type LoadNext = (specifier: string, context: LoadHookContext) => Promise<LoadFnOutput>;
+
+export async function resolve(specifier: string, context: ResolveHookContext, nextResolve: ResolveNext): Promise<ResolveFnOutput> {
+    // Allow specifying ?t=timestamp directly in await import()
+    // to force cache busting
+    let ts = 0;
+    if (specifier.includes('?t=')) {
+        const [_spec, _ts] = specifier.split('?t=');
+        specifier = _spec;
+        ts = Number(_ts) || 0;
+    }
+    // Resolve the module as usual,
+    // this results either in file:// or custom specifiers like `node:`
+    const resolved = await nextResolve(specifier, context);
+    if (!resolved.url.startsWith('file://')) {
+        return resolved;
+    }
+    // For file modules, if ts is provided explicitly in resolve request,
+    // then use that, otherwise use the file's mtime
+    const file = fileURLToPath(resolved.url);
+    if (ts > 0) {
+        return {
+            format: resolved.format,
+            url: resolved.url + `?t=${ts}`,
+        };
+    }
+    const { mtime } = await stat(file);
+    return {
+        format: resolved.format,
+        url: resolved.url + `?t=${mtime.getTime()}`,
+    };
+}
+
+export async function load(specifier: string, context: LoadHookContext, nextLoad: LoadNext): Promise<LoadFnOutput> {
+    if (/\.md($|\?)/.test(specifier)) {
+        const file = fileURLToPath(specifier);
+        const source = await readFile(file, 'utf-8');
+        return {
+            format: 'module',
+            shortCircuit: true,
+            source: `export default ${JSON.stringify(source)}`,
+        };
+    }
+    return nextLoad(specifier, context);
+}

package/src/main/register.ts

@@ -0,0 +1,3 @@
+import { register } from 'node:module';
+
+register(new URL('./module-loader.js', import.meta.url));

package/tsconfig.json

@@ -0,0 +1,8 @@
+{
+    "extends": "../../tsconfig.base.json",
+    "compilerOptions": {
+        "outDir": "./out",
+        "rootDir": "./src"
+    },
+    "include": ["src"]
+}