settings: Add async system salt retrieval logic in DeviceOAuth2TokenServiceFactory

chrome/browser/chromeos/settings/device_oauth2_token_service_factory.cc

 // Copyright (c) 2012 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.
 
 #include "chrome/browser/chromeos/settings/device_oauth2_token_service_factory.h"
 
+#include "base/bind.h"
+#include "base/callback.h"
 #include "base/message_loop/message_loop.h"
 #include "base/tracked_objects.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/settings/device_oauth2_token_service.h"
 #include "chrome/browser/chromeos/settings/token_encryptor.h"
+#include "chromeos/cryptohome/system_salt_getter.h"
 #include "content/public/browser/browser_thread.h"
 
 namespace chromeos {
 namespace {
 
 DeviceOAuth2TokenServiceFactory* g_factory = NULL;
 
 }  // namespace
 
 DeviceOAuth2TokenServiceFactory::DeviceOAuth2TokenServiceFactory()
-    : token_service_(new DeviceOAuth2TokenService(
-        g_browser_process->system_request_context(),
-        g_browser_process->local_state(),
-        new CryptohomeTokenEncryptor)) {
+    : token_service_(NULL),
+      weak_ptr_factory_(this) {
 }
 
 DeviceOAuth2TokenServiceFactory::~DeviceOAuth2TokenServiceFactory() {
   delete token_service_;
 }
 
 // static
 void DeviceOAuth2TokenServiceFactory::Get(const GetCallback& callback) {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 
-  DeviceOAuth2TokenService* token_service = NULL;
-  if (g_factory)
-    token_service = g_factory->token_service_;
+  if (!g_factory) {
+    base::MessageLoop::current()->PostTask(
+        FROM_HERE,
+        base::Bind(callback,
+                   static_cast<DeviceOAuth2TokenService*>(NULL)));
+    return;
+  }
 
-  // TODO(satorux): Implement async initialization logic for
-  // DeviceOAuth2TokenService. crbug.com/309959.
-  // Here's how that should work:
-  //
-  // if token_service is ready:
-  //   run callback asynchronously via MessageLoop
-  //   return
-  //
-  // add callback to the pending callback list
-  //
-  // if there is only one pending callback:
-  //   start getting the system salt asynchronously...

[satorux1, 2013/10/24 06:25:18]

I originally thinking about initiating the system salt retrieval here in Get(),
but I followed a suggestion from pneubeck@ [1], and decided to initiate it in
Initialize(). pneubeck@, thank you for giving me the great suggestion.

[1]
https://codereview.chromium.org/34523003/diff/40001/chrome/browser/chromeos/s...

-  //
-  // upon receiving the system salt:
-  //   create CryptohomeTokenEncryptor with that key
-  //   create DeviceOAuth2TokenService
-  //   run all the pending callbacks
-  base::MessageLoop::current()->PostTask(
-      FROM_HERE,
-      base::Bind(callback, token_service));
+  g_factory->RunAsync(callback);
 }
 
 // static
 void DeviceOAuth2TokenServiceFactory::Initialize() {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 
   DCHECK(!g_factory);
   g_factory = new DeviceOAuth2TokenServiceFactory;
+  g_factory->CreateTokenService();
 }
 
 // static
 void DeviceOAuth2TokenServiceFactory::Shutdown() {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 
   if (g_factory) {
     delete g_factory;
     g_factory = NULL;
   }
 }
 
+void DeviceOAuth2TokenServiceFactory::CreateTokenService() {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+  SystemSaltGetter::Get()->GetSystemSalt(
+      base::Bind(&DeviceOAuth2TokenServiceFactory::DidGetSystemSalt,
+                 weak_ptr_factory_.GetWeakPtr()));
+}
+
+void DeviceOAuth2TokenServiceFactory::DidGetSystemSalt(
+    const std::string& system_salt) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+  DCHECK(!token_service_);
+
+  token_service_= new DeviceOAuth2TokenService(
+      g_browser_process->system_request_context(),
+      g_browser_process->local_state(),
+      new CryptohomeTokenEncryptor(system_salt));
+
+  for (size_t i = 0; i < pending_callbacks_.size(); ++i)
+    pending_callbacks_[i].Run(token_service_);
+  pending_callbacks_.clear();
+}
+
+void DeviceOAuth2TokenServiceFactory::RunAsync(const GetCallback& callback) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+  if (token_service_) {
+    DCHECK(pending_callbacks_.empty());
+    base::MessageLoop::current()->PostTask(
+        FROM_HERE,
+        base::Bind(callback, token_service_));
+    return;
+  }
+
+  pending_callbacks_.push_back(callback);
+}
+
 }  // namespace chromeos