@ait-co/console-cli 0.1.17 → 0.1.18

package/dist/cli.mjs

@@ -1,13 +1,14 @@
 #!/usr/bin/env node
 import { defineCommand, runMain } from "citty";
-import { access, chmod, mkdir, mkdtemp, readFile, rename, rm, unlink, writeFile } from "node:fs/promises";
+import { access, chmod, copyFile, mkdir, mkdtemp, readFile, rename, rm, unlink, writeFile } from "node:fs/promises";
 import { basename, dirname, isAbsolute, join, resolve, win32 } from "node:path";
 import { homedir, tmpdir } from "node:os";
 import { unzipSync } from "fflate";
 import { parse } from "yaml";
 import { imageSize } from "image-size";
-import { spawn } from "node:child_process";
+import { execFile, spawn } from "node:child_process";
 import { constants, createReadStream } from "node:fs";
+import { promisify } from "node:util";
 import { createHash } from "node:crypto";
 //#region src/api/http.ts
 var TossApiError = class extends Error {
@@ -720,7 +721,8 @@ const ExitCode = {
 	ApiError: 17,
 	UpgradeUnavailable: 20,
 	UpgradeAlreadyLatest: 21,
-	UpgradeChecksumFailed: 22
+	UpgradeChecksumFailed: 22,
+	UpgradeSmokeTestFailed: 23
 };
 //#endregion
 //#region src/flush.ts
@@ -5559,13 +5561,14 @@ function resolveVersion() {
 		if (typeof injected === "string" && injected.length > 0) return injected;
 	} catch {}
 	try {
-		return "0.1.17";
+		return "0.1.18";
 	} catch {}
 	return "0.0.0-dev";
 }
 const VERSION = resolveVersion();
 //#endregion
 //#region src/commands/upgrade.ts
+const execFileP = promisify(execFile);
 function isStandaloneBinary() {
 	return basename(process.execPath).toLowerCase().startsWith("aitcc");
 }
@@ -5724,12 +5727,26 @@ const upgradeCommand = defineCommand({
 			process.exit(ExitCode.UpgradeChecksumFailed);
 		}
 		if (!args.json) process.stdout.write("Checksum OK.\n");
+		const backupPath = process.platform === "win32" ? null : `${exePath}.bak.${Date.now()}`;
+		if (backupPath) try {
+			await copyFile(exePath, backupPath);
+		} catch (err) {
+			await unlink(stagingPath).catch(() => {});
+			emitError({
+				reason: "backup-failed",
+				message: err.message,
+				exePath,
+				backupPath
+			}, `Failed to create rollback backup at ${backupPath}: ${err.message}`);
+			process.exit(ExitCode.Generic);
+		}
 		try {
 			if (process.platform === "win32") {
 				await rename(exePath, `${exePath}.old`);
 				await rename(stagingPath, exePath);
 			} else await rename(stagingPath, exePath);
 		} catch (err) {
+			if (backupPath) await unlink(backupPath).catch(() => {});
 			emitError({
 				reason: "replace-failed",
 				message: err.message,
@@ -5738,6 +5755,45 @@ const upgradeCommand = defineCommand({
 			}, `Failed to replace binary at ${exePath}: ${err.message}`);
 			process.exit(ExitCode.Generic);
 		}
+		let smokeFailure = null;
+		try {
+			const { stdout } = await execFileP(exePath, ["--version"], {
+				timeout: 1e4,
+				windowsHide: true
+			});
+			if (!stdout.trim()) smokeFailure = "empty stdout from --version";
+		} catch (err) {
+			smokeFailure = err.message;
+		}
+		if (smokeFailure) {
+			let rollbackError = null;
+			let recoveryHint = null;
+			try {
+				if (process.platform === "win32") {
+					try {
+						await unlink(exePath);
+					} catch (err) {
+						recoveryHint = `Failed to remove broken binary at ${exePath}; remove it manually then rename ${exePath}.old back to ${exePath}.`;
+						throw err;
+					}
+					await rename(`${exePath}.old`, exePath);
+				} else if (backupPath) await rename(backupPath, exePath);
+			} catch (err) {
+				rollbackError = err.message;
+				if (!recoveryHint) recoveryHint = process.platform === "win32" ? `Rename ${exePath}.old back to ${exePath} to restore the previous binary.` : `Rename ${backupPath} back to ${exePath} to restore the previous binary.`;
+			}
+			emitError({
+				reason: "smoke-test-failed",
+				message: smokeFailure,
+				exePath,
+				...rollbackError ? {
+					rollbackError,
+					backupPath
+				} : { rolledBack: true }
+			}, rollbackError ? `New binary failed --version smoke test: ${smokeFailure}\nRollback also failed: ${rollbackError}\n${recoveryHint}` : `New binary failed --version smoke test: ${smokeFailure}\nReverted to previous binary.`);
+			process.exit(ExitCode.UpgradeSmokeTestFailed);
+		}
+		if (backupPath) await unlink(backupPath).catch(() => {});
 		emit({
 			ok: true,
 			status: "upgraded",