Implemented in-memory TokenStore and added opportunity to save user credentials into file

oauth2_http/java/com/google/auth/oauth2/GoogleCredentials.java

@@ -37,8 +37,7 @@
 import com.google.api.client.util.Preconditions;
 import com.google.auth.http.HttpTransportFactory;
 
-import java.io.IOException;
-import java.io.InputStream;
+import java.io.*;
 import java.util.Collection;
 
 /**
@@ -165,6 +164,24 @@ public static GoogleCredentials fromStream(InputStream credentialsStream,
         fileType, USER_FILE_TYPE, SERVICE_ACCOUNT_FILE_TYPE));
   }
 
+  /**
+   * Saves the end user credentials in specified file
+   *
+   * @param credentials InputStream containing user credentials in JSON format
+   * @param credentialsFileName Name of file where to store the credentials
+   * @throws IOException An error saving the credentials.
+   */
+  public void saveCredentialsToFile(InputStream credentials, String credentialsFileName) throws IOException {
+    byte[] buffer = new byte[credentials.available()];
+    credentials.read(buffer);
+
+    String userDirectory = System.getProperty("user.dir");
+    File file = new File(userDirectory, credentialsFileName);
+    OutputStream fileOutputStream = new FileOutputStream(file);
+    fileOutputStream.write(buffer);
+    fileOutputStream.close();
+  }
+
   /**
    * Default constructor.
    **/

oauth2_http/java/com/google/auth/oauth2/UserAuthorizer.java

@@ -42,6 +42,7 @@
 import com.google.api.client.util.Joiner;
 import com.google.api.client.util.Preconditions;
 import com.google.auth.http.HttpTransportFactory;
+import com.google.auth.oauth2.storage.TokenStore;
 import com.google.common.collect.ImmutableList;
 
 import java.io.IOException;

oauth2_http/java/com/google/auth/oauth2/UserCredentials.java

@@ -31,6 +31,8 @@
 
 package com.google.auth.oauth2;
 
+import static com.google.auth.oauth2.OAuth2Utils.JSON_FACTORY;
+import static com.google.auth.oauth2.OAuth2Utils.UTF_8;
 import static com.google.common.base.MoreObjects.firstNonNull;
 
 import com.google.api.client.http.GenericUrl;
@@ -45,9 +47,7 @@
 import com.google.api.client.util.Preconditions;
 import com.google.auth.http.HttpTransportFactory;
 
-import java.io.IOException;
-import java.io.InputStream;
-import java.io.ObjectInputStream;
+import java.io.*;
 import java.net.URI;
 import java.util.Date;
 import java.util.Map;
