Add schnauzer-v2 settings generator migrations

Release.toml

@@ -216,4 +216,6 @@ version = "1.15.0"
     "migrate_v1.14.2_ecs-images-cleanup.lz4",
 ]
 "(1.14.2, 1.14.3)" = []
-"(1.14.3, 1.15.0)" = []
+"(1.14.3, 1.15.0)" = [
+    "migrate_v1.15.0_schnauzer-v2-generators.lz4",
+]

sources/Cargo.toml

@@ -53,6 +53,7 @@ members = [
     "api/migration/migrations/v1.14.0/aws-control-container-v0-7-2",
     "api/migration/migrations/v1.14.0/public-control-container-v0-7-2",
     "api/migration/migrations/v1.14.2/ecs-images-cleanup",
+    "api/migration/migrations/v1.15.0/schnauzer-v2-generators",
 
     "bloodhound",
 

sources/api/migration/migrations/v1.15.0/schnauzer-v2-generators/Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "schnauzer-v2-generators"
+version = "0.1.0"
+authors = ["Sean P. Kelly <seankell@amazon.com>"]
+license = "Apache-2.0 OR MIT"
+edition = "2021"
+publish = false
+# Don't rebuild crate just because of changes to README.
+exclude = ["README.md"]
+
+[dependencies]
+migration-helpers = { path = "../../../migration-helpers", version = "0.1.0" }
+
+[build-dependencies]
+bottlerocket-variant = { version = "0.1", path = "../../../../../bottlerocket-variant" }

sources/api/migration/migrations/v1.15.0/schnauzer-v2-generators/build.rs

@@ -0,0 +1,6 @@
+use bottlerocket_variant::Variant;
+
+fn main() {
+    let variant = Variant::from_env().unwrap();
+    variant.emit_cfgs();
+}

sources/api/migration/migrations/v1.15.0/schnauzer-v2-generators/src/main.rs

@@ -0,0 +1,78 @@
+use migration_helpers::common_migrations::{MetadataReplacement, ReplaceMetadataMigration};
+use migration_helpers::{migrate, Result};
+use std::process;
+
+fn build_metadata_migrations() -> Vec<MetadataReplacement> {
+    let mut migrations = vec![];
+
+    // On AWS platforms, we use regional ECR repositories.
+    // Elsewhere, we use ecr-public, which is global.
+    #[cfg(variant_platform = "aws")]
+    {
+        migrations.append(&mut vec![
+            MetadataReplacement {
+                setting: "settings.host-containers.admin.source",
+                metadata: "setting-generator",
+                old_val: "schnauzer settings.host-containers.admin.source",
+                new_val: "schnauzer-v2 render --requires 'aws@v1(helpers[ecr-prefix])' --template '{{ ecr-prefix settings.aws.region }}/bottlerocket-admin:v0.10.1'",
+            },
+            MetadataReplacement {
+                setting: "settings.host-containers.control.source",
+                metadata: "setting-generator",
+                old_val: "schnauzer settings.host-containers.control.source",
+                new_val: "schnauzer-v2 render --requires 'aws@v1(helpers=[ecr-prefix])' --template '{{ ecr-prefix settings.aws.region }}/bottlerocket-control:v0.7.2'",
+            },
+            MetadataReplacement {
+                setting: "settings.updates.metadata-base-url",
+                metadata: "setting-generator",
+                old_val: "schnauzer settings.updates.metadata-base-url",
+                new_val: "schnauzer-v2 render --requires 'aws@v1' --requires 'updates@v1(helpers=[metadata-prefix, tuf-prefix])' --template '{{ tuf-prefix settings.aws.region }}{{ metadata-prefix settings.aws.region }}/2020-07-07/{{ os.variant_id }}/{{ os.arch }}/'",
+            },
+            MetadataReplacement {
+                setting: "settings.updates.targets-base-url",
+                metadata: "setting-generator",
+                old_val: "schnauzer settings.updates.targets-base-url",
+                new_val: "schnauzer-v2 render --requires 'aws@v1' --requires 'updates@v1(helpers=[tuf-prefix])' --template '{{ tuf-prefix settings.aws.region }}/targets/'",
+            },
+        ]);
+    }
+    #[cfg(not(variant_platform = "aws"))]
+    {
+        migrations.append(&mut vec![
+            MetadataReplacement {
+                setting: "settings.updates.metadata-base-url",
+                metadata: "setting-generator",
+                old_val: "schnauzer settings.updates.metadata-base-url",
+                new_val: "schnauzer-v2 render --template 'https://updates.bottlerocket.aws/2020-07-07/{{ os.variant_id }}/{{ os.arch }}/'",
+            },
+        ]);
+    }
+
+    #[cfg(variant_family = "aws-k8s")]
+    {
+        migrations.append(&mut vec![
+            MetadataReplacement {
+                setting: "settings.kubernetes.pod-infra-container-image",
+                metadata: "setting-generator",
+                old_val: "schnauzer settings.kubernetes.pod-infra-container-image",
+                new_val: "schnauzer-v2 render --requires 'aws@v1' --requires 'kubernetes@v1(helpers=[pause-prefix])' --template '{{ pause-prefix settings.aws.region }}/eks/pause:3.1-eksbuild.1'",
+            },
+        ]);
+    }
+
+    migrations
+}
+
+fn run() -> Result<()> {
+    migrate(ReplaceMetadataMigration(build_metadata_migrations()))
+}
+
+// Returning a Result from main makes it print a Debug representation of the error, but with Snafu
+// we have nice Display representations of the error, so we wrap "main" (run) and print any error.
+// https://github.com/shepmaster/snafu/issues/110
+fn main() {
+    if let Err(e) = run() {
+        eprintln!("{}", e);
+        process::exit(1);
+    }
+}