src/chrome/browser/android/shortcut_helper.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/android/shortcut_helper.h"
   6
   7 #include <jni.h>
   8
   9 #include "base/android/jni_android.h"
  10 #include "base/android/jni_string.h"
  11 #include "base/basictypes.h"
  12 #include "base/location.h"
  13 #include "base/strings/string16.h"
  14 #include "base/task/cancelable_task_tracker.h"
  15 #include "base/threading/worker_pool.h"
  16 #include "chrome/browser/android/tab_android.h"
  17 #include "chrome/browser/favicon/favicon_service.h"
  18 #include "chrome/browser/favicon/favicon_service_factory.h"
  19 #include "chrome/common/render_messages.h"
  20 #include "content/public/browser/user_metrics.h"
  21 #include "content/public/browser/web_contents.h"
  22 #include "content/public/browser/web_contents_observer.h"
  23 #include "content/public/common/frame_navigate_params.h"
  24 #include "jni/ShortcutHelper_jni.h"
  25 #include "ui/gfx/android/java_bitmap.h"
  26 #include "ui/gfx/codec/png_codec.h"
  27 #include "ui/gfx/color_analysis.h"
  28 #include "ui/gfx/favicon_size.h"
  29 #include "url/gurl.h"
  30
  31 ShortcutBuilder::ShortcutBuilder(content::WebContents* web_contents,
  32                                  const base::string16& title,
  33                                  int launcher_large_icon_size)
  34     : launcher_large_icon_size_(launcher_large_icon_size),
  35       shortcut_type_(BOOKMARK) {
  36   Observe(web_contents);
  37   url_ = web_contents->GetURL();
  38   if (title.length() > 0)
  39     title_ = title;
  40   else
  41     title_ = web_contents->GetTitle();
  42
  43   // Send a message to the renderer to retrieve information about the page.
  44   Send(new ChromeViewMsg_RetrieveWebappInformation(routing_id(), url_));
  45 }
  46
  47 void ShortcutBuilder::OnDidRetrieveWebappInformation(
  48     bool success,
  49     bool is_mobile_webapp_capable,
  50     bool is_apple_mobile_webapp_capable,
  51     const GURL& expected_url) {
  52   Profile* profile =
  53       Profile::FromBrowserContext(web_contents()->GetBrowserContext());
  54   Observe(NULL);
  55
  56   if (!success) {
  57     LOG(ERROR) << "Failed to parse webpage.";
  58     Destroy();
  59     return;
  60   } else if (expected_url != url_) {
  61     LOG(ERROR) << "Unexpected URL returned.";
  62     Destroy();
  63     return;
  64   }
  65
  66   if (is_apple_mobile_webapp_capable && !is_mobile_webapp_capable) {
  67     shortcut_type_ = APP_SHORTCUT_APPLE;
  68   } else if (is_apple_mobile_webapp_capable || is_mobile_webapp_capable) {
  69     shortcut_type_ = APP_SHORTCUT;
  70   } else {
  71     shortcut_type_ = BOOKMARK;
  72   }
  73
  74   // Grab the best, largest icon we can find to represent this bookmark.
  75   // TODO(dfalcantara): Try combining with the new BookmarksHandler once its
  76   //                    rewrite is further along.
  77   std::vector<int> icon_types;
  78   icon_types.push_back(favicon_base::FAVICON);
  79   icon_types.push_back(favicon_base::TOUCH_PRECOMPOSED_ICON |
  80                        favicon_base::TOUCH_ICON);
  81   FaviconService* favicon_service = FaviconServiceFactory::GetForProfile(
  82       profile, Profile::EXPLICIT_ACCESS);
  83
  84   // Using favicon if its size is not smaller than platform required size,
  85   // otherwise using the largest icon among all avaliable icons.
  86   int threshold_to_get_any_largest_icon = launcher_large_icon_size_ - 1;
  87   favicon_service->GetLargestRawFaviconForURL(profile, url_, icon_types,
  88       threshold_to_get_any_largest_icon,
  89       base::Bind(&ShortcutBuilder::FinishAddingShortcut,
  90                  base::Unretained(this)),
  91       &cancelable_task_tracker_);
  92 }
  93
  94 void ShortcutBuilder::FinishAddingShortcut(
  95     const favicon_base::FaviconBitmapResult& bitmap_result) {
  96   base::WorkerPool::PostTask(
  97       FROM_HERE,
  98       base::Bind(&ShortcutHelper::AddShortcutInBackground,
  99                  url_,
 100                  title_,
 101                  shortcut_type_,
 102                  bitmap_result),
 103       true);
 104   Destroy();
 105 }
 106
 107 bool ShortcutBuilder::OnMessageReceived(const IPC::Message& message) {
 108   bool handled = true;
 109   IPC_BEGIN_MESSAGE_MAP(ShortcutBuilder, message)
 110     IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DidRetrieveWebappInformation,
 111                         OnDidRetrieveWebappInformation)
 112     IPC_MESSAGE_UNHANDLED(handled = false)
 113   IPC_END_MESSAGE_MAP()
 114   return handled;
 115 }
 116
 117 void ShortcutBuilder::WebContentsDestroyed() {
 118   Destroy();
 119 }
 120
 121 void ShortcutBuilder::Destroy() {
 122   if (cancelable_task_tracker_.HasTrackedTasks()) {
 123     cancelable_task_tracker_.TryCancelAll();
 124   }
 125   delete this;
 126 }
 127
 128 void ShortcutHelper::AddShortcut(content::WebContents* web_contents,
 129                                  const base::string16& title,
 130                                  int launcher_large_icon_size) {
 131   // The ShortcutBuilder deletes itself when it's done.
 132   new ShortcutBuilder(web_contents, title, launcher_large_icon_size);
 133 }
 134
 135 bool ShortcutHelper::RegisterShortcutHelper(JNIEnv* env) {
 136   return RegisterNativesImpl(env);
 137 }
 138
 139 void ShortcutHelper::AddShortcutInBackground(
 140     const GURL& url,
 141     const base::string16& title,
 142     ShortcutBuilder::ShortcutType shortcut_type,
 143     const favicon_base::FaviconBitmapResult& bitmap_result) {
 144   DCHECK(base::WorkerPool::RunsTasksOnCurrentThread());
 145
 146   // Grab the average color from the bitmap.
 147   SkColor color = SK_ColorWHITE;
 148   SkBitmap favicon_bitmap;
 149   if (bitmap_result.is_valid()) {
 150     if (gfx::PNGCodec::Decode(bitmap_result.bitmap_data->front(),
 151                               bitmap_result.bitmap_data->size(),
 152                               &favicon_bitmap))
 153       color = color_utils::CalculateKMeanColorOfBitmap(favicon_bitmap);
 154   }
 155
 156   int r_value = SkColorGetR(color);
 157   int g_value = SkColorGetG(color);
 158   int b_value = SkColorGetB(color);
 159
 160   // Send the data to the Java side to create the shortcut.
 161   JNIEnv* env = base::android::AttachCurrentThread();
 162   ScopedJavaLocalRef<jstring> java_url =
 163       base::android::ConvertUTF8ToJavaString(env, url.spec());
 164   ScopedJavaLocalRef<jstring> java_title =
 165       base::android::ConvertUTF16ToJavaString(env, title);
 166   ScopedJavaLocalRef<jobject> java_bitmap;
 167   if (favicon_bitmap.getSize())
 168     java_bitmap = gfx::ConvertToJavaBitmap(&favicon_bitmap);
 169
 170   Java_ShortcutHelper_addShortcut(env,
 171                                   base::android::GetApplicationContext(),
 172                                   java_url.obj(),
 173                                   java_title.obj(),
 174                                   java_bitmap.obj(),
 175                                   r_value,
 176                                   g_value,
 177                                   b_value,
 178                                   shortcut_type != ShortcutBuilder::BOOKMARK);
 179
 180   // Record what type of shortcut was added by the user.
 181   switch (shortcut_type) {
 182     case ShortcutBuilder::APP_SHORTCUT:
 183       content::RecordAction(
 184           base::UserMetricsAction("webapps.AddShortcut.AppShortcut"));
 185       break;
 186     case ShortcutBuilder::APP_SHORTCUT_APPLE:
 187       content::RecordAction(
 188           base::UserMetricsAction("webapps.AddShortcut.AppShortcutApple"));
 189       break;
 190     case ShortcutBuilder::BOOKMARK:
 191       content::RecordAction(
 192           base::UserMetricsAction("webapps.AddShortcut.Bookmark"));
 193       break;
 194     default:
 195       NOTREACHED();
 196   }
 197 }
 198
 199 // Adds a shortcut to the current URL to the Android home screen, firing
 200 // background tasks to pull all the data required.
 201 // Note that we don't actually care about the tab here -- we just want
 202 // its otherwise inaccessible WebContents.
 203 static void AddShortcut(JNIEnv* env,
 204                         jclass clazz,
 205                         jlong tab_android_ptr,
 206                         jstring title,
 207                         jint launcher_large_icon_size) {
 208   TabAndroid* tab = reinterpret_cast<TabAndroid*>(tab_android_ptr);
 209   ShortcutHelper::AddShortcut(
 210       tab->web_contents(),
 211       base::android::ConvertJavaStringToUTF16(env, title),
 212       launcher_large_icon_size);
 213 }