// Vercel Routing Middleware (https://vercel.com/docs/routing-middleware):
// serves the markdown mirror of a docs page when the client prefers it via
// `Accept: text/markdown`. Runs only on Vercel deployments.
import type { MiddlewareHandler } from 'astro'
import { next, rewrite } from '@vercel/functions'
// .js extensions: Vercel compiles this file separately with node16 module
// resolution, which requires them; TS and vite both resolve .js -> .ts.
import { preferredFormat } from './lib/accept.js'
import { markdownUrlFromDocsUrl } from './lib/markdown-url.js'
import { redirects } from './lib/redirects.js'

// Astro picks this file up as its own middleware (`srcDir` is the project
// root) and requires an `onRequest` export; the real logic is Vercel-only.
export const onRequest: MiddlewareHandler = (_context, nextHandler) => nextHandler()

export const config = {
  // extensionless paths outside the asset directories — the html docs pages
  matcher: '/((?!(?:dist|preview|img|static|css|js)/)(?!.*\\.).*)',
}

const alternate = (path: string, type: string) => `<${path}>; rel="alternate"; type="${type}"`

export default function middleware(request: Request) {
  const url = new URL(request.url)
  // redirect keys have no trailing slash, so "/ui/base/markdown/" must match "/ui/base/markdown"
  const path = url.pathname.length > 1 ? url.pathname.replace(/\/+$/, '') : url.pathname

  // renamed pages fall through to their 301; negotiation reruns on the target
  // not Object.hasOwn — Vercel's middleware type-check runs with a pre-es2022 lib
  if (Object.prototype.hasOwnProperty.call(redirects, path)) return next()

  const format = preferredFormat(request.headers.get('accept'))

  if (format === 'markdown') {
    const htmlPath = url.pathname
    url.pathname = markdownUrlFromDocsUrl(url.pathname)
    return rewrite(url, {
      headers: {
        Link: alternate(htmlPath, 'text/html'),
        Vary: 'Accept',
      },
    })
  }

  if (format === 'none') {
    return new Response('Not acceptable. This page is available as text/html or text/markdown.\n', {
      status: 406,
      headers: {
        'Content-Type': 'text/plain; charset=utf-8',
        'Vary': 'Accept',
      },
    })
  }

  return next({
    headers: {
      Link: alternate(markdownUrlFromDocsUrl(url.pathname), 'text/markdown'),
      Vary: 'Accept',
    },
  })
}