Add avatar syncing for supervised users.

chrome/browser/managed_mode/managed_user_registration_utility.cc

 // 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.
 
 #include "chrome/browser/managed_mode/managed_user_registration_utility.h"
 
 #include "base/base64.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/pref_service.h"
 #include "base/rand_util.h"
 #include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/managed_mode/managed_user_constants.h"
 #include "chrome/browser/managed_mode/managed_user_refresh_token_fetcher.h"
+#include "chrome/browser/managed_mode/managed_user_shared_settings_service.h"
+#include "chrome/browser/managed_mode/managed_user_shared_settings_service_factory.h"
 #include "chrome/browser/managed_mode/managed_user_sync_service.h"
 #include "chrome/browser/managed_mode/managed_user_sync_service_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/signin/profile_oauth2_token_service.h"
 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h"
 #include "chrome/browser/sync/glue/device_info.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "google_apis/gaia/gaia_urls.h"
 #include "google_apis/gaia/google_service_auth_error.h"
 
 using base::DictionaryValue;
 
 namespace {
 
 ManagedUserRegistrationUtility* g_instance_for_tests = NULL;
 
 // Actual implementation of ManagedUserRegistrationUtility.
 class ManagedUserRegistrationUtilityImpl
     : public ManagedUserRegistrationUtility,
       public ManagedUserSyncServiceObserver {
  public:
   ManagedUserRegistrationUtilityImpl(
       PrefService* prefs,
       scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher,
-      ManagedUserSyncService* service);
+      ManagedUserSyncService* service,
+      ManagedUserSharedSettingsService* shared_settings_service);
 
   virtual ~ManagedUserRegistrationUtilityImpl();
 
   // Registers a new managed user with the server. |managed_user_id| is a new
   // unique ID for the new managed user. If its value is the same as that of
   // of one of the existing managed users, then the same user will be created
   // on this machine (and if he has no avatar in sync, his avatar will
   // be updated). |info| contains necessary information like
   // the display name of the user and his avatar. |callback| is called
   // with the result of the registration. We use the info here and not the
@@ skipping 33 matching lines @@
   // Cancels any registration currently in progress, without calling the
   // callback or reporting an error.
   void CancelPendingRegistration();
 
   PrefService* prefs_;
   scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher_;
 
   // A |BrowserContextKeyedService| owned by the custodian profile.
   ManagedUserSyncService* managed_user_sync_service_;
 
+  // A |BrowserContextKeyedService| owned by the custodian profile.
+  ManagedUserSharedSettingsService* managed_user_shared_settings_service_;
+
   std::string pending_managed_user_id_;
   std::string pending_managed_user_token_;
   bool pending_managed_user_acknowledged_;
   bool is_existing_managed_user_;
   bool avatar_updated_;
   RegistrationCallback callback_;
 
   base::WeakPtrFactory<ManagedUserRegistrationUtilityImpl> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ManagedUserRegistrationUtilityImpl);
@@ skipping 30 matching lines @@
 
   ProfileOAuth2TokenService* token_service =
       ProfileOAuth2TokenServiceFactory::GetForProfile(profile);
   scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher =
       ManagedUserRefreshTokenFetcher::Create(
           token_service,
           token_service->GetPrimaryAccountId(),
           profile->GetRequestContext());
   ManagedUserSyncService* managed_user_sync_service =
       ManagedUserSyncServiceFactory::GetForProfile(profile);
+  ManagedUserSharedSettingsService* managed_user_shared_settings_service =
+      ManagedUserSharedSettingsServiceFactory::GetForBrowserContext(profile);
   return make_scoped_ptr(ManagedUserRegistrationUtility::CreateImpl(
       profile->GetPrefs(),
       token_fetcher.Pass(),
-      managed_user_sync_service));
+      managed_user_sync_service,
+      managed_user_shared_settings_service));
 }
 
 // static
 std::string ManagedUserRegistrationUtility::GenerateNewManagedUserId() {
   std::string new_managed_user_id;
   base::Base64Encode(base::RandBytesAsString(8), &new_managed_user_id);
   return new_managed_user_id;
 }
 
 // static
 void ManagedUserRegistrationUtility::SetUtilityForTests(
     ManagedUserRegistrationUtility* utility) {
   if (g_instance_for_tests)
     delete g_instance_for_tests;
   g_instance_for_tests = utility;
 }
 
 // static
 ManagedUserRegistrationUtility* ManagedUserRegistrationUtility::CreateImpl(
       PrefService* prefs,
       scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher,
-      ManagedUserSyncService* service) {
+      ManagedUserSyncService* service,
+      ManagedUserSharedSettingsService* shared_settings_service) {
   return new ManagedUserRegistrationUtilityImpl(prefs,
                                                 token_fetcher.Pass(),
-                                                service);
+                                                service,
+                                                shared_settings_service);
 }
 
 namespace {
 
 ManagedUserRegistrationUtilityImpl::ManagedUserRegistrationUtilityImpl(
     PrefService* prefs,
     scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher,
-    ManagedUserSyncService* service)
+    ManagedUserSyncService* service,
+    ManagedUserSharedSettingsService* shared_settings_service)
     : prefs_(prefs),
       token_fetcher_(token_fetcher.Pass()),
       managed_user_sync_service_(service),
+      managed_user_shared_settings_service_(shared_settings_service),
       pending_managed_user_acknowledged_(false),
       is_existing_managed_user_(false),
       avatar_updated_(false),
       weak_ptr_factory_(this) {
   managed_user_sync_service_->AddObserver(this);
 }
 
 ManagedUserRegistrationUtilityImpl::~ManagedUserRegistrationUtilityImpl() {
   managed_user_sync_service_->RemoveObserver(this);
   CancelPendingRegistration();
@@ skipping 17 matching lines @@
                                                info.avatar_index);
   } else {
     avatar_updated_ =
         managed_user_sync_service_->UpdateManagedUserAvatarIfNeeded(
             managed_user_id,
             info.avatar_index);
 
     // User already exists, don't wait for acknowledgment.
     OnManagedUserAcknowledged(managed_user_id);
   }
+#if defined(OS_CHROMEOS)
+  const char* kAvatarKey = managed_users::kChromeOSAvatarIndex;
+#else
+  const char* kAvatarKey = managed_users::kChromeAvatarIndex;
+#endif
+  managed_user_shared_settings_service_->SetValue(
+      pending_managed_user_id_, kAvatarKey,
+      base::FundamentalValue(info.avatar_index));
 
   browser_sync::DeviceInfo::GetClientName(
       base::Bind(&ManagedUserRegistrationUtilityImpl::FetchToken,
                  weak_ptr_factory_.GetWeakPtr()));
 }
 
 void ManagedUserRegistrationUtilityImpl::CancelPendingRegistration() {
   AbortPendingRegistration(
       false,  // Don't run the callback. The error will be ignored.
       GoogleServiceAuthError(GoogleServiceAuthError::NONE));
@@ skipping 75 matching lines @@
           pending_managed_user_id_);
     }
   }
 
   if (run_callback)
     callback_.Run(error, pending_managed_user_token_);
   callback_.Reset();
 }
 
 } // namespace