@rohal12/spindle 0.51.2 → 0.51.3

package/dist/pkg/headless.js

@@ -2809,18 +2809,23 @@ async function getCurrentPlaythroughId(ifid) {
 }
 /**
 * Run a save through the `beforesave` / `aftersave` hooks. `beforesave` fires
-* before `getPayload()` captures the state, so data a hook sets is part of the
-* save; `aftersave` fires once `write` has stored it. Every user-facing save
-* path goes through here.
+* before the payload is captured, so data a hook sets is part of the save;
+* `aftersave` fires once `write` has stored it. Every user-facing save path
+* goes through here.
+*
+* `beginCapture` is called just before the hooks and returns the function
+* that captures the payload after them, so the capture can tell which
+* variables the hooks changed (the store's `beginSave`).
 *
 * The hook and capture run synchronously, but an error they throw (a failing
 * hook, an unserializable payload) rejects the returned promise like a failed
 * write rather than escaping to the caller.
 */
-function saveWithHooks(slot, custom, getPayload, write) {
+function saveWithHooks(slot, custom, beginCapture, write) {
 	try {
+		const capture = beginCapture();
 		emit("beforesave", slot, custom);
-		return write(getPayload()).then((result) => {
+		return write(capture()).then((result) => {
 			emit("aftersave", slot);
 			return result;
 		});
@@ -3703,6 +3708,22 @@ function loadedEntryMoment(payload) {
 	if (typeof moment.variables !== "object" || moment.variables === null) return;
 	return moment;
 }
+/**
+* Copy the variables that changed between `before` and `after` (the live
+* variables around the `beforesave` hooks) into the payload's snapshot of the
+* saved moment. A load restores that snapshot, the state on entering the
+* passage, and runs the passage again, so data a hook adds to a save would
+* otherwise be lost on load (#227). Only the payload's copy changes: the live
+* history keeps the recorded snapshot (#159). Store updates are immutable, so
+* a value a hook did not touch keeps its identity.
+*/
+function keepHookWrites(payload, before, after) {
+	const moment = payload.history[payload.historyIndex];
+	if (!moment) return;
+	const has = (vars, key) => Object.prototype.hasOwnProperty.call(vars, key);
+	for (const key of new Set([...Object.keys(before), ...Object.keys(after)])) if (!has(after, key)) delete moment.variables[key];
+	else if (!has(before, key) || after[key] !== before[key]) moment.variables[key] = deepClone(after[key]);
+}
 /** Restore or reset PRNG from a history moment's snapshot. */
 function restorePRNGFromMoment(moment) {
 	if (moment?.prng) restorePRNG(moment.prng.seed, moment.prng.pull);
@@ -3985,7 +4006,7 @@ var useStoryStore = create()(immer((set, get) => ({
 	save: (slot, custom) => {
 		const { storyData, playthroughId } = get();
 		if (!storyData) return Promise.resolve();
-		return handled(saveWithHooks(slot, custom, () => get().getSavePayload(), async (payload) => {
+		return handled(saveWithHooks(slot, custom, get().beginSave, async (payload) => {
 			set((state) => {
 				state.saveError = null;
 			});
@@ -4127,6 +4148,14 @@ var useStoryStore = create()(immer((set, get) => ({
 			prng: snapshotPRNG()
 		};
 	},
+	beginSave: () => {
+		const before = get().variables;
+		return () => {
+			const payload = get().getSavePayload();
+			keepHookWrites(payload, before, get().variables);
+			return payload;
+		};
+	},
 	loadFromPayload: (payload, slot) => {
 		if (payload.history.length === 0) {
 			console.warn("loadFromPayload: rejecting payload with empty history");
@@ -6424,7 +6453,7 @@ function SaveManagerContent() {
 	const closeDialog = useContext(DialogCloseContext);
 	const storyData = useStoryStore((s) => s.storyData);
 	const playthroughId = useStoryStore((s) => s.playthroughId);
-	const getSavePayload = useStoryStore((s) => s.getSavePayload);
+	const beginSave = useStoryStore((s) => s.beginSave);
 	const loadFromPayload = useStoryStore((s) => s.loadFromPayload);
 	const ifid = storyData?.ifid ?? "";
 	const refresh = useCallback(async () => {
@@ -6464,7 +6493,7 @@ function SaveManagerContent() {
 	const handleNewSave = async () => {
 		if (!ifid || !playthroughId) return;
 		try {
-			await saveWithHooks(void 0, void 0, getSavePayload, (payload) => createSave(ifid, playthroughId, payload));
+			await saveWithHooks(void 0, void 0, beginSave, (payload) => createSave(ifid, playthroughId, payload));
 			showStatus("Save created");
 			await refresh();
 		} catch {
@@ -6473,7 +6502,7 @@ function SaveManagerContent() {
 	};
 	const handleOverwrite = async (saveId) => {
 		try {
-			await saveWithHooks(void 0, void 0, getSavePayload, async (payload) => {
+			await saveWithHooks(void 0, void 0, beginSave, async (payload) => {
 				if (!await overwriteSave(saveId, payload)) throw new Error("Save not found");
 			});
 			showStatus("Save overwritten");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rohal12/spindle",
-  "version": "0.51.2",
+  "version": "0.51.3",
   "type": "module",
   "description": "A Preact-based story format for Twine 2.",
   "license": "Unlicense",

package/src/components/macros/SaveManager.tsx

@@ -62,7 +62,7 @@ export function SaveManagerContent() {
 
   const storyData = useStoryStore((s) => s.storyData);
   const playthroughId = useStoryStore((s) => s.playthroughId);
-  const getSavePayload = useStoryStore((s) => s.getSavePayload);
+  const beginSave = useStoryStore((s) => s.beginSave);
   const loadFromPayload = useStoryStore((s) => s.loadFromPayload);
   const ifid = storyData?.ifid ?? '';
 
@@ -108,7 +108,7 @@ export function SaveManagerContent() {
   const handleNewSave = async () => {
     if (!ifid || !playthroughId) return;
     try {
-      await saveWithHooks(undefined, undefined, getSavePayload, (payload) =>
+      await saveWithHooks(undefined, undefined, beginSave, (payload) =>
         createSave(ifid, playthroughId, payload),
       );
       showStatus('Save created');
@@ -120,16 +120,11 @@ export function SaveManagerContent() {
 
   const handleOverwrite = async (saveId: string) => {
     try {
-      await saveWithHooks(
-        undefined,
-        undefined,
-        getSavePayload,
-        async (payload) => {
-          if (!(await overwriteSave(saveId, payload))) {
-            throw new Error('Save not found');
-          }
-        },
-      );
+      await saveWithHooks(undefined, undefined, beginSave, async (payload) => {
+        if (!(await overwriteSave(saveId, payload))) {
+          throw new Error('Save not found');
+        }
+      });
       showStatus('Save overwritten');
       await refresh();
     } catch {

package/src/saves/save-manager.ts

@@ -99,9 +99,13 @@ export async function getCurrentPlaythroughId(
 
 /**
  * Run a save through the `beforesave` / `aftersave` hooks. `beforesave` fires
- * before `getPayload()` captures the state, so data a hook sets is part of the
- * save; `aftersave` fires once `write` has stored it. Every user-facing save
- * path goes through here.
+ * before the payload is captured, so data a hook sets is part of the save;
+ * `aftersave` fires once `write` has stored it. Every user-facing save path
+ * goes through here.
+ *
+ * `beginCapture` is called just before the hooks and returns the function
+ * that captures the payload after them, so the capture can tell which
+ * variables the hooks changed (the store's `beginSave`).
  *
  * The hook and capture run synchronously, but an error they throw (a failing
  * hook, an unserializable payload) rejects the returned promise like a failed
@@ -110,12 +114,13 @@ export async function getCurrentPlaythroughId(
 export function saveWithHooks<T>(
   slot: string | undefined,
   custom: Record<string, unknown> | undefined,
-  getPayload: () => SavePayload,
+  beginCapture: () => () => SavePayload,
   write: (payload: SavePayload) => Promise<T>,
 ): Promise<T> {
   try {
+    const capture = beginCapture();
     emit('beforesave', slot, custom);
-    const payload = getPayload();
+    const payload = capture();
     return write(payload).then((result) => {
       emit('aftersave', slot);
       return result;

package/src/store.ts

@@ -297,6 +297,33 @@ function loadedEntryMoment(
   return moment;
 }
 
+/**
+ * Copy the variables that changed between `before` and `after` (the live
+ * variables around the `beforesave` hooks) into the payload's snapshot of the
+ * saved moment. A load restores that snapshot, the state on entering the
+ * passage, and runs the passage again, so data a hook adds to a save would
+ * otherwise be lost on load (#227). Only the payload's copy changes: the live
+ * history keeps the recorded snapshot (#159). Store updates are immutable, so
+ * a value a hook did not touch keeps its identity.
+ */
+function keepHookWrites(
+  payload: SavePayload,
+  before: Record<string, unknown>,
+  after: Record<string, unknown>,
+): void {
+  const moment = payload.history[payload.historyIndex];
+  if (!moment) return;
+  const has = (vars: Record<string, unknown>, key: string) =>
+    Object.prototype.hasOwnProperty.call(vars, key);
+  for (const key of new Set([...Object.keys(before), ...Object.keys(after)])) {
+    if (!has(after, key)) {
+      delete moment.variables[key];
+    } else if (!has(before, key) || after[key] !== before[key]) {
+      moment.variables[key] = deepClone(after[key]);
+    }
+  }
+}
+
 /** Restore or reset PRNG from a history moment's snapshot. */
 function restorePRNGFromMoment(moment: HistoryMoment | undefined): void {
   if (moment?.prng) {
@@ -384,6 +411,12 @@ export interface StoryState {
   clearAllData: () => void;
   deletePlaythrough: (playthroughId: string) => void;
   getSavePayload: () => SavePayload;
+  /**
+   * Start capturing a save, before the `beforesave` hooks run. The returned
+   * function builds the payload after them, and keeps the variables the
+   * hooks changed in the saved moment's snapshot, so a load restores them.
+   */
+  beginSave: () => () => SavePayload;
   /**
    * Replace the game state with a live (deserialized) payload. `slot` is
    * passed to the `beforeload`/`afterload` events.
@@ -799,29 +832,18 @@ export const useStoryStore = create<StoryState>()(
       if (!storyData) return Promise.resolve();
 
       return handled(
-        saveWithHooks(
-          slot,
-          custom,
-          () => get().getSavePayload(),
-          async (payload) => {
-            set((state) => {
-              state.saveError = null;
-            });
-            await quickSave(
-              storyData.ifid,
-              playthroughId,
-              payload,
-              slot,
-              custom,
-            );
-            set((state) => {
-              state.knownSaves = {
-                ...state.knownSaves,
-                [slot ?? '']: true,
-              };
-            });
-          },
-        ).catch((err) => {
+        saveWithHooks(slot, custom, get().beginSave, async (payload) => {
+          set((state) => {
+            state.saveError = null;
+          });
+          await quickSave(storyData.ifid, playthroughId, payload, slot, custom);
+          set((state) => {
+            state.knownSaves = {
+              ...state.knownSaves,
+              [slot ?? '']: true,
+            };
+          });
+        }).catch((err) => {
           console.error('spindle: failed to save', err);
           set((state) => {
             state.saveError =
@@ -997,6 +1019,15 @@ export const useStoryStore = create<StoryState>()(
       };
     },
 
+    beginSave: () => {
+      const before = get().variables;
+      return () => {
+        const payload = get().getSavePayload();
+        keepHookWrites(payload, before, get().variables);
+        return payload;
+      };
+    },
+
     loadFromPayload: (payload: SavePayload, slot?: string) => {
       if (payload.history.length === 0) {
         console.warn('loadFromPayload: rejecting payload with empty history');