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
13DownloadRequestInfoBarDelegate::FakeCreateCallback*
14    DownloadRequestInfoBarDelegate::callback_ = NULL;
15
16DownloadRequestInfoBarDelegate::~DownloadRequestInfoBarDelegate() {
17  if (!responded_ && host_)
18    host_->CancelOnce();
19}
20
21// static
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
46void DownloadRequestInfoBarDelegate::SetCallbackForTesting(
47    FakeCreateCallback* callback) {
48  DownloadRequestInfoBarDelegate::callback_ = callback;
49}
50
51DownloadRequestInfoBarDelegate::DownloadRequestInfoBarDelegate(
52    base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host)
53    : ConfirmInfoBarDelegate(),
54      responded_(false),
55      host_(host) {
56}
57
58int DownloadRequestInfoBarDelegate::GetIconID() const {
59  return IDR_INFOBAR_MULTIPLE_DOWNLOADS;
60}
61
62base::string16 DownloadRequestInfoBarDelegate::GetMessageText() const {
63  return l10n_util::GetStringUTF16(IDS_MULTI_DOWNLOAD_WARNING);
64}
65
66base::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
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
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