Popular sites on the NTP: Favicons!

chrome/browser/android/favicon_helper.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/android/favicon_helper.h"
 
 #include <jni.h>
 
+#include <vector>
+
 #include "base/android/jni_android.h"
 #include "base/android/jni_array.h"
 #include "base/android/jni_string.h"
 #include "base/android/scoped_java_ref.h"
 #include "base/bind.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/favicon/favicon_service_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_android.h"
 #include "chrome/browser/sync/profile_sync_service.h"
 #include "chrome/browser/sync/profile_sync_service_factory.h"
-#include "chrome/browser/sync/profile_sync_service_factory.h"
 #include "components/favicon/core/favicon_service.h"
+#include "components/favicon_base/favicon_util.h"
+#include "components/favicon_base/select_favicon_frames.h"
 #include "components/sync_driver/open_tabs_ui_delegate.h"
+#include "content/public/browser/web_contents.h"
 #include "jni/FaviconHelper_jni.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/gfx/android/java_bitmap.h"
 #include "ui/gfx/codec/png_codec.h"
 #include "ui/gfx/color_analysis.h"
 #include "ui/gfx/color_utils.h"
+#include "ui/gfx/favicon_size.h"
+#include "ui/gfx/image/image_skia.h"
 
 using base::android::ScopedJavaGlobalRef;
 using base::android::ScopedJavaLocalRef;
 using base::android::AttachCurrentThread;
 using base::android::ConvertJavaStringToUTF16;
 using base::android::ConvertJavaStringToUTF8;
 using base::android::ConvertUTF8ToJavaString;
 
 namespace {
 
@@ skipping 41 matching lines @@
   if (!favicon_bitmap.isNull())
     j_favicon_bitmap = gfx::ConvertToJavaBitmap(&favicon_bitmap);
 
   // Call java side OnLocalFaviconAvailable method.
   Java_FaviconImageCallback_onFaviconAvailable(env,
                                                j_favicon_image_callback->obj(),
                                                j_favicon_bitmap.obj(),
                                                j_icon_url.obj());
 }
 
+void OnFaviconDownloaded(
+    const ScopedJavaGlobalRef<jobject>& j_availability_callback,
+    Profile* profile,
+    const GURL& page_url,
+    int download_request_id,
+    int http_status_code,
+    const GURL& image_url,
+    const std::vector<SkBitmap>& bitmaps,
+    const std::vector<gfx::Size>& original_sizes) {
+  bool success = !bitmaps.empty();
+  if (success) {
+    gfx::Image image = gfx::Image(CreateFaviconImageSkia(bitmaps,
+                                                         original_sizes,
+                                                         gfx::kFaviconSize,
+                                                         nullptr));
+    favicon_base::SetFaviconColorSpace(&image);
+    favicon::FaviconService* service = FaviconServiceFactory::GetForProfile(
+        profile, ServiceAccessType::IMPLICIT_ACCESS);
+    service->SetFavicons(page_url, image_url, favicon_base::FAVICON, image);
+  }
+
+  JNIEnv* env = AttachCurrentThread();
+  Java_FaviconAvailabilityCallback_onFaviconAvailabilityChecked(
+      env, j_availability_callback.obj(), success);
+}
+
+void OnFaviconImageResultAvailable(
+    const ScopedJavaGlobalRef<jobject>& j_availability_callback,
+    Profile* profile,
+    content::WebContents* web_contents,
+    const GURL& page_url,
+    const GURL& favicon_url,
+    const favicon_base::FaviconImageResult& result) {
+  // If there already is a favicon, return immediately.
+  if (!result.image.IsEmpty()) {
+    JNIEnv* env = AttachCurrentThread();
+    Java_FaviconAvailabilityCallback_onFaviconAvailabilityChecked(
+        env, j_availability_callback.obj(), false);
+    return;
+  }
+
+  web_contents->DownloadImage(favicon_url, true, 0, false,
+                              base::Bind(OnFaviconDownloaded,
+                                         j_availability_callback,
+                                         profile, page_url));
+}
+
 }  // namespace
 
 static jlong Init(JNIEnv* env, jclass clazz) {
   return reinterpret_cast<intptr_t>(new FaviconHelper());
 }
 
 FaviconHelper::FaviconHelper() {
   cancelable_task_tracker_.reset(new base::CancelableTaskTracker());
 }
 
@@ skipping 102 matching lines @@
   gfx::Image favicon_image = gfx::Image::CreateFrom1xPNGBytes(favicon_png);
   SkBitmap favicon_bitmap = favicon_image.AsBitmap();
 
   ScopedJavaLocalRef<jobject> j_favicon_bitmap;
   if (favicon_bitmap.isNull())
     return ScopedJavaLocalRef<jobject>();
 
   return gfx::ConvertToJavaBitmap(&favicon_bitmap);
 }
 
+void FaviconHelper::EnsureFaviconIsAvailable(
+    JNIEnv* env,
+    jobject obj,
+    jobject j_profile,
+    jobject j_web_contents,
+    jstring j_page_url,
+    jstring j_favicon_url,
+    jobject j_availability_callback) {
+  Profile* profile = ProfileAndroid::FromProfileAndroid(j_profile);
+  DCHECK(profile);
+  content::WebContents* web_contents =
+      content::WebContents::FromJavaWebContents(j_web_contents);
+  DCHECK(web_contents);
+  GURL page_url(ConvertJavaStringToUTF8(env, j_page_url));
+  GURL favicon_url(ConvertJavaStringToUTF8(env, j_favicon_url));
+
+  // TODO(treib): Optimize this by creating a FaviconService::HasFavicon method
+  // so that we don't have to actually get the image.
+  ScopedJavaGlobalRef<jobject> j_scoped_callback(env, j_availability_callback);
+  favicon_base::FaviconImageCallback callback_runner =
+      base::Bind(&OnFaviconImageResultAvailable, j_scoped_callback,
+                 profile, web_contents, page_url, favicon_url);
+  favicon::FaviconService* service = FaviconServiceFactory::GetForProfile(
+      profile, ServiceAccessType::IMPLICIT_ACCESS);
+  service->GetFaviconImage(favicon_url, callback_runner,
+                           cancelable_task_tracker_.get());
+}
+
 FaviconHelper::~FaviconHelper() {}
 
 static jint GetDominantColorForBitmap(JNIEnv* env,
                                       jclass clazz,
                                       jobject bitmap) {
   if (!bitmap)
     return 0;
 
-    gfx::JavaBitmap bitmap_lock(bitmap);
-    SkBitmap skbitmap = gfx::CreateSkBitmapFromJavaBitmap(bitmap_lock);
-    return color_utils::CalculateKMeanColorOfBitmap(skbitmap);
+  gfx::JavaBitmap bitmap_lock(bitmap);
+  SkBitmap skbitmap = gfx::CreateSkBitmapFromJavaBitmap(bitmap_lock);
+  return color_utils::CalculateKMeanColorOfBitmap(skbitmap);
 }
 
 // static
 bool FaviconHelper::RegisterFaviconHelper(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }