import { NextResponse, NextRequest, userAgent } from 'next/server'
import { IFRAME_URL } from './configs/api-route'

export async function middleware(req: NextRequest) {
  let res = NextResponse.next()

  if (req.headers?.get('host')?.endsWith('.blog')) {
    const url = req.nextUrl.clone()
    // blog write page
    url.pathname = `/blog${req.nextUrl.pathname}`
    res = NextResponse.rewrite(url)
  } else if (req.nextUrl.pathname === '/api/iframe' && !userAgent(req).isBot) {
    const { searchParams } = req.nextUrl
    const u = searchParams.get('url')
    const base = searchParams.get('baseHref')

    if (u) {
      res = NextResponse.rewrite(
        IFRAME_URL + `?url=${encodeURIComponent(u)}&baseHref=${base || true}`
      )
    }
  }

  return res
}

export const config = {
  matcher: [
    '/',
    '/blog',
    '/blog/:path*',
    '/categories/:path*',
    '/authors/:path*',
    '/api/iframe/:path*',
  ],
}