[antlir2][docker_archive] add support for adding RepoTags

Differential Revision: D85365439

fbshipit-source-id: 8863adde11cfffeb5c824dff1d8871df0c31d63f

Author: vmagro

antlir/antlir2/antlir2_packager/src/docker_archive.rs

@@ -24,6 +24,7 @@ use crate::run_cmd;
 pub struct DockerArchive {
     build_appliance: BuildAppliance,
     oci: PathBuf,
+    repo_tags: Vec<String>,
 }
 
 impl DockerArchive {
@@ -50,6 +51,11 @@ impl DockerArchive {
                 .command("skopeo")?
                 .arg("copy")
                 .arg(oci_arg)
+                .args(
+                    self.repo_tags
+                        .iter()
+                        .map(|t| format!("--additional-tag={t}")),
+                )
                 .arg("docker-archive:/__antlir2__/out/docker_archive"),
         )
         .context("Failed to convert to docker-archive")?;

antlir/antlir2/bzl/package/docker_archive.bzl

@@ -22,6 +22,7 @@ def _impl(ctx: AnalysisContext) -> Promise:
             {"docker_archive": {
                 "build_appliance": build_appliance[BuildApplianceInfo].dir,
                 "oci": oci,
+                "repo_tags": ctx.attrs.repo_tags,
             }},
             with_inputs = True,
         )
@@ -51,7 +52,9 @@ def _impl(ctx: AnalysisContext) -> Promise:
 
 _docker_archive = rule(
     impl = _impl,
-    attrs = oci_attrs | layer_attrs | default_attrs | common_attrs,
+    attrs = {
+        "repo_tags": attrs.list(attrs.string(), default = []),
+    } | oci_attrs | layer_attrs | default_attrs | common_attrs,
     cfg = package_cfg,
 )