homeio

1"use client";
2
3import { useEffect, useState } from "react";
4import { useDesktopPreferences } from "@/hooks/useDesktopPreferences";
5import {
6  UPDATE_REMINDER_SNOOZE_MS,
7  readUpdateReminderSnooze,
8  updateReminderSnoozeRemaining,
9  writeUpdateReminderSnooze,
10  type UpdateReminderSnooze,
11} from "@/lib/desktop/update-reminder";
12import { applySystemUpdateRequest } from "@/modules/settings/hooks/backend/api";
13import { persistPowerActionState } from "@/modules/settings/hooks/useUpdateRecoveryState";
14import { useSystemUpdateStatus } from "@/modules/system/hooks/useSystemUpdateStatus";
15
16function browserStorage(): Storage | null {
17  try {
18    return window.localStorage;
19  } catch {
20    return null;
21  }
22}
23
24/**
25 * Drives the update bar above the dock: it shows while an update is available,
26 * "Later" hides it for an hour per version, and it respects the existing
27 * "Update notifications" preference.
28 */
29export function useUpdateReminder({ enabled }: { enabled: boolean }) {
30  const { data: status } = useSystemUpdateStatus();
31  const { notificationPreferences, isHydrated } = useDesktopPreferences();
32  const [snooze, setSnooze] = useState<UpdateReminderSnooze | null>(null);
33  const [isSnoozeLoaded, setIsSnoozeLoaded] = useState(false);
34  const [now, setNow] = useState(() => Date.now());
35  const [isApplying, setIsApplying] = useState(false);
36  const [error, setError] = useState<string | null>(null);
37
38  useEffect(() => {
39    const storage = browserStorage();
40    setSnooze(storage ? readUpdateReminderSnooze(storage) : null);
41    setIsSnoozeLoaded(true);
42  }, []);
43
44  const latestVersion = status?.updateAvailable ? status.latestVersion : null;
45  const snoozeRemaining = latestVersion
46    ? updateReminderSnoozeRemaining(snooze, latestVersion, now)
47    : 0;
48
49  // Wake up when the snooze runs out, so the reminder comes back without a reload.
50  useEffect(() => {
51    if (snoozeRemaining <= 0) return;
52    const timer = window.setTimeout(() => setNow(Date.now()), snoozeRemaining);
53    return () => window.clearTimeout(timer);
54  }, [snoozeRemaining]);
55
56  const visible =
57    enabled &&
58    isHydrated &&
59    isSnoozeLoaded &&
60    notificationPreferences.updateNotificationsEnabled &&
61    latestVersion !== null &&
62    snoozeRemaining === 0;
63
64  function later() {
65    if (!latestVersion) return;
66    const timestamp = Date.now();
67    const storage = browserStorage();
68    setSnooze(
69      storage
70        ? writeUpdateReminderSnooze(storage, latestVersion, timestamp)
71        : { version: latestVersion, snoozedUntil: timestamp + UPDATE_REMINDER_SNOOZE_MS },
72    );
73    setNow(timestamp);
74    setError(null);
75  }
76
77  async function update() {
78    if (isApplying) return;
79    setIsApplying(true);
80    setError(null);
81
82    try {
83      await applySystemUpdateRequest();
84    } catch (err) {
85      setError(err instanceof Error ? err.message : "Failed to start the update");
86      setIsApplying(false);
87      return;
88    }
89
90    // Same hand-off as Settings → Updates: the /updating page follows the restart.
91    persistPowerActionState("update", {
92      requestDispatchedAt: new Date().toISOString(),
93    });
94    window.location.replace("/updating");
95  }
96
97  return {
98    visible,
99    currentVersion: status?.currentVersion ?? null,
100    latestVersion,
101    canSelfUpdate: status?.canSelfUpdate !== false,
102    isApplying,
103    error,
104    later,
105    update,
106  };
107}
108