| OLD | NEW |
| (Empty) |
| 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 } | |
| OLD | NEW |