import path from "pathe";
import { exists } from "../../util/exists.ts";
import type { Assets } from "../assets.ts";
import type { Bindings } from "../bindings.ts";
import { Vite, type ViteProps } from "../vite/vite.ts";
import type { Worker } from "../worker.ts";

export interface TanStackStartProps<
  B extends Bindings,
  RPC extends Rpc.WorkerEntrypointBranded = Rpc.WorkerEntrypointBranded,
> extends ViteProps<B, RPC> {}

// don't allow the ASSETS to be overriden
export type TanStackStart<
  B extends Bindings,
  RPC extends Rpc.WorkerEntrypointBranded = Rpc.WorkerEntrypointBranded,
> = B extends { ASSETS: any } ? never : Worker<B & { ASSETS: Assets }, RPC>;

export async function TanStackStart<
  B extends Bindings,
  RPC extends Rpc.WorkerEntrypointBranded = Rpc.WorkerEntrypointBranded,
>(
  id: string,
  props?: Partial<TanStackStartProps<B, RPC>>,
): Promise<TanStackStart<B, RPC>> {
  const main =
    props?.wrangler?.main ??
    ((await exists(path.resolve(props?.cwd ?? process.cwd(), "src/server.ts")))
      ? "src/server.ts"
      : undefined);
  return await Vite(id, {
    entrypoint: "dist/server/index.js",
    assets: "dist/client",
    compatibility: "node",
    noBundle: true,
    spa: false,
    ...props,
    wrangler: {
      ...props?.wrangler,
      main,
      transform: async (spec) => {
        const transformed = (await props?.wrangler?.transform?.(spec)) ?? spec;
        if (!main) {
          transformed.main = "@tanstack/react-start/server-entry";
        }
        return transformed;
      },
    },
  });
}