entry.server.tsx

import { PassThrough } from 'node:stream'
import type { EntryContext } from '@remix-run/node'
import { Response } from '@remix-run/node'
import { RemixServer } from '@remix-run/react'
import { isbot } from 'isbot'
/* @ts-ignore */
import { renderToPipeableStream } from 'react-dom/server'
import { server } from './mocks/node'

const ABORT_DELAY = 5_000

if (process.env.NODE_ENV === 'development') {
  server.listen({
      onUnhandledRequest: 'bypass',
  })
}

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) => {
    const { pipe, abort } = renderToPipeableStream(
        <RemixServer
            context={remixContext}
            url={request.url}
            abortDelay={ABORT_DELAY}
        />,
        {
          onAllReady() {
            const body = new PassThrough()

            responseHeaders.set('Content-Type', 'text/html')

            resolve(
                new Response(body, {
                  headers: responseHeaders,
                  status: responseStatusCode,
                }),
            )

            pipe(body)
          },
          onShellError(error: unknown) {
            reject(error)
          },
          onError(error: unknown) {
            responseStatusCode = 500
            console.error(error)
          },
        },
    )

    setTimeout(abort, ABORT_DELAY)
  })
}

function handleBrowserRequest(
    request: Request,
    responseStatusCode: number,
    responseHeaders: Headers,
    remixContext: EntryContext,
) {
  return new Promise((resolve, reject) => {
    const { pipe, abort } = renderToPipeableStream(
        <RemixServer
            context={remixContext}
            url={request.url}
            abortDelay={ABORT_DELAY}
        />,
        {
          onShellReady() {
            const body = new PassThrough()

            responseHeaders.set('Content-Type', 'text/html')

            resolve(
                new Response(body, {
                  headers: responseHeaders,
                  status: responseStatusCode,
                }),
            )

            pipe(body)
          },
          onShellError(error: unknown) {
            reject(error)
          },
          onError(error: unknown) {
            console.error(error)
            responseStatusCode = 500
          },
        },
    )

    setTimeout(abort, ABORT_DELAY)
  })
}