quay/endpoints/gitlabtrigger.py

import logging

from flask import Blueprint, request, redirect, url_for
from flask.ext.login import current_user

from app import app, gitlab_trigger
from auth.auth import require_session_login
from auth.permissions import AdministerRepositoryPermission
from data import model
from endpoints.common import route_show_if
from util.http import abort
from util.names import parse_repository_name

import features


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.get_repository(namespace, repository)
    if not repo:
      msg = 'Invalid repository: %s/%s' % (namespace, repository)
      abort(404, message=msg)

    trigger = model.create_build_trigger(repo, 'gitlab', token, current_user.db_user())
    repo_path = '%s/%s' % (namespace, repository)
    full_url = '%s%s%s' % (url_for('web.repository', path=repo_path), '?tab=builds&newtrigger=', trigger.uuid)

    logger.debug('Redirecting to full url: %s', full_url)
    return redirect(full_url)

  abort(403)