From 0f6f258629b957424e79786bbba5d4543dae272a Mon Sep 17 00:00:00 2001
From: Raymond Mauge <rmauge@gmail.com>
Date: Thu, 27 Feb 2014 13:38:13 -0500
Subject: [PATCH 1/2] Adds Linkedin OAuth 2.0 provider

---
 velruse/providers/linkedin_oauth2.py | 176 +++++++++++++++++++++++++++
 1 file changed, 176 insertions(+)
 create mode 100644 velruse/providers/linkedin_oauth2.py

diff --git a/velruse/providers/linkedin_oauth2.py b/velruse/providers/linkedin_oauth2.py
new file mode 100644
index 0000000..0161f23
--- /dev/null
+++ b/velruse/providers/linkedin_oauth2.py
@@ -0,0 +1,176 @@
+import uuid
+
+from pyramid.httpexceptions import HTTPFound
+from pyramid.security import NO_PERMISSION_REQUIRED
+
+import requests
+
+from ..api import (
+    AuthenticationComplete,
+    AuthenticationDenied,
+    register_provider,
+)
+from ..exceptions import CSRFError
+from ..exceptions import ThirdPartyFailure
+from ..settings import ProviderSettings
+from ..utils import flat_url
+
+AUTH_URL = "https://www.linkedin.com/uas/oauth2/authorization"
+ACCESS_URL = "https://www.linkedin.com/uas/oauth2/accessToken"
+
+class LinkedinAuthenticationComplete(AuthenticationComplete):
+    """Google OAuth 2.0 auth complete"""
+
+def includeme(config):
+    """Activate the ``linkedin_oauth2`` Pyramid plugin via
+    ``config.include('velruse.providers.linkedin_oauth2')``. After included,
+    two new methods will be available to configure new providers.
+
+    ``config.add_linkedin_oauth2_login()``
+        See :func:`~velruse.providers.linkedin_oauth2.add_linkedin_login`
+        for the supported options.
+
+    ``config.add_linkedin_oauth2_login_from_settings()``
+
+    """
+    config.add_directive('add_linkedin_oauth2_login', add_linkedin_login)
+    config.add_directive('add_linkedin_oauth2_login_from_settings',
+                         add_linkedin_login_from_settings)
+
+def add_linkedin_login_from_settings(config, prefix='velruse.linkedin.'):
+    settings = config.registry.settings
+    p = ProviderSettings(settings, prefix)
+    p.update('consumer_key', required=True)
+    p.update('consumer_secret', required=True)
+    p.update('scope')
+    p.update('login_path')
+    p.update('callback_path')
+    config.add_linkedin_oauth2_login(**p.kwargs)
+
+def add_linkedin_login(config,
+                     consumer_key=None,
+                     consumer_secret=None,
+                     scope='',
+                     login_path='/login/linkedin',
+                     callback_path='/login/linkedin/callback',
+                     name='linkedin'):
+    """
+    Add a Linkedin login provider to the application supporting the new
+    OAuth2 protocol.
+    """
+    provider = LinkedinOAuth2Provider(
+        name,
+        consumer_key,
+        consumer_secret,
+        scope)
+
+    config.add_route(provider.login_route, login_path)
+    config.add_view(provider, attr='login', route_name=provider.login_route,
+                    permission=NO_PERMISSION_REQUIRED)
+
+    config.add_route(provider.callback_route, callback_path,
+                     use_global_views=True,
+                     factory=provider.callback)
+
+    register_provider(config, name, provider)
+
+class LinkedinOAuth2Provider(object):
+
+    def __init__(self,
+                 name,
+                 consumer_key,
+                 consumer_secret,
+                 scope):
+        self.name = name
+        self.type = 'linkedin_oauth2'
+        self.consumer_key = consumer_key
+        self.consumer_secret = consumer_secret
+        self.scope = scope
+        self.login_route = 'velruse.%s-login' % name
+        self.callback_route = 'velruse.%s-callback' % name
+
+    def login(self, request):
+        """Initiate a Linkedin login"""
+        #Overwrites scope from settings if found in login form
+        scope = request.POST.get('scope', self.scope) 
+        request.session['velruse.state'] = state = uuid.uuid4().hex
+
+        auth_url = flat_url(
+            AUTH_URL,
+            scope=scope,
+            response_type='code',
+            client_id=self.consumer_key,
+            redirect_uri=request.route_url(self.callback_route),
+            state=state)
+        return HTTPFound(location=auth_url)
+
+    def callback(self, request):
+        """Process the Linkedin redirect"""
+        sess_state = request.session.pop('velruse.state', None)
+        req_state = request.GET.get('state')
+        if not sess_state or sess_state != req_state:
+            raise CSRFError(
+                'CSRF Validation check failed. Request state {req_state} is '
+                'not the same as session state {sess_state}'.format(
+                    req_state=req_state,
+                    sess_state=sess_state
+                )
+            )
+        code = request.GET.get('code')
+        if not code:
+            reason = request.GET.get('error', 'No reason provided.')
+            description = request.GET.get('error_description', 'No description provided.')
+            return AuthenticationDenied(reason='Error: %s, Error description: %s' % (reason, description),
+                                        provider_name=self.name,
+                                        provider_type=self.type)
+
+        # Now retrieve the access token with the code
+        r = requests.post(
+            ACCESS_URL,
+            dict(client_id=self.consumer_key,
+                 client_secret=self.consumer_secret,
+                 redirect_uri=request.route_url(self.callback_route),
+                 code=code,
+                 grant_type='authorization_code')
+        )
+        if r.status_code != 200:
+            raise ThirdPartyFailure("Status %s: %s" % (
+                r.status_code, r.content))
+        token_data = r.json()
+        access_token = token_data['access_token']
+        expires_in = token_data['expires_in']
+
+        # Retrieve profile data if scopes allow
+        profile_url = 'https://api.linkedin.com/v1/people/~'
+        profile_url += (':(first-name,last-name,id,picture-url,email-address)')
+        profile = {}
+        user_url = flat_url(
+            profile_url,
+            format='json',
+            oauth2_access_token=access_token)
+        r = requests.get(user_url)
+
+        if r.status_code == 200:
+            data = r.json()
+            profile['displayName'] = u'%s %s' % (data['firstName'], data['lastName'])
+            profile['name'] = {
+                'givenName': data['firstName'],
+                'familyName': data['lastName'],
+                'formatted': u'%s %s' % (data['firstName'], data['lastName'])
+            }
+            if data.get('emailAddress'):
+                profile['emails'] = [{'value': data.get('emailAddress')}]
+            if data.get('pictureUrl'):
+                profile['photos'] = [{'value': data.get('pictureUrl')}]
+
+            profile['accounts'] = [{
+                'domain': 'linkedin.com',
+                'userid': data['id']
+            }]
+
+        cred = {'oauthAccessToken': access_token,
+                'oauthExpiresIn': expires_in}
+        return LinkedinAuthenticationComplete(profile=profile,
+                                              credentials=cred,
+                                              provider_name=self.name,
+                                              provider_type=self.type)
\ No newline at end of file

