@fedify/express
Version:
Integrate Fedify with Express
76 lines (74 loc) • 2.28 kB
JavaScript
import { Buffer } from "node:buffer";
import { Readable } from "node:stream";
//#region src/index.ts
function integrateFederation(federation, contextDataFactory) {
return (req, res, next) => {
const request = fromERequest(req);
const contextData = contextDataFactory(req);
const contextDataPromise = contextData instanceof Promise ? contextData : Promise.resolve(contextData);
contextDataPromise.then(async (contextData$1) => {
let notFound = false;
let notAcceptable = false;
const response = await federation.fetch(request, {
contextData: contextData$1,
onNotFound: () => {
notFound = true;
next();
return new Response("Not found", { status: 404 });
},
onNotAcceptable: () => {
notAcceptable = true;
next();
return new Response("Not acceptable", {
status: 406,
headers: {
"Content-Type": "text/plain",
Vary: "Accept"
}
});
}
});
if (notFound || notAcceptable && req.route != null) return;
await setEResponse(res, response);
res.end();
res.status = () => res;
res.send = () => res;
res.end = () => res;
res.json = () => res;
res.removeHeader = () => res;
res.setHeader = () => res;
});
};
}
function fromERequest(req) {
const url = `${req.protocol}://${req.header("Host") ?? req.hostname}${req.url}`;
const headers = new Headers();
for (const [key, value] of Object.entries(req.headers)) if (Array.isArray(value)) for (const v of value) headers.append(key, v);
else if (typeof value === "string") headers.append(key, value);
return new Request(url, {
method: req.method,
headers,
duplex: "half",
body: req.method === "GET" || req.method === "HEAD" ? void 0 : Readable.toWeb(req)
});
}
function setEResponse(res, response) {
res.status(response.status);
response.headers.forEach((value, key) => res.setHeader(key, value));
if (response.body == null) return Promise.resolve();
const body = response.body;
return new Promise((resolve) => {
const reader = body.getReader();
reader.read().then(function read({ done, value }) {
if (done) {
reader.releaseLock();
resolve();
return;
}
res.write(Buffer.from(value));
reader.read().then(read);
});
});
}
//#endregion
export { integrateFederation };