@glassly/engine 0.1.0-dev.61 → 0.1.0-dev.89

package/src/react/useMentraLiveOta.ts

@@ -28,7 +28,9 @@ export type MentraLiveOtaFlowPage = "check" | "progress"
 export type MentraLiveOtaScreen =
   | "initializing"
   | "checking"
+  | "finishing"
   | "update_available"
+  | "battery_required"
   | "wifi_required"
   | "up_to_date"
   | "dev_build"
@@ -69,6 +71,7 @@ export type MentraLiveOtaReleaseTransition = {
 export type MentraLiveOtaState = {
   screen: MentraLiveOtaScreen
   connected: boolean
+  batteryLevel: number | null
   transport: MentraLiveOtaTransport | null
   updateRequired: boolean
   versionChange: boolean
@@ -94,6 +97,8 @@ export type MentraLiveOtaState = {
   canDiscard: boolean
   canOpenWifiSetup: boolean
   continueDisabled: boolean
+  /** True only after an approved update session reaches a final no-update check. */
+  completedUpdate: boolean
   /** Release labels for the offered or just-completed coordinated update. */
   releaseTransition: MentraLiveOtaReleaseTransition | null
   /** Release notes crossed by this update, newest first. Populated on completion. */
@@ -128,6 +133,7 @@ type CheckState = "checking" | "update_available" | "no_update" | "dev_build" |
 
 const AUTO_CHAIN_NETWORK_RETRY_DELAY_MS = 5000
 const AUTO_CHAIN_COMPLETE_DELAY_MS = 750
+export const MINIMUM_OTA_BATTERY_LEVEL = 25
 
 function releaseRangeTargetVersion(result: OtaCheckCurrentGlassesResult): string | null {
   return result.releaseVersion ?? result.updateInfo?.versionName ?? result.latestVersionInfo?.versionName ?? null
@@ -163,10 +169,10 @@ function installProgress(snapshot: OtaInstallSnapshot): number | null {
   const isDownload = otaStatus?.phase === "download"
   const totalSteps = otaStatus?.totalSteps ?? 1
   const rawPercent = isDownload
-    ? otaStatus?.stepPercent ?? 0
+    ? (otaStatus?.stepPercent ?? 0)
     : totalSteps >= 2
-    ? otaStatus?.overallPercent ?? 0
-    : otaStatus?.stepPercent ?? 0
+      ? (otaStatus?.overallPercent ?? 0)
+      : (otaStatus?.stepPercent ?? 0)
   return Math.min(Math.max(rawPercent, snapshot.mtkInstallStallSimulatedPercent ?? 0, 0), 100)
 }
 
@@ -215,7 +221,9 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
   const [completedReleaseTransition, setCompletedReleaseTransition] = useState<MentraLiveOtaReleaseTransition | null>(
     null,
   )
+  const [completedUpdate, setCompletedUpdate] = useState(false)
   const [completedChangelogs, setCompletedChangelogs] = useState<ReleaseChangelog[]>([])
+  const [batteryBlocked, setBatteryBlocked] = useState(false)
   const [checkGeneration, setCheckGeneration] = useState(0)
   const performCheckGenerationRef = useRef(0)
   const activeCheckKeyRef = useRef<string | null>(null)
@@ -249,6 +257,7 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
   const returnToCheck = useCallback(() => {
     installActionPendingRef.current = false
     autoChainAdvancedRef.current = false
+    setCheckState("checking")
     setPage("check")
     setCheckGeneration((generation) => generation + 1)
   }, [])
@@ -259,6 +268,28 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
     setPage("progress")
   }, [])
 
+  const continueApprovedChain = useCallback(
+    (result: OtaCheckCurrentGlassesResult): boolean => {
+      if (installActionPendingRef.current || !isOtaAutoChainActive() || !result.updateInfo) return false
+      const snapshot = ota.snapshot()
+      if (!snapshot.wifiStatusKnown || (!snapshot.wifiConnected && snapshot.hotspotOtaVersion !== 1)) return false
+
+      const admission = tryAdvanceOtaAutoChain(
+        otaAutoChainFingerprint(result),
+        result.updateInfo.isDowngrade === true,
+        releaseRangeTargetVersion(result),
+        result.releaseVersion,
+      )
+      if (!admission.advance) return false
+
+      checkCompletedRef.current = true
+      ota.installSession.prepare(result)
+      navigateToProgress()
+      return true
+    },
+    [navigateToProgress],
+  )
+
   useEffect(() => {
     if (!runtimeReady || page !== "check") return
     const MIN_DISPLAY_TIME_MS = 1100
@@ -279,18 +310,18 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
 
     const performCheck = async () => {
       if (checkCompletedRef.current) return
+      if (isOtaAutoChainActive() && !otaSnapshot.ready) {
+        const remainingMs = otaAutoChainReconnectWaitRemaining()
+        if (remainingMs === null) return
+        reconnectTimeout = setTimeout(() => {
+          if (cancelled || myGeneration !== performCheckGenerationRef.current || ota.snapshot().ready) return
+          stopOtaAutoChain()
+          checkCompletedRef.current = true
+          setCheckState("error")
+        }, remainingMs)
+        return
+      }
       if (!otaSnapshot.connected) {
-        if (isOtaAutoChainActive()) {
-          const remainingMs = otaAutoChainReconnectWaitRemaining()
-          if (remainingMs === null) return
-          reconnectTimeout = setTimeout(() => {
-            if (cancelled || myGeneration !== performCheckGenerationRef.current || ota.snapshot().connected) return
-            stopOtaAutoChain()
-            checkCompletedRef.current = true
-            setCheckState("error")
-          }, remainingMs)
-          return
-        }
         if (checkStartedRef.current) {
           stopOtaAutoChain()
           checkCompletedRef.current = true
@@ -364,34 +395,21 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
             const fromVersion = currentReleaseVersion(ota.snapshot().appVersion)
             setOfferedReleaseTransition(result.releaseVersion ? {fromVersion, toVersion: result.releaseVersion} : null)
           }
-          if (isOtaAutoChainActive()) {
-            const snapshot = ota.snapshot()
-            if (!snapshot.wifiStatusKnown) return
-            if (!snapshot.wifiConnected && snapshot.hotspotOtaVersion !== 1) {
-              stopOtaAutoChain()
-            } else {
-              const admission = tryAdvanceOtaAutoChain(
-                fingerprint,
-                result.updateInfo.isDowngrade === true,
-                releaseRangeTargetVersion(result),
-                result.releaseVersion,
-              )
-              if (admission.advance) {
-                checkCompletedRef.current = true
-                ota.installSession.prepare(result)
-                navigateToProgress()
-                return
-              }
-            }
+          if (isOtaAutoChainActive() && !ota.snapshot().wifiStatusKnown) {
+            checkCompletedRef.current = true
+            return
           }
+          if (continueApprovedChain(result)) return
           checkCompletedRef.current = true
           setCheckState("update_available")
           return
         }
-        if (isOtaAutoChainActive()) {
+        const completedApprovedSession = isOtaAutoChainActive()
+        if (completedApprovedSession) {
           setCompletedChangelogs(releaseChangelogsForActiveChain())
           setCompletedReleaseTransition(releaseTransitionFromRange(otaAutoChainReleaseRange()))
         }
+        setCompletedUpdate(completedApprovedSession)
         stopOtaAutoChain()
         checkCompletedRef.current = true
         ota.clearUpdateAvailable()
@@ -412,7 +430,42 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
       cancelled = true
       if (reconnectTimeout) clearTimeout(reconnectTimeout)
     }
-  }, [checkGeneration, navigateToProgress, otaSnapshot.connected, otaSnapshot.wifiStatusKnown, page, runtimeReady])
+  }, [
+    checkGeneration,
+    continueApprovedChain,
+    otaSnapshot.connected,
+    otaSnapshot.ready,
+    otaSnapshot.wifiStatusKnown,
+    page,
+    runtimeReady,
+  ])
+
+  useEffect(() => {
+    if (
+      !runtimeReady ||
+      page !== "check" ||
+      (checkState !== "checking" && checkState !== "update_available") ||
+      !otaSnapshot.wifiStatusKnown ||
+      !isOtaAutoChainActive()
+    ) {
+      return
+    }
+    const result = selectedCheckResultRef.current
+    if (!result?.updateAvailable || !result.updateInfo) return
+    if (!otaSnapshot.wifiConnected && otaSnapshot.hotspotOtaVersion !== 1) {
+      setCheckState("update_available")
+      return
+    }
+    continueApprovedChain(result)
+  }, [
+    checkState,
+    continueApprovedChain,
+    otaSnapshot.hotspotOtaVersion,
+    otaSnapshot.wifiConnected,
+    otaSnapshot.wifiStatusKnown,
+    page,
+    runtimeReady,
+  ])
 
   useEffect(() => {
     if (!runtimeReady || page !== "progress") return
@@ -457,8 +510,10 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
   }, [installSnapshot.connected, installSnapshot.displayState, page, returnToCheck])
 
   const check = useCallback(() => {
+    setBatteryBlocked(false)
     setOfferedReleaseTransition(null)
     setCompletedReleaseTransition(null)
+    setCompletedUpdate(false)
     setCompletedChangelogs([])
     setPage("check")
     setCheckState("checking")
@@ -474,6 +529,10 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
       return
     }
     const snapshot = ota.snapshot()
+    if (snapshot.batteryLevel !== null && snapshot.batteryLevel < MINIMUM_OTA_BATTERY_LEVEL) {
+      setBatteryBlocked(true)
+      return
+    }
     if (!snapshot.wifiStatusKnown) {
       check()
       return
@@ -483,6 +542,8 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
       return
     }
     ota.installSession.prepare(result)
+    setBatteryBlocked(false)
+    setCompletedUpdate(false)
     setCompletedChangelogs([])
     if (updateFingerprint) {
       const fromVersion = offeredReleaseTransition
@@ -497,6 +558,15 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
     navigateToProgress()
   }, [check, isVersionChange, navigateToProgress, offeredReleaseTransition, page, updateFingerprint])
 
+  useEffect(() => {
+    if (
+      batteryBlocked &&
+      (otaSnapshot.batteryLevel === null || otaSnapshot.batteryLevel >= MINIMUM_OTA_BATTERY_LEVEL)
+    ) {
+      setBatteryBlocked(false)
+    }
+  }, [batteryBlocked, otaSnapshot.batteryLevel])
+
   const retryInstall = useCallback(() => {
     if (page !== "progress") return
     onFirmwareRestartingChangeRef.current?.(false, true)
@@ -508,7 +578,11 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
       onFinishedRef.current?.()
       return
     }
-    if (installSnapshot.displayState === "restarting") stopOtaAutoChain()
+    const restartStillInProgress =
+      installSnapshot.displayState === "restarting" ||
+      installSnapshot.versionChangePhase === "restarting" ||
+      installSnapshot.versionChangePhase === "verifying"
+    if (restartStillInProgress) return
     const requiresGlassesReboot = shouldRequireGlassesRebootForBesFailure(
       installSnapshot.otaStatus,
       installSnapshot.otaProgress,
@@ -534,10 +608,12 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
   const state = useMemo<MentraLiveOtaState>(() => {
     const hotspotSupported = otaSnapshot.hotspotOtaVersion === 1
     const canInstall = otaSnapshot.wifiStatusKnown && (otaSnapshot.wifiConnected || hotspotSupported)
+    const autoChainActive = isOtaAutoChainActive()
     if (!runtimeReady) {
       return {
         screen: "initializing",
         connected: otaSnapshot.connected,
+        batteryLevel: otaSnapshot.batteryLevel,
         transport: null,
         updateRequired: isUpdateRequired,
         versionChange: isVersionChange,
@@ -563,6 +639,7 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
         canDiscard: false,
         canOpenWifiSetup: false,
         continueDisabled: false,
+        completedUpdate: false,
         releaseTransition: null,
         changelogs: [],
       }
@@ -570,29 +647,27 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
 
     if (page === "check") {
       const wifiRequired = otaSnapshot.wifiStatusKnown && !otaSnapshot.wifiConnected && !hotspotSupported
-      const screen: MentraLiveOtaScreen =
-        checkState === "checking"
-          ? "checking"
-          : checkState === "update_available"
-          ? wifiRequired
-            ? "wifi_required"
-            : "update_available"
-          : checkState === "no_update"
-          ? "up_to_date"
-          : checkState === "dev_build"
-          ? "dev_build"
-          : errorKind === "pin_unavailable"
-          ? "update_info_unavailable"
-          : "check_failed"
-      const error: MentraLiveOtaError | null =
-        screen === "check_failed"
-          ? {code: "check_failed", message: "Couldn't check for updates. Please check your connection and try again."}
-          : screen === "update_info_unavailable"
-          ? {code: "update_info_unavailable", message: "Update information for this app version is unavailable."}
-          : null
+      let screen: MentraLiveOtaScreen
+      if (checkState === "checking") screen = autoChainActive ? "finishing" : "checking"
+      else if (checkState === "update_available") {
+        screen = wifiRequired ? "wifi_required" : batteryBlocked ? "battery_required" : "update_available"
+      } else if (checkState === "no_update") screen = "up_to_date"
+      else if (checkState === "dev_build") screen = "dev_build"
+      else screen = errorKind === "pin_unavailable" ? "update_info_unavailable" : "check_failed"
+
+      let error: MentraLiveOtaError | null = null
+      if (screen === "check_failed") {
+        error = {
+          code: "check_failed",
+          message: "Couldn't check for updates. Please check your connection and try again.",
+        }
+      } else if (screen === "update_info_unavailable") {
+        error = {code: "update_info_unavailable", message: "Update information for this app version is unavailable."}
+      }
       return {
         screen,
         connected: otaSnapshot.connected,
+        batteryLevel: otaSnapshot.batteryLevel,
         transport: otaSnapshot.wifiConnected ? "wifi" : hotspotSupported ? "hotspot" : null,
         updateRequired: isUpdateRequired,
         versionChange: isVersionChange,
@@ -614,13 +689,19 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
         canInstall: screen === "update_available" && canInstall,
         canRetry: screen === "check_failed",
         canFinish: screen === "up_to_date" || screen === "dev_build" || screen === "update_info_unavailable",
-        canDismiss: (screen === "update_available" || screen === "wifi_required") && !isUpdateRequired,
+        canDismiss:
+          (screen === "update_available" || screen === "wifi_required" || screen === "battery_required") &&
+          !isUpdateRequired &&
+          !autoChainActive,
         canDiscard: false,
         canOpenWifiSetup: screen === "wifi_required",
         continueDisabled: false,
+        completedUpdate: screen === "up_to_date" && completedUpdate,
         releaseTransition:
           screen === "update_available" || screen === "wifi_required"
-            ? offeredReleaseTransition
+            ? autoChainActive
+              ? releaseTransitionFromRange(otaAutoChainReleaseRange())
+              : offeredReleaseTransition
             : screen === "up_to_date"
               ? completedReleaseTransition
               : null,
@@ -641,7 +722,8 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
     const displayedError = requiresGlassesReboot
       ? BES_INSTALL_RESTART_MESSAGE
       : installSnapshot.errorMsg || getOtaErrorMessage(installSnapshot.otaStatus?.error)
-    const screen = progressScreen(installSnapshot)
+    const progressState = progressScreen(installSnapshot)
+    const screen = progressState === "complete" && autoChainActive ? "finishing" : progressState
     const error =
       screen === "failed"
         ? {
@@ -654,6 +736,7 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
     return {
       screen,
       connected: installSnapshot.connected,
+      batteryLevel: otaSnapshot.batteryLevel,
       transport: installSnapshot.transport,
       updateRequired: isUpdateRequired,
       versionChange: installSnapshot.isVersionChange,
@@ -677,18 +760,21 @@ export function useMentraLiveOta(options: UseMentraLiveOtaOptions = {}): MentraL
       error,
       canInstall: false,
       canRetry: screen === "failed" && !requiresGlassesReboot,
-      canFinish: screen === "complete" || screen === "restarting" || (screen === "failed" && requiresGlassesReboot),
+      canFinish: screen === "complete" || (screen === "failed" && requiresGlassesReboot),
       canDismiss: false,
       canDiscard: screen === "disconnected",
       canOpenWifiSetup: screen === "failed" && showChangeWifi,
       continueDisabled: installSnapshot.continueButtonDisabled,
+      completedUpdate: false,
       releaseTransition: releaseTransitionFromRange(otaAutoChainReleaseRange()),
       changelogs,
     }
   }, [
     checkState,
+    batteryBlocked,
     completedChangelogs,
     completedReleaseTransition,
+    completedUpdate,
     errorKind,
     firmwareRestarting,
     installSnapshot,