From 5c1c25de7cfa75554a90d8d005449a7bb9653529 Mon Sep 17 00:00:00 2001
From: Raymond Mauge <rmauge@gmail.com>
Date: Fri, 7 Mar 2014 20:03:58 -0500
Subject: [PATCH 2/2] Adds tests and renames provider

---
 tests/selenium/testapp/__init__.py          |  7 +++++
 tests/selenium/testapp/templates/login.mako |  1 +
 tests/selenium/tests.py                     | 33 +++++++++++++++++++++
 velruse/providers/linkedin_oauth2.py        |  2 +-
 4 files changed, 42 insertions(+), 1 deletion(-)

diff --git a/tests/selenium/testapp/__init__.py b/tests/selenium/testapp/__init__.py
index 764619c..f019cd9 100644
--- a/tests/selenium/testapp/__init__.py
+++ b/tests/selenium/testapp/__init__.py
@@ -134,6 +134,13 @@ def main(global_conf, **settings):
             settings['velruse.linkedin.consumer_key'],
             settings['velruse.linkedin.consumer_secret'],
         )
+        
+    if 'linkedin_oauth2' in providers:
+        config.include('velruse.providers.linkedin_oauth2')
+        config.add_linkedin_oauth2_login(
+            settings['velruse.linkedin.consumer_key'],
+            settings['velruse.linkedin.consumer_secret'],
+        )
 
     config.scan(__name__)
     return config.make_wsgi_app()
