From 3a606c6c41b83996d5f23860062548a63a08546c Mon Sep 17 00:00:00 2001
From: Lukas <134181853+bo2themax@users.noreply.github.com>
Date: Sat, 1 Aug 2026 17:32:48 +0200
Subject: [PATCH] macOS: install update with same code path

---
 .../TerminalCommandPalette.swift              |  2 +-
 .../Features/Update/UpdateController.swift    | 40 +++----------------
 2 files changed, 6 insertions(+), 36 deletions(-)

diff --git a/macos/Sources/Features/Command Palette/TerminalCommandPalette.swift b/macos/Sources/Features/Command Palette/TerminalCommandPalette.swift
index 09e369d4a..264e67cf8 100644
--- a/macos/Sources/Features/Command Palette/TerminalCommandPalette.swift	
+++ b/macos/Sources/Features/Command Palette/TerminalCommandPalette.swift	
@@ -104,7 +104,7 @@ struct TerminalCommandPaletteView: View {
             badge: updateViewModel.badge,
             emphasis: true
         ) {
-            (NSApp.delegate as? AppDelegate)?.updateController.installUpdate()
+            (NSApp.delegate as? AppDelegate)?.updateController.viewModel.state.confirm()
         })
 
         options.append(CommandOption(
diff --git a/macos/Sources/Features/Update/UpdateController.swift b/macos/Sources/Features/Update/UpdateController.swift
index 1ca218c8b..af99b4e86 100644
--- a/macos/Sources/Features/Update/UpdateController.swift
+++ b/macos/Sources/Features/Update/UpdateController.swift
@@ -10,7 +10,6 @@ import Combine
 class UpdateController {
     private(set) var updater: SPUUpdater
     private let userDriver: UpdateDriver
-    private var installCancellable: AnyCancellable?
 
     var viewModel: UpdateViewModel {
         userDriver.viewModel
@@ -18,7 +17,11 @@ class UpdateController {
 
     /// True if we're installing an update.
     var isInstalling: Bool {
-        installCancellable != nil
+        if case .installing = viewModel.state {
+            return true
+        } else {
+            return false
+        }
     }
 
     /// Initialize a new update controller.
@@ -35,10 +38,6 @@ class UpdateController {
         )
     }
 
-    deinit {
-        installCancellable?.cancel()
-    }
-
     /// Start the updater.
     ///
     /// This must be called before the updater can check for updates. If starting fails,
@@ -60,34 +59,6 @@ class UpdateController {
         }
     }
 
-    /// Force install the current update. As long as we're in some "update available" state this will
-    /// trigger all the steps necessary to complete the update.
-    func installUpdate() {
-        // Must be in an installable state
-        guard viewModel.state.isInstallable else { return }
-
-        // If we're already force installing then do nothing.
-        guard installCancellable == nil else { return }
-
-        // Setup a combine listener to listen for state changes and to always
-        // confirm them. If we go to a non-installable state, cancel the listener.
-        // The sink runs immediately with the current state, so we don't need to
-        // manually confirm the first state.
-        installCancellable = viewModel.$state.sink { [weak self] state in
-            guard let self else { return }
-
-            // If we move to a non-installable state (error, idle, etc.) then we
-            // stop force installing.
-            guard state.isInstallable else {
-                self.installCancellable = nil
-                return
-            }
-
-            // Continue the `yes` chain!
-            state.confirm()
-        }
-    }
-
     /// Check for updates.
     ///
     /// This is typically connected to a menu item action.
@@ -99,7 +70,6 @@ class UpdateController {
         }
 
         // If we're not idle then we need to cancel any prior state.
-        installCancellable?.cancel()
         viewModel.state.cancel()
 
         // The above will take time to settle, so we delay the check for some time.