Add tag/untag commands for container repositories

fixes: pulp#423

CHANGES/423.feature

@@ -0,0 +1 @@
+Added tag/untag commands to add and remove tags from images in container repositories.

pulpcore/cli/container/context.py

@@ -1,4 +1,5 @@
 import gettext
+from typing import ClassVar
 
 from pulpcore.cli.common.context import (
     EntityDefinition,
@@ -75,6 +76,8 @@ class PulpContainerRepositoryContext(PulpRepositoryContext):
     UPDATE_ID = "repositories_container_container_partial_update"
     DELETE_ID = "repositories_container_container_delete"
     SYNC_ID = "repositories_container_container_sync"
+    TAG_ID: ClassVar[str] = "repositories_container_container_tag"
+    UNTAG_ID: ClassVar[str] = "repositories_container_container_untag"
     VERSION_CONTEXT = PulpContainerRepositoryVersionContext
     CAPABILITIES = {
         "sync": [PluginRequirement("container")],
@@ -89,6 +92,8 @@ class PulpContainerPushRepositoryContext(PulpRepositoryContext):
     # CREATE_ID = "repositories_container_container_push_create"
     # UPDATE_ID = "repositories_container_container_push_update"
     # DELETE_ID = "repositories_container_container_push_delete"
+    TAG_ID: ClassVar[str] = "repositories_container_push_tag"
+    UNTAG_ID: ClassVar[str] = "repositories_container_push_untag"
     VERSION_CONTEXT = PulpContainerPushRepositoryVersionContext
 
 

pulpcore/cli/container/repository.py

@@ -119,3 +119,42 @@ def sync(
         href=repository_href,
         body=body,
     )
+
+
+@repository.command(name="tag")
+@name_option
+@href_option
+@click.option("--tag", help=_("Name to tag an image with"), required=True)
+@click.option("--digest", help=_("SHA256 digest of the Manifest file"), required=True)
+@pass_repository_context
+def add_tag(repository_ctx: PulpRepositoryContext, digest: str, tag: str,) -> None:
+    if len(tag) == 0:
+        raise click.ClickException("Please pass a non empty tag name.")
+
+    digest = digest.strip()
+    if not digest.startswith("sha256:"):
+        digest = f"sha256:{digest}"
+    if len(digest) != 71:  # len("sha256:" + 64
+        raise click.ClickException("Improper SHA256, please provide a valid 64 digit digest.")
+
+    repository_ctx.pulp_ctx.call(
+        repository_ctx.TAG_ID,
+        parameters={repository_ctx.HREF: repository_ctx.pulp_href},
+        body={"tag": tag, "digest": digest},
+    )
+
+
+@repository.command(name="untag")
+@name_option
+@href_option
+@click.option("--tag", help=_("Name of tag to remove"), required=True)
+@pass_repository_context
+def remove_tag(repository_ctx: PulpRepositoryContext, tag: str) -> None:
+    if len(tag) == 0:
+        raise click.ClickException("Please pass a non empty tag name.")
+
+    repository_ctx.pulp_ctx.call(
+        repository_ctx.UNTAG_ID,
+        parameters={repository_ctx.HREF: repository_ctx.pulp_href},
+        body={"tag": tag},
+    )