quay/endpoints/appr/cnr_backend.py

import base64

from cnr.exception import raise_package_not_found
from cnr.models.blob_base import BlobBase
from cnr.models.channel_base import ChannelBase
from cnr.models.db_base import CnrDB
from cnr.models.package_base import PackageBase, manifest_media_type

from flask import request
from app import storage
from endpoints.appr.models_oci import model


class Blob(BlobBase):
  @classmethod
  def upload_url(cls, digest):
    return "cnr/blobs/sha256/%s/%s" % (digest[0:2], digest)

  def save(self, content_media_type):
    model.store_blob(self, content_media_type)

  @classmethod
  def delete(cls, package_name, digest):
    pass

  @classmethod
  def _fetch_b64blob(cls, package_name, digest):
    blobpath = cls.upload_url(digest)
    locations = model.get_blob_locations(digest)
    if not locations:
      raise_package_not_found(package_name, digest)
    return base64.b64encode(storage.get_content(locations, blobpath))

  @classmethod
  def download_url(cls, package_name, digest):
    blobpath = cls.upload_url(digest)
    locations = model.get_blob_locations(digest)
    if not locations:
      raise_package_not_found(package_name, digest)
    return storage.get_direct_download_url(locations, blobpath, request.remote_addr)


class Channel(ChannelBase):
  """ CNR Channel model implemented against the Quay data model. """

  def __init__(self, name, package, current=None):
    super(Channel, self).__init__(name, package, current=current)
    self._channel_data = None

  def _exists(self):
    """ Check if the channel is saved already """
    return model.channel_exists(self.package, self.name)

  @classmethod
  def get(cls, name, package):
    chanview = model.fetch_channel(package, name, with_releases=False)
    return cls(name, package, chanview.current)

  def save(self):
    model.update_channel(self.package, self.name, self.current)

  def delete(self):
    model.delete_channel(self.package, self.name)

  @classmethod
  def all(cls, package_name):
    return [
      Channel(c.name, package_name, c.current) for c in model.list_channels(package_name)
    ]

  @property
  def _channel(self):
    if self._channel_data is None:
      self._channel_data = model.fetch_channel(self.package, self.name)
    return self._channel_data

  def releases(self):
    """ Returns the list of versions """
    return self._channel.releases

  def _add_release(self, release):
    return model.update_channel(self.package, self.name, release)._asdict

  def _remove_release(self, release):
    model.delete_channel(self.package, self.name)


class User(object):
  """ User in CNR models """

  @classmethod
  def get_user(cls, username, password):
    """ Returns True if user creds is valid """
    return model.get_user(username, password)


class Package(PackageBase):
  """ CNR Package model implemented against the Quay data model. """

  @classmethod
  def _apptuple_to_dict(cls, apptuple):
    return {
      'release': apptuple.release,
      'created_at': apptuple.created_at,
      'digest': apptuple.manifest.digest,
      'mediaType': apptuple.manifest.mediaType,
      'package': apptuple.name,
      'content': apptuple.manifest.content._asdict()
    }

  @classmethod
  def create_repository(cls, package_name, visibility, owner):
    model.create_application(package_name, visibility, owner)

  @classmethod
  def exists(cls, package_name):
    return model.application_exists(package_name)

  @classmethod
  def all(cls, organization=None, media_type=None, search=None, username=None, **kwargs):
    return [
      dict(x._asdict())
      for x in model.list_applications(namespace=organization, media_type=media_type,
                                               search=search, username=username)
    ]

  @classmethod
  def _fetch(cls, package_name, release, media_type):
    data = model.fetch_release(package_name, release, manifest_media_type(media_type))
    return cls._apptuple_to_dict(data)

  @classmethod
  def all_releases(cls, package_name, media_type=None):
    return model.list_releases(package_name, media_type)

  @classmethod
  def search(cls, query, username=None):
    return model.basic_search(query, username=username)

  def _save(self, force=False, **kwargs):
    user = kwargs['user']
    visibility = kwargs['visibility']
    model.create_release(self, user, visibility, force)

  @classmethod
  def _delete(cls, package_name, release, media_type):
    model.delete_release(package_name, release, manifest_media_type(media_type))

  @classmethod
  def isdeleted_release(cls, package, release):
    return model.release_exists(package, release)

  def channels(self, channel_class, iscurrent=True):
    return [
      c.name
      for c in model.list_release_channels(self.package, self.release, active=iscurrent)
    ]

  @classmethod
  def manifests(cls, package, release=None):
    return model.list_manifests(package, release)

  @classmethod
  def dump_all(cls, blob_cls):
    raise NotImplementedError


class QuayDB(CnrDB):
  """ Wrapper Class to embed all CNR Models """
  Channel = Channel
  Package = Package
  Blob = Blob

  @classmethod
  def reset_db(cls, force=False):
    pass