@senguoyun/dsh-arkme 0.1.12 → 0.1.14

package/lib/extension-profile-restart-helper.js

@@ -1,5 +1,5 @@
 import { n as prepareProfilePackageManager, t as localExtensionPnpmArgs } from "./profile-package-manager-DvzGgkqe.js";
-import { t as ArkmeExtensionInstallStore } from "./install-store-D6P-dH2i.js";
+import { t as ArkmeExtensionInstallStore } from "./install-store-Sn3Ccxfu.js";
 import { closeSync, openSync } from "node:fs";
 import { relative, resolve } from "node:path";
 import { fileURLToPath } from "node:url";
@@ -11,6 +11,9 @@ const HEALTH_TIMEOUT_MS = 45e3;
 function stringValue(value) {
 	return typeof value === "string" && value.trim() !== "" && value.length <= 4096 ? value : void 0;
 }
+function isPluginUpdateRestartPlan(value) {
+	return value !== null && typeof value === "object" && !Array.isArray(value) && typeof value.targetArtifactPath === "string" && typeof value.targetVersion === "string" && typeof value.jobId === "string";
+}
 function parseExtensionProfileRestartPlan(value) {
 	if (value === null || typeof value !== "object") throw new Error("extension restart plan must be an object");
 	const source = value;
@@ -188,9 +191,22 @@ async function rollbackManagedExtensionProfileRestart(planPath, overrides = {})
 }
 async function main() {
 	const mode = process.argv[2];
-	const planPath = mode === "--managed-finalize" || mode === "--managed-rollback" ? process.argv[3] : mode;
+	const managed = mode === "--managed-finalize" || mode === "--managed-rollback";
+	const planPath = managed ? process.argv[3] : mode;
 	if (planPath === void 0) throw new Error("extension restart plan path is required");
-	closeSync(openSync(parseExtensionProfileRestartPlan(JSON.parse(await readFile(planPath, "utf8"))).logPath, "a", 384));
+	const rawPlan = JSON.parse(await readFile(planPath, "utf8"));
+	if (managed && isPluginUpdateRestartPlan(rawPlan)) {
+		const pluginUpdate = await import("./plugin-updater-helper.js");
+		if (mode === "--managed-finalize") {
+			const healthUrl = process.argv[4];
+			if (healthUrl === void 0) throw new Error("managed plugin update health URL is required");
+			await pluginUpdate.finalizeManagedPluginUpdate(planPath, healthUrl);
+			return;
+		}
+		await pluginUpdate.rollbackManagedPluginUpdate(planPath);
+		return;
+	}
+	closeSync(openSync(parseExtensionProfileRestartPlan(rawPlan).logPath, "a", 384));
 	if (mode === "--managed-finalize") {
 		const healthUrl = process.argv[4];
 		if (healthUrl === void 0) throw new Error("managed restart health URL is required");