Allow configuring enablePlatformTags via plugins

jib-core/src/integration-test/java/com/google/cloud/tools/jib/api/ContainerizerIntegrationTest.java

@@ -19,13 +19,15 @@
 import com.google.cloud.tools.jib.Command;
 import com.google.cloud.tools.jib.api.buildplan.AbsoluteUnixPath;
 import com.google.cloud.tools.jib.api.buildplan.FileEntriesLayer;
+import com.google.cloud.tools.jib.api.buildplan.Platform;
 import com.google.cloud.tools.jib.event.events.ProgressEvent;
 import com.google.cloud.tools.jib.event.progress.ProgressEventHandler;
 import com.google.cloud.tools.jib.global.JibSystemProperties;
 import com.google.cloud.tools.jib.registry.LocalRegistry;
 import com.google.common.base.Splitter;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableSet;
 import com.google.common.io.Resources;
 import java.io.IOException;
 import java.net.URISyntaxException;
@@ -35,6 +37,8 @@
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
+import java.util.Set;
 import java.util.concurrent.ExecutionException;
 import java.util.stream.Stream;
 import org.hamcrest.CoreMatchers;
@@ -83,6 +87,8 @@ private void checkCompletion() {
   private static final Logger logger = LoggerFactory.getLogger(ContainerizerIntegrationTest.class);
   private static final String DISTROLESS_DIGEST =
       "sha256:f488c213f278bc5f9ffe3ddf30c5dbb2303a15a74146b738d12453088e662880";
+  private static final String MULTI_PLATFORM_DIGEST =
+      "sha256:c737fc29fc2556d3377d6a719a9842a500777fce35a7f1299acd569c73f65247";
   private static final double DOUBLE_ERROR_MARGIN = 1e-10;
 
   public static ImmutableList<FileEntriesLayer> fakeLayerConfigurations;
@@ -228,6 +234,30 @@ public void testSteps_forBuildToDockerRegistry_multipleTags()
         new Command("docker", "run", "--rm", imageReference3).run());
   }
 
+  @Test
+  public void testSteps_forBuildToDockerRegistry_multiplePlatforms()
+      throws IOException, InterruptedException, ExecutionException, RegistryException,
+          CacheDirectoryCreationException, InvalidImageReferenceException {
+    buildImage(
+        ImageReference.of("gcr.io", "distroless/java17-debian11", MULTI_PLATFORM_DIGEST),
+        Containerizer.to(RegistryImage.named(dockerHost + ":5000/testimage:testtag")),
+        Collections.singletonList("testtag"),
+        ImmutableSet.of(new Platform("amd64", "linux"), new Platform("arm64", "linux")));
+
+    String imageReference = dockerHost + ":5000/testimage:testtag-amd64";
+    localRegistry.pull(imageReference);
+    assertDockerInspect(imageReference);
+    Assert.assertEquals(
+        "Hello, world. An argument.\n", new Command("docker", "run", "--rm", imageReference).run());
+
+    String imageReference2 = dockerHost + ":5000/testimage:testtag-arm64";
+    localRegistry.pull(imageReference2);
+    assertDockerInspect(imageReference2);
+    Assert.assertEquals(
+        "Hello, world. An argument.\n",
+        new Command("docker", "run", "--rm", imageReference2).run());
+  }
+
   @Test
   public void testSteps_forBuildToDockerRegistry_skipExistingDigest()
       throws IOException, InterruptedException, ExecutionException, RegistryException,
