Add RebootRequired to UpdateCategory, use for reboot button logic

model/update.go

 package model
 
 type UpdateCategory struct {
-	Name         string
-	Enabled      bool // User can toggle this category
-	Available    bool // Has updates available
-	Packages     []Package
-	DownloadSize uint64
-	InstallSize  int64
-	LastChecked  string // RFC3339 timestamp
+	Name           string
+	Enabled        bool // User can toggle this category
+	Available      bool // Has updates available
+	RebootRequired bool
+	Packages       []Package
+	DownloadSize   uint64
+	InstallSize    int64
+	LastChecked    string // RFC3339 timestamp
 }
 
 type KernelUpdateInfo struct {

service/update_service.go

@@ -148,7 +148,13 @@ func (us *UpdateService) RunUpdates(ctx context.Context) error {
 	startTime := time.Now()
 	packagesUpdated := 0
 	var categoryErrors []string
-	systemOrKernelSucceeded := false
+	rebootRequired := false
+
+	categoryMap := map[string]*model.UpdateCategory{
+		"System": state.SystemUpdates,
+		"Kernel": state.KernelUpdates,
+		"Play":   state.PlayUpdates,
+	}
 
 	// Execute updates sequentially
 	for _, category := range categories {
@@ -158,13 +164,11 @@ func (us *UpdateService) RunUpdates(ctx context.Context) error {
 			continue
 		}
 
-		switch category {
-		case "System":
-			packagesUpdated += len(state.SystemUpdates.Packages)
-			systemOrKernelSucceeded = true
-		case "Kernel":
-			packagesUpdated += len(state.KernelUpdates.Packages)
-			systemOrKernelSucceeded = true
+		if cat, ok := categoryMap[category]; ok {
+			packagesUpdated += len(cat.Packages)
+			if cat.RebootRequired {
+				rebootRequired = true
+			}
 		}
 	}
 
@@ -174,9 +178,9 @@ func (us *UpdateService) RunUpdates(ctx context.Context) error {
 	// Finish update
 	success := len(categoryErrors) == 0
 	us.store.Dispatch(&store.FinishUpdateAction{
-		Success:                 success,
-		Error:                   strings.Join(categoryErrors, "; "),
-		SystemOrKernelSucceeded: systemOrKernelSucceeded,
+		Success:        success,
+		Error:          strings.Join(categoryErrors, "; "),
+		RebootRequired: rebootRequired,
 	})
 
 	// Record in history

store/actions.go

@@ -28,6 +28,7 @@ func (a *LoadSystemUpdatesAction) Apply(state *State) error {
 	state.SystemUpdates.Packages = append(state.SystemUpdates.Packages, a.Changes.Install...)
 	state.SystemUpdates.DownloadSize = a.Changes.DownloadSize
 	state.SystemUpdates.InstallSize = a.Changes.InstallSize
+	state.SystemUpdates.RebootRequired = true
 	state.SystemUpdates.LastChecked = time.Now().Format(time.RFC3339)
 	return nil
 }
@@ -42,6 +43,7 @@ func (a *LoadKernelUpdatesAction) Apply(state *State) error {
 	state.KernelUpdates.Packages = append(state.KernelUpdates.Packages, a.Info.Packages.Install...)
 	state.KernelUpdates.DownloadSize = a.Info.Packages.DownloadSize
 	state.KernelUpdates.InstallSize = a.Info.Packages.InstallSize
+	state.KernelUpdates.RebootRequired = true
 	state.KernelUpdates.LastChecked = time.Now().Format(time.RFC3339)
 	return nil
 }
@@ -124,9 +126,9 @@ func (a *UpdateProgressAction) Apply(state *State) error {
 }
 
 type FinishUpdateAction struct {
-	Success                 bool
-	Error                   string
-	SystemOrKernelSucceeded bool
+	Success        bool
+	Error          string
+	RebootRequired bool
 }
 
 func (a *FinishUpdateAction) Apply(state *State) error {

ui/window.go

@@ -136,8 +136,7 @@ func (w *Window) onStateChange(change store.StateChange) {
 	case store.ChangeUpdateFinish:
 		action, ok := change.Data.(*store.FinishUpdateAction)
 		if ok {
-			showReboot := action.Success || action.SystemOrKernelSucceeded
-			w.processPage.ShowComplete(action.Success, showReboot)
+			w.processPage.ShowComplete(action.Success, action.RebootRequired)
 		}
 
 	case store.ChangeHistory: