64 lines
2.0 KiB
TypeScript
64 lines
2.0 KiB
TypeScript
import { NextResponse } from 'next/server'
|
||
import type { NextRequest } from 'next/server'
|
||
|
||
const PRODUCTION_DOMAIN = 'rentaldrivego.com'
|
||
const DEFAULT_SLUG = process.env.NEXT_PUBLIC_COMPANY_SLUG ?? 'demo'
|
||
|
||
/**
|
||
* Resolve the company slug from the request host.
|
||
*
|
||
* Production: {slug}.rentaldrivego.com → slug
|
||
* Development: localhost / 127.0.0.1 → DEFAULT_SLUG (env var fallback)
|
||
* Staging/preview URLs that don't match the production domain also fall back.
|
||
*/
|
||
function resolveSlug(host: string | null): string {
|
||
if (!host) return DEFAULT_SLUG
|
||
|
||
// Strip port number if present
|
||
const hostname = host.split(':')[0]
|
||
|
||
// Production subdomain: slug.rentaldrivego.com
|
||
if (hostname.endsWith(`.${PRODUCTION_DOMAIN}`)) {
|
||
const subdomain = hostname.slice(0, hostname.length - PRODUCTION_DOMAIN.length - 1)
|
||
// Ignore www / bare domain
|
||
if (subdomain && subdomain !== 'www') return subdomain
|
||
}
|
||
|
||
return DEFAULT_SLUG
|
||
}
|
||
|
||
export function middleware(request: NextRequest) {
|
||
const host = request.headers.get('host')
|
||
const slug = resolveSlug(host)
|
||
|
||
const response = NextResponse.next()
|
||
|
||
// Forward the slug as a request header so server components can read it
|
||
// via `headers()` without needing cookies.
|
||
response.headers.set('x-company-slug', slug)
|
||
|
||
// Also set a short-lived cookie so client components / route handlers
|
||
// can pick it up without another round-trip.
|
||
response.cookies.set('x-company-slug', slug, {
|
||
path: '/',
|
||
maxAge: 60, // 1 minute – intentionally short; re-derived on every request
|
||
sameSite: 'lax',
|
||
httpOnly: false, // readable from JS for client-side fetches
|
||
})
|
||
|
||
return response
|
||
}
|
||
|
||
export const config = {
|
||
matcher: [
|
||
/*
|
||
* Match all request paths except:
|
||
* - _next/static (static files)
|
||
* - _next/image (image optimisation)
|
||
* - favicon.ico
|
||
* - public folder files (png, svg, jpg, …)
|
||
*/
|
||
'/((?!_next/static|_next/image|favicon.ico|.*\\.(?:png|jpg|jpeg|gif|svg|ico|webp|woff2?|ttf|otf|css|js)$).*)',
|
||
],
|
||
}
|