chrome/browser/download/bubble/download_display_controller.cc

// Copyright 2021 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/download/bubble/download_display_controller.h"

#include "base/numerics/safe_conversions.h"
#include "base/time/time.h"
#include "base/timer/timer.h"
#include "chrome/browser/download/bubble/download_display.h"
#include "chrome/browser/download/bubble/download_icon_state.h"
#include "chrome/browser/download/download_item_model.h"
#include "chrome/browser/download/download_prefs.h"

namespace {

// The amount of time for the toolbar icon to be visible after a download is
// completed.
constexpr base::TimeDelta kToolbarIconVisibilityTimeInterval = base::Hours(24);
}

DownloadDisplayController::DownloadDisplayController(
    DownloadDisplay* display,
    content::DownloadManager* download_manager)
    : display_(display),
      download_manager_(download_manager),
      download_notifier_(download_manager, this) {
  MaybeShowButtonWhenCreated();
}

DownloadDisplayController::~DownloadDisplayController() = default;

void DownloadDisplayController::OnDownloadCreated(
    content::DownloadManager* manager,
    download::DownloadItem* item) {
  UpdateToolbarButtonState();
}

void DownloadDisplayController::OnDownloadUpdated(
    content::DownloadManager* manager,
    download::DownloadItem* item) {
  DownloadItemModel item_model(item);

  if (item_model.IsDone()) {
    ScheduleToolbarDisappearance(kToolbarIconVisibilityTimeInterval);
  }
  UpdateToolbarButtonState();
}

void DownloadDisplayController::OnManagerGoingDown(
    content::DownloadManager* manager) {
  if (download_manager_ == manager) {
    download_manager_ = nullptr;
  }
}

void DownloadDisplayController::ShowToolbarButton(bool show_details) {
  if (!display_->IsShowing()) {
    display_->Enable();
    display_->Show();
    if (show_details) {
      display_->ShowDetails();
    }
  }
}

void DownloadDisplayController::HideToolbarButton() {
  if (display_->IsShowing()) {
    display_->Hide();
  }
}

void DownloadDisplayController::UpdateToolbarButtonState() {
  download::DownloadIconState icon_state;

  if (download_manager_->InProgressCount() > 0) {
    ShowToolbarButton(/*show_details=*/true);
    icon_state = download::DownloadIconState::kProgress;
  } else {
    icon_state = download::DownloadIconState::kComplete;
  }

  display_->UpdateDownloadIcon(icon_state);
}

void DownloadDisplayController::ScheduleToolbarDisappearance(
    base::TimeDelta interval) {
  icon_disappearance_timer_.Stop();
  icon_disappearance_timer_.Start(
      FROM_HERE, interval, this, &DownloadDisplayController::HideToolbarButton);
}

void DownloadDisplayController::MaybeShowButtonWhenCreated() {
  base::Time last_complete_time =
      DownloadPrefs::FromDownloadManager(download_manager_)
          ->GetLastCompleteTime();
  base::Time current_time = base::Time::Now();
  base::TimeDelta time_since_last_completion =
      current_time - last_complete_time;
  // If the last download complete time is less than
  // `kToolbarIconVisibilityTimeInterval` ago, show the button immediately. Also
  // check that the current time is not smaller than the last complete time,
  // this can happen if the system clock has moved backward.
  if (time_since_last_completion < kToolbarIconVisibilityTimeInterval &&
      current_time >= last_complete_time) {
    ShowToolbarButton(/*show_details=*/false);
    ScheduleToolbarDisappearance(kToolbarIconVisibilityTimeInterval -
                                 time_since_last_completion);
  }
}

DownloadDisplayController::ProgressInfo
DownloadDisplayController::GetProgress() {
  DownloadDisplayController::ProgressInfo progress_info;

  int64_t received_bytes = 0;
  int64_t total_bytes = 0;

  content::DownloadManager::DownloadVector items;
  download_manager_->GetAllDownloads(&items);
  for (auto* item : items) {
    if (item->GetState() == download::DownloadItem::IN_PROGRESS) {
      ++progress_info.download_count;
      if (item->GetTotalBytes() <= 0) {
        // There may or may not be more data coming down this pipe.
        progress_info.progress_certain = false;
      } else {
        received_bytes += item->GetReceivedBytes();
        total_bytes += item->GetTotalBytes();
      }
    }
  }

  if (total_bytes > 0) {
    progress_info.progress_percentage =
        base::ClampFloor(received_bytes * 100.0 / total_bytes);
  }

  return progress_info;
}