import type { NextFunction, Response } from 'express'

import languages from '@/languages/lib/languages-server'
import { defaultCacheControl } from '@/frame/middleware/cache-control'
import { ExtendedRequest } from '@/types'

const redirectPatterns = Object.values(languages)
  .map((language) => language.redirectPatterns || [])
  .flat()
// If the enabled languages have `.redirectPatterns`, combine them
// into one which we only need to use to determine if we should bother
// doing the redirect at all.
const combinedRedirectPatternRegex =
  redirectPatterns.length > 0
    ? new RegExp(redirectPatterns.map((rex) => rex.source).join('|'))
    : null

const allRedirectPatterns = Object.values(languages)
  .map((language) =>
    (language.redirectPatterns || []).map((redirectPattern) => [language.code, redirectPattern]),
  )
  .flat() as [string, RegExp][] // Seems TypeScript didn't understand the .flat()

// This middleware handles redirects for mistyped language codes
//
// Examples:
// /jp*    -> /ja*
// /zh-TW* -> /zh*
export default function languageCodeRedirects(
  req: ExtendedRequest,
  res: Response,
  next: NextFunction,
) {
  // Only in the unlikely event that the `req.path` starts with one of these
  // prefixes do we bother looking up what the redirect should be.
  if (req.path.startsWith('/_next/static')) return next()
  if (!combinedRedirectPatternRegex) return next()
  if (!combinedRedirectPatternRegex.test(req.path)) return next()

  // This loop is almost never ever used so it doesn't have to be
  // particularly smart or fast.
  const matched = allRedirectPatterns.find(([, pattern]) => pattern.test(req.path))
  if (matched) {
    const [code, pattern] = matched
    if (code && pattern) {
      defaultCacheControl(res)
      return res.redirect(301, req.path.replace(pattern, `/${code}`))
    }
  }
  return next()
}