Upstream: Add additional checks before creating a WebAPK after clicking "Add to Homescreen"

chrome/browser/android/webapps/add_to_homescreen_data_fetcher.cc

 // Copyright 2015 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/webapps/add_to_homescreen_data_fetcher.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
+#include "base/command_line.h"
 #include "base/location.h"
 #include "base/strings/string16.h"
 #include "base/task/cancelable_task_tracker.h"
 #include "chrome/browser/android/offline_pages/offline_page_utils.h"
 #include "chrome/browser/android/shortcut_helper.h"
+#include "chrome/browser/banners/app_banner_debug_log.h"
+#include "chrome/browser/banners/webapp_manifest_validator.h"
 #include "chrome/browser/favicon/favicon_service_factory.h"
 #include "chrome/browser/manifest/manifest_icon_downloader.h"
 #include "chrome/browser/manifest/manifest_icon_selector.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 "chrome/common/web_application_info.h"
 #include "components/dom_distiller/core/url_utils.h"
 #include "components/favicon/core/favicon_service.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/service_worker_context.h"
+#include "content/public/browser/storage_partition.h"
 #include "content/public/browser/user_metrics.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/common/frame_navigate_params.h"
 #include "content/public/common/manifest.h"
 #include "third_party/WebKit/public/platform/modules/screen_orientation/WebScreenOrientationLockType.h"
 #include "ui/display/display.h"
 #include "ui/display/screen.h"
 #include "ui/gfx/codec/png_codec.h"
 #include "ui/gfx/favicon_size.h"
@@ skipping 14 matching lines @@
       is_icon_saved_(false),
       is_ready_(false),
       icon_timeout_timer_(false, false),
       shortcut_info_(GetShortcutUrl(web_contents->GetURL())),
       ideal_icon_size_in_dp_(ideal_icon_size_in_dp),
       minimum_icon_size_in_dp_(minimum_icon_size_in_dp),
       ideal_splash_image_size_in_dp_(ideal_splash_image_size_in_dp),
       minimum_splash_image_size_in_dp_(minimum_splash_image_size_in_dp) {
   DCHECK(minimum_icon_size_in_dp <= ideal_icon_size_in_dp);
   DCHECK(minimum_splash_image_size_in_dp <= ideal_splash_image_size_in_dp);
+  shortcut_url_ = shortcut_info_.url;
 
   // Send a message to the renderer to retrieve information about the page.
   is_waiting_for_web_application_info_ = true;
   Send(new ChromeViewMsg_GetWebApplicationInfo(routing_id()));
 }
 
 void AddToHomescreenDataFetcher::OnDidGetWebApplicationInfo(
     const WebApplicationInfo& received_web_app_info) {
   is_waiting_for_web_application_info_ = false;
   if (!web_contents() || !weak_observer_) return;
@@ skipping 42 matching lines @@
     const content::Manifest& manifest) {
   if (!web_contents() || !weak_observer_) return;
 
   if (!manifest.IsEmpty()) {
       content::RecordAction(
           base::UserMetricsAction("webapps.AddShortcut.Manifest"));
       shortcut_info_.UpdateFromManifest(manifest);
       shortcut_info_.manifest_url = manifest_url;
   }
 
+  if (!IsManifestWebApkCapable(manifest)) {
+    banners::OutputDeveloperMessageCode code =
+        banners::OutputDeveloperMessageCode::kNone;
+    if (banners::webapp_manifest_validator::IsWebappCapable(manifest, &code)) {
+      shortcut_info_.type = ShortcutInfo::Type::WEBAPP;
+    }
+    OnCheckIsWebApk(manifest, false);
+    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);
+  storage_partition->GetServiceWorkerContext()->CheckHasServiceWorker(
+      shortcut_url_, manifest.start_url,
+      base::Bind(&AddToHomescreenDataFetcher::OnDidCheckServiceWorkerForWebApk,
+                 this, manifest));
+}
+
+void AddToHomescreenDataFetcher::OnDidCheckServiceWorkerForWebApk(
+    const content::Manifest& manifest,
+    bool has_service_worker) {
+  if (!web_contents() || !weak_observer_) return;
+  OnCheckIsWebApk(manifest, has_service_worker);
+}
+
+void AddToHomescreenDataFetcher::OnCheckIsWebApk(
+    const content::Manifest& manifest,
+    bool is_webapk) {
+  if (is_webapk)
+    shortcut_info_.type = ShortcutInfo::Type::WEBAPK;
+
   GURL icon_src = ManifestIconSelector::FindBestMatchingIcon(
       manifest.icons, ideal_icon_size_in_dp_, minimum_icon_size_in_dp_);
 
   // If fetching the Manifest icon fails, fallback to the best favicon
   // for the page.
   if (!ManifestIconDownloader::Download(
         web_contents(),
         icon_src,
         ideal_icon_size_in_dp_,
         minimum_icon_size_in_dp_,
@@ skipping 142 matching lines @@
 
   // If URL points to an offline content, get original URL.
   GURL online_url =
       offline_pages::OfflinePageUtils::MaybeGetOnlineURLForOfflineURL(
           web_contents()->GetBrowserContext(), original_url);
   if (online_url.is_valid())
     return online_url;
 
   return original_url;
 }
+
+bool AddToHomescreenDataFetcher::IsManifestWebApkCapable(
+    const content::Manifest& manifest) const {
+  banners::OutputDeveloperMessageCode code =
+      banners::OutputDeveloperMessageCode::kNone;
+  return banners::webapp_manifest_validator::IsManifestGoodForWebapp(manifest,
+                                                                     &code) &&
+         base::CommandLine::ForCurrentProcess()->HasSwitch(
+             switches::kEnableWebApk);
+}