@@ -58,6 +58,8 @@
  */
 public class UserCredentials extends GoogleCredentials {
 
+  public static final String USER_CREDENTIALS_FILE_NAME = "GOOGLE_AUTH_USER_CREDENTIALS_FILE_NAME";
+
   private static final String GRANT_TYPE = "refresh_token";
   private static final String PARSE_ERROR_PREFIX = "Error parsing token refresh response. ";
   private static final long serialVersionUID = -4800758775038679176L;
@@ -170,7 +172,7 @@ public static UserCredentials fromStream(InputStream credentialsStream,
     Preconditions.checkNotNull(credentialsStream);
     Preconditions.checkNotNull(transportFactory);
 
-    JsonFactory jsonFactory = OAuth2Utils.JSON_FACTORY;
+    JsonFactory jsonFactory = JSON_FACTORY;
     JsonObjectParser parser = new JsonObjectParser(jsonFactory);
     GenericJson fileContents = parser.parseAndClose(
         credentialsStream, OAuth2Utils.UTF_8, GenericJson.class);
@@ -206,7 +208,7 @@ public AccessToken refreshAccessToken() throws IOException {
     HttpRequestFactory requestFactory = transportFactory.create().createRequestFactory();
     HttpRequest request =
         requestFactory.buildPostRequest(new GenericUrl(tokenServerUri), content);
-    request.setParser(new JsonObjectParser(OAuth2Utils.JSON_FACTORY));
+    request.setParser(new JsonObjectParser(JSON_FACTORY));
     HttpResponse response = request.execute();
     GenericData responseData = response.parseAs(GenericData.class);
     String accessToken =
@@ -244,6 +246,41 @@ public final String getRefreshToken() {
     return refreshToken;
   }
 
+
+  /**
+   * Returns the instance of InputStream containing user credentials in JSON format
+   *
+   * @return user credentials stream
+   */
+  private InputStream getUserCredentialsStream() throws IOException {
+    GenericJson json = new GenericJson();
+    json.put("type", GoogleCredentials.USER_FILE_TYPE);
+    if (refreshToken != null) {
+      json.put("refresh_token", refreshToken);
+    }
+    if (tokenServerUri != null) {
+      json.put("token_server_uri", tokenServerUri);
+    }
+    if (clientId != null) {
+      json.put("client_id", clientId);
+    }
+    if (clientSecret != null) {
+      json.put("client_secret", clientSecret);
+    }
+    json.setFactory(JSON_FACTORY);
+    String text = json.toPrettyString();
+    return new ByteArrayInputStream(text.getBytes(UTF_8));
+  }
+
+  /**
+   * Saves the end user credentials in file
+   *
+   * @throws IOException An error storing the credentials.
+   */
+  public void saveUserCredentials() throws IOException {
+    super.saveCredentialsToFile(getUserCredentialsStream(), USER_CREDENTIALS_FILE_NAME);
+  }
+
   @Override
   public int hashCode() {
     return Objects.hash(super.hashCode(), clientId, clientSecret, refreshToken, tokenServerUri,

oauth2_http/java/com/google/auth/oauth2/storage/MemoryTokensStorage.java

@@ -0,0 +1,56 @@
+/*
+ * Copyright 2017, Google Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ *    * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *    * Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ *
+ *    * Neither the name of Google Inc. nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+package com.google.auth.oauth2.storage;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+
+public class MemoryTokensStorage implements TokenStore {
+
+    private final Map<String, String> tokensStorage = new HashMap<>();
+
+    @Override
+    public String load(String id) throws IOException {
+        return tokensStorage.get(id);
+    }
+
+    @Override
+    public void store(String id, String tokens) throws IOException {
+        tokensStorage.put(id, tokens);
+    }
+
+    @Override
+    public void delete(String id) throws IOException {
+        tokensStorage.remove(id);
+    }
+}

oauth2_http/java/com/google/auth/oauth2/TokenStore.java → oauth2_http/java/com/google/auth/oauth2/storage/TokenStore.java

@@ -29,7 +29,7 @@
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-package com.google.auth.oauth2;
+package com.google.auth.oauth2.storage;
 
 import java.io.IOException;
 

oauth2_http/javatests/com/google/auth/oauth2/UserAuthorizerTest.java

@@ -40,9 +40,12 @@
 import com.google.auth.TestUtils;
 import com.google.auth.oauth2.GoogleCredentialsTest.MockTokenServerTransportFactory;
 
+import com.google.auth.oauth2.storage.MemoryTokensStorage;
+import com.google.auth.oauth2.storage.TokenStore;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
+import sun.tools.jstat.Token;
 
 import java.io.IOException;
 import java.net.URI;
@@ -75,7 +78,7 @@ public class UserAuthorizerTest {
 
   @Test
   public void constructorMinimum() {
-    TestTokenStore store = new TestTokenStore();
+    TokenStore store = new MemoryTokensStorage();
 
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
@@ -91,7 +94,7 @@ public void constructorMinimum() {
 
   @Test
   public void constructorCommon() {
-    TestTokenStore store = new TestTokenStore();
+    TokenStore store = new MemoryTokensStorage();
 
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
@@ -172,7 +175,7 @@ public void getCredentials_noCredentials_returnsNull() throws IOException {
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
-        .setTokenStore(new TestTokenStore())
+        .setTokenStore(new MemoryTokensStorage())
         .build();
 
     UserCredentials credentials = authorizer.getCredentials(USER_ID);
@@ -182,7 +185,7 @@ public void getCredentials_noCredentials_returnsNull() throws IOException {
 
   @Test
   public void getCredentials_storedCredentials_returnsStored() throws IOException {
-    TestTokenStore tokenStore = new TestTokenStore();
+    TokenStore tokenStore = new MemoryTokensStorage();
 
     UserCredentials initialCredentials = UserCredentials.newBuilder()
         .setClientId(CLIENT_ID_VALUE)
@@ -207,7 +210,7 @@ public void getCredentials_storedCredentials_returnsStored() throws IOException
 
   @Test(expected = NullPointerException.class)
   public void getCredentials_nullUserId_throws() throws IOException {
-    TestTokenStore tokenStore = new TestTokenStore();
+    TokenStore tokenStore = new MemoryTokensStorage();
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
@@ -236,7 +239,7 @@ public void getCredentials_refreshedToken_stored() throws IOException {
     MockTokenServerTransportFactory transportFactory = new MockTokenServerTransportFactory();
     transportFactory.transport.addClient(CLIENT_ID_VALUE, CLIENT_SECRET);
     transportFactory.transport.addRefreshToken(REFRESH_TOKEN, accessTokenValue2);
-    TestTokenStore tokenStore = new TestTokenStore();
+    TokenStore tokenStore = new MemoryTokensStorage();
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
@@ -277,7 +280,7 @@ public void getCredentialsFromCode_conevertsCodeToTokens() throws IOException {
     MockTokenServerTransportFactory transportFactory = new MockTokenServerTransportFactory();
     transportFactory.transport.addClient(CLIENT_ID_VALUE, CLIENT_SECRET);
     transportFactory.transport.addAuthorizationCode(CODE, REFRESH_TOKEN, ACCESS_TOKEN_VALUE);
-    TestTokenStore tokenStore = new TestTokenStore();
+    TokenStore tokenStore = new MemoryTokensStorage();
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
@@ -296,7 +299,7 @@ public void getCredentialsFromCode_nullCode_throws() throws IOException {
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
-        .setTokenStore(new TestTokenStore())
+        .setTokenStore(new MemoryTokensStorage())
         .build();
 
     authorizer.getCredentialsFromCode(null, BASE_URI);
@@ -309,7 +312,7 @@ public void getAndStoreCredentialsFromCode_getAndStoresCredentials() throws IOEx
     MockTokenServerTransportFactory transportFactory = new MockTokenServerTransportFactory();
     transportFactory.transport.addClient(CLIENT_ID_VALUE, CLIENT_SECRET);
     transportFactory.transport.addAuthorizationCode(CODE, REFRESH_TOKEN, accessTokenValue1);
-    TestTokenStore tokenStore = new TestTokenStore();
+    TokenStore tokenStore = new MemoryTokensStorage();
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
@@ -342,7 +345,7 @@ public void getAndStoreCredentialsFromCode_nullCode_throws() throws IOException
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
-        .setTokenStore(new TestTokenStore())
+        .setTokenStore(new MemoryTokensStorage())
         .build();
 
     authorizer.getAndStoreCredentialsFromCode(USER_ID, null, BASE_URI);
@@ -353,15 +356,15 @@ public void getAndStoreCredentialsFromCode_nullUserId_throws() throws IOExceptio
     UserAuthorizer authorizer = UserAuthorizer.newBuilder()
         .setClientId(CLIENT_ID)
         .setScopes(SCOPES)
-        .setTokenStore(new TestTokenStore())
+        .setTokenStore(new MemoryTokensStorage())
         .build();
 
     authorizer.getAndStoreCredentialsFromCode(null, CODE, BASE_URI);
   }
 
   @Test
   public void revokeAuthorization_revokesAndClears() throws IOException {
-    TestTokenStore tokenStore = new TestTokenStore();
+    TokenStore tokenStore = new MemoryTokensStorage();
     MockTokenServerTransportFactory transportFactory = new MockTokenServerTransportFactory();
     transportFactory.transport.addClient(CLIENT_ID_VALUE, CLIENT_SECRET);
     transportFactory.transport.addRefreshToken(REFRESH_TOKEN, ACCESS_TOKEN_VALUE);
@@ -398,24 +401,4 @@ public void revokeAuthorization_revokesAndClears() throws IOException {
     UserCredentials credentials2 = authorizer.getCredentials(USER_ID);
     assertNull(credentials2);
   }
-
-  private static class TestTokenStore implements TokenStore {
-
-    private final Map<String, String> map = new HashMap<>();
-
-    @Override
-    public String load(String id) throws IOException {
-      return map.get(id);
-    }
-
-    @Override
-    public void store(String id, String tokens) throws IOException {
-      map.put(id, tokens);
-    }
-
-    @Override
-    public void delete(String id) throws IOException {
-      map.remove(id);
-    }
-  }
 }

oauth2_http/javatests/com/google/auth/oauth2/UserCredentialsTest.java

@@ -51,9 +51,7 @@
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
-import java.io.ByteArrayInputStream;
-import java.io.IOException;
-import java.io.InputStream;
+import java.io.*;
 import java.net.URI;
 import java.util.Collection;
 import java.util.Collections;
@@ -492,6 +490,36 @@ public void fromStream_userNoRefreshToken_throws() throws IOException {
     testFromStreamException(userStream, "refresh_token");
   }
 
+  @Test
+  public void saveUserCredentials_saved_throws() throws IOException {
+    UserCredentials userCredentials = UserCredentials.newBuilder()
+            .setClientId(CLIENT_ID)
+            .setClientSecret(CLIENT_SECRET)
+            .setRefreshToken(REFRESH_TOKEN)
+            .build();
+    userCredentials.saveUserCredentials();
+  }
+
+  @Test
+  public void saveAndRestoreUserCredential_saveAndRestored_throws() throws IOException {
+    UserCredentials userCredentials = UserCredentials.newBuilder()
+            .setClientId(CLIENT_ID)
+            .setClientSecret(CLIENT_SECRET)
+            .setRefreshToken(REFRESH_TOKEN)
+            .build();
+    userCredentials.saveUserCredentials();
+
+    String userDirectory = System.getProperty("user.dir");
+    System.out.println(userDirectory);
+    FileInputStream inputStream = new FileInputStream(new File(userDirectory, UserCredentials.USER_CREDENTIALS_FILE_NAME));
+
+    UserCredentials restoredCredentials = UserCredentials.fromStream(inputStream);
+
+    assertEquals(userCredentials.getClientId(), restoredCredentials.getClientId());
+    assertEquals(userCredentials.getClientSecret(), restoredCredentials.getClientSecret());
+    assertEquals(userCredentials.getRefreshToken(), restoredCredentials.getRefreshToken());
+  }
+
   static GenericJson writeUserJson(String clientId, String clientSecret, String refreshToken) {
     GenericJson json = new GenericJson();
     if (clientId != null) {