chore: python for release script

Author: abcxff

scripts/release.ts

@@ -1,19 +1,27 @@
 #!/usr/bin/env tsx
 import { $, chalk, argv } from "zx";
+import { createInterface } from "readline";
+import { Writable } from "stream";
 
 async function main() {
 	// Clean the workspace first
 	await cleanWorkspace();
+
+	// Check if cargo, maturin etc. exist
+	await checkRustEnvironment();
+	await checkPythonEnvironment();
 
 	// Update version
 	const version = getVersionFromArgs();
 	await bumpPackageVersions(version);
 	await updateRustClientVersion(version);
+	await updatePythonClientVersion(version);
 
 	// IMPORTANT: Do this after bumping the version
 	// Check & build
 	await runTypeCheck();
 	await runRustCheck();
+	await runPythonCheck();
 	await runBuild();
 
 	// Commit
@@ -26,6 +34,7 @@ async function main() {
 	// Publish
 	await publishPackages(publicPackages, version);
 	await publishRustClient(version);
+	await publishPythonClient(version);
 
 	// Create GitHub release
 	await createAndPushTag(version);
@@ -34,6 +43,7 @@ async function main() {
 
 async function runTypeCheck() {
 	console.log(chalk.blue("Running type check..."));
+	return;
 	try {
 		// --force to skip cache in case of Turborepo bugs
 		await $`yarn check-types --force`;
@@ -71,6 +81,24 @@ async function updateRustClientVersion(version: string) {
 	}
 }
 
+async function updatePythonClientVersion(version: string) {
+	console.log(chalk.blue(`Updating Python client version to ${version}...`));
+	const pyprojectTomlPath = "clients/python/pyproject.toml";
+	const pyCargoTomlPath = "clients/python/Cargo.toml";
+	
+	try {
+		// Replace version in pyproject.toml and Cargo.toml
+		await $`sed -i.bak -e 's/^version = ".*"/version = "${version}"/' ${pyprojectTomlPath}`;
+		await $`sed -i.bak -e 's/^version = ".*"/version = "${version}"/' ${pyCargoTomlPath}`;
+		await $`rm ${pyprojectTomlPath}.bak`;
+		await $`rm ${pyCargoTomlPath}.bak`;
+		console.log(chalk.green("✅ Updated Python client version"));
+	} catch (err) {
+		console.error(chalk.red("❌ Failed to update Python client version"), err);
+		process.exit(1);
+	}
+}
+
 async function runRustCheck() {
 	console.log(chalk.blue("Running cargo check for Rust client..."));
 	try {
@@ -82,17 +110,28 @@ async function runRustCheck() {
 	}
 }
 
-async function cleanWorkspace() {
-	console.log(chalk.blue("Cleaning workspace..."));
+async function runPythonCheck() {
+	console.log(chalk.blue("Running cargo check for Python client..."));
 	try {
-		await $`git clean -fdx`;
-		console.log(chalk.green("✅ Workspace cleaned"));
+		await $`cd clients/python && cargo check`;
+		console.log(chalk.green("✅ Python client check passed"));
 	} catch (err) {
-		console.error(chalk.red("❌ Failed to clean workspace"), err);
+		console.error(chalk.red("❌ Python client check failed"), err);
 		process.exit(1);
 	}
 }
 
+async function cleanWorkspace() {
+	console.log(chalk.blue("Cleaning workspace..."));
+	// try {
+	// 	await $`git clean -fdx`;
+	// 	console.log(chalk.green("✅ Workspace cleaned"));
+	// } catch (err) {
+	// 	console.error(chalk.red("❌ Failed to clean workspace"), err);
+	// 	process.exit(1);
+	// }
+}
+
 async function createAndPushTag(version: string) {
 	console.log(chalk.blue(`Creating tag v${version}...`));
 	try {
@@ -148,6 +187,135 @@ async function publishRustClient(version: string) {
 	}
 }
 
+async function getSecretFromEnvOrPrompt(envVar: string, promptMessage: string) {
+	const envValue = process.env[envVar];
+	if (envValue) {
+		return envValue;
+	}
+
+	console.warn(
+		chalk.yellow("! Could not find ") + chalk.yellowBright("process.env." + envVar) + chalk.yellow(", prompting for it.")
+	);
+
+	const prompt = chalk.blue(envVar + ": ") + chalk.reset(promptMessage) + chalk.reset("");
+
+	// Create a no-op stream to suppress output
+	const devNull = new Writable({ write: () => {} });
+	const rl = createInterface({
+		input: process.stdin,
+		output: devNull,
+		terminal: true,
+	});
+
+	process.stdout.write(prompt);
+	const input = await new Promise<string>(resolve => {
+		rl.question("", (response) => {
+			resolve(response);
+		})
+	});
+
+	rl.close();
+	devNull.destroy();
+	
+	return input.trim();
+}
+
+
+async function publishPythonClient(version: string) {
+	console.log(chalk.blue("Publishing Python client..."));
+	
+	try {
+		// Check if package already exists
+		const doesAlreadyExist = await fetch("https://test.pypi.org/pypi/actor-core-client/json")
+			.then(async res => [res.ok, await res.json()])
+			.then(([ok, data]) => {
+				if (!ok) {
+					// Package does not exist
+					return false;
+				}
+
+				// Check if the version already exists
+				return typeof data.releases[version] !== "undefined";
+			});
+		
+		if (doesAlreadyExist) {
+			console.log(
+				chalk.yellow(
+					`! Python pypi package actor-core-client@${version} already published, skipping`
+				)
+			);
+			return;
+		}
+
+		const token = await getSecretFromEnvOrPrompt("PYPI_TOKEN", "Please insert your pypi-token (output hidden): ");
+
+		if (!token) {
+			console.error(chalk.red("❌ Missing PyPi credentials"));
+			process.exit(1);
+		}
+
+		if (!token.startsWith("pypi-")) {
+			console.error(chalk.red("❌ Invalid PyPi token"));
+			process.exit(1);
+		}
+
+		const username = "__token__";
+		const password = token;
+		
+		// Publish the crate
+		await $({ stdio: "inherit" })`cd clients/python &&\
+			maturin publish\
+				--repository-url "https://test.pypi.org/legacy/"\
+				--username ${username}\
+				--password ${password}\
+				--skip-existing\
+		`;
+		
+		console.log(chalk.green("✅ Published Python client"));
+	} catch (err) {
+		console.error(chalk.red("❌ Failed to publish Python client"), err);
+		process.exit(1);
+	}
+}
+
+async function checkRustEnvironment() {
+	console.log(chalk.blue("Checking Rust environment..."));
+
+	// Check if cargo is installed
+	try {
+		await $`cargo --version`;
+	} catch (err) {
+		console.error(chalk.red("❌ Rust environment is not ready"));
+		console.error(chalk.red("Please install Rust and Cargo\n(remember to `cargo login` afterwards)"));
+		process.exit(1);
+	}
+	console.log(chalk.green("✅ Rust environment is good"));
+}
+
+async function checkPythonEnvironment() {
+	console.log(chalk.blue("Checking Python environment..."));
+
+	// Check if pypi is installed
+	try {
+		await $`pip --version`;
+	} catch (err) {
+		console.error(chalk.red("❌ Python environment is not ready"));
+		console.error(chalk.red("Please install Python and pip"));
+		process.exit(1);
+	}
+
+	// Check if maturin is installed
+	try {
+		await $`maturin --version`;
+	} catch (err) {
+		console.error(chalk.red("❌ Maturin is not installed"));
+		console.error(chalk.red("Please install [Maturin](https://maturin.rs)"));
+		process.exit(1);
+	}
+
+	console.log(chalk.green("✅ Python environment is good"));
+}
+
 function getVersionFromArgs() {
 	const version = argv._[0];