import chalk from "chalk";
import { Command } from "commander";
import { importJson } from "./lib/importJson.js";
import { Server } from "./lib/Server.js";

async function runStageOperations(
	packageName: string,
	version: string,
	hostname: string,
	operations = ["download"],
	state: { bundle?: any; namespace?: any } = {},
) {
	let lastSuccessfulStep = "none";
	try {
		let server = new Server(hostname);

		let bundle = state.bundle ?? (await server.getBundle(packageName));
		if (!bundle || !bundle.ID) throw Error(`Bundle with name '${packageName}' not found.`);

		state.bundle = bundle;

		let { ID: id, Project_ID, Name, Namespace_ID } = bundle;
		if (!Namespace_ID) {
			throw Error(`Bundle '${packageName}' is not assigned to a namespace`);
		}

		let namespace = state.namespace ?? (await server.getNamespace(Namespace_ID));
		server.logServerMessage(
			`Found bundle '${Name}' with Namespace/Project/Version: ${namespace.Name}/${id}/${Project_ID}`,
		);
		lastSuccessfulStep = "getBundleInformation";

		state.namespace = namespace;

		if (operations.includes("publish")) {
			await server.publishBundle(Project_ID, id, `${packageName}@${version}`);
			lastSuccessfulStep = "publish";
		}

		if (operations.includes("download")) {
			await server.download(namespace.Name);
			lastSuccessfulStep = "download";
		}

		if (operations.includes("generate")) {
			await server.generate(namespace.ID);
			lastSuccessfulStep = "generate";
		}

		if (operations.includes("apply")) {
			await server.assertOnlyOneTransaction("apply", packageName, namespace.Name);
			await server.apply(namespace.ID);
			lastSuccessfulStep = "apply";
		}

		if (operations.includes("deploy")) {
			await server.assertOnlyOneTransaction("deploy", packageName, namespace.Name);
			await server.deploy(namespace.ID);
			lastSuccessfulStep = "deploy";
		}
	} catch (error: any) {
		error.lastSuccessfulStep = lastSuccessfulStep;
		throw error;
	}
}

async function publishFromDev(packageName?: string) {
	try {
		let name: string;
		let version: string;
		if (!packageName) {
			const pkg = await importJson("./package.json", true);
			name = pkg.name;
			version = pkg.version;
		} else {
			name = packageName;
			version = "latest";
		}

		let crossServerState = {};

		await runStageOperations(name, version, "dev.obet.no", ["publish", "generate", "deploy"], crossServerState);
		await runStageOperations(name, version, "stage.obet.no", ["download", "apply", "deploy"], crossServerState);
		await runStageOperations(name, version, "test.obet.no", ["download"], crossServerState);
	} catch (error: any) {
		console.log(
			chalk.red(error.message),
			error.lastSuccessfulStep && `(Last step completed: ${error.lastSuccessfulStep})`,
		);
		process.exit(1);
	}
}

const cliPkg = await importJson("../package.json");
let program = new Command();
program
	.version(cliPkg.version)
	.argument("[package]", "Specify name of the package to publish, otherwise the name in package.json is used")
	.action(publishFromDev);

await program.parseAsync();