[TUF] Add flag for TUF autoupdater rollout (#1524)

ee/agent/flags/flag_controller.go

@@ -500,6 +500,15 @@ func (fc *FlagController) UpdateDirectory() string {
 	).get(fc.getControlServerValue(keys.UpdateDirectory))
 }
 
+func (fc *FlagController) SetUseTUFAutoupdater(enabled bool) error {
+	return fc.setControlServerValue(keys.UseTUFAutoupdater, boolToBytes(enabled))
+}
+func (fc *FlagController) UseTUFAutoupdater() bool {
+	return NewBoolFlagValue(
+		WithDefaultBool(false),
+	).get(fc.getControlServerValue(keys.UseTUFAutoupdater))
+}
+
 func (fc *FlagController) SetExportTraces(enabled bool) error {
 	return fc.setControlServerValue(keys.ExportTraces, boolToBytes(enabled))
 }

ee/agent/flags/keys/keys.go

@@ -44,6 +44,7 @@ const (
 	MirrorServerURL                 FlagKey = "mirror_url"
 	AutoupdateInterval              FlagKey = "autoupdate_interval"
 	UpdateChannel                   FlagKey = "update_channel"
+	UseTUFAutoupdater               FlagKey = "use_tuf_autoupdater"
 	NotaryPrefix                    FlagKey = "notary_prefix"
 	AutoupdateInitialDelay          FlagKey = "autoupdater_initial_delay"
 	UpdateDirectory                 FlagKey = "update_directory"

ee/agent/knapsack/knapsack.go

@@ -419,6 +419,13 @@ func (k *knapsack) UpdateDirectory() string {
 	return k.flags.UpdateDirectory()
 }
 
+func (k *knapsack) SetUseTUFAutoupdater(enabled bool) error {
+	return k.flags.SetUseTUFAutoupdater(enabled)
+}
+func (k *knapsack) UseTUFAutoupdater() bool {
+	return k.flags.UseTUFAutoupdater()
+}
+
 func (k *knapsack) SetExportTraces(enabled bool) error {
 	return k.flags.SetExportTraces(enabled)
 }

ee/agent/types/flags.go

@@ -185,6 +185,10 @@ type Flags interface {
 	SetUpdateDirectory(directory string) error
 	UpdateDirectory() string
 
+	// UseTUFAutoupdater controls whether launcher uses the new TUF autoupdater instead of the legacy autoupdater
+	SetUseTUFAutoupdater(enabled bool) error
+	UseTUFAutoupdater() bool
+
 	// ExportTraces enables exporting our traces
 	SetExportTraces(enabled bool) error
 	SetExportTracesOverride(value bool, duration time.Duration)