Add GAIA info to profile info cache

chrome/browser/profiles/profile_info_cache.cc

 // Copyright (c) 2011 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/profiles/profile_info_cache.h"
 
+#include "base/bind.h"
+#include "base/file_util.h"
 #include "base/format_macros.h"
+#include "base/i18n/case_conversion.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/rand_util.h"
+#include "base/stl_util.h"
 #include "base/string_number_conversions.h"
 #include "base/stringprintf.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/prefs/scoped_user_pref_update.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/pref_names.h"
+#include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_service.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources.h"
+#include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
+#include "ui/gfx/codec/png_codec.h"
+#include "ui/gfx/image/image.h"
+#include "ui/gfx/image/image_util.h"
+
+using content::BrowserThread;
 
 namespace {
 
 const char kNameKey[] = "name";
+const char kGAIANameKey[] = "gaia_name";
+const char kUseGAIANameKey[] = "use_gaia_name";
 const char kUserNameKey[] = "user_name";
 const char kAvatarIconKey[] = "avatar_icon";
+const char kUseGAIAPictureKey[] = "use_gaia_picture";
 const char kBackgroundAppsKey[] = "background_apps";
+const char kHasMigratedToGAIAInfoKey[] = "has_migrated_to_gaia_info";
 const char kDefaultUrlPrefix[] = "chrome://theme/IDR_PROFILE_AVATAR_";
+const char kGAIAPictureFileName[] = "Google Picture.png";
 
 const int kDefaultAvatarIconResources[] = {
   IDR_PROFILE_AVATAR_0,
   IDR_PROFILE_AVATAR_1,
   IDR_PROFILE_AVATAR_2,
   IDR_PROFILE_AVATAR_3,
   IDR_PROFILE_AVATAR_4,
   IDR_PROFILE_AVATAR_5,
   IDR_PROFILE_AVATAR_6,
   IDR_PROFILE_AVATAR_7,
@@ skipping 36 matching lines @@
   IDS_DEFAULT_AVATAR_NAME_17,
   IDS_DEFAULT_AVATAR_NAME_18,
   IDS_DEFAULT_AVATAR_NAME_19,
   IDS_DEFAULT_AVATAR_NAME_20,
   IDS_DEFAULT_AVATAR_NAME_21,
   IDS_DEFAULT_AVATAR_NAME_22,
   IDS_DEFAULT_AVATAR_NAME_23,
   IDS_DEFAULT_AVATAR_NAME_24,
   IDS_DEFAULT_AVATAR_NAME_25
 };
 

[Miranda Callahan, 2011/11/17 15:21:10]

These two methods seem really out of place in ProfileInfoCache; is there a
utility class for image manipulation or saving things to disk where it could be
located? If not, maybe it's time to make one -- that way we could directly
combine this code and the code from user_manager.cc, and anybody else
reading/writing images could use it.

[sail, 2011/11/21 23:52:47]

I think this is a good idea. Mind if I do this later as a part of the overall
refactoring?

+// Writes the given bitmap as a PNG to disk. |callback| is invoked
+// on completion with true on success and false on failure.
+void SaveBitmap(gfx::Image image,
+                FilePath image_path,
+                base::Callback<void(bool)> callback) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+  std::vector<unsigned char> encoded_image;
+  if (!gfx::PNGCodec::EncodeBGRASkBitmap(image, true, &encoded_image)) {
+    LOG(ERROR) << "Failed to PNG encode the image.";
+    callback.Run(false);
+    return;
+  }
+
+  if (file_util::WriteFile(image_path,
+                           reinterpret_cast<char*>(&encoded_image[0]),
+                           encoded_image.size()) == -1) {
+    LOG(ERROR) << "Failed to save image to file.";
+    callback.Run(false);
+    return;
+  }
+
+  callback.Run(true);
+}
+
+// Reads a PNG from disk and decodes it. |callback| is invoked on
+// completion. If the bitmap was successfully read from disk the then callback
+// parameter will contain the bitmap image, otherwise it will be NULL.
+void ReadBitmap(FilePath image_path,
+                base::Callback<void(gfx::Image*)> callback) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+
+  std::string image_data;
+  if (!file_util::ReadFileToString(image_path, &image_data)) {
+    LOG(ERROR) << "Failed to read PNG file from disk.";
+    callback.Run(NULL);
+    return;
+  }
+
+  const unsigned char* data =
+      reinterpret_cast<const unsigned char*>(image_data.data());
+  gfx::Image* image = gfx::ImageFromPNGEncodedData(data, image_data.length());
+  if (image == NULL) {
+    LOG(ERROR) << "Failed to decode PNG file.";
+    callback.Run(NULL);
+    return;
+  }
+
+  callback.Run(image);
+}
+
 } // namespace
 
 ProfileInfoCache::ProfileInfoCache(PrefService* prefs,
                                    const FilePath& user_data_dir)
     : prefs_(prefs),
       user_data_dir_(user_data_dir) {
   // Populate the cache
   const DictionaryValue* cache =
       prefs_->GetDictionary(prefs::kProfileInfoCache);
   for (DictionaryValue::key_iterator it = cache->begin_keys();
        it != cache->end_keys(); ++it) {
     std::string key = *it;
     DictionaryValue* info = NULL;
     cache->GetDictionary(key, &info);
     string16 name;
     info->GetString(kNameKey, &name);
     sorted_keys_.insert(FindPositionForProfile(key, name), key);
   }
 }
 
 ProfileInfoCache::~ProfileInfoCache() {
+  STLDeleteContainerPairSecondPointers(
+      gaia_pictures_.begin(), gaia_pictures_.end());
 }
 
 void ProfileInfoCache::AddProfileToCache(const FilePath& profile_path,
                                          const string16& name,
                                          const string16& username,
                                          size_t icon_index) {
   std::string key = CacheKeyFromProfilePath(profile_path);
   DictionaryPrefUpdate update(prefs_, prefs::kProfileInfoCache);
   DictionaryValue* cache = update.Get();
 
@@ skipping 37 matching lines @@
     return std::string::npos;
   std::string search_key = CacheKeyFromProfilePath(profile_path);
   for (size_t i = 0; i < sorted_keys_.size(); ++i) {
     if (sorted_keys_[i] == search_key)
       return i;
   }
   return std::string::npos;
 }
 
 string16 ProfileInfoCache::GetNameOfProfileAtIndex(size_t index) const {
+  if (GetIsUsingGAIANameOfProfileAtIndex(index))
+    return GetGAIANameOfProfileAtIndex(index);
+
   string16 name;
   GetInfoForProfileAtIndex(index)->GetString(kNameKey, &name);
   return name;
 }
 
 FilePath ProfileInfoCache::GetPathOfProfileAtIndex(size_t index) const {
-  FilePath::StringType base_name;
-#if defined(OS_POSIX)
-  base_name = sorted_keys_[index];
-#elif defined(OS_WIN)
-  base_name = ASCIIToWide(sorted_keys_[index]);
-#endif
-  return user_data_dir_.Append(base_name);
+  return user_data_dir_.AppendASCII(sorted_keys_[index]);
 }
 
 string16 ProfileInfoCache::GetUserNameOfProfileAtIndex(size_t index) const {
   string16 user_name;
   GetInfoForProfileAtIndex(index)->GetString(kUserNameKey, &user_name);
   return user_name;
 }
 
 const gfx::Image& ProfileInfoCache::GetAvatarIconOfProfileAtIndex(
     size_t index) const {
+  if (GetIsUsingGAIAPictureOfProfileAtIndex(index))
+    return GetGAIAPictureOfProfileAtIndex(index);
+
   int resource_id = GetDefaultAvatarIconResourceIDAtIndex(
       GetAvatarIconIndexOfProfileAtIndex(index));
   return ResourceBundle::GetSharedInstance().GetImageNamed(resource_id);
 }
 
 bool ProfileInfoCache::GetBackgroundStatusOfProfileAtIndex(
     size_t index) const {
   bool background_app_status;
   GetInfoForProfileAtIndex(index)->GetBoolean(kBackgroundAppsKey,
                                               &background_app_status);
   return background_app_status;
 }
 
