import compression from "compression"
import express, { Router } from "express"
import { readFile } from "node:fs/promises"
import { join, resolve } from "node:path"
import { createServer as createViteServer } from "vite"
import type * as entryModule from "./src/entry.server"

async function createDevelopmentRouter() {
  const vite = await createViteServer({
    server: { middlewareMode: "ssr" },
  })

  return Router()
    .use(vite.middlewares)
    .use("*", async (req, res) => {
      const url = req.originalUrl

      try {
        const { render } = (await vite.ssrLoadModule(
          "/src/entry.server.tsx",
        )) as typeof entryModule

        const htmlTemplate = await readFile(
          join(vite.config.root, "index.html"),
          "utf8",
        )

        const html = await vite.transformIndexHtml(
          url,
          render(url, htmlTemplate),
        )

        res.status(200).set({ "Content-Type": "text/html" }).end(html)
      } catch (error: any) {
        vite.ssrFixStacktrace(error)
        console.error(error)
        res.status(500).end(error.stack || error.message)
      }
    })
}

function createProductionRouter() {
  return Router()
    .use(compression())
    .use(express.static(resolve("dist/client"), { index: false }))
    .use("*", async (req, res) => {
      try {
        const { render }: typeof entryModule = await import(
          "./dist/server/entry.server"
        )

        const htmlTemplate = await readFile("dist/client/index.html", "utf8")

        res
          .status(200)
          .set({ "Content-Type": "text/html" })
          .end(render(req.originalUrl, htmlTemplate))
      } catch (error: any) {
        console.error(error)
        res.status(500).end(error.stack || error.message)
      }
    })
}

const app = express()

if (process.env.NODE_ENV === "production") {
  app.use(createProductionRouter())
} else {
  app.use(await createDevelopmentRouter())
}

const port = process.env.PORT || 3000
app.listen(port, () => {
  console.log(`listening on http://localhost:${port}`)
})