makefx 1.6.6 → 1.6.7

package/README.md

@@ -61,4 +61,4 @@ IDs, or runs. Retired commands and aliases are not redirected.
 
 Run `makefx --help` or `makefx <noun> --help` for current options.
 
-Version: 1.6.6
+Version: 1.6.7

package/makefx.mjs

@@ -4442,7 +4442,7 @@ require_websocket_server();
 var wrapper_default = import_websocket.default;
 //#endregion
 //#region src/cli/version.ts
-var CLI_VERSION = "1.6.6+b1c0c6d222d7";
+var CLI_VERSION = "1.6.7+0439271d1461";
 var CLI_VERSION_HEADER = "X-MakeFX-CLI-Version";
 function cliVersionHeaders() {
 	return {
@@ -4461,6 +4461,7 @@ function cliVersionHeaders() {
 var GENERATION_REQUEST_TIMEOUT_MS = 3e5;
 var VIDEO_GENERATION_REQUEST_TIMEOUT_MS = 72e4;
 var VARIANT_SYNC_POLL_INTERVAL_MS = 15e3;
+var GENERATION_RECONNECT_RETRY_MS = 1e3;
 function getGenerationRequestTimeoutMs(mediaKind) {
 	return mediaKind === "video" ? VIDEO_GENERATION_REQUEST_TIMEOUT_MS : GENERATION_REQUEST_TIMEOUT_MS;
 }
@@ -4493,6 +4494,8 @@ var WebSocketClient = class WebSocketClient {
 	env;
 	spaceId;
 	connectionLoggingEnabled = false;
+	disconnectRequested = false;
+	generationReconnect = null;
 	chatHandlers = /* @__PURE__ */ new Map();
 	generateHandlers = /* @__PURE__ */ new Map();
 	earlyTerminalVariants = /* @__PURE__ */ new Map();
@@ -4535,6 +4538,7 @@ var WebSocketClient = class WebSocketClient {
 	* Connect to the WebSocket endpoint
 	*/
 	async connect() {
+		this.disconnectRequested = false;
 		return new Promise((resolve, reject) => {
 			const protocol = this.baseUrl.startsWith("https") ? "wss" : "ws";
 			const url = `${protocol}://${this.baseUrl.replace(/^https?:\/\//, "")}/api/spaces/${this.spaceId}/ws`;
@@ -4563,7 +4567,7 @@ var WebSocketClient = class WebSocketClient {
 			});
 			this.ws.on("close", (code, reason) => {
 				if (this.connectionLoggingEnabled) console.log(`[WebSocketClient] Disconnected: ${code} - ${reason}`);
-				this.ws = null;
+				this.handleSocketClose();
 			});
 		});
 	}
@@ -4571,11 +4575,35 @@ var WebSocketClient = class WebSocketClient {
 	* Disconnect from the WebSocket
 	*/
 	disconnect() {
+		this.disconnectRequested = true;
 		if (this.ws) {
 			this.ws.close();
 			this.ws = null;
 		}
 	}
+	handleSocketClose() {
+		this.ws = null;
+		if (!this.disconnectRequested && this.hasPendingGenerationWaits()) this.reconnectPendingGeneration();
+	}
+	hasPendingGenerationWaits() {
+		return this.generateHandlers.size > 0 || this.variantCompletionHandlers.size > 0;
+	}
+	reconnectPendingGeneration() {
+		if (this.generationReconnect || this.disconnectRequested || !this.hasPendingGenerationWaits()) return;
+		this.generationReconnect = this.runGenerationReconnect().finally(() => {
+			this.generationReconnect = null;
+		});
+	}
+	async runGenerationReconnect() {
+		while (!this.disconnectRequested && this.hasPendingGenerationWaits()) try {
+			await this.connect();
+			if (this.hasPendingGenerationWaits()) this.requestSync();
+			return;
+		} catch {
+			if (this.disconnectRequested || !this.hasPendingGenerationWaits()) return;
+			await new Promise((resolve) => setTimeout(resolve, GENERATION_RECONNECT_RETRY_MS));
+		}
+	}
 	/**
 	* Set error handler
 	*/

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "makefx",
-  "version": "1.6.6",
+  "version": "1.6.7",
   "description": "Command-line interface for AI-assisted game asset production with Make Effects.",
   "license": "MIT",
   "type": "module",