+string16 ProfileInfoCache::GetGAIANameOfProfileAtIndex(size_t index) const {
+  string16 name;
+  GetInfoForProfileAtIndex(index)->GetString(kGAIANameKey, &name);
+  return name;
+}
+
+bool ProfileInfoCache::GetIsUsingGAIANameOfProfileAtIndex(size_t index) const {
+  bool value = false;
+  GetInfoForProfileAtIndex(index)->GetBoolean(kUseGAIANameKey, &value);
+  return value;
+}
+
+const gfx::Image& ProfileInfoCache::GetGAIAPictureOfProfileAtIndex(
+    size_t index) const {
+  FilePath path = GetPathOfProfileAtIndex(index);
+  std::string key = CacheKeyFromProfilePath(path);
+  if (gaia_pictures_.count(key)) {
+    return *gaia_pictures_[key];
+  }
+
+  // The GAIA picture is not in the cache yet. Load it from disk and return
+  // a blank picture for now.
+  gaia_pictures_[key] = new gfx::Image(new SkBitmap());
+
+  FilePath image_path = path.AppendASCII(kGAIAPictureFileName);
+  base::Callback<void(gfx::Image*)> callback = base::Bind(
+      &ProfileInfoCache::OnGAIAPictureLoaded, base::Unretained(this), path);

[Miranda Callahan, 2011/11/17 15:21:10]

Could we be heading for trouble by using "Unretained" here (and in other places
in this class)? 

[Robert Sesek, 2011/11/17 20:42:04]

Definitely. You need to use WeakPtrFactory.

[sail, 2011/11/21 23:52:47]

Done.

+  BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE,
+      base::Bind(&ReadBitmap, image_path, callback));
+
+  return *gaia_pictures_[key];
+}
+
+void ProfileInfoCache::OnGAIAPictureLoaded(FilePath path,
+                                           gfx::Image* image) const {
+  if (!image)
+    return;
+
+  std::string key = CacheKeyFromProfilePath(path);
+  gaia_pictures_[key] = image;
+
+  content::NotificationService::current()->Notify(
+    chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED,

[Robert Sesek, 2011/11/17 20:42:04]

nit: indent 4

[sail, 2011/11/21 23:52:47]

Done.

+    content::NotificationService::AllSources(),
+    content::NotificationService::NoDetails());
+}
+
+void ProfileInfoCache::OnGAIAPictureSaved(FilePath path, bool success) const {
+  if (!success)
+    return;
+
+  content::NotificationService::current()->Notify(
+    chrome::NOTIFICATION_PROFILE_CACHE_PICTURE_SAVED,

[Robert Sesek, 2011/11/17 20:42:04]

nit: indent 4

[sail, 2011/11/21 23:52:47]

Done.

+    content::NotificationService::AllSources(),
+    content::NotificationService::NoDetails());
+}
+
+bool ProfileInfoCache::GetIsUsingGAIAPictureOfProfileAtIndex(
+    size_t index) const {
+  bool value = false;
+  GetInfoForProfileAtIndex(index)->GetBoolean(kUseGAIAPictureKey, &value);
+  return value;
+}
+
 size_t ProfileInfoCache::GetAvatarIconIndexOfProfileAtIndex(size_t index)
     const {
   std::string icon_url;
   GetInfoForProfileAtIndex(index)->GetString(kAvatarIconKey, &icon_url);
   size_t icon_index = 0;
   if (IsDefaultAvatarIconUrl(icon_url, &icon_index))
     return icon_index;
 
   DLOG(WARNING) << "Unknown avatar icon: " << icon_url;
   return GetDefaultAvatarIconResourceIDAtIndex(0);
 }
 
 void ProfileInfoCache::SetNameOfProfileAtIndex(size_t index,
                                                const string16& name) {
+  if (name == GetNameOfProfileAtIndex(index))
+    return;
+
   scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
   info->SetString(kNameKey, name);
   // This takes ownership of |info|.
   SetInfoForProfileAtIndex(index, info.release());
-
-  // Remove and reinsert key in |sorted_keys_| to alphasort.
-  std::string key = CacheKeyFromProfilePath(GetPathOfProfileAtIndex(index));
-  std::vector<std::string>::iterator key_it =
-      std::find(sorted_keys_.begin(), sorted_keys_.end(), key);
-  DCHECK(key_it != sorted_keys_.end());
-  sorted_keys_.erase(key_it);
-  sorted_keys_.insert(FindPositionForProfile(key, name), key);
-
-  content::NotificationService::current()->Notify(
-    chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED,
-    content::NotificationService::AllSources(),
-    content::NotificationService::NoDetails());
+  UpdateSortForProfileIndex(index);
 }
 
 void ProfileInfoCache::SetUserNameOfProfileAtIndex(size_t index,
                                                    const string16& user_name) {
-  string16 old_user_name;
-  const base::DictionaryValue* old_info = GetInfoForProfileAtIndex(index);
-  old_info->GetString(kUserNameKey, &old_user_name);
-  if (old_user_name == user_name)
+  if (user_name == GetUserNameOfProfileAtIndex(index))
     return;
 
-  scoped_ptr<DictionaryValue> info(old_info->DeepCopy());
+  scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
   info->SetString(kUserNameKey, user_name);
   // This takes ownership of |info|.
   SetInfoForProfileAtIndex(index, info.release());
 }
 
 void ProfileInfoCache::SetAvatarIconOfProfileAtIndex(size_t index,
                                                      size_t icon_index) {
   scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
   info->SetString(kAvatarIconKey, GetDefaultAvatarIconUrl(icon_index));
   // This takes ownership of |info|.
   SetInfoForProfileAtIndex(index, info.release());
 }
 
 void ProfileInfoCache::SetBackgroundStatusOfProfileAtIndex(
     size_t index,
     bool running_background_apps) {
   if (GetBackgroundStatusOfProfileAtIndex(index) == running_background_apps)
     return;
   scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
   info->SetBoolean(kBackgroundAppsKey, running_background_apps);
   // This takes ownership of |info|.
   SetInfoForProfileAtIndex(index, info.release());
 }
 
