homeio

1"use client";
2
3import { reloadBrowserWindow } from "@/lib/desktop/browser-reload";
4import {
5  clearPersistedPowerActionState,
6  readPersistedPowerActionState,
7  writePersistedPowerActionCompletion,
8  writePersistedPowerActionState,
9} from "@/lib/desktop/reboot-state";
10import { StatusScreen } from "@/modules/shell/components/status-screen";
11import { useRouter } from "next/navigation";
12import { useEffect, useMemo, useState } from "react";
13
14const HEALTH_POLL_MS = 2_000;
15const STARTING_PHASE_MS = 4_000;
16
17export type UpdateRecoveryPhase = "starting" | "waiting" | "reconnecting" | "failed";
18
19type UpdateRecoveryScreenProps = {
20  manageRecovery?: boolean;
21  phaseOverride?: Exclude<UpdateRecoveryPhase, "failed">;
22};
23
24function getUpdateRecoveryCopy(phase: UpdateRecoveryPhase, fatalError: string | null) {
25  if (phase === "failed") {
26    return {
27      title: "Homeio update could not start",
28      body:
29        fatalError ??
30        "The update request failed before the service restart began. Return to Homeio and try again.",
31    };
32  }
33
34  if (phase === "reconnecting") {
35    return {
36      title: "Reconnecting to Homeio...",
37      body: "The update completed and Homeio is coming back. This page will refresh automatically when the new build is ready.",
38    };
39  }
40
41  if (phase === "waiting") {
42    return {
43      title: "Applying Homeio update...",
44      body: "The updater is rebuilding and restarting Homeio now. This page will reconnect automatically when the updated service is available again.",
45    };
46  }
47
48  return {
49    title: "Starting Homeio update...",
50    body: "Preparing the updater now. This page will stay in recovery mode until Homeio finishes restarting.",
51  };
52}
53
54export function UpdateRecoveryScreen({
55  manageRecovery = true,
56  phaseOverride,
57}: UpdateRecoveryScreenProps) {
58  const router = useRouter();
59  const [phase, setPhase] = useState<UpdateRecoveryPhase>("starting");
60  const [fatalError, setFatalError] = useState<string | null>(null);
61
62  const phaseCopy = useMemo(() => {
63    return getUpdateRecoveryCopy(phaseOverride ?? phase, fatalError);
64  }, [fatalError, phase, phaseOverride]);
65
66  useEffect(() => {
67    if (manageRecovery) return;
68    setFatalError(null);
69    setPhase(phaseOverride ?? "starting");
70  }, [manageRecovery, phaseOverride]);
71
72  useEffect(() => {
73    if (!manageRecovery) return;
74    if (typeof window === "undefined") return;
75
76    const actionState = readPersistedPowerActionState(window.localStorage);
77    if (actionState?.action !== "update") {
78      router.replace("/");
79      return;
80    }
81
82    let cancelled = false;
83    let timer: ReturnType<typeof setTimeout> | null = null;
84    let hasObservedUnavailable = false;
85
86    const scheduleNextCheck = () => {
87      if (cancelled) return;
88      timer = setTimeout(() => {
89        void pollRecovery();
90      }, HEALTH_POLL_MS);
91    };
92
93    const pollRecovery = async () => {
94      if (cancelled) return;
95
96      const latestState = readPersistedPowerActionState(window.localStorage);
97      if (!latestState || latestState.action !== "update") {
98        router.replace("/");
99        return;
100      }
101
102      if (!latestState.requestDispatchedAt) {
103        writePersistedPowerActionState(window.localStorage, {
104          ...latestState,
105          requestDispatchedAt: latestState.startedAt,
106        });
107      }
108
109      const startedAtMs = Date.parse(latestState.startedAt);
110      const elapsedMs = Number.isFinite(startedAtMs)
111        ? Date.now() - startedAtMs
112        : STARTING_PHASE_MS;
113      setPhase(elapsedMs < STARTING_PHASE_MS ? "starting" : "waiting");
114
115      try {
116        const healthResponse = await fetch("/api/health", {
117          cache: "no-store",
118        });
119
120        if (!healthResponse.ok) {
121          hasObservedUnavailable = true;
122          scheduleNextCheck();
123          return;
124        }
125
126        if (!hasObservedUnavailable && elapsedMs < STARTING_PHASE_MS) {
127          scheduleNextCheck();
128          return;
129        }
130
131        setPhase("reconnecting");
132
133        const authResponse = await fetch("/api/auth/me", {
134          cache: "no-store",
135        });
136
137        if (authResponse.status === 401) {
138          clearPersistedPowerActionState(window.localStorage);
139          router.replace("/login");
140          return;
141        }
142
143        if (!authResponse.ok) {
144          scheduleNextCheck();
145          return;
146        }
147
148        writePersistedPowerActionCompletion(window.localStorage, {
149          action: "update",
150          completedAt: new Date().toISOString(),
151        });
152        clearPersistedPowerActionState(window.localStorage);
153        reloadBrowserWindow();
154      } catch {
155        hasObservedUnavailable = true;
156        scheduleNextCheck();
157      }
158    };
159
160    void pollRecovery();
161
162    return () => {
163      cancelled = true;
164      if (timer) {
165        clearTimeout(timer);
166      }
167    };
168  }, [manageRecovery, router]);
169
170  return (
171    <StatusScreen
172      title={phaseCopy.title}
173      body={phaseCopy.body}
174      failed={phase === "failed"}
175      action={
176        phase === "failed" ? (
177          <button
178            type="button"
179            onClick={() => router.replace("/")}
180            className="mt-6 inline-flex items-center rounded-[var(--radius)] bg-primary px-5 py-2.5 text-sm font-medium text-primary-foreground transition hover:brightness-110"
181          >
182            Return to Homeio
183          </button>
184        ) : null
185      }
186    />
187  );
188}
189