56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
|
from collections import namedtuple
|
||
|
|
|
||
|
|
from flask import jsonify
|
||
|
|
|
||
|
|
import features
|
||
|
|
|
||
|
|
from app import model_cache
|
||
|
|
from auth.auth_context import get_authenticated_user, get_authenticated_context
|
||
|
|
from auth.registry_jwt_auth import process_registry_jwt_auth
|
||
|
|
from data import model
|
||
|
|
from data.cache import cache_key
|
||
|
|
from endpoints.decorators import anon_protect
|
||
|
|
from endpoints.v2 import v2_bp, paginate
|
||
|
|
|
||
|
|
|
||
|
|
class Repository(namedtuple('Repository', ['id', 'namespace_name', 'name'])):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
@v2_bp.route('/_catalog', methods=['GET'])
|
||
|
|
@process_registry_jwt_auth()
|
||
|
|
@anon_protect
|
||
|
|
@paginate()
|
||
|
|
def catalog_search(start_id, limit, pagination_callback):
|
||
|
|
def _load_catalog():
|
||
|
|
include_public = bool(features.PUBLIC_CATALOG)
|
||
|
|
if not include_public and not get_authenticated_user():
|
||
|
|
return []
|
||
|
|
|
||
|
|
username = get_authenticated_user().username if get_authenticated_user() else None
|
||
|
|
if username and not get_authenticated_user().enabled:
|
||
|
|
return []
|
||
|
|
|
||
|
|
query = model.repository.get_visible_repositories(username,
|
||
|
|
kind_filter='image',
|
||
|
|
include_public=include_public,
|
||
|
|
start_id=start_id,
|
||
|
|
limit=limit + 1)
|
||
|
|
# NOTE: The repository ID is in `rid` (not `id`) here, as per the requirements of
|
||
|
|
# the `get_visible_repositories` call.
|
||
|
|
return [Repository(repo.rid, repo.namespace_user.username, repo.name)._asdict()
|
||
|
|
for repo in query]
|
||
|
|
|
||
|
|
context_key = get_authenticated_context().unique_key if get_authenticated_context() else None
|
||
|
|
catalog_cache_key = cache_key.for_catalog_page(context_key, start_id, limit)
|
||
|
|
visible_repositories = [Repository(**repo_dict) for repo_dict
|
||
|
|
in model_cache.retrieve(catalog_cache_key, _load_catalog)]
|
||
|
|
|
||
|
|
response = jsonify({
|
||
|
|
'repositories': ['%s/%s' % (repo.namespace_name, repo.name)
|
||
|
|
for repo in visible_repositories][0:limit],
|
||
|
|
})
|
||
|
|
|
||
|
|
pagination_callback(visible_repositories, response)
|
||
|
|
return response
|