Support PermissionRequestManager for multiple download permissions on Android

chrome/browser/download/download_request_limiter.cc

 // Copyright (c) 2012 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/download/download_request_limiter.h"
 
 #include "base/bind.h"
 #include "base/stl_util.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
+#include "chrome/browser/download/download_permission_request.h"
 #include "chrome/browser/infobars/infobar_service.h"
+#include "chrome/browser/permissions/permission_request_manager.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/tab_contents/tab_util.h"
 #include "components/content_settings/core/browser/content_settings_details.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/resource_dispatcher_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "url/gurl.h"
 
 #if defined(OS_ANDROID)
 #include "chrome/browser/download/download_request_infobar_delegate_android.h"
-#else
-#include "chrome/browser/download/download_permission_request.h"
-#include "chrome/browser/permissions/permission_request_manager.h"
 #endif
 
 using content::BrowserThread;
 using content::NavigationController;
 using content::NavigationEntry;
 
 namespace {
 
 ContentSetting GetSettingFromStatus(
     DownloadRequestLimiter::DownloadStatus status) {
@@ skipping 124 matching lines @@
 }
 
 void DownloadRequestLimiter::TabDownloadState::DidGetUserInteraction(
     const blink::WebInputEvent::Type type) {
   if (is_showing_prompt() ||
       type == blink::WebInputEvent::kGestureScrollBegin) {
     // Don't change state if a prompt is showing or if the user has scrolled.
     return;
   }
 
-#if defined(OS_ANDROID)
-  bool promptable = InfoBarService::FromWebContents(web_contents()) != nullptr;
-#else
-  bool promptable =
-      PermissionRequestManager::FromWebContents(web_contents()) != nullptr;
-#endif
+  bool promptable;
+  if (PermissionRequestManager::IsEnabled()) {
+    promptable =
+        PermissionRequestManager::FromWebContents(web_contents()) != nullptr;
+  } else {
+    promptable = InfoBarService::FromWebContents(web_contents()) != nullptr;
+  }
 
   // See PromptUserForDownload(): if there's no InfoBarService, then
   // DOWNLOADS_NOT_ALLOWED is functionally equivalent to PROMPT_BEFORE_DOWNLOAD.
   if ((status_ != DownloadRequestLimiter::ALLOW_ALL_DOWNLOADS) &&
       (!promptable ||
        (status_ != DownloadRequestLimiter::DOWNLOADS_NOT_ALLOWED))) {
     // Revert to default status.
     host_->Remove(this, web_contents());
     // WARNING: We've been deleted.
   }
 }
 
 void DownloadRequestLimiter::TabDownloadState::WebContentsDestroyed() {
   // Tab closed, no need to handle closing the dialog as it's owned by the
   // WebContents.
 
   NotifyCallbacks(false);
   host_->Remove(this, web_contents());
   // WARNING: We've been deleted.
 }
 
 void DownloadRequestLimiter::TabDownloadState::PromptUserForDownload(
     const DownloadRequestLimiter::Callback& callback) {
   callbacks_.push_back(callback);
   DCHECK(web_contents_);
   if (is_showing_prompt())
     return;
 
+  if (PermissionRequestManager::IsEnabled()) {
+    PermissionRequestManager* permission_request_manager =
+        PermissionRequestManager::FromWebContents(web_contents_);
+    if (permission_request_manager) {
+      permission_request_manager->AddRequest(
+          new DownloadPermissionRequest(factory_.GetWeakPtr()));
+    } else {
+      Cancel();
+    }
+  } else {
 #if defined(OS_ANDROID)
-  DownloadRequestInfoBarDelegateAndroid::Create(
-      InfoBarService::FromWebContents(web_contents_), factory_.GetWeakPtr());
-#else
-  PermissionRequestManager* permission_request_manager =
-      PermissionRequestManager::FromWebContents(web_contents_);
-  if (permission_request_manager) {
-    permission_request_manager->AddRequest(
-        new DownloadPermissionRequest(factory_.GetWeakPtr()));
-  } else {
-    Cancel();
+    DownloadRequestInfoBarDelegateAndroid::Create(
+        InfoBarService::FromWebContents(web_contents_), factory_.GetWeakPtr());
+#endif
   }
-#endif
 }
 
 void DownloadRequestLimiter::TabDownloadState::SetContentSetting(
     ContentSetting setting) {
   if (!web_contents_)
     return;
   HostContentSettingsMap* settings =
       DownloadRequestLimiter::GetContentSettings(web_contents_);
   if (!settings)
     return;
@@ skipping 281 matching lines @@
       NOTREACHED();
   }
 }
 
 void DownloadRequestLimiter::Remove(TabDownloadState* state,
                                     content::WebContents* contents) {
   DCHECK(base::ContainsKey(state_map_, contents));
   state_map_.erase(contents);
   delete state;
 }