From 73b5d5514e16bf2f1b763b53c28f7e719da23440 Mon Sep 17 00:00:00 2001
From: Aarnav Tale <git@tale.me>
Date: Mon, 30 Mar 2026 14:02:39 -0400
Subject: [PATCH] fix: don't nuke agent working dir

---
 app/server/hp-agent.ts | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/app/server/hp-agent.ts b/app/server/hp-agent.ts
index 9122abb..4cabd1a 100644
--- a/app/server/hp-agent.ts
+++ b/app/server/hp-agent.ts
@@ -148,13 +148,21 @@ export async function createAgentManager(
       try {
         stdout = await runAgent(authKey);
       } catch (err) {
-        if (stateExists) {
-          log.info("agent", "Agent failed with existing state, clearing and retrying");
+        if (!stateExists) {
+          throw err;
+        }
+
+        // Retry once with existing state (e.g. stale lock from a previous run)
+        log.info("agent", "Agent failed with existing state, retrying");
+        try {
+          const retryKey = await generateAuthKey();
+          stdout = await runAgent(retryKey);
+        } catch {
+          // Only clear identity as a last resort
+          log.warn("agent", "Retry failed, clearing state and starting fresh");
           await rm(join(workDir, "tailscaled.state"), { force: true });
           const freshKey = await generateAuthKey();
           stdout = await runAgent(freshKey);
-        } else {
-          throw err;
         }
       }