fleet/orbit/pkg/update/disk_encryption.go

package update

import (
	"sync/atomic"

	"github.com/fleetdm/fleet/v4/orbit/pkg/useraction"
	"github.com/fleetdm/fleet/v4/server/fleet"
	"github.com/rs/zerolog/log"
)

const maxRetries = 2

type DiskEncryptionRunner struct {
	fetcher   OrbitConfigFetcher
	isRunning atomic.Bool
}

func ApplyDiskEncryptionRunnerMiddleware(f OrbitConfigFetcher) *DiskEncryptionRunner {
	return &DiskEncryptionRunner{fetcher: f}
}

func (d *DiskEncryptionRunner) GetConfig() (*fleet.OrbitConfig, error) {
	cfg, err := d.fetcher.GetConfig()
	if err != nil {
		log.Info().Err(err).Msg("calling GetConfig from DiskEncryptionFetcher")
		return nil, err
	}

	log.Debug().Msgf("running disk encryption fetcher middleware, notification: %v, isIdle: %v", cfg.Notifications.RotateDiskEncryptionKey, d.isRunning.Load())

	if cfg.Notifications.RotateDiskEncryptionKey && !d.isRunning.Swap(true) {
		go func() {
			defer d.isRunning.Store(false)
			if err := useraction.RotateDiskEncryptionKey(maxRetries); err != nil {
				log.Error().Err(err).Msg("rotating encryption key")
			}
		}()
	}

	return cfg, nil
}