diff --git a/tests/selenium/testapp/templates/login.mako b/tests/selenium/testapp/templates/login.mako
index f1396c0..fd0df3b 100644
--- a/tests/selenium/testapp/templates/login.mako
+++ b/tests/selenium/testapp/templates/login.mako
@@ -30,6 +30,7 @@ ${form('google_hybrid', 'Login with Google OpenID+OAuth',
        openid_identifier='google.com')}
 ${form('google_oauth2', 'Login with Google OAuth2')}
 ${form('linkedin', 'Login with Linkedin')}
+${form('linkedin_oauth2', 'Login with Linkedin OAuth2')}
 ${form('live', 'Login with Windows Live')}
 ${form('openid', 'Login with OpenID',
        openid_identifier='myopenid.com')}
diff --git a/tests/selenium/tests.py b/tests/selenium/tests.py
index 6e5b6db..3b0aad0 100644
--- a/tests/selenium/tests.py
+++ b/tests/selenium/tests.py
@@ -492,3 +492,36 @@ def test_it(self):
         creds = result['credentials']
         self.assertTrue('oauthAccessToken' in creds)
         self.assertTrue('oauthAccessTokenSecret' in creds)
+
+class TestLinkedinOAuth2(ProviderTests, unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.require_provider('linkedin_oauth2')
+        cls.login = config['linkedin_oauth2.login']
+        cls.password = config['linkedin_oauth2.password']
+        cls.login_url = find_login_url(config, 'linkedin_oauth2.login_url')
+
+    def test_it(self):
+        browser.get(self.login_url)
+        self.assertEqual(browser.title, 'Auth Page')
+        browser.find_element_by_id('linkedin_oauth2').submit()
+        self.assertEqual(browser.title, 'Authorize | LinkedIn')
+        form = browser.find_element_by_name('oauth2SAuthorizeForm')
+        login = form.find_element_by_id('session_key-oauth2SAuthorizeForm')
+        login.send_keys(self.login)
+        passwd = form.find_element_by_id('session_password-oauth2SAuthorizeForm')
+        passwd.send_keys(self.password)
+        form.find_element_by_name('authorize').submit()
+        result = WebDriverWait(browser, 2).until(
+            EC.presence_of_element_located((By.ID, 'result')))
+        self.assertEqual(browser.title, 'Result Page')
+        result = json.loads(result.text)
+        self.assertTrue('profile' in result)
+        self.assertTrue('credentials' in result)
+        profile = result['profile']
+        self.assertTrue('displayName' in profile)
+        self.assertTrue('accounts' in profile)
+        creds = result['credentials']
+        self.assertTrue('oauthAccessToken' in creds)
+        self.assertTrue('oauthExpiresIn' in creds)
\ No newline at end of file
diff --git a/velruse/providers/linkedin_oauth2.py b/velruse/providers/linkedin_oauth2.py
index 0161f23..9560d5b 100644
--- a/velruse/providers/linkedin_oauth2.py
+++ b/velruse/providers/linkedin_oauth2.py
@@ -53,7 +53,7 @@ def add_linkedin_login(config,
                      scope='',
                      login_path='/login/linkedin',
                      callback_path='/login/linkedin/callback',
-                     name='linkedin'):
+                     name='linkedin_oauth2'):
     """
     Add a Linkedin login provider to the application supporting the new
     OAuth2 protocol.