Underpinnings of the ability to connect a GitHub account to an existing Quay account.

endpoints/web.py

@@ -4,7 +4,7 @@ import stripe
 
 from flask import (abort, redirect, request, url_for, render_template,
                    make_response, Response)
-from flask.ext.login import current_user
+from flask.ext.login import login_required, current_user
 from urlparse import urlparse
 
 from data import model
@@ -155,8 +155,7 @@ def receipt():
   abort(404)
 
 
-@app.route('/oauth2/github/callback', methods=['GET'])
-def github_oauth_callback():
+def exchange_github_code_for_token(code):
   code = request.args.get('code')
   payload = {
     'client_id': app.config['GITHUB_CLIENT_ID'],
@@ -171,13 +170,23 @@ def github_oauth_callback():
                                    params=payload, headers=headers)
 
   token = get_access_token.json()['access_token']
+  return token
 
+
+def get_github_user(token):
   token_param = {
     'access_token': token,
   }
   get_user = requests.get(app.config['GITHUB_USER_URL'], params=token_param)
 
-  user_data = get_user.json()
+  return get_user.json()
+
+
+@app.route('/oauth2/github/callback', methods=['GET'])
+def github_oauth_callback():
+  token = exchange_github_code_for_token(request.args.get('code'))
+  user_data = get_github_user(token)
+
   username = user_data['login']
   github_id = user_data['id']
 
@@ -218,6 +227,16 @@ def github_oauth_callback():
   return render_page_template('githuberror.html')
 
 
+@app.route('/oauth2/github/callback/attach', methods=['GET'])
+@login_required
+def github_oauth_attach():
+  token = exchange_github_code_for_token(request.args.get('code'))
+  user_data = get_github_user(token)
+  github_id = user_data['id']
+  user_obj = current_user.db_user()
+  model.attach_federated_login(user_obj, 'github', github_id)
+
+
 @app.route('/confirm', methods=['GET'])
 def confirm_email():
   code = request.values['code']