import type { AppLoadContext, ServerBuild } from '@netlify/remix-runtime'
import { createRequestHandler as createRemixRequestHandler } from '@netlify/remix-runtime'
import type { Context } from '@netlify/edge-functions'

type LoadContext = AppLoadContext & Context

/**
 * A function that returns the value to use as `context` in route `loader` and
 * `action` functions.
 *
 * You can think of this as an escape hatch that allows you to pass
 * environment/platform-specific values through to your loader/action.
 */
export type GetLoadContextFunction = (request: Request, context: Context) => Promise<LoadContext> | LoadContext

export type RequestHandler = (request: Request, context: LoadContext) => Promise<Response | void>

export function createRequestHandler({
  build,
  mode,
  getLoadContext,
}: {
  build: ServerBuild
  mode?: string
  getLoadContext?: GetLoadContextFunction
}): RequestHandler {
  const remixHandler = createRemixRequestHandler(build, mode)

  const assetPath = build.assets.url.split('/').slice(0, -1).join('/')

  return async (request: Request, context: LoadContext): Promise<Response | void> => {
    const { pathname } = new URL(request.url)
    // Skip the handler for static files
    if (pathname.startsWith(`${assetPath}/`)) {
      // Temporary fix - passing the request to the Netlify static asset handler causes a 203 Not Content error. Passing the through remix works, but I asssume isn't ideal
      // console.log('Skipping Remix handler for static file', pathname)
      // return;
    }
    try {
      const loadContext = (await getLoadContext?.(request, context)) || context

      const response = await remixHandler(request, loadContext)

      // A useful header for debugging
      response.headers.set('x-nf-runtime', 'Edge')

      if (response.status === 404) {
        // Check if there is a matching static file
        const originResponse = await context.next({
          sendConditionalRequest: true,
        })
        if (originResponse.status !== 404) {
          return originResponse
        }
      }
      return response
    } catch (error: unknown) {
      console.error(error)

      return new Response('Internal Error', { status: 500 })
    }
  }
}