Show GAIA profile info in NTP

chrome/browser/ui/webui/ntp/ntp_login_handler.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/ui/webui/ntp/ntp_login_handler.h"
 
 #include <string>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/metrics/histogram.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/google/google_util.h"
 #include "chrome/browser/prefs/pref_notifier.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_info_cache.h"
+#include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/profiles/profile_metrics.h"
 #include "chrome/browser/sync/profile_sync_service.h"
 #include "chrome/browser/sync/sync_setup_flow.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/browser/ui/webui/sync_promo_ui.h"
+#include "chrome/browser/ui/webui/web_ui_util.h"
 #include "chrome/browser/web_resource/promo_resource_service.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "content/public/browser/notification_details.h"
+#include "content/public/browser/notification_service.h"
 #include "grit/chromium_strings.h"
 #include "grit/generated_resources.h"
+#include "skia/ext/image_operations.h"
 #include "ui/base/l10n/l10n_util.h"
+#include "ui/gfx/canvas_skia.h"
+#include "ui/gfx/image/image.h"
+
+namespace {
+
+SkBitmap GetGAIAPictureForNTP(const gfx::Image& image) {
+  // This value must match the width and height value of login-status-icon
+  // in new_tab.css.
+  const int length = 27;
+  SkBitmap bmp = skia::ImageOperations::Resize(
+      image, skia::ImageOperations::RESIZE_BEST, length, length);
+
+  gfx::CanvasSkia canvas(length, length, false);
+  canvas.DrawBitmapInt(bmp, 0, 0);
+
+  // Draw a gray border on the inside of the icon.
+  SkColor color = SkColorSetARGB(83, 0, 0, 0);
+  canvas.DrawRectInt(color, 0, 0, length - 1, length - 1);
+
+  return canvas.ExtractBitmap();
+}
+
+} // namespace
 
 NTPLoginHandler::NTPLoginHandler() {
 }
 
 NTPLoginHandler::~NTPLoginHandler() {
 }
 
 WebUIMessageHandler* NTPLoginHandler::Attach(WebUI* web_ui) {
   PrefService* pref_service = Profile::FromWebUI(web_ui)->GetPrefs();
   username_pref_.Init(prefs::kGoogleServicesUsername, pref_service, this);
 
+  registrar_.Add(this, chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED,
+                 content::NotificationService::AllSources());
+
   return WebUIMessageHandler::Attach(web_ui);
 }
 
 void NTPLoginHandler::RegisterMessages() {
   web_ui_->RegisterMessageCallback("initializeSyncLogin",
       base::Bind(&NTPLoginHandler::HandleInitializeSyncLogin,
                  base::Unretained(this)));
   web_ui_->RegisterMessageCallback("showSyncLoginUI",
       base::Bind(&NTPLoginHandler::HandleShowSyncLoginUI,
                  base::Unretained(this)));
   web_ui_->RegisterMessageCallback("loginMessageSeen",
       base::Bind(&NTPLoginHandler::HandleLoginMessageSeen,
                  base::Unretained(this)));
   web_ui_->RegisterMessageCallback("showAdvancedLoginUI",
       base::Bind(&NTPLoginHandler::HandleShowAdvancedLoginUI,
                  base::Unretained(this)));
 }
 
 void NTPLoginHandler::Observe(int type,
                               const content::NotificationSource& source,
                               const content::NotificationDetails& details) {
-  DCHECK(type == chrome::NOTIFICATION_PREF_CHANGED);
-  std::string* name = content::Details<std::string>(details).ptr();
-  if (prefs::kGoogleServicesUsername == *name)
+  if (type == chrome::NOTIFICATION_PROFILE_CACHED_INFO_CHANGED) {
     UpdateLogin();
+  } else if (type == chrome::NOTIFICATION_PREF_CHANGED) {
+    std::string* name = content::Details<std::string>(details).ptr();
+    if (prefs::kGoogleServicesUsername == *name)
+      UpdateLogin();
+  } else {
+    NOTREACHED();
+  }
 }
 
 void NTPLoginHandler::HandleInitializeSyncLogin(const ListValue* args) {
   UpdateLogin();
 }
 
 void NTPLoginHandler::HandleShowSyncLoginUI(const ListValue* args) {
   Profile* profile = Profile::FromWebUI(web_ui_);
   std::string username = profile->GetPrefs()->GetString(
       prefs::kGoogleServicesUsername);
@@ skipping 49 matching lines @@
 void NTPLoginHandler::HandleShowAdvancedLoginUI(const ListValue* args) {
   Profile::FromWebUI(web_ui_)->GetProfileSyncService()->ShowConfigure(false);
 }
 
 void NTPLoginHandler::UpdateLogin() {
   Profile* profile = Profile::FromWebUI(web_ui_);
   std::string username = profile->GetPrefs()->GetString(
       prefs::kGoogleServicesUsername);
 
   string16 header, sub_header;
+  std::string icon_url;
   if (!username.empty()) {
-    header = UTF8ToUTF16(username);
+    ProfileInfoCache& cache =
+        g_browser_process->profile_manager()->GetProfileInfoCache();
+    size_t profile_index = cache.GetIndexOfProfileWithPath(profile->GetPath());
+    if (profile_index != std::string::npos) {
+      // Only show the profile picture and full name for the single profile
+      // case. In the multi-profile case the profile picture is visible in the
+      // title bar and the full name can be ambiguous.
+      if (cache.GetNumberOfProfiles() == 1) {
+        header = cache.GetGAIANameOfProfileAtIndex(profile_index);
+        const gfx::Image* image =
+            cache.GetGAIAPictureOfProfileAtIndex(profile_index);
+        if (image) {
+          fprintf(stderr, "%s - setting image\n", __func__);
+          icon_url = web_ui_util::GetImageDataUrl(GetGAIAPictureForNTP(*image));
+        }
+      }
+      if (header.empty())
+        header = UTF8ToUTF16(username);
+    }
   } else if (SyncPromoUI::ShouldShowSyncPromo(profile) &&
              (SyncPromoUI::UserHasSeenSyncPromoAtStartup(profile) ||
               PromoResourceService::CanShowNTPSignInPromo(profile))) {
     string16 signed_in_link = l10n_util::GetStringUTF16(
         IDS_SYNC_PROMO_NOT_SIGNED_IN_STATUS_LINK);
     signed_in_link = ASCIIToUTF16("<span class='link-span'>") + signed_in_link +
                      ASCIIToUTF16("</span>");
     header = l10n_util::GetStringFUTF16(
         IDS_SYNC_PROMO_NOT_SIGNED_IN_STATUS_HEADER,
         l10n_util::GetStringUTF16(IDS_SHORT_PRODUCT_NAME));
     sub_header = l10n_util::GetStringFUTF16(
         IDS_SYNC_PROMO_NOT_SIGNED_IN_STATUS_SUB_HEADER, signed_in_link);
     // Record that the user was shown the promo.
     RecordInHistogram(NTP_SIGN_IN_PROMO_VIEWED);
   }
 
   StringValue header_value(header);
   StringValue sub_header_value(sub_header);
+  StringValue icon_url_value(icon_url);
   web_ui_->CallJavascriptFunction(
-      "updateLogin", header_value, sub_header_value);
+      "updateLogin", header_value, sub_header_value, icon_url_value);
 }
 
 // static
 bool NTPLoginHandler::ShouldShow(Profile* profile) {
 #if defined(OS_CHROMEOS)
   // For now we don't care about showing sync status on Chrome OS. The promo
   // UI and the avatar menu don't exist on that platform.
   return false;
 #else
   if (profile->IsOffTheRecord())
@@ skipping 17 matching lines @@
           l10n_util::GetStringUTF16(IDS_SHORT_PRODUCT_NAME)));
   values->SetString("login_status_url",
       google_util::StringAppendGoogleLocaleParam(chrome::kSyncLearnMoreURL));
   values->SetString("login_status_learn_more",
       l10n_util::GetStringUTF16(IDS_LEARN_MORE));
   values->SetString("login_status_advanced",
       l10n_util::GetStringUTF16(IDS_SYNC_PROMO_NTP_BUBBLE_ADVANCED));
   values->SetString("login_status_dismiss",
       l10n_util::GetStringUTF16(IDS_SYNC_PROMO_NTP_BUBBLE_OK));
 }