1
0
mirror of https://github.com/quay/quay.git synced 2026-01-26 06:21:37 +03:00
Files
quay/endpoints/gitlabtrigger.py
2019-12-02 12:23:08 -05:00

59 lines
1.8 KiB
Python

import logging
from flask import Blueprint, request, redirect, url_for
from flask_login import current_user
import features
from app import app, gitlab_trigger
from auth.decorators import require_session_login
from auth.permissions import AdministerRepositoryPermission
from data import model
from endpoints.decorators import route_show_if
from util.http import abort
logger = logging.getLogger(__name__)
client = app.config["HTTPCLIENT"]
gitlabtrigger = Blueprint("gitlab", __name__)
@gitlabtrigger.route("/gitlab/callback/trigger", methods=["GET"])
@route_show_if(features.GITLAB_BUILD)
@require_session_login
def attach_gitlab_build_trigger():
state = request.args.get("state", None)
if not state:
abort(400)
state = state[len("repo:") :]
try:
[namespace, repository] = state.split("/")
except ValueError:
abort(400)
permission = AdministerRepositoryPermission(namespace, repository)
if permission.can():
code = request.args.get("code")
token = gitlab_trigger.exchange_code_for_token(
app.config, client, code, redirect_suffix="/trigger"
)
if not token:
msg = "Could not exchange token. It may have expired."
abort(404, message=msg)
repo = model.repository.get_repository(namespace, repository)
if not repo:
msg = "Invalid repository: %s/%s" % (namespace, repository)
abort(404, message=msg)
elif repo.kind.name != "image":
abort(501)
trigger = model.build.create_build_trigger(repo, "gitlab", token, current_user.db_user())
repo_path = "%s/%s" % (namespace, repository)
full_url = url_for("web.buildtrigger", path=repo_path, trigger=trigger.uuid)
logger.debug("Redirecting to full url: %s", full_url)
return redirect(full_url)
abort(403)