improve superseeding, fix linux persistence stuff

2026-06-14 02:12:26 -05:00
parent 41f1228fe9
commit c1fefec1b5
10 changed files with 588 additions and 12 deletions
@@ -0,0 +1,57 @@
+package main
+
+import (
+	"errors"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"overlord-client/cmd/agent/config"
+	"overlord-client/cmd/agent/persistence"
+
+	"nhooyr.io/websocket"
+)
+
+const supersededCloseCode = 4004
+
+var exitProcess = os.Exit
+
+func isSupersededError(err error) bool {
+	if err == nil {
+		return false
+	}
+	var closeErr *websocket.CloseError
+	if !errors.As(err, &closeErr) {
+		return false
+	}
+	return closeErr.Code == supersededCloseCode || strings.EqualFold(strings.TrimSpace(closeErr.Reason), "superseded")
+}
+
+func handleSuperseded(cfg config.Config) {
+	if !cfg.EnablePersistence || !builtWithPersistenceDefault() {
+		return
+	}
+
+	currentExe, err := os.Executable()
+	if err != nil {
+		log.Printf("superseded: failed to resolve executable for cleanup: %v", err)
+		return
+	}
+	if realPath, err := filepath.EvalSymlinks(currentExe); err == nil {
+		currentExe = realPath
+	}
+	currentExe, err = filepath.Abs(currentExe)
+	if err != nil {
+		log.Printf("superseded: failed to resolve absolute executable path for cleanup: %v", err)
+		return
+	}
+
+	if err := persistence.RemoveCurrentInstall(currentExe); err != nil {
+		log.Printf("superseded: current-install cleanup warning: %v", err)
+	}
+}
+
+func builtWithPersistenceDefault() bool {
+	return strings.EqualFold(strings.TrimSpace(config.DefaultPersistence), "true")
+}