initial import for Open Source 🎉

endpoints/v1/__init__.py

@@ -0,0 +1,76 @@
+import logging
+
+from functools import wraps
+
+from flask import Blueprint, make_response, jsonify
+
+import features
+
+from app import metric_queue, app
+from data.readreplica import ReadOnlyModeException
+from endpoints.decorators import anon_protect, anon_allowed
+from util.metrics.metricqueue import time_blueprint
+from util.http import abort
+
+v1_bp = Blueprint('v1', __name__)
+time_blueprint(v1_bp, metric_queue)
+
+logger = logging.getLogger(__name__)
+
+# Note: This is *not* part of the Docker index spec. This is here for our own health check,
+# since we have nginx handle the _ping below.
+@v1_bp.route('/_internal_ping')
+@anon_allowed
+def internal_ping():
+  return make_response('true', 200)
+
+
+@v1_bp.route('/_ping')
+@anon_allowed
+def ping():
+  # NOTE: any changes made here must also be reflected in the nginx config
+  response = make_response('true', 200)
+  response.headers['X-Docker-Registry-Version'] = '0.6.0'
+  response.headers['X-Docker-Registry-Standalone'] = '0'
+  return response
+
+
+@v1_bp.app_errorhandler(ReadOnlyModeException)
+def handle_readonly(ex):
+  response = jsonify({'message': 'System is currently read-only. Pulls will succeed but all ' +
+                                 'write operations are currently suspended.',
+                      'is_readonly': True})
+  response.status_code = 503
+  return response
+
+
+def check_v1_push_enabled(namespace_name_kwarg='namespace_name'):
+  """ Decorator which checks if V1 push is enabled for the current namespace. The first argument
+      to the wrapped function must be the namespace name or there must be a kwarg with the
+      name `namespace_name`.
+  """
+  def wrapper(wrapped):
+    @wraps(wrapped)
+    def decorated(*args, **kwargs):
+      if namespace_name_kwarg in kwargs:
+        namespace_name = kwargs[namespace_name_kwarg]
+      else:
+        namespace_name = args[0]
+
+      if features.RESTRICTED_V1_PUSH:
+        whitelist = app.config.get('V1_PUSH_WHITELIST') or []
+        logger.debug('V1 push is restricted to whitelist: %s', whitelist)
+        if namespace_name not in whitelist:
+          abort(405,
+                message=('V1 push support has been deprecated. To enable for this ' +
+                         'namespace, please contact support.'))
+
+      return wrapped(*args, **kwargs)
+    return decorated
+  return wrapper
+
+
+from endpoints.v1 import (
+  index,
+  registry,
+  tag,)