diff --git a/main.go b/main.go index e9931d46..ef6af293 100644 --- a/main.go +++ b/main.go @@ -14,6 +14,7 @@ import ( var appCtx context.Context func Main() { + logger.Log().Msg("JetKVM Starting Up") LoadConfig() var cancel context.CancelFunc @@ -78,16 +79,16 @@ func Main() { initDisplay() go func() { + // wait for 15 minutes before starting auto-update checks + // this is to avoid interfering with initial setup processes + // and to ensure the system is stable before checking for updates time.Sleep(15 * time.Minute) - for { - logger.Debug().Bool("auto_update_enabled", config.AutoUpdateEnabled).Msg("UPDATING") - if !config.AutoUpdateEnabled { - return - } - if isTimeSyncNeeded() || !timeSync.IsSyncSuccess() { - logger.Debug().Msg("system time is not synced, will retry in 30 seconds") - time.Sleep(30 * time.Second) + for { + logger.Info().Bool("auto_update_enabled", config.AutoUpdateEnabled).Msg("auto-update check") + if !config.AutoUpdateEnabled { + logger.Debug().Msg("auto-update disabled") + time.Sleep(5 * time.Minute) // we'll check if auto-updates are enabled in five minutes continue } @@ -97,6 +98,12 @@ func Main() { continue } + if isTimeSyncNeeded() || !timeSync.IsSyncSuccess() { + logger.Debug().Msg("system time is not synced, will retry in 30 seconds") + time.Sleep(30 * time.Second) + continue + } + includePreRelease := config.IncludePreRelease err = TryUpdate(context.Background(), GetDeviceID(), includePreRelease) if err != nil { @@ -106,6 +113,7 @@ func Main() { time.Sleep(1 * time.Hour) } }() + //go RunFuseServer() go RunWebServer() @@ -122,7 +130,8 @@ func Main() { sigs := make(chan os.Signal, 1) signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM) <-sigs - logger.Info().Msg("JetKVM Shutting Down") + + logger.Log().Msg("JetKVM Shutting Down") //if fuseServer != nil { // err := setMassStorageImage(" ") // if err != nil { diff --git a/ota.go b/ota.go index bf0828dc..62273732 100644 --- a/ota.go +++ b/ota.go @@ -176,7 +176,7 @@ func downloadFile(ctx context.Context, path string, url string, downloadProgress if nr > 0 { nw, ew := file.Write(buf[0:nr]) if nw < nr { - return fmt.Errorf("short write: %d < %d", nw, nr) + return fmt.Errorf("short file write: %d < %d", nw, nr) } written += int64(nw) if ew != nil { @@ -240,7 +240,7 @@ func verifyFile(path string, expectedHash string, verifyProgress *float32, scope if nr > 0 { nw, ew := hash.Write(buf[0:nr]) if nw < nr { - return fmt.Errorf("short write: %d < %d", nw, nr) + return fmt.Errorf("short hash write: %d < %d", nw, nr) } verified += int64(nw) if ew != nil { @@ -260,11 +260,14 @@ func verifyFile(path string, expectedHash string, verifyProgress *float32, scope } } - hashSum := hash.Sum(nil) - scopedLogger.Info().Str("path", path).Str("hash", hex.EncodeToString(hashSum)).Msg("SHA256 hash of") + // close the file so we can rename below + fileToHash.Close() - if hex.EncodeToString(hashSum) != expectedHash { - return fmt.Errorf("hash mismatch: %x != %s", hashSum, expectedHash) + hashSum := hex.EncodeToString(hash.Sum(nil)) + scopedLogger.Info().Str("path", path).Str("hash", hashSum).Msg("SHA256 hash of") + + if hashSum != expectedHash { + return fmt.Errorf("hash mismatch: %s != %s", hashSum, expectedHash) } if err := os.Rename(unverifiedPath, path); err != nil { @@ -296,6 +299,8 @@ type OTAState struct { AppUpdatedAt *time.Time `json:"appUpdatedAt,omitempty"` SystemUpdateProgress float32 `json:"systemUpdateProgress,omitempty"` //TODO: port rk_ota, then implement SystemUpdatedAt *time.Time `json:"systemUpdatedAt,omitempty"` + RebootNeeded bool `json:"rebootNeeded,omitempty"` + Rebooting bool `json:"rebooting,omitempty"` } var otaState = OTAState{} @@ -313,7 +318,7 @@ func triggerOTAStateUpdate() { func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) error { scopedLogger := otaLogger.With(). Str("deviceId", deviceId). - Str("includePreRelease", fmt.Sprintf("%v", includePreRelease)). + Bool("includePreRelease", includePreRelease). Logger() scopedLogger.Info().Msg("Trying to update...") @@ -322,7 +327,8 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err } otaState = OTAState{ - Updating: true, + Updating: true, + RebootNeeded: false, } triggerOTAStateUpdate() @@ -335,7 +341,7 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err if err != nil { otaState.Error = fmt.Sprintf("Error checking for updates: %v", err) scopedLogger.Error().Err(err).Msg("Error checking for updates") - return fmt.Errorf("error checking for updates: %w", err) + return err } now := time.Now() @@ -349,8 +355,6 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err appUpdateAvailable := updateStatus.AppUpdateAvailable systemUpdateAvailable := updateStatus.SystemUpdateAvailable - rebootNeeded := false - if appUpdateAvailable { scopedLogger.Info(). Str("local", local.AppVersion). @@ -361,7 +365,6 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err if err != nil { otaState.Error = fmt.Sprintf("Error downloading app update: %v", err) scopedLogger.Error().Err(err).Msg("Error downloading app update") - triggerOTAStateUpdate() return err } downloadFinished := time.Now() @@ -378,18 +381,20 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err if err != nil { otaState.Error = fmt.Sprintf("Error verifying app update hash: %v", err) scopedLogger.Error().Err(err).Msg("Error verifying app update hash") - triggerOTAStateUpdate() return err } verifyFinished := time.Now() otaState.AppVerifiedAt = &verifyFinished otaState.AppVerificationProgress = 1 + triggerOTAStateUpdate() + otaState.AppUpdatedAt = &verifyFinished otaState.AppUpdateProgress = 1 triggerOTAStateUpdate() scopedLogger.Info().Msg("App update downloaded") - rebootNeeded = true + otaState.RebootNeeded = true + triggerOTAStateUpdate() } else { scopedLogger.Info().Msg("App is up to date") } @@ -404,7 +409,6 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err if err != nil { otaState.Error = fmt.Sprintf("Error downloading system update: %v", err) scopedLogger.Error().Err(err).Msg("Error downloading system update") - triggerOTAStateUpdate() return err } downloadFinished := time.Now() @@ -421,7 +425,6 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err if err != nil { otaState.Error = fmt.Sprintf("Error verifying system update hash: %v", err) scopedLogger.Error().Err(err).Msg("Error verifying system update hash") - triggerOTAStateUpdate() return err } scopedLogger.Info().Msg("System update downloaded") @@ -441,6 +444,7 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err scopedLogger.Error().Err(err).Msg("Error starting rk_ota command") return fmt.Errorf("error starting rk_ota command: %w", err) } + ctx, cancel := context.WithCancel(context.Background()) defer cancel() @@ -481,14 +485,19 @@ func TryUpdate(ctx context.Context, deviceId string, includePreRelease bool) err otaState.SystemUpdateProgress = 1 otaState.SystemUpdatedAt = &verifyFinished triggerOTAStateUpdate() - rebootNeeded = true + + otaState.RebootNeeded = true + triggerOTAStateUpdate() } else { scopedLogger.Info().Msg("System is up to date") } - if rebootNeeded { + if otaState.RebootNeeded { scopedLogger.Info().Msg("System Rebooting in 10s") time.Sleep(10 * time.Second) + otaState.Rebooting = true + triggerOTAStateUpdate() + cmd := exec.Command("reboot") err := cmd.Start() if err != nil { diff --git a/ui/src/hooks/stores.ts b/ui/src/hooks/stores.ts index bfbbb26e..1a7e9668 100644 --- a/ui/src/hooks/stores.ts +++ b/ui/src/hooks/stores.ts @@ -518,6 +518,7 @@ export type UpdateModalViews = | "upToDate" | "updateAvailable" | "updateCompleted" + | "rebooting" | "error"; export interface OtaState { @@ -549,19 +550,26 @@ export interface OtaState { systemUpdateProgress: number; systemUpdatedAt: string | null; + + rebootNeeded: boolean; + rebooting: boolean; }; export interface UpdateState { isUpdatePending: boolean; setIsUpdatePending: (isPending: boolean) => void; + updateDialogHasBeenMinimized: boolean; + setUpdateDialogHasBeenMinimized: (hasBeenMinimized: boolean) => void; + otaState: OtaState; setOtaState: (state: OtaState) => void; - setUpdateDialogHasBeenMinimized: (hasBeenMinimized: boolean) => void; + modalView: UpdateModalViews setModalView: (view: UpdateModalViews) => void; - setUpdateErrorMessage: (errorMessage: string) => void; + updateErrorMessage: string | null; + setUpdateErrorMessage: (errorMessage: string) => void; } export const useUpdateStore = create(set => ({ @@ -587,13 +595,17 @@ export const useUpdateStore = create(set => ({ appUpdatedAt: null, systemUpdateProgress: 0, systemUpdatedAt: null, + rebootNeeded: false, + rebooting: false, }, updateDialogHasBeenMinimized: false, setUpdateDialogHasBeenMinimized: (hasBeenMinimized: boolean) => set({ updateDialogHasBeenMinimized: hasBeenMinimized }), + modalView: "loading", setModalView: (view: UpdateModalViews) => set({ modalView: view }), + updateErrorMessage: null, setUpdateErrorMessage: (errorMessage: string) => set({ updateErrorMessage: errorMessage }), })); diff --git a/ui/src/routes/devices.$id.settings.general.update.tsx b/ui/src/routes/devices.$id.settings.general.update.tsx index 38c15412..5847ddfe 100644 --- a/ui/src/routes/devices.$id.settings.general.update.tsx +++ b/ui/src/routes/devices.$id.settings.general.update.tsx @@ -1,10 +1,11 @@ import { useLocation, useNavigate } from "react-router"; import { useCallback, useEffect, useRef, useState } from "react"; import { CheckCircleIcon } from "@heroicons/react/20/solid"; +import { MdConnectWithoutContact, MdRestartAlt } from "react-icons/md"; import Card from "@/components/Card"; import { useJsonRpc } from "@/hooks/useJsonRpc"; -import { Button } from "@components/Button"; +import { Button, LinkButton } from "@components/Button"; import { UpdateState, useUpdateStore } from "@/hooks/stores"; import LoadingSpinner from "@/components/LoadingSpinner"; import { useDeviceUiNavigation } from "@/hooks/useAppNavigation"; @@ -33,7 +34,7 @@ export default function SettingsGeneralUpdateRoute() { } else { setModalView("loading"); } - }, [otaState.updating, otaState.error, setModalView, updateSuccess]); + }, [otaState.error, otaState.updating, setModalView, updateSuccess]); { /* TODO: Migrate to using URLs instead of the global state. To simplify the refactoring, we'll keep the global state for now. */ @@ -41,8 +42,6 @@ export default function SettingsGeneralUpdateRoute() { return navigate("..")} onConfirmUpdate={onConfirmUpdate} />; } - - export function Dialog({ onClose, onConfirmUpdate, @@ -239,14 +238,18 @@ function UpdatingDeviceState({ if (!otaState.metadataFetchedAt) { return "Fetching update information..."; + } else if (otaState.rebooting) { + return "Rebooting..."; } else if (!downloadFinishedAt) { return `Downloading ${type} update...`; } else if (!verfiedAt) { return `Verifying ${type} update...`; } else if (!updatedAt) { return `Installing ${type} update...`; + } else if (otaState.rebootNeeded) { + return "Reboot needed"; } else { - return `Awaiting reboot`; + return "Awaiting reboot"; } }; @@ -278,12 +281,47 @@ function UpdatingDeviceState({ {areAllUpdatesComplete() ? (
- -
- - Rebooting to complete the update... - -
+ + {otaState.rebooting ? ( +
+ + Rebooting the device to complete the update... + +

+ This may take a few minutes. The device will automatically + reconnect once it is back online. If it doesn{"'"}t, you can + manually reconnect. + +

+
+ ) : ( + otaState.rebootNeeded && ( +
+ + Device reboot is pending... + +

+ The JetKVM is preparing to reboot. This may take a while. If it doesn{"'"}t automatically reboot + after a few minutes, you can manually request a reboot. + +

+
+ ) + )}
) : ( <>