WIP - add rollbacks available to upgrade marker

Author: pchila

internal/pkg/agent/application/application.go

@@ -120,7 +120,7 @@ func New(
 
 	// monitoring is not supported in bootstrap mode https://github.com/elastic/elastic-agent/issues/1761
 	isMonitoringSupported := !disableMonitoring && cfg.Settings.V1MonitoringEnabled
-	upgrader, err := upgrade.NewUpgrader(log, cfg.Settings.DownloadConfig, agentInfo)
+	upgrader, err := upgrade.NewUpgrader(log, cfg.Settings.DownloadConfig, cfg.Settings.Upgrade, agentInfo)
 	if err != nil {
 		return nil, nil, nil, fmt.Errorf("failed to create upgrader: %w", err)
 	}

internal/pkg/agent/application/coordinator/coordinator_unit_test.go

@@ -461,11 +461,7 @@ func TestCoordinatorReportsInvalidPolicy(t *testing.T) {
 		}
 	}()
 
-	upgradeMgr, err := upgrade.NewUpgrader(
-		log,
-		&artifact.Config{},
-		&info.AgentInfo{},
-	)
+	upgradeMgr, err := upgrade.NewUpgrader(log, &artifact.Config{}, nil, &info.AgentInfo{})
 	require.NoError(t, err, "errored when creating a new upgrader")
 
 	// Channels have buffer length 1, so we don't have to run on multiple

internal/pkg/agent/application/upgrade/details/details.go

@@ -61,6 +61,17 @@ type Metadata struct {
 	Reason string `json:"reason,omitempty" yaml:"reason,omitempty"`
 }
 
