port label support to refactored v2 registry

data/database.py

@@ -957,7 +957,7 @@ class ServiceKey(BaseModel):
   rotation_duration = IntegerField(null=True)
   approval = ForeignKeyField(ServiceKeyApproval, null=True)
 
-'''
+
 class MediaType(BaseModel):
   """ MediaType is an enumeration of the possible formats of various objects in the data model. """
   name = CharField(index=True, unique=True)
@@ -992,6 +992,21 @@ class TagManifestLabel(BaseModel):
       (('annotated', 'label'), True),
     )
 
+
+'''
+
+class ManifestLabel(BaseModel):
+  repository = ForeignKeyField(Repository, index=True)
+  annotated = ForeignKeyField(Manifest, index=True)
+  label = ForeignKeyField(Label)
+
+  class Meta:
+    database = db
+    read_slaves = (read_slave,)
+    indexes = (
+      (('repository', 'annotated', 'label'), True),
+    )
+
 class Blob(BaseModel):
   """ Blob represents a content-addressable object stored outside of the database. """
   digest = CharField(index=True, unique=True)

data/interfaces/v2.py

@@ -4,7 +4,7 @@ from namedlist import namedlist
 from peewee import IntegrityError
 
 from data import model, database
-from data.model import DataModelException
+from data.model import DataModelException, TagAlreadyCreatedException
 from image.docker.v1 import DockerV1Metadata
 
 _MEDIA_TYPE = "application/vnd.docker.distribution.manifest.v1+prettyjws"
@@ -48,6 +48,11 @@ class RepositoryReference(namedtuple('RepositoryReference', ['id', 'name', 'name
   RepositoryReference represents a reference to a Repository, without its full metadata.
   """
 
+class Label(namedtuple('Label', ['key', 'value', 'source_type', 'media_type'])):
+  """
+  Label represents a key-value pair that describes a particular Manifest.
+  """
+
 
 class DockerRegistryV2DataInterface(object):
   """
@@ -158,6 +163,8 @@ class DockerRegistryV2DataInterface(object):
     """
     Saves a manifest pointing to the given leaf image, with the given manifest, under the matching
     repository as a tag with the given name.
+
+    Returns a boolean whether or not the tag was newly created or not.
     """
     raise NotImplementedError()
 
@@ -246,6 +253,14 @@ class DockerRegistryV2DataInterface(object):
     """
     raise NotImplementedError()
 
+  @classmethod
+  def create_manifest_labels(cls, namespace_name, repo_name, manifest_digest, labels):
+    """
+    Creates a new labels for the provided manifest.
+    """
+    raise NotImplementedError()
+
+
   @classmethod
   def get_blob_path(cls, blob):
     """
@@ -407,8 +422,10 @@ class PreOCIModel(DockerRegistryV2DataInterface):
   @classmethod
   def save_manifest(cls, namespace_name, repo_name, tag_name, leaf_layer_docker_id, manifest_digest,
                     manifest_bytes):
-    model.tag.store_tag_manifest(namespace_name, repo_name, tag_name, leaf_layer_docker_id,
-                                 manifest_digest, manifest_bytes)
+    (_, newly_created) = model.tag.store_tag_manifest(namespace_name, repo_name, tag_name,
+                                                      leaf_layer_docker_id, manifest_digest,
+                                                      manifest_bytes)
+    return newly_created
 
   @classmethod
   def repository_tags(cls, namespace_name, repo_name, limit, offset):
@@ -540,6 +557,17 @@ class PreOCIModel(DockerRegistryV2DataInterface):
     blob_record = model.storage.get_storage_by_uuid(blob.uuid)
     model.storage.save_torrent_info(blob_record, piece_size, piece_bytes)
 
+  @classmethod
+  def create_manifest_labels(cls, namespace_name, repo_name, manifest_digest, labels):
+    if not labels:
+      # No point in doing anything more.
+      return
+
+    tag_manifest = model.tag.load_manifest_by_digest(namespace_name, repo_name, manifest_digest)
+    for label in labels:
+      model.label.create_manifest_label(tag_manifest, label.key, label.value, label.source_type,
+                                        label.media_type)
+
   @classmethod
   def get_blob_path(cls, blob):
     blob_record = model.storage.get_storage_by_uuid(blob.uuid)

data/model/label.py

@@ -1,11 +1,12 @@
 import logging
 
+from cachetools import lru_cache
+
 from data.database import Label, TagManifestLabel, MediaType, LabelSourceType, db_transaction
 from data.model import InvalidLabelKeyException, InvalidMediaTypeException, DataModelException
 from data.model._basequery import prefix_search
 from util.validation import validate_label_key
 from util.validation import is_json
-from cachetools import lru_cache
 
 logger = logging.getLogger(__name__)