@@ -336,6 +366,16 @@ private JibContainer buildImage(
       ImageReference baseImage, Containerizer containerizer, List<String> additionalTags)
       throws IOException, InterruptedException, RegistryException, CacheDirectoryCreationException,
           ExecutionException {
+    return buildImage(baseImage, containerizer, additionalTags, null);
+  }
+
+  private JibContainer buildImage(
+      ImageReference baseImage,
+      Containerizer containerizer,
+      List<String> additionalTags,
+      Set<Platform> additionalPlatforms)
+      throws IOException, InterruptedException, RegistryException, CacheDirectoryCreationException,
+          ExecutionException {
     JibContainerBuilder containerBuilder =
         Jib.from(baseImage)
             .setEntrypoint(
@@ -346,6 +386,9 @@ private JibContainer buildImage(
             .setExposedPorts(Ports.parse(Arrays.asList("1000", "2000-2002/tcp", "3000/udp")))
             .setLabels(ImmutableMap.of("key1", "value1", "key2", "value2"))
             .setFileEntriesLayers(fakeLayerConfigurations);
+    if (Objects.nonNull(additionalPlatforms)) {
+      containerBuilder.setPlatforms(additionalPlatforms);
+    }
 
     containerizer
         .setAllowInsecureRegistries(true)

jib-core/src/main/java/com/google/cloud/tools/jib/api/Containerizer.java

@@ -159,6 +159,7 @@ public static Containerizer to(DockerClient dockerClient, DockerDaemonImage dock
   @Nullable private String toolVersion = DEFAULT_TOOL_VERSION;
   private boolean alwaysCacheBaseImage = false;
   private ListMultimap<String, String> registryMirrors = ArrayListMultimap.create();
+  private boolean enablePlatformTags = false;
 
   /** Instantiate with {@link #to}. */
   private Containerizer(
@@ -339,6 +340,18 @@ public Containerizer addRegistryMirrors(String registry, List<String> mirrors) {
     return this;
   }
 
+  /**
+   * Enables adding platform tags to images.
+   *
+   * @param enablePlatformTags if {@code true}, adds platform tags to images. If {@code false}
+   *     images are not tagged with platform tags.
+   * @return this
+   */
+  public Containerizer setEnablePlatformTags(boolean enablePlatformTags) {
+    this.enablePlatformTags = enablePlatformTags;
+    return this;
+  }
+
   Set<String> getAdditionalTags() {
     return ImmutableSet.copyOf(additionalTags);
   }
@@ -394,6 +407,10 @@ boolean getAlwaysCacheBaseImage() {
     return alwaysCacheBaseImage;
   }
 
+  boolean getEnablePlatformTags() {
+    return enablePlatformTags;
+  }
+
   String getDescription() {
     return description;
   }

jib-core/src/main/java/com/google/cloud/tools/jib/api/JibContainerBuilder.java

@@ -736,6 +736,7 @@ BuildContext toBuildContext(Containerizer containerizer) throws CacheDirectoryCr
         .setEventHandlers(containerizer.buildEventHandlers())
         .setAlwaysCacheBaseImage(containerizer.getAlwaysCacheBaseImage())
         .setRegistryMirrors(containerizer.getRegistryMirrors())
+        .setEnablePlatformTags(containerizer.getEnablePlatformTags())
         .build();
   }
 

jib-core/src/test/java/com/google/cloud/tools/jib/api/ContainerizerTest.java

@@ -66,7 +66,8 @@ private void verifyTo(Containerizer containerizer) throws CacheDirectoryCreation
         .setBaseImageLayersCache(Paths.get("base/image/layers"))
         .setApplicationLayersCache(Paths.get("application/layers"))
         .setAllowInsecureRegistries(true)
-        .setToolName("tool");
+        .setToolName("tool")
+        .setEnablePlatformTags(true);
 
     Assert.assertEquals(ImmutableSet.of("tag1", "tag2"), containerizer.getAdditionalTags());
     Assert.assertTrue(containerizer.getExecutorService().isPresent());
@@ -77,6 +78,7 @@ private void verifyTo(Containerizer containerizer) throws CacheDirectoryCreation
         Paths.get("application/layers"), containerizer.getApplicationLayersCacheDirectory());
     Assert.assertTrue(containerizer.getAllowInsecureRegistries());
     Assert.assertEquals("tool", containerizer.getToolName());
+    Assert.assertTrue(containerizer.getEnablePlatformTags());
   }
 
   @Test

jib-core/src/test/java/com/google/cloud/tools/jib/api/JibContainerBuilderTest.java

@@ -149,7 +149,8 @@ public void testToBuildContext()
             .setApplicationLayersCache(Paths.get("application/layers"))
             .setExecutorService(executorService)
             .addEventHandler(mockJibEventConsumer)
-            .setAlwaysCacheBaseImage(false);
+            .setAlwaysCacheBaseImage(false)
+            .setEnablePlatformTags(true);
 
     ImageConfiguration baseImageConfiguration =
         ImageConfiguration.builder(ImageReference.parse("base/image"))
@@ -220,6 +221,8 @@ public void testToBuildContext()
         ImmutableSet.of("latest", "tag1", "tag2"), buildContext.getAllTargetImageTags());
     Assert.assertEquals("toolName", buildContext.getToolName());
     Assert.assertFalse(buildContext.getAlwaysCacheBaseImage());
+
+    Assert.assertTrue(buildContext.getEnablePlatformTags());
   }
 
   @Test

