Check if there is a SW controlling the start page for app banner.

chrome/browser/android/banners/app_banner_manager.cc

 // Copyright 2014 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/banners/app_banner_manager.h"
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_string.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/metrics/histogram.h"
 #include "base/threading/worker_pool.h"
 #include "chrome/browser/android/banners/app_banner_infobar_delegate.h"
 #include "chrome/browser/android/banners/app_banner_metrics_ids.h"
 #include "chrome/browser/android/banners/app_banner_utilities.h"
 #include "chrome/browser/android/manifest_icon_selector.h"
 #include "chrome/browser/android/shortcut_helper.h"
 #include "chrome/browser/android/shortcut_info.h"
 #include "chrome/browser/banners/app_banner_settings_helper.h"
 #include "chrome/browser/bitmap_fetcher/bitmap_fetcher.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/render_messages.h"
 #include "content/public/browser/android/content_view_core.h"
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/service_worker_context.h"
+#include "content/public/browser/storage_partition.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/frame_navigate_params.h"
 #include "content/public/common/manifest.h"
 #include "jni/AppBannerManager_jni.h"
 #include "net/base/load_flags.h"
 #include "ui/gfx/android/java_bitmap.h"
 #include "ui/gfx/screen.h"
 
 using base::android::ConvertJavaStringToUTF8;
 using base::android::ConvertUTF8ToJavaString;
 using base::android::ConvertUTF16ToJavaString;
 
 namespace {
 const char kBannerTag[] = "google-play-id";
 }
 
 namespace banners {
 
+// The AppBannerManager::IOThreadHandler class is used to hold onto the
+// weak pointer to the AppBannerManager. It catches the callbacks on the
+// IO thread, and forwards on the calls to the UI thread. This is required
+// as the weak pointer can only be dereferenced on the UI thread.
+class AppBannerManager::IOThreadHandler
+    : public base::RefCounted<AppBannerManager::IOThreadHandler> {
+ public:
+  explicit IOThreadHandler(
+      const base::WeakPtr<AppBannerManager>& banner_manager)
+      : banner_manager_(banner_manager) {}
+
+  void OnDidCheckHasSameServiceWorkerOnIOThread(bool has_same) {
+    DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
+    content::BrowserThread::PostTask(
+        content::BrowserThread::UI, FROM_HERE,
+        base::Bind(&AppBannerManager::OnDidCheckHasSameServiceWorker,
+                   banner_manager_, has_same));
+  }
+
+ private:
+  base::WeakPtr<AppBannerManager> banner_manager_;
+};
+
 AppBannerManager::AppBannerManager(JNIEnv* env, jobject obj)
-    : weak_java_banner_view_manager_(env, obj) {}
+    : weak_java_banner_view_manager_(env, obj), weak_factory_(this) {
+  io_thread_handler_ =
+      make_scoped_refptr(new IOThreadHandler(weak_factory_.GetWeakPtr()));
+}
 
 AppBannerManager::~AppBannerManager() {
 }
 
 void AppBannerManager::Destroy(JNIEnv* env, jobject obj) {
   delete this;
 }
 
 void AppBannerManager::BlockBanner(JNIEnv* env,
                                    jobject obj,
@@ skipping 74 matching lines @@
   if (manifest.IsEmpty()
       || !manifest.start_url.is_valid()
       || (manifest.name.is_null() && manifest.short_name.is_null())) {
     // No usable manifest, see if there is a play store meta tag.
     Send(new ChromeViewMsg_RetrieveMetaTagContent(routing_id(),
                                                   validated_url_,
                                                   kBannerTag));
     return;
   }
 
-  // TODO(benwells): Check triggering parameters here and if there is a meta
-  // tag.
-
-  // Create an infobar to promote the manifest's app.
   manifest_ = manifest;
 
-  GURL icon_url =
-      ManifestIconSelector::FindBestMatchingIcon(
-          manifest.icons,
-          GetPreferredIconSize(),
-          gfx::Screen::GetScreenFor(web_contents()->GetNativeView()));
-  if (icon_url.is_empty())
-    return;
+  // Check to see if there is a single service worker controlling this page
+  // and the manifest's start url.
+  Profile* profile =
+      Profile::FromBrowserContext(web_contents()->GetBrowserContext());
+  content::StoragePartition* storage_partition =
+      content::BrowserContext::GetStoragePartition(
+          profile, web_contents()->GetSiteInstance());
+  DCHECK(storage_partition);
 
-  FetchIcon(icon_url);
+  storage_partition->GetServiceWorkerContext()->CheckHasSameServiceWorker(
+      validated_url_, manifest.start_url,
+      base::Bind(&AppBannerManager::IOThreadHandler::
+                     OnDidCheckHasSameServiceWorkerOnIOThread,
+                 io_thread_handler_));
+}
+
+void AppBannerManager::OnDidCheckHasSameServiceWorker(bool has_same) {
+  if (has_same) {
+    // TODO(benwells): Check triggering parameters.
+    GURL icon_url =
+        ManifestIconSelector::FindBestMatchingIcon(
+            manifest_.icons,
+            GetPreferredIconSize(),
+            gfx::Screen::GetScreenFor(web_contents()->GetNativeView()));
+    if (icon_url.is_empty())
+      return;
+
+    FetchIcon(icon_url);
+  }
 }
 
 bool AppBannerManager::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(AppBannerManager, message)
     IPC_MESSAGE_HANDLER(ChromeViewHostMsg_DidRetrieveMetaTagContent,
                         OnDidRetrieveMetaTagContent)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
@@ skipping 134 matching lines @@
   return base::CommandLine::ForCurrentProcess()->HasSwitch(
       switches::kEnableAppInstallAlerts);
 }
 
 // Register native methods
 bool RegisterAppBannerManager(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 }  // namespace banners