Basic labels support

Adds basic labels support to the registry code (V2), and the API. Note that this does not yet add any UI related support.

initdb.py

@@ -12,9 +12,6 @@ from peewee import (SqliteDatabase, create_model_tables, drop_model_tables, save
 from itertools import count
 from uuid import UUID, uuid4
 from threading import Event
-from hashlib import sha256
-from Crypto.PublicKey import RSA
-from jwkest.jwk import RSAKey
 
 from email.utils import formatdate
 from data.database import (db, all_models, Role, TeamRole, Visibility, LoginService,
@@ -22,7 +19,7 @@ from data.database import (db, all_models, Role, TeamRole, Visibility, LoginServ
                            ImageStorageTransformation, ImageStorageSignatureKind,
                            ExternalNotificationEvent, ExternalNotificationMethod, NotificationKind,
                            QuayRegion, QuayService, UserRegion, OAuthAuthorizationCode,
-                           ServiceKeyApprovalType)
+                           ServiceKeyApprovalType, MediaType, LabelSourceType)
 from data import model
 from data.queue import WorkQueue
 from app import app, storage as store, tf
@@ -345,6 +342,9 @@ def initialize_database():
 
   LogEntryKind.create(name='take_ownership')
 
+  LogEntryKind.create(name='manifest_label_add')
+  LogEntryKind.create(name='manifest_label_delete')
+
   ImageStorageLocation.create(name='local_eu')
   ImageStorageLocation.create(name='local_us')
 
@@ -389,6 +389,13 @@ def initialize_database():
   QuayRegion.create(name='us')
   QuayService.create(name='quay')
 
+  MediaType.create(name='text/plain')
+  MediaType.create(name='application/json')
+
+  LabelSourceType.create(name='manifest')
+  LabelSourceType.create(name='api', mutable=True)
+  LabelSourceType.create(name='internal')
+
 
 def wipe_database():
   logger.debug('Wiping all data from the DB.')
@@ -474,6 +481,24 @@ def populate_database(minimal=False, with_storage=False):
 
   simple_repo = __generate_repository(with_storage, new_user_1, 'simple', 'Simple repository.', False,
                                       [], (4, [], ['latest', 'prod']))
+
+  # Add some labels to the latest tag's manifest.
+  tag_manifest = model.tag.load_tag_manifest(new_user_1.username, 'simple', 'latest')
+  first_label = model.label.create_manifest_label(tag_manifest, 'foo', 'bar', 'manifest')
+  model.label.create_manifest_label(tag_manifest, 'foo', 'baz', 'api')
+  model.label.create_manifest_label(tag_manifest, 'anotherlabel', '1234', 'internal')
+
+  label_metadata = {
+    'key': 'foo',
+    'value': 'bar',
+    'id': first_label.id,
+    'manifest_digest': tag_manifest.digest
+  }
+
+  model.log.log_action('manifest_label_add', new_user_1.username, performer=new_user_1,
+                       timestamp=datetime.now(), metadata=label_metadata,
+                       repository=tag_manifest.tag.repository)
+
   model.blob.initiate_upload(new_user_1.username, simple_repo.name, str(uuid4()), 'local_us', {})
   model.notification.create_repo_notification(simple_repo, 'repo_push', 'quay_notification', {}, {})