UNPKG

@fedify/express

Version:

Integrate Fedify with Express

76 lines (74 loc) 2.28 kB
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 };