vite.config.ts

import path from "node:path";
import { defineConfig, type Plugin } from "vite";
import react from "@vitejs/plugin-react";
import { cloudflare } from "@cloudflare/vite-plugin";
import tailwindcss from "@tailwindcss/vite";

// Plugin to stub out server-only modules in the browser
function stubServerModules(): Plugin {
  return {
    name: "stub-server-modules",
    enforce: "pre",
    resolveId(id, importer, options) {
      // Stub gateway for all browser code (not SSR)
      if (id === "@ai-sdk/gateway" && !options.ssr) {
        return "\0virtual:@ai-sdk/gateway";
      }
    },
    load(id) {
      if (id === "\0virtual:@ai-sdk/gateway") {
        // Provide stub exports for all gateway exports (not used in browser)
        return `
          export const gateway = null;
          export const createGateway = () => { throw new Error('Gateway is not available in browser'); };
          export const createGatewayProvider = () => { throw new Error('Gateway is not available in browser'); };

          // Stub error classes
          class GatewayError extends Error {}
          export class GatewayAuthenticationError extends GatewayError {}
          export class GatewayInvalidRequestError extends GatewayError {}
          export class GatewayRateLimitError extends GatewayError {}
          export class GatewayModelNotFoundError extends GatewayError {}
          export class GatewayInternalServerError extends GatewayError {}
          export class GatewayResponseError extends GatewayError {}

          // Stub helper functions
          export const asGatewayError = (error) => error;
          export const wrapGatewayError = (error) => error;
        `;
      }
    }
  };
}

export default defineConfig({
  plugins: [cloudflare(), react(), tailwindcss(), stubServerModules()],
  resolve: {
    alias: {
      "@": path.resolve(__dirname, "./src")
    }
  }
});