jib-gradle-plugin/README.md

@@ -225,6 +225,7 @@ Property | Type | Default | Description
 `auth` | [`auth`](#auth-closure) | *None* | Specifies credentials directly (alternative to `credHelper`).
 `credHelper` | `String` | *None* | Specifies a credential helper that can authenticate pushing the target image. This parameter can either be configured as an absolute path to the credential helper executable or as a credential helper suffix (following `docker-credential-`).
 `tags` | `List<String>` | *None* | Additional tags to push to.
+`enablePlatformTags` | `boolean` | `false` | When creating multi-platform images takes the tags, suffixes them with the platform, and tags the image.
 
 <a name="auth-closure"></a>`auth` is a closure with the following properties (see [Using Specific Credentials](#using-specific-credentials)):
 

jib-gradle-plugin/src/main/java/com/google/cloud/tools/jib/gradle/GradleRawConfiguration.java

@@ -222,4 +222,9 @@ public List<? extends ExtensionConfiguration> getPluginExtensions() {
   public List<? extends PlatformConfiguration> getPlatforms() {
     return jibExtension.getFrom().getPlatforms().get();
   }
+
+  @Override
+  public boolean getEnablePlatformTags() {
+    return jibExtension.getTo().getEnablePlatformTags();
+  }
 }

jib-gradle-plugin/src/main/java/com/google/cloud/tools/jib/gradle/TargetImageParameters.java

@@ -39,6 +39,7 @@ public class TargetImageParameters {
   private final Property<String> image;
   private final SetProperty<String> tags;
   private final CredHelperParameters credHelper;
+  private boolean enablePlatformTags;
 
   @Inject
   public TargetImageParameters(ObjectFactory objectFactory) {
@@ -123,4 +124,16 @@ public AuthParameters getAuth() {
   public void auth(Action<? super AuthParameters> action) {
     action.execute(auth);
   }
+
+  @Input
+  public boolean getEnablePlatformTags() {
+    if (System.getProperty(PropertyNames.ENABLE_PLATFORM_TAGS) != null) {
+      return Boolean.parseBoolean(System.getProperty(PropertyNames.ENABLE_PLATFORM_TAGS));
+    }
+    return enablePlatformTags;
+  }
+
+  public void setEnablePlatformTags(boolean expand) {
+    enablePlatformTags = expand;
+  }
 }

jib-gradle-plugin/src/test/java/com/google/cloud/tools/jib/gradle/GradleRawConfigurationTest.java

@@ -81,6 +81,7 @@ public void testGetters() {
     Mockito.when(toCredHelperParameters.getEnvironment())
         .thenReturn(Collections.singletonMap("ENV_VARIABLE", "Value2"));
     Mockito.when(targetImageParameters.getCredHelper()).thenReturn(toCredHelperParameters);
+    Mockito.when(targetImageParameters.getEnablePlatformTags()).thenReturn(true);
 
     Mockito.when(containerParameters.getAppRoot()).thenReturn("/app/root");
     Mockito.when(containerParameters.getArgs()).thenReturn(Arrays.asList("--log", "info"));
@@ -150,5 +151,6 @@ public void testGetters() {
     Assert.assertEquals(Paths.get("id/path"), rawConfiguration.getImageIdOutputPath());
     Assert.assertEquals(Paths.get("json/path"), rawConfiguration.getImageJsonOutputPath());
     Assert.assertEquals(Paths.get("tar/path"), rawConfiguration.getTarOutputPath());
+    Assert.assertTrue(rawConfiguration.getEnablePlatformTags());
   }
 }

jib-gradle-plugin/src/test/java/com/google/cloud/tools/jib/gradle/JibExtensionTest.java

@@ -453,6 +453,8 @@ public void testProperties() {
     assertThat(testJibExtension.getTo().getTags()).containsExactly("tag1", "tag2", "tag3");
     System.setProperty("jib.to.credHelper", "credHelper");
     assertThat(testJibExtension.getTo().getCredHelper().getHelper()).isEqualTo("credHelper");
+    System.setProperty("jib.to.enablePlatformTags", "true");
+    assertThat(testJibExtension.getTo().getEnablePlatformTags()).isTrue();
 
     System.setProperty("jib.container.appRoot", "appRoot");
     assertThat(testJibExtension.getContainer().getAppRoot()).isEqualTo("appRoot");

jib-plugins-common/src/main/java/com/google/cloud/tools/jib/plugins/common/PluginConfigurationProcessor.java

@@ -1054,6 +1054,8 @@ private static void configureContainerizer(
                 PropertyNames.APPLICATION_CACHE, projectProperties.getDefaultCacheDirectory()));
 
     rawConfiguration.getToTags().forEach(containerizer::withAdditionalTag);
+
+    containerizer.setEnablePlatformTags(rawConfiguration.getEnablePlatformTags());
   }
 
   /**

jib-plugins-common/src/main/java/com/google/cloud/tools/jib/plugins/common/PropertyNames.java

@@ -27,6 +27,7 @@ public class PropertyNames {
   public static final String TO_IMAGE = "jib.to.image";
   public static final String TO_IMAGE_ALTERNATE = "image";
   public static final String TO_TAGS = "jib.to.tags";
+  public static final String ENABLE_PLATFORM_TAGS = "jib.to.enablePlatformTags";
   public static final String TO_CRED_HELPER = "jib.to.credHelper";
   public static final String TO_AUTH_USERNAME = "jib.to.auth.username";
   public static final String TO_AUTH_PASSWORD = "jib.to.auth.password";

jib-plugins-common/src/main/java/com/google/cloud/tools/jib/plugins/common/RawConfiguration.java

@@ -134,4 +134,8 @@ interface CredHelperConfiguration {
   Path getImageJsonOutputPath();
 
   List<? extends ExtensionConfiguration> getPluginExtensions();
+
+  default boolean getEnablePlatformTags() {
+    return false;
+  }
 }

jib-plugins-common/src/test/java/com/google/cloud/tools/jib/plugins/common/PluginConfigurationProcessorTest.java

@@ -216,6 +216,8 @@ public void testPluginConfigurationProcessor_defaults()
     verify(containerizer).setBaseImageLayersCache(Containerizer.DEFAULT_BASE_CACHE_DIRECTORY);
     verify(containerizer).setApplicationLayersCache(appCacheDirectory);
 
+    verify(containerizer).setEnablePlatformTags(false);
+
     ArgumentMatcher<LogEvent> isLogWarn = logEvent -> logEvent.getLevel() == LogEvent.Level.WARN;
     verify(logger, never()).accept(argThat(isLogWarn));
   }