castle-web-sdk 0.4.2 → 0.4.4

package/dist/castle.d.ts

@@ -2,7 +2,7 @@ export { isEdit } from "./context";
 export { CastleError } from "./errors";
 export { Leaderboard } from "./leaderboard";
 export type { LeaderboardData, LeaderboardEntry, LeaderboardOptions, LeaderboardScope, LeaderboardSort, } from "./leaderboard";
-export { CARD_RATIO, initCard, setup, writeFile } from "./runtime";
+export { CARD_RATIO, initCard, onBeforeRestart, setup, writeFile } from "./runtime";
 export { SharedStorage, Storage } from "./storage";
 export { Time } from "./time";
 export type { CastleClockZone, CastleDateParts, CastleTimeApi } from "./time";

package/dist/castle.js

@@ -2,7 +2,7 @@
 export { isEdit } from "./context";
 export { CastleError } from "./errors";
 export { Leaderboard } from "./leaderboard";
-export { CARD_RATIO, initCard, setup, writeFile } from "./runtime";
+export { CARD_RATIO, initCard, onBeforeRestart, setup, writeFile } from "./runtime";
 export { SharedStorage, Storage } from "./storage";
 export { Time } from "./time";
 export { User } from "./user";

package/dist/runtime.d.ts

@@ -9,4 +9,5 @@ interface LocalResponse {
 export declare function setup(): void;
 export declare function writeFile(path: string, contents: string): Promise<LocalResponse>;
 export declare function initCard(): HTMLDivElement;
+export declare function onBeforeRestart(hook: () => void | Promise<void>): () => void;
 export {};

package/dist/runtime.js

@@ -275,12 +275,38 @@ function handleLocalMessage(msg) {
         });
     }
     else if (msg.type === "restart") {
-        location.reload();
+        scheduleRestart();
     }
     else if (msg.type === "write_file_response") {
         resolveLocalRequest(msg);
     }
 }
+// Restart (from `castle-web restart` / task agents) is debounced so a burst
+// of reload requests -- several tasks finishing close together -- produces
+// one reload. Before reloading, registered hooks run (the kit editor flushes
+// its debounced unsaved edits there) so in-flight work isn't lost.
+const RESTART_DEBOUNCE_MS = 1500;
+let restartTimer = null;
+const beforeRestartHooks = new Set();
+export function onBeforeRestart(hook) {
+    beforeRestartHooks.add(hook);
+    return () => beforeRestartHooks.delete(hook);
+}
+function scheduleRestart() {
+    if (restartTimer !== null)
+        clearTimeout(restartTimer);
+    restartTimer = setTimeout(() => {
+        void (async () => {
+            try {
+                await Promise.all([...beforeRestartHooks].map((hook) => hook()));
+            }
+            catch {
+                // a failed flush shouldn't block the reload
+            }
+            location.reload();
+        })();
+    }, RESTART_DEBOUNCE_MS);
+}
 function localWsUrl(path) {
     const url = new URL(path, location.href);
     url.protocol = location.protocol === "https:" ? "wss:" : "ws:";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "castle-web-sdk",
-  "version": "0.4.2",
+  "version": "0.4.4",
   "type": "module",
   "main": "dist/castle.js",
   "types": "dist/castle.d.ts",

package/src/castle.ts

@@ -10,7 +10,7 @@ export type {
   LeaderboardScope,
   LeaderboardSort,
 } from "./leaderboard";
-export { CARD_RATIO, initCard, setup, writeFile } from "./runtime";
+export { CARD_RATIO, initCard, onBeforeRestart, setup, writeFile } from "./runtime";
 export { SharedStorage, Storage } from "./storage";
 export { Time } from "./time";
 export type { CastleClockZone, CastleDateParts, CastleTimeApi } from "./time";

package/src/runtime.ts

@@ -332,12 +332,39 @@ function handleLocalMessage(msg: IncomingMessage): void {
         });
       });
   } else if (msg.type === "restart") {
-    location.reload();
+    scheduleRestart();
   } else if (msg.type === "write_file_response") {
     resolveLocalRequest(msg);
   }
 }
 
+// Restart (from `castle-web restart` / task agents) is debounced so a burst
+// of reload requests -- several tasks finishing close together -- produces
+// one reload. Before reloading, registered hooks run (the kit editor flushes
+// its debounced unsaved edits there) so in-flight work isn't lost.
+const RESTART_DEBOUNCE_MS = 1500;
+let restartTimer: ReturnType<typeof setTimeout> | null = null;
+const beforeRestartHooks = new Set<() => void | Promise<void>>();
+
+export function onBeforeRestart(hook: () => void | Promise<void>): () => void {
+  beforeRestartHooks.add(hook);
+  return () => beforeRestartHooks.delete(hook);
+}
+
+function scheduleRestart(): void {
+  if (restartTimer !== null) clearTimeout(restartTimer);
+  restartTimer = setTimeout(() => {
+    void (async () => {
+      try {
+        await Promise.all([...beforeRestartHooks].map((hook) => hook()));
+      } catch {
+        // a failed flush shouldn't block the reload
+      }
+      location.reload();
+    })();
+  }, RESTART_DEBOUNCE_MS);
+}
+
 function localWsUrl(path: string): string {
   const url = new URL(path, location.href);
   url.protocol = location.protocol === "https:" ? "wss:" : "ws:";