Get OAuth2TokenService working on Android.

chrome/android/java/src/org/chromium/chrome/browser/sync/ProfileSyncService.java

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.chrome.browser.sync;
 
 import android.accounts.Account;
+import android.accounts.AccountManager;

[nyquist, 2013/03/25 21:01:21]

Should not import this.

[Patrick Dubroy, 2013/03/25 21:15:49]

Done.

 import android.content.Context;
 import android.os.AsyncTask;
 import android.util.Log;
 
 import com.google.common.annotations.VisibleForTesting;
 
 import org.chromium.base.CalledByNative;
 import org.chromium.base.ThreadUtils;
 import org.chromium.chrome.browser.identity.UniqueIdentificationGenerator;
 import org.chromium.sync.internal_api.pub.SyncDecryptionPassphraseType;
@@ skipping 64 matching lines @@
         ThreadUtils.assertOnUiThread();
         // We should store the application context, as we outlive any activity which may create us.
         mContext = context.getApplicationContext();
 
         // This may cause us to create ProfileSyncService even if sync has not
         // been set up, but ProfileSyncService::Startup() won't be called until
         // credentials are available.
         mNativeProfileSyncServiceAndroid = nativeInit();
     }
 
+    @CalledByNative
+    private static int getProfileSyncServiceAndroid(Context context) {
+        return get(context).mNativeProfileSyncServiceAndroid;
+    }
+
     /**
      * If we are currently in the process of setting up sync, this method clears the
      * sync setup in progress flag.
      */
     @VisibleForTesting
     public void finishSyncFirstSetupIfNeeded() {
         if (isFirstSetupInProgress()) {
             setSyncSetupCompleted();
             setSetupInProgress(false);
         }
@@ skipping 54 matching lines @@
                     "This may lead to unexpected tab sync behavior.");
             return;
         }
         String sessionTag = SESSION_TAG_PREFIX + uniqueTag;
         if (!nativeSetSyncSessionsId(mNativeProfileSyncServiceAndroid, sessionTag)) {
             Log.e(TAG, "Unable to write session sync tag. " +
                     "This may lead to unexpected tab sync behavior.");
         }
     }
 
+    private Account getAccountOrNullFromUsername(String username) {
+        if (username == null) {
+            Log.e(TAG, "username is null");
+            return null;
+        }
+
+        final AccountManagerHelper accountManagerHelper = AccountManagerHelper.get(mContext);
+        final Account account = accountManagerHelper.getAccountFromName(username);
+        if (account == null) {
+            Log.e(TAG, "Account not found for provided username.");
+            return null;
+        }
+        return account;
+    }
+
     /**
      * Requests a new auth token from the AccountManager. Invalidates the old token
      * if |invalidAuthToken| is not empty.
      */
     @CalledByNative
     public void getNewAuthToken(final String username, final String invalidAuthToken) {
-        if (username == null) {
-            Log.e(TAG, "username is null");
-            return;
-        }
-
-        final AccountManagerHelper accountManagerHelper = AccountManagerHelper.get(mContext);
-        final Account account = accountManagerHelper.getAccountFromName(username);
-        if (account == null) {
-            Log.e(TAG, "Account not found for provided username.");
-            return;
-        }
+        final Account account = getAccountOrNullFromUsername(username);
+        if (account == null) return;
 
         // Since this is blocking, do it in the background.
         new AsyncTask<Void, Void, String>() {
 
             @Override
             public String doInBackground(Void... params) {
                 // Invalidate our old auth token and fetch a new one.
+                AccountManagerHelper accountManagerHelper = AccountManagerHelper.get(mContext);
                 return accountManagerHelper.getNewAuthToken(
-                        account, invalidAuthToken, SyncStatusHelper.AUTH_TOKEN_TYPE_SYNC);
+                    account, invalidAuthToken, SyncStatusHelper.AUTH_TOKEN_TYPE_SYNC);
             }
 
             @Override
             public void onPostExecute(String authToken) {
                 if (authToken == null) {
-                    // DO NOT COMMIT do we really need this TODO? We trigger a call to
-                    // requestSyncFromNativeChrome() when an account changes and sync is setup.
                     // TODO(sync): Need to hook LOGIN_ACCOUNTS_CHANGED_ACTION (http://b/5354713).
                     Log.d(TAG, "Auth token for sync was null.");
                 } else {
                     Log.d(TAG, "Successfully retrieved sync auth token.");
                     nativeTokenAvailable(mNativeProfileSyncServiceAndroid, username, authToken);
                 }
             }
         }.execute();
     }
 
