From c0603c9e4e36f0cd6329e8500364057e96236e72 Mon Sep 17 00:00:00 2001
From: Jun Rama <j.rama@arcanys.com>
Date: Mon, 16 Nov 2015 14:59:05 +0800
Subject: [PATCH] add token to user entity if available.

---
 Security/SSOAuthenticator.php     | 4 ++--
 Security/User/SSOUserProvider.php | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/Security/SSOAuthenticator.php b/Security/SSOAuthenticator.php
index 89a4a33..a4931d7 100644
--- a/Security/SSOAuthenticator.php
+++ b/Security/SSOAuthenticator.php
@@ -35,8 +35,6 @@ public function supportsToken(TokenInterface $token, $providerKey)
 
     public function authenticateToken(TokenInterface $token, UserProviderInterface $userProvider, $providerKey)
     {
-
-
         $secret = $token->getCredentials();
         $userData = $this->session->getFlashBag()->get('arcanys_sso_auth.user_data');
         if($userData){
@@ -45,6 +43,7 @@ public function authenticateToken(TokenInterface $token, UserProviderInterface $
             $email = reset($userData['email']);
             $firstname = reset($userData['firstname']);
             $lastname = reset($userData['lastname']);
+            $token = reset($userData['token']);
             $roles = $userData['rights'];
             if(!$roles) $roles = ['ROLE_USER'];
         }else{
@@ -62,6 +61,7 @@ public function authenticateToken(TokenInterface $token, UserProviderInterface $
             'email'=>$email,
             'firstname'=>$firstname,
             'lastname'=>$lastname,
+            'token'=>$token,
             'roles'=>$roles,
         ]);
 
diff --git a/Security/User/SSOUserProvider.php b/Security/User/SSOUserProvider.php
index 2e4c5f2..b7e6a6a 100644
--- a/Security/User/SSOUserProvider.php
+++ b/Security/User/SSOUserProvider.php
@@ -40,6 +40,8 @@ public function loadUserByUsername($userData)
             $user->setLastname($userData['lastname']);
         if(isset($userData['roles']))
             $user->setRoles($userData['roles']);
+        if(isset($userData['token']))
+            $user->setToken($userData['token']);
         
        if($new){
            $this->doctrine->getManager()->persist($user);