vite.config.ts

import { defineConfig, type Plugin } from "vite";
import react from "@vitejs/plugin-react";
import type { IncomingMessage, ServerResponse } from "node:http";

function readBody(req: IncomingMessage): Promise<string> {
  return new Promise((resolve, reject) => {
    const chunks: Buffer[] = [];
    req.on("data", (c) => chunks.push(Buffer.from(c)));
    req.on("end", () => resolve(Buffer.concat(chunks).toString("utf8")));
    req.on("error", reject);
  });
}

function apiProxyPlugin(): Plugin {
  const handler = async (req: IncomingMessage, res: ServerResponse) => {
    if (req.method === "OPTIONS") {
      res.statusCode = 204;
      res.end();
      return;
    }
    try {
      const raw = await readBody(req);
      const payload = JSON.parse(raw || "{}") as {
        url?: string;
        method?: string;
        headers?: Record<string, string>;
        body?: string;
      };
      if (!payload.url) {
        res.statusCode = 400;
        res.setHeader("content-type", "application/json");
        res.end(JSON.stringify({ error: "url required" }));
        return;
      }
      const headers = { ...(payload.headers || {}) };
      // Never forward hop-by-hop / browser-only headers
      delete headers["host"];
      delete headers["origin"];
      delete headers["referer"];
      delete headers["content-length"];

      const upstream = await fetch(payload.url, {
        method: payload.method || "GET",
        headers,
        body:
          payload.body &&
          !["GET", "HEAD"].includes((payload.method || "GET").toUpperCase())
            ? payload.body
            : undefined,
      });

      const text = await upstream.text();
      let body: unknown = text;
      try {
        body = JSON.parse(text);
      } catch {
        /* keep text */
      }

      const outHeaders: Record<string, string> = {};
      upstream.headers.forEach((v, k) => {
        outHeaders[k] = v;
      });

      res.statusCode = 200;
      res.setHeader("content-type", "application/json");
      res.end(
        JSON.stringify({
          status: upstream.status,
          headers: outHeaders,
          body,
        }),
      );
    } catch (e) {
      res.statusCode = 500;
      res.setHeader("content-type", "application/json");
      res.end(
        JSON.stringify({
          status: 0,
          error: e instanceof Error ? e.message : "proxy failed",
          body: null,
        }),
      );
    }
  };

  return {
    name: "chisel-api-proxy",
    configureServer(server) {
      server.middlewares.use("/api/proxy", (req, res) => {
        void handler(req, res);
      });
    },
    configurePreviewServer(server) {
      server.middlewares.use("/api/proxy", (req, res) => {
        void handler(req, res);
      });
    },
  };
}

export default defineConfig({
  plugins: [react(), apiProxyPlugin()],
});