+   /**
+    * Called by native to invalidate an OAuth2 token.
+    */
+    @CalledByNative
+    public void invalidateOAuth2AuthToken(String scope, String access_token) {

[nyquist, 2013/03/25 21:01:21]

camelCase

[Patrick Dubroy, 2013/03/25 21:15:49]

Done.

+      AccountManager.get(mContext).invalidateAuthToken(scope, access_token);

[nyquist, 2013/03/25 21:01:21]

Do not use the AccountManager directly. Always go through the
AccountManagerHelper. Otherwise it is a pain to test this. We don't want our
tests to hit the real auth backends.

Just add a simple method to AccountManagerHelper that invalidates an auth token.

[Patrick Dubroy, 2013/03/25 21:15:49]

Done.

+    }
+
+    /**
+     * Called by native to retrieve OAuth2 tokens.
+     *
+     * @param username the native username (full address)
+     * @param scope the scope to get an auth token for (without Android-style 'oauth2:' prefix).
+     * @param oldAuthToken if provided, the token will be invalidated before getting a new token.
+     * @param nativeCallback the pointer to the native callback that should be run upon completion.
+     */
+    @CalledByNative
+    public void getOAuth2AuthToken(
+            String username, String scope, final int nativeCallback) {

[nyquist, 2013/03/25 21:01:21]

Nit: Does this fit on the line above?

[Patrick Dubroy, 2013/03/25 21:15:49]

Done.

+        final Account account = getAccountOrNullFromUsername(username);
+        if (account == null) {
+            nativeOAuth2TokenFetched(
+                mNativeProfileSyncServiceAndroid, nativeCallback, null, false);
+            return;
+        }
+        final String oauth2Scope = "oauth2:" + scope;
+
+        new AsyncTask<Void, Void, String>() {
+            @Override
+            public String doInBackground(Void... params) {
+                AccountManagerHelper accountManagerHelper = AccountManagerHelper.get(mContext);
+                return accountManagerHelper.getAuthTokenFromBackground(account, oauth2Scope);
+            }
+
+            @Override
+            public void onPostExecute(String authToken) {
+                nativeOAuth2TokenFetched(
+                    mNativeProfileSyncServiceAndroid, nativeCallback, authToken, authToken != null);
+            }
+        }.execute();
+    }
+
     /**
      * Checks if a password or a passphrase is required for decryption of sync data.
      * <p/>
      * Returns NONE if the state is unavailable, or decryption passphrase/password is not required.
      *
      * @return the enum describing the decryption passphrase type required
      */
     public SyncDecryptionPassphraseType getSyncDecryptionPassphraseTypeIfRequired() {
         // ProfileSyncService::IsUsingSecondaryPassphrase() requires the sync backend to be
         // initialized, and that happens just after OnPassphraseRequired(). Therefore, we need to
@@ skipping 308 matching lines @@
     private native void nativeSetSyncSetupCompleted(int nativeProfileSyncServiceAndroid);
     private native boolean nativeHasSyncSetupCompleted(int nativeProfileSyncServiceAndroid);
     private native boolean nativeHasKeepEverythingSynced(int nativeProfileSyncServiceAndroid);
     private native boolean nativeIsAutofillSyncEnabled(int nativeProfileSyncServiceAndroid);
     private native boolean nativeIsBookmarkSyncEnabled(int nativeProfileSyncServiceAndroid);
     private native boolean nativeIsPasswordSyncEnabled(int nativeProfileSyncServiceAndroid);
     private native boolean nativeIsTypedUrlSyncEnabled(int nativeProfileSyncServiceAndroid);
     private native boolean nativeIsSessionSyncEnabled(int nativeProfileSyncServiceAndroid);
     private native boolean nativeHasUnrecoverableError(int nativeProfileSyncServiceAndroid);
     private native String nativeGetAboutInfoForTest(int nativeProfileSyncServiceAndroid);
+    private native void nativeOAuth2TokenFetched(
+            int nativeProfileSyncServiceAndroid, int nativeCallback, String authToken,
+            boolean result);
 }