import { existsSync, readFileSync } from 'node:fs'
import { resolve } from 'node:path'
import { defineConfig, type Plugin } from 'vite'
import react from '@vitejs/plugin-react'

function serveBuiltDocs(): Plugin {
  return {
    name: 'serve-built-docs',
    configureServer(server) {
      const publicDir = resolve('public')

      server.middlewares.use((req, res, next) => {
        const url = req.url?.split('?')[0]
        if (!url) return next()

        const filePath = resolve(publicDir, url.slice(1))

        if ((url.endsWith('.txt') || url.endsWith('.md')) && existsSync(filePath)) {
          res.setHeader(
            'Content-Type',
            url.endsWith('.md')
              ? 'text/markdown; charset=utf-8'
              : 'text/plain; charset=utf-8',
          )
          res.end(readFileSync(filePath))
          return
        }

        const [, site] = url.split('/')
        if (!site || site.includes('.')) return next()

        const siteDir = resolve(publicDir, site)
        if (!existsSync(resolve(siteDir, 'index.html'))) return next()

        if (url === `/${site}`) {
          req.url = `/${site}/index.html`
          return next()
        }

        if (url.endsWith('/')) {
          if (existsSync(resolve(filePath, 'index.html'))) {
            req.url = `${url}index.html`
            return next()
          }
        }

        if (!url.split('/').pop()?.includes('.')) {
          if (existsSync(`${filePath}.html`)) {
            req.url = `${url}.html`
            return next()
          }
        }

        next()
      })
    },
  }
}

// https://vite.dev/config/
export default defineConfig({
  plugins: [react(), serveBuiltDocs()],
})