vite.config.ts

import { defineConfig, Plugin } from "vite";
import react from "@vitejs/plugin-react-swc";
import path from "path";
import { copyFileSync } from "fs";
import type { IncomingMessage, ServerResponse } from "http";
import { createServer } from "./server";

// https://vitejs.dev/config/
export default defineConfig(({ mode }) => ({
  // Use root path for custom domain (aof.sh)
  // If deploying to GitHub Pages subdomain, change to "/aofdotsh/"
  base: process.env.VITE_BASE_PATH || "/",
  server: {
    host: "::",
    port: 3040,
    fs: {
      allow: ["./client", "./shared"],
      deny: [".env", ".env.*", "*.{crt,pem}", "**/.git/**", "server/**"],
    },
  },
  build: {
    outDir: "dist/spa",
    rollupOptions: {
      input: {
        main: path.resolve(__dirname, "index.html"),
      },
    },
  },
  publicDir: "public",
  plugins: [react(), expressPlugin(), copy404Plugin()],
  resolve: {
    alias: {
      "@": path.resolve(__dirname, "./client"),
      "@shared": path.resolve(__dirname, "./shared"),
    },
  },
}));

function expressPlugin(): Plugin {
  return {
    name: "express-plugin",
    apply: "serve", // Only apply during development (serve mode)
    configureServer(server) {
      const app = createServer();

      // Add Express app as middleware to Vite dev server
      // Only handle /api routes, let Vite handle everything else
      server.middlewares.use((req: IncomingMessage, res: ServerResponse, next: () => void) => {
        if (req.url?.startsWith("/api")) {
          // Express is compatible with Connect-style middleware (IncomingMessage/ServerResponse)
          // Type assertion needed because Express types are more specific
          app(req as any, res as any, next);
        } else {
          next();
        }
      });
    },
  };
}

function copy404Plugin(): Plugin {
  return {
    name: "copy-404",
    apply: "build",
    closeBundle() {
      // Copy index.html to 404.html for GitHub Pages SPA routing
      const indexPath = path.resolve(__dirname, "dist/spa/index.html");
      const notFoundPath = path.resolve(__dirname, "dist/spa/404.html");
      try {
        copyFileSync(indexPath, notFoundPath);
        console.log("✓ Copied index.html to 404.html for GitHub Pages");
      } catch (error) {
        console.warn("Failed to copy index.html to 404.html:", error);
      }
    },
  };
}