mirror of https://github.com/msgbyte/tailchat
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.5 KiB
TypeScript
112 lines
2.5 KiB
TypeScript
import { PassThrough } from 'stream';
|
|
import type { EntryContext } from '@remix-run/node';
|
|
import { Response } from '@remix-run/node';
|
|
import { RemixServer } from '@remix-run/react';
|
|
import isbot from 'isbot';
|
|
import { renderToPipeableStream } from 'react-dom/server';
|
|
|
|
const ABORT_DELAY = 5000;
|
|
|
|
export default function handleRequest(
|
|
request: Request,
|
|
responseStatusCode: number,
|
|
responseHeaders: Headers,
|
|
remixContext: EntryContext
|
|
) {
|
|
return isbot(request.headers.get('user-agent'))
|
|
? handleBotRequest(
|
|
request,
|
|
responseStatusCode,
|
|
responseHeaders,
|
|
remixContext
|
|
)
|
|
: handleBrowserRequest(
|
|
request,
|
|
responseStatusCode,
|
|
responseHeaders,
|
|
remixContext
|
|
);
|
|
}
|
|
|
|
function handleBotRequest(
|
|
request: Request,
|
|
responseStatusCode: number,
|
|
responseHeaders: Headers,
|
|
remixContext: EntryContext
|
|
) {
|
|
return new Promise((resolve, reject) => {
|
|
let didError = false;
|
|
|
|
const { pipe, abort } = renderToPipeableStream(
|
|
<RemixServer context={remixContext} url={request.url} />,
|
|
{
|
|
onAllReady() {
|
|
const body = new PassThrough();
|
|
|
|
responseHeaders.set('Content-Type', 'text/html');
|
|
|
|
resolve(
|
|
new Response(body, {
|
|
headers: responseHeaders,
|
|
status: didError ? 500 : responseStatusCode,
|
|
})
|
|
);
|
|
|
|
pipe(body);
|
|
},
|
|
onShellError(error: unknown) {
|
|
reject(error);
|
|
},
|
|
onError(error: unknown) {
|
|
didError = true;
|
|
|
|
console.error(error);
|
|
},
|
|
}
|
|
);
|
|
|
|
setTimeout(abort, ABORT_DELAY);
|
|
});
|
|
}
|
|
|
|
function handleBrowserRequest(
|
|
request: Request,
|
|
responseStatusCode: number,
|
|
responseHeaders: Headers,
|
|
remixContext: EntryContext
|
|
) {
|
|
return new Promise((resolve, reject) => {
|
|
let didError = false;
|
|
|
|
const { pipe, abort } = renderToPipeableStream(
|
|
<RemixServer context={remixContext} url={request.url} />,
|
|
{
|
|
onShellReady() {
|
|
const body = new PassThrough();
|
|
|
|
responseHeaders.set('Content-Type', 'text/html');
|
|
|
|
resolve(
|
|
new Response(body, {
|
|
headers: responseHeaders,
|
|
status: didError ? 500 : responseStatusCode,
|
|
})
|
|
);
|
|
|
|
pipe(body);
|
|
},
|
|
onShellError(err: unknown) {
|
|
reject(err);
|
|
},
|
|
onError(error: unknown) {
|
|
didError = true;
|
|
|
|
console.error(error);
|
|
},
|
|
}
|
|
);
|
|
|
|
setTimeout(abort, ABORT_DELAY);
|
|
});
|
|
}
|