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 "components/infobars/core/infobar.h" 9 #include "grit/generated_resources.h" 10 #include "grit/theme_resources.h" 11 #include "ui/base/l10n/l10n_util.h" 12 13 DownloadRequestInfoBarDelegate::FakeCreateCallback* 14 DownloadRequestInfoBarDelegate::callback_ = NULL; 15 ~DownloadRequestInfoBarDelegate()16DownloadRequestInfoBarDelegate::~DownloadRequestInfoBarDelegate() { 17 if (!responded_ && host_) 18 host_->CancelOnce(); 19 } 20 21 // static Create(InfoBarService * infobar_service,base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host)22void 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(ConfirmInfoBarDelegate::CreateInfoBar( 40 scoped_ptr<ConfirmInfoBarDelegate>( 41 new DownloadRequestInfoBarDelegate(host)))); 42 } 43 } 44 45 // static SetCallbackForTesting(FakeCreateCallback * callback)46void DownloadRequestInfoBarDelegate::SetCallbackForTesting( 47 FakeCreateCallback* callback) { 48 DownloadRequestInfoBarDelegate::callback_ = callback; 49 } 50 DownloadRequestInfoBarDelegate(base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host)51DownloadRequestInfoBarDelegate::DownloadRequestInfoBarDelegate( 52 base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host) 53 : ConfirmInfoBarDelegate(), 54 responded_(false), 55 host_(host) { 56 } 57 GetIconID() const58int DownloadRequestInfoBarDelegate::GetIconID() const { 59 return IDR_INFOBAR_MULTIPLE_DOWNLOADS; 60 } 61 GetMessageText() const62base::string16 DownloadRequestInfoBarDelegate::GetMessageText() const { 63 return l10n_util::GetStringUTF16(IDS_MULTI_DOWNLOAD_WARNING); 64 } 65 GetButtonLabel(InfoBarButton button) const66base::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_DENY); 70 } 71 Accept()72bool 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 Cancel()82bool DownloadRequestInfoBarDelegate::Cancel() { 83 DCHECK(!responded_); 84 responded_ = true; 85 if (host_) { 86 // This may invalidate |host_|. 87 host_->Cancel(); 88 } 89 return !host_; 90 } 91