chrome/browser/download/download_request_infobar_delegate.cc

   1 // Copyright (c) 2012 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/download/download_request_infobar_delegate.h"
   6
   7 #include "chrome/browser/infobars/infobar_service.h"
   8 #include "chrome/grit/generated_resources.h"
   9 #include "components/infobars/core/infobar.h"
  10 #include "grit/theme_resources.h"
  11 #include "ui/base/l10n/l10n_util.h"
  12
  13 DownloadRequestInfoBarDelegate::FakeCreateCallback*
  14     DownloadRequestInfoBarDelegate::callback_ = NULL;
  15
  16 DownloadRequestInfoBarDelegate::~DownloadRequestInfoBarDelegate() {
  17   if (!responded_ && host_)
  18     host_->CancelOnce();
  19 }
  20
  21 // static
  22 void DownloadRequestInfoBarDelegate::Create(
  23     InfoBarService* infobar_service,
  24     base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host) {
  25   if (DownloadRequestInfoBarDelegate::callback_ &&
  26       !DownloadRequestInfoBarDelegate::callback_->is_null()) {
  27     DownloadRequestInfoBarDelegate::callback_->Run(infobar_service, host);
  28   } else if (!infobar_service) {
  29     // |web_contents| may not have a InfoBarService if it's actually a
  30     // WebContents like those used for extension popups/bubbles and hosted apps
  31     // etc.
  32     // TODO(benjhayden): If this is an automatic download from an extension,
  33     // it would be convenient for the extension author if we send a message to
  34     // the extension's DevTools console (as we do for CSP) about how
  35     // extensions should use chrome.downloads.download() (requires the
  36     // "downloads" permission) to automatically download >1 files.
  37     host->Cancel();
  38   } else {
  39     infobar_service->AddInfoBar(infobar_service->CreateConfirmInfoBar(
  40         scoped_ptr<ConfirmInfoBarDelegate>(
  41             new DownloadRequestInfoBarDelegate(host))));
  42   }
  43 }
  44
  45 // static
  46 void DownloadRequestInfoBarDelegate::SetCallbackForTesting(
  47     FakeCreateCallback* callback) {
  48   DownloadRequestInfoBarDelegate::callback_ = callback;
  49 }
  50
  51 DownloadRequestInfoBarDelegate::DownloadRequestInfoBarDelegate(
  52     base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host)
  53     : ConfirmInfoBarDelegate(),
  54       responded_(false),
  55       host_(host) {
  56 }
  57
  58 int DownloadRequestInfoBarDelegate::GetIconID() const {
  59   return IDR_INFOBAR_MULTIPLE_DOWNLOADS;
  60 }
  61
  62 base::string16 DownloadRequestInfoBarDelegate::GetMessageText() const {
  63   return l10n_util::GetStringUTF16(IDS_MULTI_DOWNLOAD_WARNING);
  64 }
  65
  66 base::string16 DownloadRequestInfoBarDelegate::GetButtonLabel(
  67     InfoBarButton button) const {
  68   return l10n_util::GetStringUTF16((button == BUTTON_OK) ?
  69       IDS_MULTI_DOWNLOAD_WARNING_ALLOW : IDS_MULTI_DOWNLOAD_WARNING_BLOCK);
  70 }
  71
  72 bool DownloadRequestInfoBarDelegate::Accept() {
  73   DCHECK(!responded_);
  74   responded_ = true;
  75   if (host_) {
  76     // This may invalidate |host_|.
  77     host_->Accept();
  78   }
  79   return !host_;
  80 }
  81
  82 bool DownloadRequestInfoBarDelegate::Cancel() {
  83   DCHECK(!responded_);
  84   responded_ = true;
  85   if (host_) {
  86     // This may invalidate |host_|.
  87     host_->Cancel();
  88   }
  89   return !host_;
  90 }