File size: 2,914 Bytes
d44b3c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import fs from 'node:fs';
import path from 'node:path';
import { fileURLToPath } from 'node:url';
import mime from 'mime-types';
import { serverDirectory } from './server-directory.js';
const originalFetch = globalThis.fetch;
const ALLOWED_EXTENSIONS = [
'.wasm',
];
/**
* Checks if a child path is under a parent path.
* @param {string} parentPath Parent path
* @param {string} childPath Child path
* @returns {boolean} Returns true if the child path is under the parent path, false otherwise
*/
function isPathUnderParent(parentPath, childPath) {
const normalizedParent = path.normalize(parentPath);
const normalizedChild = path.normalize(childPath);
const relativePath = path.relative(normalizedParent, normalizedChild);
return !relativePath.startsWith('..') && !path.isAbsolute(relativePath);
}
/**
* Checks if the given request is a file URL.
* @param {string | URL | Request} request The request to check
* @return {boolean} Returns true if the request is a file URL, false otherwise
*/
function isFileURL(request) {
if (typeof request === 'string') {
return request.startsWith('file://');
}
if (request instanceof URL) {
return request.protocol === 'file:';
}
if (request instanceof Request) {
return request.url.startsWith('file://');
}
return false;
}
/**
* Gets the URL from the request.
* @param {string | URL | Request} request The request to get the URL from
* @return {string} The URL of the request
*/
function getRequestURL(request) {
if (typeof request === 'string') {
return request;
}
if (request instanceof URL) {
return request.href;
}
if (request instanceof Request) {
return request.url;
}
throw new TypeError('Invalid request type');
}
// Patched fetch function that handles file URLs
globalThis.fetch = async (/** @type {string | URL | Request} */ request, /** @type {RequestInit | undefined} */ options) => {
if (!isFileURL(request)) {
return originalFetch(request, options);
}
const url = getRequestURL(request);
const filePath = path.resolve(fileURLToPath(url));
const isUnderServerDirectory = isPathUnderParent(serverDirectory, filePath);
if (!isUnderServerDirectory) {
throw new Error('Requested file path is outside of the server directory.');
}
const parsedPath = path.parse(filePath);
if (!ALLOWED_EXTENSIONS.includes(parsedPath.ext)) {
throw new Error('Unsupported file extension.');
}
const fileName = parsedPath.base;
const buffer = await fs.promises.readFile(filePath);
const response = new Response(buffer, {
status: 200,
statusText: 'OK',
headers: {
'Content-Type': mime.lookup(fileName) || 'application/octet-stream',
'Content-Length': buffer.length.toString(),
},
});
return response;
};
|