chrome/browser/download/android/dangerous_download_infobar_delegate.cc

// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/download/android/dangerous_download_infobar_delegate.h"

#include <memory>

#include "base/memory/ptr_util.h"
#include "base/metrics/histogram_functions.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/android/android_theme_resources.h"
#include "chrome/grit/generated_resources.h"
#include "components/infobars/android/confirm_infobar.h"
#include "components/infobars/content/content_infobar_manager.h"
#include "components/infobars/core/infobar.h"
#include "ui/base/l10n/l10n_util.h"

namespace {
// Records user interactions with the dangerous download infobar.
// Used in UMA, do not remove, change or reuse existing entries.
// Update histograms.xml and enums.xml when adding entries.
enum class DangerousDownloadInfobarEvent {
  // Infobar was shown.
  kShown = 0,
  // Accepted the dangerous download.
  kAccepted = 1,
  // Canceled the dangerous download.
  kCanceled = 2,
  // Dismissed the dangerous download.
  kDismissed = 3,
  kMaxValue = kDismissed
};

void RecordDangerousDownloadInfobarEvent(DangerousDownloadInfobarEvent event) {
  base::UmaHistogramEnumeration("Download.Mobile.DangerousDownloadInfobarEvent",
                                event);
}
}  // namespace

// static
void DangerousDownloadInfoBarDelegate::Create(
    infobars::ContentInfoBarManager* infobar_manager,
    download::DownloadItem* download_item) {
  if (infobar_manager->AddInfoBar(
          std::make_unique<infobars::ConfirmInfoBar>(base::WrapUnique(
              new DangerousDownloadInfoBarDelegate(download_item))))) {
    RecordDangerousDownloadInfobarEvent(DangerousDownloadInfobarEvent::kShown);
  }
}

DangerousDownloadInfoBarDelegate::DangerousDownloadInfoBarDelegate(
    download::DownloadItem* download_item)
    : download_item_(download_item) {
  download_item_->AddObserver(this);
  message_text_ = l10n_util::GetStringFUTF16(
      IDS_PROMPT_DANGEROUS_DOWNLOAD,
      base::UTF8ToUTF16(download_item_->GetFileNameToReportUser().value()));
}

DangerousDownloadInfoBarDelegate::~DangerousDownloadInfoBarDelegate() {
  if (download_item_)
    download_item_->RemoveObserver(this);
}

void DangerousDownloadInfoBarDelegate::OnDownloadDestroyed(
    download::DownloadItem* download_item) {
  DCHECK_EQ(download_item, download_item_);
  download_item_ = nullptr;
}

infobars::InfoBarDelegate::InfoBarIdentifier
DangerousDownloadInfoBarDelegate::GetIdentifier() const {
  return DANGEROUS_DOWNLOAD_INFOBAR_DELEGATE_ANDROID;
}

int DangerousDownloadInfoBarDelegate::GetIconId() const {
  return IDR_ANDROID_INFOBAR_WARNING;
}

bool DangerousDownloadInfoBarDelegate::ShouldExpire(
    const NavigationDetails& details) const {
  return false;
}

void DangerousDownloadInfoBarDelegate::InfoBarDismissed() {
  if (download_item_)
    download_item_->Remove();
  RecordDangerousDownloadInfobarEvent(
      DangerousDownloadInfobarEvent::kDismissed);
}

std::u16string DangerousDownloadInfoBarDelegate::GetMessageText() const {
  return message_text_;
}

bool DangerousDownloadInfoBarDelegate::Accept() {
  if (download_item_)
    download_item_->ValidateDangerousDownload();
  RecordDangerousDownloadInfobarEvent(DangerousDownloadInfobarEvent::kAccepted);
  return true;
}

bool DangerousDownloadInfoBarDelegate::Cancel() {
  if (download_item_)
    download_item_->Remove();
  RecordDangerousDownloadInfobarEvent(DangerousDownloadInfobarEvent::kCanceled);
  return true;
}