import path from "node:path";
import fs from "fs-extra";
import {
	type CallExpression,
	Node,
	type ObjectLiteralExpression,
	SyntaxKind,
} from "ts-morph";
import type { PackageManager } from "../../../types";
import { addPackageDependency } from "../../../utils/add-package-deps";
import { ensureArrayProperty, tsProject } from "../../../utils/ts-morph";

export async function setupTanstackStartWorkersDeploy(
	projectDir: string,
	packageManager: PackageManager,
) {
	const webAppDir = path.join(projectDir, "apps/web");
	if (!(await fs.pathExists(webAppDir))) return;

	await addPackageDependency({
		devDependencies: ["wrangler"],
		projectDir: webAppDir,
	});

	const pkgPath = path.join(webAppDir, "package.json");
	if (await fs.pathExists(pkgPath)) {
		const pkg = await fs.readJson(pkgPath);
		pkg.scripts = {
			...pkg.scripts,
			deploy: `${packageManager} run build && wrangler deploy`,
			"cf-typegen": "wrangler types --env-interface Env",
		};
		await fs.writeJson(pkgPath, pkg, { spaces: 2 });
	}

	const viteConfigPath = path.join(webAppDir, "vite.config.ts");
	if (!(await fs.pathExists(viteConfigPath))) return;

	const sourceFile = tsProject.addSourceFileAtPathIfExists(viteConfigPath);
	if (!sourceFile) return;

	const defineCall = sourceFile
		.getDescendantsOfKind(SyntaxKind.CallExpression)
		.find((expr) => {
			const expression = expr.getExpression();
			return (
				Node.isIdentifier(expression) && expression.getText() === "defineConfig"
			);
		}) as CallExpression | undefined;

	if (!defineCall) return;

	const configObj = defineCall.getArguments()[0] as
		| ObjectLiteralExpression
		| undefined;
	if (!configObj) return;

	const pluginsArray = ensureArrayProperty(configObj, "plugins");

	const tanstackPluginIndex = pluginsArray
		.getElements()
		.findIndex((el) => el.getText().includes("tanstackStart("));

	const tanstackPluginText = 'tanstackStart({ target: "cloudflare-module" })';

	if (tanstackPluginIndex === -1) {
		pluginsArray.addElement(tanstackPluginText);
	} else {
		pluginsArray
			.getElements()
			[tanstackPluginIndex].replaceWithText(tanstackPluginText);
	}

	await tsProject.save();
}