| author | |
| committer | |
| log | b9e22de6a354a97c2ecc772ad00754bfbd7333e6 |
| tree | f83e2f30a8d78cdaa86fd274cef81fcdd18bd874 |
| parent | f3b43a148b9e39192be5d43415289ac06eca1443 |
| signature | Signed by PGP key B5690EEEBB952194 |
Co-authored-by: codex <codex@users.noreply.github.com>2 files changed, 52 insertions(+), 5 deletions(-)
apps/web/src/state/desktopUpdate.test.ts+20-2| ... | ... | @@ -1,7 +1,7 @@ |
| 1 | 1 | import type { DesktopUpdateState } from "@t3tools/contracts"; |
| 2 | 2 | import * as AsyncResult from "effect/unstable/reactivity/AsyncResult"; |
| 3 | 3 | import { AtomRegistry } from "effect/unstable/reactivity"; |
| 4 | import { describe, expect, it, vi } from "vite-plus/test"; | |
| 4 | import { afterEach, describe, expect, it, vi } from "vite-plus/test"; | |
| 5 | 5 | |
| 6 | 6 | import { createDesktopUpdateStateAtom } from "./desktopUpdate"; |
| 7 | 7 | |
| ... | ... | @@ -22,6 +22,10 @@ const baseState: DesktopUpdateState = { |
| 22 | 22 | canRetry: false, |
| 23 | 23 | }; |
| 24 | 24 | |
| 25 | afterEach(() => { | |
| 26 | vi.restoreAllMocks(); | |
| 27 | }); | |
| 28 | ||
| 25 | 29 | describe("desktopUpdateStateAtom", () => { |
| 26 | 30 | it("loads once, retains state, and follows desktop update events", async () => { |
| 27 | 31 | let listener: ((state: DesktopUpdateState) => void) | undefined; |
| ... | ... | @@ -91,8 +95,11 @@ describe("desktopUpdateStateAtom", () => { |
| 91 | 95 | |
| 92 | 96 | it("keeps listening when the initial desktop state read fails", async () => { |
| 93 | 97 | let listener: ((state: DesktopUpdateState) => void) | undefined; |
| 98 | const cause = new Error("IPC unavailable"); | |
| 99 | const reportError = vi.spyOn(console, "log").mockImplementation(() => undefined); | |
| 100 | const getUpdateState = vi.fn(async () => Promise.reject(cause)); | |
| 94 | 101 | const atom = createDesktopUpdateStateAtom(() => ({ |
| 95 | getUpdateState: async () => Promise.reject(new Error("IPC unavailable")), | |
| 102 | getUpdateState, | |
| 96 | 103 | onUpdateState: (nextListener) => { |
| 97 | 104 | listener = nextListener; |
| 98 | 105 | return () => undefined; |
| ... | ... | @@ -102,6 +109,17 @@ describe("desktopUpdateStateAtom", () => { |
| 102 | 109 | registry.mount(atom); |
| 103 | 110 | |
| 104 | 111 | await vi.waitFor(() => expect(listener).toBeDefined()); |
| 112 | await vi.waitFor(() => expect(reportError).toHaveBeenCalledOnce()); | |
| 113 | expect(getUpdateState).toHaveBeenCalledTimes(3); | |
| 114 | const [, errorMessage, errorContext] = reportError.mock.calls[0] ?? []; | |
| 115 | expect(errorMessage).toBe("Failed to read the initial desktop update state after 3 attempts."); | |
| 116 | expect(errorContext).toMatchObject({ | |
| 117 | errorTag: "DesktopUpdateStateReadError", | |
| 118 | attemptCount: 3, | |
| 119 | }); | |
| 120 | expect(errorContext).not.toHaveProperty("error"); | |
| 121 | expect(errorContext).not.toHaveProperty("cause"); | |
| 122 | ||
| 105 | 123 | listener?.(baseState); |
| 106 | 124 | await vi.waitFor(() => { |
| 107 | 125 | expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(baseState); |
apps/web/src/state/desktopUpdate.ts+32-3| ... | ... | @@ -2,12 +2,27 @@ import { useAtomValue } from "@effect/atom-react"; |
| 2 | 2 | import type { DesktopBridge, DesktopUpdateState } from "@t3tools/contracts"; |
| 3 | 3 | import * as Effect from "effect/Effect"; |
| 4 | 4 | import * as Queue from "effect/Queue"; |
| 5 | import * as Schema from "effect/Schema"; | |
| 5 | 6 | import * as Stream from "effect/Stream"; |
| 6 | 7 | import * as AsyncResult from "effect/unstable/reactivity/AsyncResult"; |
| 7 | 8 | import { Atom } from "effect/unstable/reactivity"; |
| 8 | 9 | |
| 9 | 10 | type DesktopUpdateBridge = Pick<DesktopBridge, "getUpdateState" | "onUpdateState">; |
| 10 | 11 | |
| 12 | const INITIAL_STATE_READ_ATTEMPT_COUNT = 3; | |
| 13 | ||
| 14 | export class DesktopUpdateStateReadError extends Schema.TaggedErrorClass<DesktopUpdateStateReadError>()( | |
| 15 | "DesktopUpdateStateReadError", | |
| 16 | { | |
| 17 | attemptCount: Schema.Number, | |
| 18 | cause: Schema.Defect(), | |
| 19 | }, | |
| 20 | ) { | |
| 21 | override get message(): string { | |
| 22 | return `Failed to read the initial desktop update state after ${this.attemptCount} attempts.`; | |
| 23 | } | |
| 24 | } | |
| 25 | ||
| 11 | 26 | function getDesktopUpdateBridge(): DesktopUpdateBridge | undefined { |
| 12 | 27 | return typeof window === "undefined" ? undefined : window.desktopBridge; |
| 13 | 28 | } |
| ... | ... | @@ -32,9 +47,23 @@ export function createDesktopUpdateStateAtom(getBridge: () => DesktopUpdateBridg |
| 32 | 47 | (unsubscribe) => Effect.sync(unsubscribe), |
| 33 | 48 | ); |
| 34 | 49 | |
| 35 | const initialState = yield* Effect.tryPromise(() => bridge.getUpdateState()).pipe( | |
| 36 | Effect.retry({ times: 2 }), | |
| 37 | Effect.orElseSucceed(() => null), | |
| 50 | const initialState = yield* Effect.tryPromise({ | |
| 51 | try: () => bridge.getUpdateState(), | |
| 52 | catch: (cause) => | |
| 53 | new DesktopUpdateStateReadError({ | |
| 54 | attemptCount: INITIAL_STATE_READ_ATTEMPT_COUNT, | |
| 55 | cause, | |
| 56 | }), | |
| 57 | }).pipe( | |
| 58 | Effect.retry({ times: INITIAL_STATE_READ_ATTEMPT_COUNT - 1 }), | |
| 59 | Effect.catchTags({ | |
| 60 | DesktopUpdateStateReadError: (error) => | |
| 61 | Effect.logError(error.message, { | |
| 62 | errorTag: error._tag, | |
| 63 | attemptCount: error.attemptCount, | |
| 64 | stack: error.stack, | |
| 65 | }).pipe(Effect.as(null)), | |
| 66 | }), | |
| 38 | 67 | ); |
| 39 | 68 | if (!receivedUpdate && initialState !== null) { |
| 40 | 69 | Queue.offerUnsafe(queue, initialState); |