gitlab oauth

endpoints/gitlabtrigger.py

@@ -0,0 +1,50 @@
+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)
+    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)