feat: implement update management with relaunch functionality and localization support

Author: PythonSmall-Q

src-tauri/src/commands/app_cmd.rs

@@ -100,6 +100,11 @@ pub fn get_install_channel_info() -> InstallChannelInfo {
     }
 }
 
+#[tauri::command]
+pub fn relaunch_app(app: AppHandle) {
+    app.restart();
+}
+
 fn default_shortcuts() -> ShortcutSettings {
     ShortcutSettings {
         open_widget_center: "Alt+W".to_string(),
@@ -110,10 +115,11 @@ fn default_shortcuts() -> ShortcutSettings {
 }
 
 #[tauri::command]
-pub fn get_app_settings(app: AppHandle, db: State<DbState>) -> Result<AppSettingsPayload, String> {
+pub fn get_app_settings(db: State<DbState>) -> Result<AppSettingsPayload, String> {
     let conn = db.lock().map_err(|e| e.to_string())?;
 
-    let launch_at_startup = app.autolaunch().is_enabled().unwrap_or(false);
+    let launch_at_startup =
+        crate::db::get_bool_setting(&conn, "launch_at_startup", false).unwrap_or(false);
     let silent_startup =
         crate::db::get_bool_setting(&conn, "silent_startup", true).map_err(|e| e.to_string())?;
     let auto_open_widgets =

src-tauri/src/lib.rs

@@ -16,6 +16,7 @@ use tauri::{
     tray::{TrayIconBuilder, TrayIconEvent},
     AppHandle, Emitter, Listener, Manager,
 };
+use tauri_plugin_autostart::ManagerExt;
 
 use commands::storage_cmd::DbState;
 use monitor::{MonitorStatus, SharedMonitorStatus};
@@ -609,6 +610,25 @@ pub fn run() {
             // Widget/main windows can start invoking commands immediately.
             app.manage(db_state.clone());
 
+            // Sync OS-level autostart with the stored user preference. Re-installing the
+            // app often clears the registry/launch-agent entry, so re-apply it here.
+            {
+                let conn = db_state.lock().unwrap();
+                match crate::db::get_bool_setting(&conn, "launch_at_startup", false) {
+                    Ok(true) => {
+                        if let Err(e) = app.autolaunch().enable() {
+                            log::warn!("Failed to enable launch at startup: {}", e);
+                        }
+                    }
+                    Ok(false) => {
+                        if let Err(e) = app.autolaunch().disable() {
+                            log::warn!("Failed to disable launch at startup: {}", e);
+                        }
+                    }
+                    Err(e) => log::warn!("Failed to read launch_at_startup setting: {}", e),
+                }
+            }
+
             // Widget runtime v2.2.0 rate limiters
             app.manage(commands::WidgetCallRateLimiter::new());
             app.manage(commands::WidgetEventRateLimiter::new());
@@ -1123,6 +1143,7 @@ pub fn run() {
             commands::get_browser_extension_status,
             commands::get_local_api_base_url,
             commands::get_install_channel_info,
+            commands::relaunch_app,
             commands::set_launch_at_startup,
             commands::get_tray_icon_style,
             commands::set_tray_icon_style,

src/MainApp.tsx

@@ -6,6 +6,7 @@ import { register as registerGlobalShortcut, unregisterAll as unregisterAllGloba
 import { isPermissionGranted, requestPermission, sendNotification } from "@tauri-apps/plugin-notification";
 import { open as openExternal } from "@tauri-apps/plugin-shell";
 import { check } from "@tauri-apps/plugin-updater";
+import type { Update, DownloadEvent } from "@tauri-apps/plugin-updater";
 import MainLayout from "./components/layout/MainLayout";
 import Loading from "./components/Loading";
 
@@ -86,15 +87,17 @@ export default function MainApp() {
   } = useStatsStore();
   const {
     setMonitoringActive,
-    autoCheckUpdates,
+    updateMode,
     notificationQuietHoursEnabled,
     notificationQuietStart,
     notificationQuietEnd,
     notificationCooldownMin,
   } = useSettingsStore();
   const { t } = useTranslation(["common", "limits", "browserUsage"]);
 
-  const [updateInfo, setUpdateInfo] = useState<{ version: string; notes: string; url: string } | null>(null);
+  const [updateInfo, setUpdateInfo] = useState<{ version: string; notes: string; url: string; update: Update | null } | null>(null);
+  const [updatePhase, setUpdatePhase] = useState<"available" | "downloading" | "downloaded" | "installing">("available");
+  const [downloadProgress, setDownloadProgress] = useState(0);
   const updateCloseButtonRef = useRef<HTMLButtonElement>(null);
 
   useEffect(() => {
@@ -318,7 +321,7 @@ export default function MainApp() {
 
   // Update check – once after 4 s
   useEffect(() => {
-    if (!autoCheckUpdates) return;
+    if (updateMode === "off") return;
 
     const timer = setTimeout(async () => {
       try {
@@ -329,7 +332,6 @@ export default function MainApp() {
         if (!(latest && compareVersions(latest, CURRENT_VERSION) > 0)) return;
 
         const channel = await api.getInstallChannelInfo();
-        setUpdateInfo({ version: latest, notes: data.body ?? "", url: data.html_url ?? "" });
 
         if (!channel.should_trigger_update) {
           const storeUpdateUrl = channel.update_url ?? "ms-windows-store://downloadsandupdates";
@@ -347,32 +349,46 @@ export default function MainApp() {
           return;
         }
 
+        if (updateMode === "auto") {
+          try {
+            const update = await check();
+            if (update) {
+              await update.downloadAndInstall();
+              await api.relaunchApp();
+              return;
+            }
+          } catch {
+            // fallback to release page
+          }
+          if (data.html_url) window.open(data.html_url, "_blank", "noopener,noreferrer");
+          await notifyWithNavigate(
+            t("common:updateAvailableTitle"),
+            t("common:updateAvailableBody", { version: latest, current: CURRENT_VERSION }),
+            "#/settings"
+          );
+          return;
+        }
+
+        // notify mode: show the manual download/install dialog
         try {
           const update = await check();
           if (update) {
-            await update.downloadAndInstall();
-            await notifyWithNavigate(
-              t("common:updateAvailableTitle"),
-              t("common:updateInstallReady", { version: latest }),
-              "#/settings"
-            );
+            setUpdateInfo({ version: latest, notes: data.body ?? "", url: data.html_url ?? "", update });
+            setUpdatePhase("available");
+            setDownloadProgress(0);
             return;
           }
         } catch {
-          // fallback to release page
+          // fallback to release page modal
         }
 
-        if (data.html_url) window.open(data.html_url, "_blank", "noopener,noreferrer");
-
-        await notifyWithNavigate(
-          t("common:updateAvailableTitle"),
-          t("common:updateAvailableBody", { version: latest, current: CURRENT_VERSION }),
-          "#/settings"
-        );
+        setUpdateInfo({ version: latest, notes: data.body ?? "", url: data.html_url ?? "", update: null });
+        setUpdatePhase("available");
+        setDownloadProgress(0);
       } catch { /* offline */ }
     }, 4000);
     return () => clearTimeout(timer);
-  }, [autoCheckUpdates, notifyWithNavigate, t]);
+  }, [updateMode, notifyWithNavigate, t]);
 
   useEffect(() => {
     if (updateInfo) {
@@ -447,6 +463,54 @@ export default function MainApp() {
     };
   }, [focusMainAndNavigate, setMonitorActive, setMonitoringActive, toggleWidgetsVisibility]);
 
+  const closeUpdateModal = useCallback(() => {
+    if (updateInfo?.update && updatePhase === "available") {
+      updateInfo.update.close().catch(() => {});
+    }
+    setUpdateInfo(null);
+    setUpdatePhase("available");
+    setDownloadProgress(0);
+  }, [updateInfo, updatePhase]);
+
+  const handleDownloadUpdate = useCallback(async () => {
+    if (!updateInfo?.update) return;
+    setUpdatePhase("downloading");
+    setDownloadProgress(0);
+    let downloaded = 0;
+    let contentLength = 0;
+    try {
+      await updateInfo.update.download((event: DownloadEvent) => {
+        if (event.event === "Started") {
+          contentLength = event.data.contentLength ?? 0;
+        } else if (event.event === "Progress") {
+          downloaded += event.data.chunkLength;
+          if (contentLength > 0) {
+            setDownloadProgress(Math.min(100, Math.round((downloaded / contentLength) * 100)));
+          }
+        }
+      });
+      setUpdatePhase("downloaded");
+    } catch (err) {
+      setUpdatePhase("available");
+      setDownloadProgress(0);
+      const message = err instanceof Error ? err.message : "";
+      await notifyWithNavigate(t("common:downloadUpdateFailed"), message, "#/settings");
+    }
+  }, [updateInfo, notifyWithNavigate, t]);
+
+  const handleInstallUpdate = useCallback(async () => {
+    if (!updateInfo?.update) return;
+    setUpdatePhase("installing");
+    try {
+      await updateInfo.update.install();
+      await api.relaunchApp();
+    } catch (err) {
+      setUpdatePhase("downloaded");
+      const message = err instanceof Error ? err.message : "";
+      await notifyWithNavigate(t("common:error"), message, "#/settings");
+    }
+  }, [updateInfo, notifyWithNavigate, t]);
+
   return (
     <HashRouter future={{ v7_startTransition: true, v7_relativeSplatPath: true }}>
       <MainLayout>
@@ -486,27 +550,88 @@ export default function MainApp() {
               </div>
               <button
                 ref={updateCloseButtonRef}
-                onClick={() => setUpdateInfo(null)}
+                onClick={closeUpdateModal}
                 aria-label={t("common:close")}
                 title={t("common:close")}
                 className="text-text-muted hover:text-text-primary flex-shrink-0"
               >
                 ✕
               </button>
             </div>
-            {updateInfo.notes && (
+
+            {updateInfo.notes && updatePhase === "available" && (
               <div className="bg-surface-light rounded-xl p-3 max-h-52 overflow-y-auto">
                 <p className="text-xs font-semibold text-text-secondary mb-1">{t("common:whatsNew")}</p>
                 <pre className="text-xs text-text-muted whitespace-pre-wrap font-sans leading-relaxed">{updateInfo.notes}</pre>
               </div>
             )}
+
+            {updatePhase === "downloading" && (
+              <div className="space-y-2">
+                <div className="w-full bg-surface-hover rounded-full h-2 overflow-hidden">
+                  <div
+                    className="bg-accent-blue h-full transition-all duration-150"
+                    style={{ width: `${downloadProgress}%` }}
+                  />
+                </div>
+                <p className="text-xs text-text-muted">
+                  {t("common:downloadingUpdate")}
+                  {downloadProgress > 0 ? ` ${downloadProgress}%` : ""}
+                </p>
+              </div>
+            )}
+
+            {updatePhase === "downloaded" && (
+              <p className="text-sm text-text-secondary">{t("common:updateDownloadReady")}</p>
+            )}
+
+            {updatePhase === "installing" && (
+              <p className="text-sm text-text-secondary">{t("common:installingUpdate")}</p>
+            )}
+
             <div className="flex gap-3">
-              <a href={updateInfo.url} target="_blank" rel="noopener noreferrer" className="flex-1 py-2.5 rounded-xl text-sm font-medium text-center bg-accent-blue/20 text-accent-blue hover:bg-accent-blue/30 transition-colors border border-accent-blue/30">
-                {t("common:viewOnGitHub")}
-              </a>
-              <button onClick={() => setUpdateInfo(null)} className="flex-1 py-2.5 rounded-xl text-sm font-medium border border-surface-border text-text-muted hover:text-text-secondary hover:bg-surface-hover transition-colors">
-                {t("common:remindLater")}
-              </button>
+              {updatePhase === "available" && updateInfo.update && (
+                <button
+                  onClick={handleDownloadUpdate}
+                  className="flex-1 py-2.5 rounded-xl text-sm font-medium text-center bg-accent-blue/20 text-accent-blue hover:bg-accent-blue/30 transition-colors border border-accent-blue/30"
+                >
+                  {t("common:downloadUpdate")}
+                </button>
+              )}
+              {updatePhase === "available" && !updateInfo.update && updateInfo.url && (
+                <a
+                  href={updateInfo.url}
+                  target="_blank"
+                  rel="noopener noreferrer"
+                  className="flex-1 py-2.5 rounded-xl text-sm font-medium text-center bg-accent-blue/20 text-accent-blue hover:bg-accent-blue/30 transition-colors border border-accent-blue/30"
+                >
+                  {t("common:viewOnGitHub")}
+                </a>
+              )}
+              {updatePhase === "downloaded" && updateInfo.update && (
+                <button
+                  onClick={handleInstallUpdate}
+                  className="flex-1 py-2.5 rounded-xl text-sm font-medium text-center bg-accent-green/20 text-accent-green hover:bg-accent-green/30 transition-colors border border-accent-green/30"
+                >
+                  {t("common:installUpdate")}
+                </button>
+              )}
+              {(updatePhase === "available" || updatePhase === "downloaded") && (
+                <button
+                  onClick={closeUpdateModal}
+                  className="flex-1 py-2.5 rounded-xl text-sm font-medium border border-surface-border text-text-muted hover:text-text-secondary hover:bg-surface-hover transition-colors"
+                >
+                  {updatePhase === "downloaded" ? t("common:installLater") : t("common:remindLater")}
+                </button>
+              )}
+              {(updatePhase === "downloading" || updatePhase === "installing") && (
+                <button
+                  disabled
+                  className="flex-1 py-2.5 rounded-xl text-sm font-medium border border-surface-border text-text-muted opacity-50 cursor-not-allowed"
+                >
+                  {updatePhase === "downloading" ? t("common:downloadingUpdate") : t("common:installingUpdate")}
+                </button>
+              )}
             </div>
           </div>
         </div>

src/i18n/locales/de/common.json

@@ -34,6 +34,13 @@
   "updateAvailableBody": "TimeLens v{{version}} ist verfügbar. Sie verwenden v{{current}}.",
   "updateAvailableStoreBody": "TimeLens v{{version}} ist verfügbar. Sie verwenden v{{current}}. TimeLens hat die Microsoft Store-Updates geöffnet.",
   "updateInstallReady": "TimeLens v{{version}} wurde installiert. Starten Sie die App neu, um Updates anzuwenden.",
+  "downloadUpdate": "Download update",
+  "downloadingUpdate": "Downloading update…",
+  "installUpdate": "Install update",
+  "installingUpdate": "Installing update…",
+  "installLater": "Install later",
+  "updateDownloadReady": "The update package is ready. Install now?",
+  "downloadUpdateFailed": "Failed to download update",
   "vscodeExtensionOfflineTitle": "VS Code-Erweiterung nicht verbunden",
   "vscodeExtensionOfflineBody": "VS Code-Statistiken wurden leicht reduziert, da die Erweiterung nicht läuft.",
   "viewOnGitHub": "Auf GitHub anzeigen",

src/i18n/locales/de/settings.json

@@ -348,7 +348,10 @@
     "version": "Version",
     "logs": "Logs",
     "openLogFolder": "Log-Ordner öffnen",
-    "autoCheckUpdates": "Updates automatisch prüfen",
+    "updateMode": "Update checks",
+    "updateModeOff": "Off",
+    "updateModeNotify": "Notify only",
+    "updateModeAuto": "Download and install automatically",
     "checkUpdate": "Nach Updates suchen",
     "checking": "Prüfe…",
     "checkFailed": "Update-Prüfung fehlgeschlagen",

src/i18n/locales/en/common.json

@@ -34,6 +34,13 @@
   "updateAvailableBody": "TimeLens v{{version}} is now available. You are on v{{current}}.",
   "updateAvailableStoreBody": "TimeLens v{{version}} is now available. You are on v{{current}}. TimeLens has opened Microsoft Store updates.",
   "updateInstallReady": "TimeLens v{{version}} has been installed. Restart the app to apply updates.",
+  "downloadUpdate": "Download update",
+  "downloadingUpdate": "Downloading update…",
+  "installUpdate": "Install update",
+  "installingUpdate": "Installing update…",
+  "installLater": "Install later",
+  "updateDownloadReady": "The update package is ready. Install now?",
+  "downloadUpdateFailed": "Failed to download update",
   "vscodeExtensionOfflineTitle": "VS Code extension is not connected",
   "vscodeExtensionOfflineBody": "VS Code stats have been soft-degraded because the extension is not running.",
   "viewOnGitHub": "View on GitHub",

src/i18n/locales/en/settings.json

@@ -348,7 +348,10 @@
     "version": "Version",
     "logs": "Logs",
     "openLogFolder": "Open Log Folder",
-    "autoCheckUpdates": "Auto-check updates",
+    "updateMode": "Update checks",
+    "updateModeOff": "Off",
+    "updateModeNotify": "Notify only",
+    "updateModeAuto": "Download and install automatically",
     "checkUpdate": "Check for Updates",
     "checking": "Checking…",
     "checkFailed": "Update check failed",

src/i18n/locales/es/common.json

@@ -34,6 +34,13 @@
   "updateAvailableBody": "TimeLens v{{version}} está disponible. Tienes v{{current}}.",
   "updateAvailableStoreBody": "TimeLens v{{version}} está disponible. Tienes v{{current}}. TimeLens abrió las actualizaciones de Microsoft Store.",
   "updateInstallReady": "TimeLens v{{version}} se ha instalado. Reinicia la app para aplicar las actualizaciones.",
+  "downloadUpdate": "Download update",
+  "downloadingUpdate": "Downloading update…",
+  "installUpdate": "Install update",
+  "installingUpdate": "Installing update…",
+  "installLater": "Install later",
+  "updateDownloadReady": "The update package is ready. Install now?",
+  "downloadUpdateFailed": "Failed to download update",
   "vscodeExtensionOfflineTitle": "La extensión de VS Code no está conectada",
   "vscodeExtensionOfflineBody": "Las estadísticas de VS Code se han degradado levemente porque la extensión no se está ejecutando.",
   "viewOnGitHub": "Ver en GitHub",