fix(code): keep runtime zoom state in memory

Generated-By: PostHog Code
Task-Id: ec0186a8-c8b7-4b07-b526-5d58095df15c

Author: richardsolomou

apps/code/src/main/zoom.test.ts

@@ -161,6 +161,27 @@ describe("window zoom", () => {
     });
   });
 
+  it("uses the in-memory zoom level when persistence fails", () => {
+    const window = createWindow();
+    setupWindowZoom(window);
+    store.save.mockImplementation(() => {});
+
+    window.webContents.emit("zoom-changed", { preventDefault: vi.fn() }, "in");
+    vi.runAllTimers();
+    window.webContents.emit("zoom-changed", { preventDefault: vi.fn() }, "in");
+    vi.runAllTimers();
+
+    expect({
+      persistedZoomLevel: store.state.zoomLevel,
+      zoomLevel: window.webContents.zoomLevel,
+      saved: store.save.mock.calls,
+    }).toEqual({
+      persistedZoomLevel: 0.5,
+      zoomLevel: 1.5,
+      saved: [[1], [1.5]],
+    });
+  });
+
   it("clamps invalid persisted levels before restoring", () => {
     store.get.mockReturnValue(10);
     const window = createWindow();

apps/code/src/main/zoom.ts

@@ -31,6 +31,7 @@ interface ZoomWindow {
 }
 
 interface ZoomState {
+  currentZoomLevel: number;
   deferredActions: Array<() => void>;
   wheelZoomDelta: number;
   wheelZoomTimeout: ReturnType<typeof setTimeout> | null;
@@ -46,6 +47,10 @@ function getSavedZoomLevel(): number {
   return clampZoomLevel(windowStateStore.get("zoomLevel", 0));
 }
 
+function getCurrentZoomLevel(window: ZoomWindow): number {
+  return zoomStates.get(window)?.currentZoomLevel ?? getSavedZoomLevel();
+}
+
 function runAfterWheelZoom(window: ZoomWindow, action: () => void): void {
   const state = zoomStates.get(window);
   if (!state?.wheelZoomTimeout) {
@@ -58,6 +63,8 @@ function runAfterWheelZoom(window: ZoomWindow, action: () => void): void {
 
 export function setWindowZoom(window: ZoomWindow, level: number): void {
   const nextLevel = clampZoomLevel(level);
+  const state = zoomStates.get(window);
+  if (state) state.currentZoomLevel = nextLevel;
   window.webContents.setZoomLevel(nextLevel);
   saveZoomLevel(nextLevel);
 }
@@ -67,19 +74,21 @@ export function adjustWindowZoom(
   delta: number | "reset",
 ): void {
   runAfterWheelZoom(window, () => {
-    const nextLevel = delta === "reset" ? 0 : getSavedZoomLevel() + delta;
+    const nextLevel =
+      delta === "reset" ? 0 : getCurrentZoomLevel(window) + delta;
     setWindowZoom(window, nextLevel);
   });
 }
 
 export function restoreWindowZoom(window: ZoomWindow): void {
   runAfterWheelZoom(window, () => {
-    window.webContents.setZoomLevel(getSavedZoomLevel());
+    window.webContents.setZoomLevel(getCurrentZoomLevel(window));
   });
 }
 
 export function setupWindowZoom(window: ZoomWindow): void {
   const state: ZoomState = {
+    currentZoomLevel: getSavedZoomLevel(),
     deferredActions: [],
     wheelZoomDelta: 0,
     wheelZoomTimeout: null,
@@ -100,7 +109,7 @@ export function setupWindowZoom(window: ZoomWindow): void {
     event.preventDefault();
     state.wheelZoomDelta += zoomDirection === "in" ? ZOOM_STEP : -ZOOM_STEP;
     state.wheelZoomTimeout ??= setTimeout(() => {
-      const nextLevel = getSavedZoomLevel() + state.wheelZoomDelta;
+      const nextLevel = state.currentZoomLevel + state.wheelZoomDelta;
       state.wheelZoomDelta = 0;
       state.wheelZoomTimeout = null;
       setWindowZoom(window, nextLevel);