+void ProfileInfoCache::SetGAIANameOfProfileAtIndex(size_t index,
+                                                   const string16& name) {
+  if (name == GetGAIANameOfProfileAtIndex(index))
+    return;
+
+  scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
+  info->SetString(kGAIANameKey, name);
+  // This takes ownership of |info|.
+  SetInfoForProfileAtIndex(index, info.release());
+  UpdateSortForProfileIndex(index);
+}
+
+void ProfileInfoCache::SetIsUsingGAIANameOfProfileAtIndex(size_t index,
+                                                          bool value) {
+  if (value == GetIsUsingGAIANameOfProfileAtIndex(index))
+    return;
+
+  scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
+  info->SetBoolean(kUseGAIANameKey, value);
+  // This takes ownership of |info|.
+  SetInfoForProfileAtIndex(index, info.release());
+  UpdateSortForProfileIndex(index);
+}
+
+void ProfileInfoCache::SetGAIAPictureOfProfileAtIndex(size_t index,
+                                                      const gfx::Image& image) {
+  FilePath path = GetPathOfProfileAtIndex(index);
+  std::string key = CacheKeyFromProfilePath(path);
+
+  delete gaia_pictures_[key];
+  gaia_pictures_[key] = new gfx::Image(image);
+
+  FilePath image_path = path.AppendASCII(kGAIAPictureFileName);
+  base::Callback<void(bool)> callback = base::Bind(
+      &ProfileInfoCache::OnGAIAPictureSaved, base::Unretained(this), path);
+  BrowserThread::PostTask(BrowserThread::FILE,
+                          FROM_HERE,
+                          base::Bind(&SaveBitmap, image, image_path, callback));
+
+  content::NotificationService::current()->Notify(
+    chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED,

[Robert Sesek, 2011/11/17 20:42:04]

nit: indent 4

[sail, 2011/11/21 23:52:47]

Done.

+    content::NotificationService::AllSources(),
+    content::NotificationService::NoDetails());
+}
+
+void ProfileInfoCache::SetIsUsingGAIAPictureOfProfileAtIndex(size_t index,
+                                                             bool value) {
+  scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
+  info->SetBoolean(kUseGAIAPictureKey, value);
+  // This takes ownership of |info|.
+  SetInfoForProfileAtIndex(index, info.release());
+}
+
 string16 ProfileInfoCache::ChooseNameForNewProfile(size_t icon_index) {
   string16 name;
   for (int name_index = 1; ; ++name_index) {
     if (icon_index < kGenericIconCount) {
       name = l10n_util::GetStringFUTF16Int(IDS_NUMBERED_PROFILE_NAME,
                                            name_index);
     } else {
       name = l10n_util::GetStringUTF16(
           kDefaultNames[icon_index - kGenericIconCount]);
       if (name_index > 1)
         name.append(UTF8ToUTF16(base::IntToString(name_index)));
     }
 
     // Loop through previously named profiles to ensure we're not duplicating.
     bool name_found = false;
     for (size_t i = 0; i < GetNumberOfProfiles(); ++i) {
       if (GetNameOfProfileAtIndex(i) == name) {
         name_found = true;
         break;
       }
     }
     if (!name_found)
       return name;
   }
 }
 
