From 9daa32894197a91fd3444c990cc95dc3d419ac32 Mon Sep 17 00:00:00 2001
From: Thomas Marchand <thomas.marchand@tuta.io>
Date: Fri, 16 Jan 2026 15:58:45 +0000
Subject: [PATCH] Fix update process: reset repo before checkout and ensure SSE
 flush

- Add git reset --hard HEAD before checkout to prevent local changes blocking updates
- Add git clean -fd to remove untracked files that might interfere
- Add 100ms delay before restart to ensure the "restarting" SSE event is flushed
---
 src/api/system.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/api/system.rs b/src/api/system.rs
index 2b160d6..7bf92f3 100644
--- a/src/api/system.rs
+++ b/src/api/system.rs
@@ -505,6 +505,20 @@ fn stream_open_agent_update() -> impl Stream<Item = Result<Event, std::convert::
             progress: Some(15),
         }).unwrap()));
 
+        // Reset any local changes before checkout to prevent conflicts
+        let _ = Command::new("git")
+            .args(["reset", "--hard", "HEAD"])
+            .current_dir(OPEN_AGENT_REPO_PATH)
+            .output()
+            .await;
+
+        // Clean untracked files that might interfere
+        let _ = Command::new("git")
+            .args(["clean", "-fd"])
+            .current_dir(OPEN_AGENT_REPO_PATH)
+            .output()
+            .await;
+
         // Checkout the tag/branch
         let checkout_result = Command::new("git")
             .args(["checkout", &latest_tag])
@@ -648,6 +662,9 @@ fn stream_open_agent_update() -> impl Stream<Item = Result<Event, std::convert::
             progress: Some(100),
         }).unwrap()));
 
+        // Small delay to ensure the SSE event is flushed before we restart
+        tokio::time::sleep(tokio::time::Duration::from_millis(100)).await;
+
         // Restart the service - this will terminate our process, so no code after this
         // will execute. The client should poll /api/health to confirm the new version.
         let _ = Command::new("systemctl")