From 02e9c87fa41be61a0c94e5796c91dcc996268133 Mon Sep 17 00:00:00 2001
From: pedroigor <pigor.craveiro@gmail.com>
Date: Mon, 9 Feb 2015 21:31:16 -0200
Subject: [PATCH] [KEYCLOAK-992] - Storing twitter tokens.

---
 .../twitter/TwitterIdentityProvider.java      | 25 ++++++++++++++-----
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/social/twitter/src/main/java/org/keycloak/social/twitter/TwitterIdentityProvider.java b/social/twitter/src/main/java/org/keycloak/social/twitter/TwitterIdentityProvider.java
index 1b7a563194..d9ee567e29 100755
--- a/social/twitter/src/main/java/org/keycloak/social/twitter/TwitterIdentityProvider.java
+++ b/social/twitter/src/main/java/org/keycloak/social/twitter/TwitterIdentityProvider.java
@@ -31,8 +31,10 @@ import org.keycloak.models.FederatedIdentityModel;
 import org.keycloak.social.SocialIdentityProvider;
 import twitter4j.Twitter;
 import twitter4j.TwitterFactory;
+import twitter4j.auth.AccessToken;
 import twitter4j.auth.RequestToken;
 
+import javax.ws.rs.core.MediaType;
 import javax.ws.rs.core.MultivaluedMap;
 import javax.ws.rs.core.Response;
 import javax.ws.rs.core.UriInfo;
@@ -98,15 +100,26 @@ public class TwitterIdentityProvider extends AbstractIdentityProvider<OAuth2Iden
 
             RequestToken requestToken = new RequestToken(twitterToken, twitterSecret);
 
-            twitter.getOAuthAccessToken(requestToken, verifier);
+            AccessToken oAuthAccessToken = twitter.getOAuthAccessToken(requestToken, verifier);
             twitter4j.User twitterUser = twitter.verifyCredentials();
 
-            FederatedIdentity user = new FederatedIdentity(Long.toString(twitterUser.getId()));
+            FederatedIdentity identity = new FederatedIdentity(Long.toString(twitterUser.getId()));
 
-            user.setUsername(twitterUser.getScreenName());
-            user.setName(twitterUser.getName());
+            identity.setUsername(twitterUser.getScreenName());
+            identity.setName(twitterUser.getName());
 
-            return AuthenticationResponse.end(user);
+            StringBuilder tokenBuilder = new StringBuilder();
+
+            tokenBuilder.append("{");
+            tokenBuilder.append("\"oauth_token\":").append("\"").append(oAuthAccessToken.getToken()).append("\"").append(",");
+            tokenBuilder.append("\"oauth_token_secret\":").append("\"").append(oAuthAccessToken.getTokenSecret()).append("\"").append(",");
+            tokenBuilder.append("\"screen_name\":").append("\"").append(oAuthAccessToken.getScreenName()).append("\"").append(",");
+            tokenBuilder.append("\"user_id\":").append("\"").append(oAuthAccessToken.getUserId()).append("\"");
+            tokenBuilder.append("}");
+
+            identity.setToken(tokenBuilder.toString());
+
+            return AuthenticationResponse.end(identity);
         } catch (Exception e) {
             throw new RuntimeException(e);
         }
@@ -114,6 +127,6 @@ public class TwitterIdentityProvider extends AbstractIdentityProvider<OAuth2Iden
 
     @Override
     public Response retrieveToken(FederatedIdentityModel identity) {
-        return Response.noContent().build();
+        return Response.ok(identity.getToken()).type(MediaType.APPLICATION_JSON).build();
     }
 }