+bool ProfileInfoCache::GetHasMigratedToGAIAInfoOfProfileAtIndex(
+      size_t index) const {
+  bool value = false;
+  GetInfoForProfileAtIndex(index)->GetBoolean(
+      kHasMigratedToGAIAInfoKey, &value);
+  return value;
+}
+
+void ProfileInfoCache::SetHasMigratedToGAIAInfoOfProfileAtIndex(
+    size_t index, bool value) {
+  scoped_ptr<DictionaryValue> info(GetInfoForProfileAtIndex(index)->DeepCopy());
+  info->SetBoolean(kHasMigratedToGAIAInfoKey, value);
+  // This takes ownership of |info|.
+  SetInfoForProfileAtIndex(index, info.release());
+}
+
 bool ProfileInfoCache::IconIndexIsUnique(size_t icon_index) const {
   for (size_t i = 0; i < GetNumberOfProfiles(); ++i) {
     if (GetAvatarIconIndexOfProfileAtIndex(i) == icon_index)
       return false;
   }
   return true;
 }
 
 bool ProfileInfoCache::ChooseAvatarIconIndexForNewProfile(
     bool allow_generic_icon,
@@ skipping 98 matching lines @@
 std::string ProfileInfoCache::CacheKeyFromProfilePath(
     const FilePath& profile_path) const {
   DCHECK(user_data_dir_ == profile_path.DirName());
   FilePath base_name = profile_path.BaseName();
   return base_name.MaybeAsASCII();
 }
 
 std::vector<std::string>::iterator ProfileInfoCache::FindPositionForProfile(
     std::string search_key,
     const string16& search_name) {
+  string16 search_name_l = base::i18n::ToLower(search_name);
   for (size_t i = 0; i < GetNumberOfProfiles(); ++i) {
-    int name_compare = search_name.compare(GetNameOfProfileAtIndex(i));
+    string16 name_l = base::i18n::ToLower(GetNameOfProfileAtIndex(i));
+    int name_compare = search_name_l.compare(name_l);
     if (name_compare < 0)
       return sorted_keys_.begin() + i;
     if (name_compare == 0) {
       int key_compare = search_key.compare(sorted_keys_[i]);
       if (key_compare < 0)
         return sorted_keys_.begin() + i;
     }
   }
   return sorted_keys_.end();
 }
 
 void ProfileInfoCache::RegisterPrefs(PrefService* prefs) {
   prefs->RegisterDictionaryPref(prefs::kProfileInfoCache);
 }
+
+void ProfileInfoCache::UpdateSortForProfileIndex(size_t index) {
+  string16 name = GetNameOfProfileAtIndex(index);
+
+  // Remove and reinsert key in |sorted_keys_| to alphasort.
+  std::string key = CacheKeyFromProfilePath(GetPathOfProfileAtIndex(index));
+  std::vector<std::string>::iterator key_it =
+      std::find(sorted_keys_.begin(), sorted_keys_.end(), key);
+  DCHECK(key_it != sorted_keys_.end());
+  sorted_keys_.erase(key_it);
+  sorted_keys_.insert(FindPositionForProfile(key, name), key);
+
+  content::NotificationService::current()->Notify(
+    chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED,

[Robert Sesek, 2011/11/17 20:42:04]

nit: indent 4

[sail, 2011/11/21 23:52:47]

Done.

+    content::NotificationService::AllSources(),
+    content::NotificationService::NoDetails());
+}