+//rollback_available:
+//- version: 8.16.0-SNAPSHOT,
+//home: data/elastic-agent-8.16.0-SNAPSHOT-b65953
+//valid_until: "2024-11-21T14:42:21Z"
+
+type RollbackAvailable struct {
+	Version    string    `json:"version" yaml:"version"`
+	Home       string    `json:"home" yaml:"home"`
+	ValidUntil time.Time `json:"valid_until" yaml:"valid_until"`
+}
+
 func NewDetails(targetVersion string, initialState State, actionID string) *Details {
 	return &Details{
 		TargetVersion: targetVersion,

internal/pkg/agent/application/upgrade/rollback_test.go

@@ -504,10 +504,6 @@ func createUpdateMarker(t *testing.T, log *logger.Logger, topDir, newAgentVersio
 		hash:          oldAgentHash,
 		versionedHome: oldAgentVersionedHome,
 	}
-	err := markUpgrade(log,
-		paths.DataFrom(topDir),
-		newAgentInstall,
-		oldAgentInstall,
-		nil, nil, OUTCOME_UPGRADE)
+	err := markUpgrade(log, paths.DataFrom(topDir), newAgentInstall, oldAgentInstall, nil, nil, OUTCOME_UPGRADE, 0)
 	require.NoError(t, err, "error writing fake update marker")
 }

internal/pkg/agent/application/upgrade/step_download_test.go

@@ -91,7 +91,7 @@ func TestDownloadWithRetries(t *testing.T) {
 			return &mockDownloader{expectedDownloadPath, nil}, nil
 		}
 
-		u, err := NewUpgrader(testLogger, &settings, &info.AgentInfo{})
+		u, err := NewUpgrader(testLogger, &settings, nil, &info.AgentInfo{})
 		require.NoError(t, err)
 
 		parsedVersion, err := agtversion.ParseVersion("8.9.0")
@@ -141,7 +141,7 @@ func TestDownloadWithRetries(t *testing.T) {
 			return nil, nil
 		}
 
-		u, err := NewUpgrader(testLogger, &settings, &info.AgentInfo{})
+		u, err := NewUpgrader(testLogger, &settings, nil, &info.AgentInfo{})
 		require.NoError(t, err)
 
 		parsedVersion, err := agtversion.ParseVersion("8.9.0")
@@ -196,7 +196,7 @@ func TestDownloadWithRetries(t *testing.T) {
 			return nil, nil
 		}
 
-		u, err := NewUpgrader(testLogger, &settings, &info.AgentInfo{})
+		u, err := NewUpgrader(testLogger, &settings, nil, &info.AgentInfo{})
 		require.NoError(t, err)
 
 		parsedVersion, err := agtversion.ParseVersion("8.9.0")
@@ -241,7 +241,7 @@ func TestDownloadWithRetries(t *testing.T) {
 			return &mockDownloader{"", errors.New("download failed")}, nil
 		}
 
-		u, err := NewUpgrader(testLogger, &settings, &info.AgentInfo{})
+		u, err := NewUpgrader(testLogger, &settings, nil, &info.AgentInfo{})
 		require.NoError(t, err)
 
 		parsedVersion, err := agtversion.ParseVersion("8.9.0")

internal/pkg/agent/application/upgrade/step_mark.go

@@ -197,7 +197,7 @@ type agentInstall struct {
 }
 
 // markUpgrade marks update happened so we can handle grace period
-func markUpgrade(log *logger.Logger, dataDirPath string, agent, previousAgent agentInstall, action *fleetapi.ActionUpgrade, upgradeDetails *details.Details, desiredOutcome UpgradeOutcome) error {
+func markUpgrade(log *logger.Logger, dataDirPath string, agent, previousAgent agentInstall, action *fleetapi.ActionUpgrade, upgradeDetails *details.Details, desiredOutcome UpgradeOutcome, rollbackWindow time.Duration) error {
 
 	if len(previousAgent.hash) > hashLen {
 		previousAgent.hash = previousAgent.hash[:hashLen]
@@ -216,6 +216,11 @@ func markUpgrade(log *logger.Logger, dataDirPath string, agent, previousAgent ag
 		DesiredOutcome:    desiredOutcome,
 	}
 
+	if rollbackWindow > 0 {
+		upgradeDetails.Metadata.
+	}
+
+
 	markerBytes, err := yaml.Marshal(newMarkerSerializer(marker))
 	if err != nil {
 		return errors.New(err, errors.TypeConfig, "failed to parse marker file")

internal/pkg/agent/application/upgrade/upgrade.go

@@ -73,12 +73,13 @@ func init() {
 
 // Upgrader performs an upgrade
 type Upgrader struct {
-	log            *logger.Logger
-	settings       *artifact.Config
-	agentInfo      info.Agent
-	upgradeable    bool
-	fleetServerURI string
-	markerWatcher  MarkerWatcher
+	log             *logger.Logger
+	settings        *artifact.Config
+	upgradeSettings *configuration.UpgradeConfig
+	agentInfo       info.Agent
+	upgradeable     bool
+	fleetServerURI  string
+	markerWatcher   MarkerWatcher
 }
 
 // IsUpgradeable when agent is installed and running as a service or flag was provided.
@@ -89,13 +90,14 @@ func IsUpgradeable() bool {
 }
 
 // NewUpgrader creates an upgrader which is capable of performing upgrade operation
-func NewUpgrader(log *logger.Logger, settings *artifact.Config, agentInfo info.Agent) (*Upgrader, error) {
+func NewUpgrader(log *logger.Logger, settings *artifact.Config, upgradeConfig *configuration.UpgradeConfig, agentInfo info.Agent) (*Upgrader, error) {
 	return &Upgrader{
-		log:           log,
-		settings:      settings,
-		agentInfo:     agentInfo,
-		upgradeable:   IsUpgradeable(),
-		markerWatcher: newMarkerFileWatcher(markerFilePath(paths.Data()), log),
+		log:             log,
+		settings:        settings,
+		upgradeSettings: upgradeConfig,
+		agentInfo:       agentInfo,
+		upgradeable:     IsUpgradeable(),
+		markerWatcher:   newMarkerFileWatcher(markerFilePath(paths.Data()), log),
 	}, nil
 }
 
@@ -147,6 +149,8 @@ func (u *Upgrader) Reload(rawConfig *config.Config) error {
 	}
 
 	u.settings = cfg.Settings.DownloadConfig
+	u.upgradeSettings = cfg.Settings.Upgrade
+
 	return nil
 }
 
@@ -347,11 +351,7 @@ func (u *Upgrader) Upgrade(ctx context.Context, version string, rollback bool, s
 		versionedHome: currentVersionedHome,
 	}
 
-	if err := markUpgrade(u.log,
-		paths.Data(), // data dir to place the marker in
-		current,      // new agent version data
-		previous,     // old agent version data
-		action, det, OUTCOME_UPGRADE); err != nil {
+	if err := markUpgrade(u.log, paths.Data(), current, previous, action, det, OUTCOME_UPGRADE, 0); err != nil {
 		u.log.Errorw("Rolling back: marking upgrade failed", "error.message", err)
 		rollbackErr := rollbackInstall(ctx, u.log, paths.Top(), hashedDir, currentVersionedHome)
 		return nil, goerrors.Join(err, rollbackErr)